diff --git a/.gitignore b/.gitignore index 46957c4..d66a6f3 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,3 @@ Vagrantfile .vagrant +checkout/ diff --git a/bin/generate_docs.rb b/bin/generate_docs.rb index a4795eb..d3a37ce 100755 --- a/bin/generate_docs.rb +++ b/bin/generate_docs.rb @@ -5,11 +5,23 @@ require 'lock_file' require 'docs_generator' require 'git_manager' +require 'fileutils' + +if ARGV.size > 1 || ARGV.first == "-h" || ARGV.first == "--help" + puts "USAGE: bin/generate_docs.rb [CHECKOUT_PATH]" + exit +end + +CHECKOUT_PATH = ARGV.first || Dir.home + +unless Dir.exists?(CHECKOUT_PATH) + FileUtils.mkdir(CHECKOUT_PATH) +end LockFile.acquiring('docs_generation.lock') do - git_manager = GitManager.new(Dir.home) + git_manager = GitManager.new(CHECKOUT_PATH) git_manager.update_master - generator = DocsGenerator.new(Dir.home, git_manager) + generator = DocsGenerator.new(CHECKOUT_PATH, git_manager) generator.generate end