captify 0.2.2 → 1.0.0

data/Gemfile

@@ -6,4 +6,7 @@ group :development do
   gem "rdoc", "~> 3.12"
   gem "bundler", "~> 1.0.0"
   gem "jeweler", "~> 1.8.3"
+  gem "minitest", "~> 3.2.0"
+  gem "debugger"
+  gem "debugger-pry", :require => "debugger/pry"
 end

data/Gemfile.lock

@@ -7,14 +7,28 @@ GEM
       net-sftp (>= 2.0.0)
       net-ssh (>= 2.0.14)
       net-ssh-gateway (>= 1.1.0)
+    coderay (1.0.7)
+    columnize (0.3.6)
+    debugger (1.1.4)
+      columnize (>= 0.3.1)
+      debugger-linecache (~> 1.1.1)
+      debugger-ruby_core_source (~> 1.1.3)
+    debugger-linecache (1.1.2)
+      debugger-ruby_core_source (>= 1.1.1)
+    debugger-pry (0.1.1)
+      debugger (~> 1)
+      pry (>= 0.9.9)
+    debugger-ruby_core_source (1.1.3)
     git (1.2.5)
-    highline (1.6.12)
-    jeweler (1.8.3)
+    highline (1.6.13)
+    jeweler (1.8.4)
       bundler (~> 1.0)
       git (>= 1.2.5)
       rake
       rdoc
     json (1.7.3)
+    method_source (0.7.1)
+    minitest (3.2.0)
     net-scp (1.0.4)
       net-ssh (>= 1.99.1)
     net-sftp (2.0.5)
@@ -22,9 +36,14 @@ GEM
     net-ssh (2.5.2)
     net-ssh-gateway (1.1.0)
       net-ssh (>= 1.99.1)
+    pry (0.9.9.6)
+      coderay (~> 1.0.5)
+      method_source (~> 0.7.1)
+      slop (>= 2.4.4, < 3)
     rake (0.9.2.2)
     rdoc (3.12)
       json (~> 1.4)
+    slop (2.4.4)
 
 PLATFORMS
   ruby
@@ -32,5 +51,8 @@ PLATFORMS
 DEPENDENCIES
   bundler (~> 1.0.0)
   capistrano
+  debugger
+  debugger-pry
   jeweler (~> 1.8.3)
+  minitest (~> 3.2.0)
   rdoc (~> 3.12)

data/README.rdoc

@@ -1,7 +1,7 @@
 = captify
 
 == Install
-  gem install capistrano-templates  # default template is stored in this gem
+  gem install capistrano-templates  # default template stored in this gem
 
 For bundler, add this to your Gemfile:
 
@@ -9,21 +9,12 @@ For bundler, add this to your Gemfile:
 
 == Usage
   cd my_project
-  bundle exec captify .   # use default template: rails-basic
+  captify -t rails-base .   # use default template: rails-basic
 
 It will:
 
 1. Run capify.
-2. Overwrite Capistrano files with default template: rails-basic
-
-To use a different template:
-
-  cd my_project
-  bundle exec captify -t sinatra-basic .
-
-=== How to load a template gem without bundler?
-  cd my_project
-  captify -g capistrano-templates -t rails-basic .
+2. Overwrite Capistrano files with template: rails-base (see capistrnao-templates[https://github.com/teohm/capistrano-templates]).
 
 === See also
   captify -h

data/Rakefile

@@ -27,6 +27,13 @@ Jeweler::RubygemsDotOrgTasks.new
 
 task :default => :test
 
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'spec'
+  test.pattern = 'spec/**/*_spec.rb'
+  test.verbose = true
+end
+
 require 'rdoc/task'
 Rake::RDocTask.new do |rdoc|
   version = File.exist?('VERSION') ? File.read('VERSION') : ""

data/VERSION

@@ -1 +1 @@
-0.2.2
+1.0.0

data/bin/captify

@@ -1,53 +1,5 @@
 #!/usr/bin/env ruby
 
-require 'captify/template'
-require 'optparse'
+require 'captify'
+Captify.run
 
-template_name = 'rails-basic'
-
-opts = OptionParser.new do |opts|
-  opts.banner = "Usage: captify [options] directory"
-
-  opts.separator ""
-  opts.separator "Options:"
-
-  opts.on("-t", "--template TEMPLATE_NAME", 
-          "Specify the template to be used") do |template|
-    template_name = template
-  end
-
-  opts.on_tail("-h", "--help", "Show this message") do 
-    puts opts
-    exit
-  end
-end
-
-opts.parse!(ARGV)
-
-
-Captify::Template.load_all
-
-if ARGV.empty?
-    abort "Please specify the directory to captify, e.g. `#{File.basename($0)} .'\n\n#{opts}"
-elsif !File.exists?(ARGV.first)
-    abort "`#{ARGV.first}' does not exist.\n\n#{opts}"
-elsif !File.directory?(ARGV.first)
-    abort "`#{ARGV.first}' is not a directory.\n\n#{opts}"
-elsif ARGV.length > 1
-    abort "Too many arguments; please specify only the directory to captify.\n\n#{opts}"
-end
-
-base_dir = ARGV.first
-
-file_paths = Captify::Template.template_files(template_name)
-
-puts `capify #{base_dir}`
-
-file_paths.each do |src_path|
-  dest_path = src_path.split( "#{template_name}" ).last
-  dest_dir  = File.dirname(dest_path)
-  puts "[captify] Overwrite file '#{base_dir}#{dest_path}'" 
-  `mkdir -p \"#{base_dir}#{dest_dir}\" && cp \"#{src_path}\" \"#{base_dir}#{dest_path}\"`
-end
-
-puts "[captify] Done!"

data/captify.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = "captify"
-  s.version = "0.2.2"
+  s.version = "1.0.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Huiming Teo"]
-  s.date = "2012-06-12"
+  s.date = "2012-07-16"
   s.description = "Capistrano capify with canned templates."
   s.email = "teohuiming@gmail.com"
   s.executables = ["captify"]
@@ -27,7 +27,37 @@ Gem::Specification.new do |s|
     "VERSION",
     "bin/captify",
     "captify.gemspec",
-    "lib/captify/template.rb"
+    "lib/captify.rb",
+    "lib/captify/cli.rb",
+    "lib/captify/runner.rb",
+    "lib/captify/template.rb",
+    "lib/captify/template_bundle.rb",
+    "lib/captify/template_loader.rb",
+    "lib/captify/template_registrar.rb",
+    "spec/captify/cli_spec.rb",
+    "spec/captify/runner_spec.rb",
+    "spec/captify/template_bundle_spec.rb",
+    "spec/captify/template_loader_spec.rb",
+    "spec/captify/template_registrar_spec.rb",
+    "spec/captify/template_spec.rb",
+    "spec/captify_spec.rb",
+    "spec/data/files/Gemfile.test_captify",
+    "spec/data/files/Gemfile.test_captify.lock",
+    "spec/data/files/load_path/not-template/lib/not_template.rb",
+    "spec/data/files/load_path/template-without-lib/template_bundle.rb",
+    "spec/data/files/load_path/template1/lib/template_bundle.rb",
+    "spec/data/files/load_path/template2/lib/template_bundle.rb",
+    "spec/data/files/rubygems/gems/not-template-1.0.0/lib/not_template.rb",
+    "spec/data/files/rubygems/gems/template1-1.0.0/lib/template_bundle.rb",
+    "spec/data/files/rubygems/gems/template1-1.0.0/templates/rails-base/config/deploy.rb",
+    "spec/data/files/rubygems/gems/template1-1.0.0/templates/rails-base/config/deploy/production.rb",
+    "spec/data/files/rubygems/gems/template1-1.0.0/templates/rails-base/config/deploy/staging.rb",
+    "spec/data/files/rubygems/gems/template1-1.0.0/templates/sinatra-base/config/deploy.rb",
+    "spec/data/files/rubygems/gems/template2-1.0.0/lib/template_bundle.rb",
+    "spec/data/files/rubygems/specifications/not-template-1.0.0.gemspec",
+    "spec/data/files/rubygems/specifications/template1-1.0.0.gemspec",
+    "spec/data/files/rubygems/specifications/template2-1.0.0.gemspec",
+    "spec/helper.rb"
   ]
   s.homepage = "http://github.com/teohm/captify"
   s.licenses = ["MIT"]
@@ -43,17 +73,26 @@ Gem::Specification.new do |s|
       s.add_development_dependency(%q<rdoc>, ["~> 3.12"])
       s.add_development_dependency(%q<bundler>, ["~> 1.0.0"])
       s.add_development_dependency(%q<jeweler>, ["~> 1.8.3"])
+      s.add_development_dependency(%q<minitest>, ["~> 3.2.0"])
+      s.add_development_dependency(%q<debugger>, [">= 0"])
+      s.add_development_dependency(%q<debugger-pry>, [">= 0"])
     else
       s.add_dependency(%q<capistrano>, [">= 0"])
       s.add_dependency(%q<rdoc>, ["~> 3.12"])
       s.add_dependency(%q<bundler>, ["~> 1.0.0"])
       s.add_dependency(%q<jeweler>, ["~> 1.8.3"])
+      s.add_dependency(%q<minitest>, ["~> 3.2.0"])
+      s.add_dependency(%q<debugger>, [">= 0"])
+      s.add_dependency(%q<debugger-pry>, [">= 0"])
     end
   else
     s.add_dependency(%q<capistrano>, [">= 0"])
     s.add_dependency(%q<rdoc>, ["~> 3.12"])
     s.add_dependency(%q<bundler>, ["~> 1.0.0"])
     s.add_dependency(%q<jeweler>, ["~> 1.8.3"])
+    s.add_dependency(%q<minitest>, ["~> 3.2.0"])
+    s.add_dependency(%q<debugger>, [">= 0"])
+    s.add_dependency(%q<debugger-pry>, [">= 0"])
   end
 end
 

data/lib/captify.rb

@@ -0,0 +1,25 @@
+require 'captify/cli'
+require 'captify/template_loader'
+require 'captify/template_registrar'
+require 'captify/template'
+require 'captify/template_bundle'
+require 'captify/runner'
+
+module Captify
+  def self.run(argv=ARGV, env=ENV)
+    opts = Cli.new.parse_argv argv
+
+    puts `capify #{opts[:target_dir]}`
+
+    begin
+      Runner.new.run(
+        opts.delete(:template_name) || ENV['CAPTIFY_TEMPLATE'],
+        opts.delete(:target_dir),
+        opts
+      )
+    rescue ArgumentError => ex
+      puts ex.message
+    end
+  end
+end
+

data/lib/captify/cli.rb

@@ -0,0 +1,59 @@
+module Captify
+
+  # Capture inputs from command line interface.
+  #
+  class Cli
+
+    def initialize(kernel=Kernel)
+      @kernel = kernel
+    end
+
+    def parse_argv(argv)
+      options = {}
+      argv = argv.clone
+      parse_options! argv, options
+      parse_target_dir! argv, options
+      return options
+    rescue AbortException => ex
+      @kernel.abort "#{ex.message}"
+    rescue ExitException => ex
+      @kernel.puts "#{ex.message}"
+      @kernel.exit
+    end
+
+    private
+
+    class AbortException < StandardError; end
+    class ExitException < StandardError; end
+
+    def parse_target_dir!(argv, options)
+      if argv.empty?
+        raise AbortException, "Please specify a target dir."
+      elsif argv.length > 1
+        raise AbortException, "Too many target dirs; please specify only the dir to captify."
+      end
+
+      options[:target_dir] = argv.delete_at(0)
+    end
+
+    def parse_options!(argv, options)
+      opts = OptionParser.new do |opts|
+        opts.banner = "Usage: captify [options] directory"
+        opts.separator ""
+        opts.separator "Options:"
+        opts.on("-t", "--template TEMPLATE_NAME", 
+                "Specify the template to be used") do |template|
+          options[:template_name] = template
+        end
+        opts.on_tail("-h", "--help", "Show this message") do 
+          raise ExitException, opts.help
+        end
+      end
+      opts.parse! argv
+      return opts.help
+    rescue OptionParser::InvalidOption => ex
+      raise AbortException, "#{ex.message}\n\n#{opts.help}"
+    end
+
+  end
+end

data/lib/captify/runner.rb

@@ -0,0 +1,31 @@
+module Captify
+  class Runner
+
+    def initialize(template_loader=TemplateLoader.new, kernel=Kernel)
+      @template_loader = template_loader
+      @kernel = kernel
+      @default_options = {:load_path_only => false}
+    end
+
+    def run(template_name, target_dir, options={})
+      raise ArgumentError, "'#{target_dir}' does not exist." unless File.exist? target_dir
+      raise ArgumentError, "'#{target_dir}' is not a directory." unless File.directory? target_dir
+
+      options = @default_options.merge(options)
+
+      @template_loader.reload! options[:load_path_only]
+
+      template = @template_loader.find(template_name)
+      unless template
+        raise ArgumentError, "template not found: '#{template_name}'"
+      end
+
+      @kernel.puts "[apply] template '#{template_name}'"
+
+      logs = template.apply_to(target_dir)
+      logs.each{|msg| @kernel.puts msg}
+
+      @kernel.puts "[done] captified!"
+    end
+  end
+end

data/lib/captify/template.rb

@@ -1,73 +1,31 @@
 module Captify
   class Template
+    attr_accessor :name, :base_dir, :files
 
-    def self.find_files_in_load_path(glob)
-      $LOAD_PATH.map { |load_path|
-        Dir["#{File.expand_path glob, load_path}#{Gem.suffix_pattern}"]
-      }.flatten.select { |file| File.file? file.untaint }
-    end
-
-    def self.find_files_in_latest_gems(glob, prelease=false)
-      Gem::Specification.latest_specs(:prelease => prelease).map { |spec|
-        spec.matches_for_glob("#{glob}#{Gem.suffix_pattern}")
-      }.flatten
-    end
-
-    def self.find_latest_files(glob, check_load_path=true)
-      files = []
-      files = find_files_in_load_path(glob) if check_load_path
-      files.concat find_files_in_latest_gems(glob)
-      return files
-    end
-
-    def self.load_all
-     find_latest_files('captify_template').each do |path|
-       load path if File.exist? path
-     end
-    end
-
-    def self.template_files(template_name)
-      @@templates ||= {}
-      @@templates.fetch(template_name, [])
-    end
+    def apply_to(target_dir)
+      msg = []
+      files.each do |file|
+        target_file = File.join(target_dir, file)
+        prefix = File.exist?(target_file) ? "[overwrite]" : '[add]'
 
-    protected
+        FileUtils.mkdir_p File.dirname(target_file)
+        FileUtils.cp File.join(base_dir, file), target_file
 
-    def self.inherited(subclass)
-      file_path = extract_file_path caller
-      templates_path = get_templates_path file_path
-      dirs = find_all_template_dir templates_path
-
-      @@templates ||= {}
-      dirs.each do |dir|
-        name, files = extract_template_name_and_files dir
-        if @@templates[name].nil?
-          @@templates[name] = files
-        else
-          puts "[captify] warn: template `#{name}' already loaded, ignore `#{file_path}'"
-        end
+        msg << "#{prefix} writing '#{target_file}'"
       end
+      return msg
     end
 
-    def self.extract_template_name_and_files(dir)
-      name  = File.basename dir
-      files = Dir[ File.join(dir, '**', '*') ]
-      files = files.select{|path| File.file? path}
-
-      [name, files]
-    end
-
-    def self.extract_file_path(caller_value)
-      caller_value.first[/^[^:]+/]
-    end
+    def self.load_from_path(path)
+      return nil unless File.exists? path
 
-    def self.get_templates_path(file_path)
-      File.realdirpath File.join(File.dirname(file_path), '..', 'templates')
-    end
-
-    def self.find_all_template_dir(templates_path)
-      Dir[ File.join(templates_path, '*') ]
+      self.new.tap do |t|
+        t.name = File.basename path
+        t.base_dir = path
+        t.files = Dir[ File.join(path, '**', '*') ].
+          select{|item| File.file? item}.
+          map{|file| file.sub(path+"/", '')}
+      end
     end
   end
 end
-

data/lib/captify/template_bundle.rb

@@ -0,0 +1,38 @@
+require 'captify/template_registrar'
+
+module Captify
+  class TemplateBundle
+    def self.register
+      templates_path = find_templates_path(file_path(Kernel.caller.first))
+      TemplateRegistrar.instance.register_templates_in_dir templates_path
+    end
+
+    def self.find_templates_path(caller_file_path, root="/")
+      base_path = find_base_path(caller_file_path, root)
+      File.join(base_path, 'templates')
+    end
+
+    def self.find_base_path(caller_file_path, root="/")
+      base_path = File.dirname(File.expand_path(caller_file_path))
+      default_base_path = base_path
+
+      while base_path && File.directory?(base_path) && !File.exist?(File.join(base_path,'lib'))
+        parent = File.dirname base_path
+        if parent == root
+          base_path = default_base_path
+          break
+        end
+        base_path = parent
+      end
+
+      return base_path
+    end
+
+    private
+
+    def self.file_path(call_stack_item)
+      extract_file = /^[^:]+/
+      File.expand_path call_stack_item[extract_file]
+    end
+  end
+end

data/lib/captify/template_loader.rb

@@ -0,0 +1,46 @@
+module Captify
+  class TemplateLoader
+
+    def initialize(opts={})
+      @file_name_pattern   = opts.fetch(:file_name_pattern,   'template_bundle')
+      @file_suffix_pattern = opts.fetch(:file_suffix_pattern, Gem.suffix_pattern)
+
+      @load_path = opts.fetch(:load_path, $LOAD_PATH)
+      @gem_spec  = opts.fetch(:gem_spec,  Gem::Specification)
+      @kernel    = opts.fetch(:kernel,    Kernel)
+      @file      = opts.fetch(:file,      File)
+    end
+
+    def reload!(load_path_only=false)
+      latest_files(@file_name_pattern, load_path_only).each do |path|
+        @kernel.load path if @file.exist? path
+      end
+    end
+
+    def find(template_name)
+      Captify::TemplateRegistrar.instance.find template_name
+    end
+
+    private
+
+    def latest_files(glob, load_path_only)
+      files = []
+      files.concat find_load_path_files(glob)
+      files.concat find_latest_gem_files(glob) unless load_path_only
+      return files
+    end
+
+    def find_load_path_files(glob)
+      @load_path.map { |load_path|
+        Dir["#{@file.expand_path glob, load_path}#{@file_suffix_pattern}"]
+      }.flatten.select { |file| @file.file? file.untaint }
+    end
+
+    def find_latest_gem_files(glob)
+      @gem_spec.latest_specs.map { |spec|
+        spec.matches_for_glob("#{glob}#{@file_suffix_pattern}")
+      }.flatten
+    end
+  end
+end
+

data/lib/captify/template_registrar.rb

@@ -0,0 +1,44 @@
+require 'captify/template'
+
+module Captify
+  class TemplateRegistrar
+
+    def initialize(template_builder=Captify::Template)
+      @template_builder = template_builder
+      @templates = {}
+    end
+
+    @@instance = TemplateRegistrar.new
+
+    def self.instance
+      @@instance
+    end
+
+    def register_templates_in_dir(root_dir)
+      template_dirs(root_dir).each do |dir|
+        if template = @template_builder.load_from_path(dir)
+          register template
+        end
+      end
+    end
+
+    def template_count
+      @templates.count
+    end
+
+    def register(template)
+      @templates[template.name] = template unless @templates.key? template.name
+    end
+
+    def find(template_name)
+      @templates[template_name]
+    end
+
+    private
+
+    def template_dirs(root_path)
+      Dir[ File.join(File.expand_path(root_path), '*') ]
+    end
+
+  end
+end

data/spec/captify/cli_spec.rb

@@ -0,0 +1,45 @@
+require 'helper'
+
+describe Captify::Cli do
+  before do
+    @kernel = MiniTest::Mock.new
+    @cli = Captify::Cli.new(@kernel)
+  end
+  describe "Parse options" do
+    it "parses all options" do
+      opts = @cli.parse_argv %w(-t rails-base .)
+      opts.must_equal(
+        :target_dir => '.',
+        :template_name => 'rails-base'
+      )
+    end
+    it "without template_name" do
+      opts = @cli.parse_argv %w(.)
+      opts.must_equal(
+        :target_dir => '.'
+      )
+    end
+  end
+
+  describe "Exit exceptions" do
+    it "exits when display help" do
+      @kernel.expect :puts, nil, [String]
+      @kernel.expect :exit, nil, []
+      @cli.parse_argv %w(-h)
+      @kernel.verify
+    end
+  end
+
+  describe "Abort exceptions" do
+    it "aborts when missing target dir" do
+      @kernel.expect :abort, nil, ["Please specify a target dir."]
+      @cli.parse_argv %w(-t rails-base)
+      @kernel.verify
+    end
+    it "aborts when more than one target dir" do
+      @kernel.expect :abort, nil, ["Too many target dirs; please specify only the dir to captify."]
+      @cli.parse_argv %w(-t rails-base dir1 dir2)
+      @kernel.verify
+    end
+  end
+end

data/spec/captify/runner_spec.rb

@@ -0,0 +1,52 @@
+require 'helper'
+require 'tmpdir'
+
+describe Captify::Runner do
+  FILES_DIR = 'spec/data/files'
+  LOAD_PATH_DIR = "#{FILES_DIR}/load_path"
+  RUBYGEMS_DIR = "#{FILES_DIR}/rubygems"
+
+  before do
+    load_path = [ ]
+    gem_spec = Object.new
+    def gem_spec.latest_specs
+      [ Gem::Specification.load("#{RUBYGEMS_DIR}/specifications/template1-1.0.0.gemspec") ]
+    end
+
+    kernel = Object.new
+    def kernel.puts(msg)
+    end
+
+    @runner = Captify::Runner.new(
+      Captify::TemplateLoader.new(
+        :load_path => load_path,
+        :gem_spec => gem_spec
+      ),
+      kernel
+    )
+  end
+
+  it "can captify the target dir" do
+    Dir.mktmpdir do |temp_dir|
+      @runner.run('rails-base', temp_dir,
+        :load_path_only => false,
+        :force => false
+      )
+      File.exist?(File.join(temp_dir, 'config/deploy.rb')).must_equal true
+      File.exist?(File.join(temp_dir, 'config/deploy/staging.rb')).must_equal true
+      File.exist?(File.join(temp_dir, 'config/deploy/production.rb')).must_equal true
+    end
+  end
+
+  it "can handle when template not found" do
+    lambda { @runner.run('not-exist', 'not-exist') }.must_raise ArgumentError
+  end
+
+  it "throws error when target dir not exists" do
+    lambda { @runner.run('rails-base', 'not-exist') }.must_raise ArgumentError
+  end
+
+  it "throws error when target dir is not a directory" do
+    lambda { @runner.run('rails-base', __FILE__) }.must_raise ArgumentError
+  end
+end

data/spec/captify/template_bundle_spec.rb

@@ -0,0 +1,30 @@
+require 'helper'
+
+describe Captify::TemplateBundle do
+
+  describe "Subclass Captify::Template" do
+    it "finds templates/ dir in root path" do
+      caller_file_path = File.expand_path('spec/data/files/rubygems/gems/template1-1.0.0/lib/template_bundle.rb')
+      path = Captify::TemplateBundle.find_templates_path(caller_file_path, File.expand_path('.'))
+      path.must_equal File.expand_path('spec/data/files/rubygems/gems/template1-1.0.0/templates')
+    end
+
+    it "treats template_bundle.rb parent dir as root path if cannot find lib/ dir" do
+      caller_file_path = File.expand_path 'spec/data/files/load_path/template-without-lib/template_bundle.rb'
+      path = Captify::TemplateBundle.find_templates_path(caller_file_path, File.expand_path('.'))
+      path.must_equal File.expand_path('spec/data/files/load_path/template-without-lib/templates')
+    end
+
+    it "passes dirname correctly into Template Registrar." do
+      received_path = ""
+      Captify::TemplateRegistrar.instance.stub :register_templates_in_dir,
+                            lambda{|path| received_path = path} do
+
+        load 'spec/data/files/load_path/template1/lib/template_bundle.rb'
+
+      end
+      received_path.must_equal File.expand_path('spec/data/files/load_path/template1/templates')
+
+    end
+  end
+end

data/spec/captify/template_loader_spec.rb

@@ -0,0 +1,57 @@
+require 'helper'
+
+describe Captify::TemplateLoader do
+
+  before do
+    @file_dir = 'spec/data/files'
+    @load_path_dir = "#{@file_dir}/load_path"
+    @rubygems_dir = "#{@file_dir}/rubygems"
+
+    @load_path = [
+      "#{@load_path_dir}/template1/lib",
+      "#{@load_path_dir}/template2/lib",
+      "#{@load_path_dir}/not-template/lib"
+    ]
+    @gem_spec = MiniTest::Mock.new
+    @gem_spec.expect :latest_specs, [
+      Gem::Specification.load("#{@rubygems_dir}/specifications/template1-1.0.0.gemspec"),
+      Gem::Specification.load("#{@rubygems_dir}/specifications/template2-1.0.0.gemspec"),
+      Gem::Specification.load("#{@rubygems_dir}/specifications/not-template-1.0.0.gemspec")
+    ]
+    @kernel = MiniTest::Mock.new
+
+    @loader = Captify::TemplateLoader.new(
+      :load_path => @load_path,
+      :gem_spec  => @gem_spec,
+      :kernel    => @kernel)
+  end
+
+  it "can load captify_template.rb from load paths, then from latest gems." do
+    [
+      "#{@load_path_dir}/template1",
+      "#{@load_path_dir}/template2",
+      "#{@rubygems_dir}/gems/template1-1.0.0",
+      "#{@rubygems_dir}/gems/template2-1.0.0"
+    ].each do |path|
+      @kernel.expect :load, nil, [File.expand_path("#{path}/lib/template_bundle.rb")]
+    end
+
+    @loader.reload!
+
+    @kernel.verify
+  end
+
+  it "can load captify_template.rb from load paths only." do
+    load_path_only = true
+    [
+      "#{@load_path_dir}/template1",
+      "#{@load_path_dir}/template2"
+    ].each do |path|
+      @kernel.expect :load, nil, [File.expand_path("#{path}/lib/template_bundle.rb")]
+    end
+
+    @loader.reload! load_path_only
+
+    @kernel.verify
+  end
+end

data/spec/captify/template_registrar_spec.rb

@@ -0,0 +1,46 @@
+require 'helper'
+
+describe "Register template" do
+
+  it "can register all templates under templates/" do
+
+    templates_path = 'spec/data/files/rubygems/gems/template1-1.0.0/templates'
+    template_builder = MiniTest::Mock.new
+    registrar = Captify::TemplateRegistrar.new(template_builder)
+
+    fake_template1 = Captify::Template.new.tap do |t|
+      t.name = 'rails-base'
+    end
+    fake_template2 = Captify::Template.new.tap do |t|
+      t.name = 'sinatra-base'
+    end
+    fake_template3 = Captify::Template.new.tap do |t|
+      t.name = 'no-file'
+    end
+    template_builder.expect :load_from_path, fake_template3,
+      [File.expand_path(File.join(templates_path, 'no-file'))]
+    template_builder.expect :load_from_path, fake_template1,
+      [File.expand_path(File.join(templates_path, 'rails-base'))]
+    template_builder.expect :load_from_path, fake_template2,
+      [File.expand_path(File.join(templates_path, 'sinatra-base'))]
+
+    registrar.register_templates_in_dir templates_path 
+
+    template_builder.verify
+    registrar.template_count.must_equal 3
+    registrar.find('rails-base').must_be_same_as fake_template1
+    registrar.find('sinatra-base').must_be_same_as fake_template2
+    registrar.find('no-file').must_be_same_as fake_template3
+  end
+
+  it "can handle when templates/ does not exist." do
+
+    templates_path = 'spec/data/files/rubygems/gems/template2-1.0.0/templates'
+    registrar = Captify::TemplateRegistrar.new
+
+    registrar.register_templates_in_dir templates_path
+
+    registrar.template_count.must_equal 0
+  end
+
+end

data/spec/captify/template_spec.rb

@@ -0,0 +1,60 @@
+require 'helper'
+require 'tmpdir'
+
+describe Captify::Template do
+  it "can create new template by parsing a dir path." do
+    template_path = 'spec/data/files/rubygems/gems/template1-1.0.0/templates/rails-base'
+
+    template = Captify::Template.load_from_path template_path
+
+    template.name.must_equal "rails-base"
+    template.base_dir.must_equal template_path
+    template.files.sort.must_equal [
+      'config/deploy.rb',
+      'config/deploy/staging.rb',
+      'config/deploy/production.rb'
+    ].sort
+  end
+
+  it "can handle when template dir path empty." do
+    template_path = 'spec/data/files/rubygems/gems/template1-1.0.0/templates/no-file'
+
+    template = Captify::Template.load_from_path template_path
+
+    template.name.must_equal "no-file"
+    template.base_dir.must_equal template_path
+    template.files.must_equal []
+  end
+
+  it "can apply template to target dir" do
+    template_path = 'spec/data/files/rubygems/gems/template1-1.0.0/templates/rails-base'
+    template = Captify::Template.load_from_path template_path
+
+    Dir.mktmpdir do |temp_dir|
+      template.apply_to temp_dir
+
+      File.file?(File.join(temp_dir, 'config/deploy.rb')).must_equal true
+      File.file?(File.join(temp_dir, 'config/deploy/staging.rb')).must_equal true
+      File.file?(File.join(temp_dir, 'config/deploy/production.rb')).must_equal true
+    end
+  end
+
+  it "always overwrites existing files in target dir" do
+    template_path = 'spec/data/files/rubygems/gems/template1-1.0.0/templates/rails-base'
+    template = Captify::Template.load_from_path template_path
+
+    Dir.mktmpdir do |temp_dir|
+
+      deploy_rb = File.join(temp_dir, 'config/deploy.rb')
+      FileUtils.mkdir_p File.dirname(deploy_rb)
+      File.open(deploy_rb, 'w') {|f| f.write("test") }
+
+      template.apply_to temp_dir
+
+      File.open(deploy_rb) {|f| f.read }.wont_equal "test"
+      File.file?(File.join(temp_dir, 'config/deploy.rb')).must_equal true
+      File.file?(File.join(temp_dir, 'config/deploy/staging.rb')).must_equal true
+      File.file?(File.join(temp_dir, 'config/deploy/production.rb')).must_equal true
+    end
+  end
+end

data/spec/captify_spec.rb

@@ -0,0 +1,18 @@
+require 'helper'
+require 'tmpdir'
+
+describe Captify do
+  it "can run captify" do
+    Dir.mktmpdir do |tmp_dir|
+      ENV['CAPTIFY_TEMPLATE'] = 'rails-base'
+      ENV['BUNDLE_GEMFILE'] = 'spec/data/files/Gemfile.test_captify'
+
+      `bundle exec ./bin/captify #{tmp_dir}`
+
+      File.exist?(File.join(tmp_dir, 'Capfile')).must_equal true
+      File.exist?(File.join(tmp_dir, 'config/deploy.rb')).must_equal true
+      File.exist?(File.join(tmp_dir, 'config/deploy/staging.rb')).must_equal true
+      File.exist?(File.join(tmp_dir, 'config/deploy/production.rb')).must_equal true
+    end
+  end
+end

data/spec/data/files/Gemfile.test_captify

@@ -0,0 +1,2 @@
+gem 'template2-1.0.0', '1.0.0', :path => 'rubygems/gems/template1-1.0.0'
+gem 'captify', :path => '../../..'

data/spec/data/files/Gemfile.test_captify.lock

@@ -0,0 +1,34 @@
+PATH
+  remote: ../../..
+  specs:
+    captify (0.2.2)
+      capistrano
+
+PATH
+  remote: rubygems/gems/template1-1.0.0
+  specs:
+    template2-1.0.0 (1.0.0)
+
+GEM
+  specs:
+    capistrano (2.12.0)
+      highline
+      net-scp (>= 1.0.0)
+      net-sftp (>= 2.0.0)
+      net-ssh (>= 2.0.14)
+      net-ssh-gateway (>= 1.1.0)
+    highline (1.6.13)
+    net-scp (1.0.4)
+      net-ssh (>= 1.99.1)
+    net-sftp (2.0.5)
+      net-ssh (>= 2.0.9)
+    net-ssh (2.5.2)
+    net-ssh-gateway (1.1.0)
+      net-ssh (>= 1.99.1)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  captify!
+  template2-1.0.0 (= 1.0.0)!

data/spec/data/files/load_path/template-without-lib/template_bundle.rb

@@ -0,0 +1 @@
+Captify::TemplateBundle.register

data/spec/data/files/load_path/template1/lib/template_bundle.rb

@@ -0,0 +1 @@
+Captify::TemplateBundle.register

data/spec/data/files/load_path/template2/lib/template_bundle.rb

@@ -0,0 +1 @@
+Captify::TemplateBundle.register

data/spec/data/files/rubygems/gems/template1-1.0.0/lib/template_bundle.rb

@@ -0,0 +1 @@
+Captify::TemplateBundle.register

data/spec/data/files/rubygems/gems/template2-1.0.0/lib/template_bundle.rb

@@ -0,0 +1 @@
+Captify::TemplateBundle.register

data/spec/data/files/rubygems/specifications/not-template-1.0.0.gemspec

@@ -0,0 +1,15 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = "not-template"
+  s.version = "1.0.0"
+
+  s.files = [
+    "lib/not_template.rb"
+  ]
+  s.require_paths = ["lib"]
+  s.rubygems_version = "1.8.24"
+  s.summary = "Not Template"
+
+end
+

data/spec/data/files/rubygems/specifications/template1-1.0.0.gemspec

@@ -0,0 +1,15 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = "template1"
+  s.version = "1.0.0"
+
+  s.files = [
+    "lib/template_bundle.rb"
+  ]
+  s.require_paths = ["lib"]
+  s.rubygems_version = "1.8.24"
+  s.summary = "Template 1"
+
+end
+

data/spec/data/files/rubygems/specifications/template2-1.0.0.gemspec

@@ -0,0 +1,15 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = "template2"
+  s.version = "1.0.0"
+
+  s.files = [
+    "lib/template_bundle.rb"
+  ]
+  s.require_paths = ["lib"]
+  s.rubygems_version = "1.8.24"
+  s.summary = "Template 2"
+
+end
+

data/spec/helper.rb

@@ -0,0 +1,20 @@
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'minitest/spec'
+require 'minitest/autorun'
+require 'minitest/pride'
+
+require 'debugger'
+require 'debugger/pry'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'captify'
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: captify
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 1.0.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-06-12 00:00:00.000000000 Z
+date: 2012-07-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: capistrano
@@ -75,6 +75,54 @@ dependencies:
     - - ~>
       - !ruby/object:Gem::Version
         version: 1.8.3
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 3.2.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 3.2.0
+- !ruby/object:Gem::Dependency
+  name: debugger
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: debugger-pry
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Capistrano capify with canned templates.
 email: teohuiming@gmail.com
 executables:
@@ -93,7 +141,37 @@ files:
 - VERSION
 - bin/captify
 - captify.gemspec
+- lib/captify.rb
+- lib/captify/cli.rb
+- lib/captify/runner.rb
 - lib/captify/template.rb
+- lib/captify/template_bundle.rb
+- lib/captify/template_loader.rb
+- lib/captify/template_registrar.rb
+- spec/captify/cli_spec.rb
+- spec/captify/runner_spec.rb
+- spec/captify/template_bundle_spec.rb
+- spec/captify/template_loader_spec.rb
+- spec/captify/template_registrar_spec.rb
+- spec/captify/template_spec.rb
+- spec/captify_spec.rb
+- spec/data/files/Gemfile.test_captify
+- spec/data/files/Gemfile.test_captify.lock
+- spec/data/files/load_path/not-template/lib/not_template.rb
+- spec/data/files/load_path/template-without-lib/template_bundle.rb
+- spec/data/files/load_path/template1/lib/template_bundle.rb
+- spec/data/files/load_path/template2/lib/template_bundle.rb
+- spec/data/files/rubygems/gems/not-template-1.0.0/lib/not_template.rb
+- spec/data/files/rubygems/gems/template1-1.0.0/lib/template_bundle.rb
+- spec/data/files/rubygems/gems/template1-1.0.0/templates/rails-base/config/deploy.rb
+- spec/data/files/rubygems/gems/template1-1.0.0/templates/rails-base/config/deploy/production.rb
+- spec/data/files/rubygems/gems/template1-1.0.0/templates/rails-base/config/deploy/staging.rb
+- spec/data/files/rubygems/gems/template1-1.0.0/templates/sinatra-base/config/deploy.rb
+- spec/data/files/rubygems/gems/template2-1.0.0/lib/template_bundle.rb
+- spec/data/files/rubygems/specifications/not-template-1.0.0.gemspec
+- spec/data/files/rubygems/specifications/template1-1.0.0.gemspec
+- spec/data/files/rubygems/specifications/template2-1.0.0.gemspec
+- spec/helper.rb
 homepage: http://github.com/teohm/captify
 licenses:
 - MIT
@@ -109,7 +187,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -4128769129857961565
+      hash: -177249211097526545
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements: