move stage and DSL to setup

lib/capistrano/setup.rb

@@ -0,0 +1,10 @@
+include Capistrano::DSL
+
+stages.each do |stage|
+  Rake::Task.define_task(stage) do
+    load "config/deploy/#{stage}.rb"
+    set(:stage, stage.to_sym)
+    configure_ssh_kit
+  end
+end
+

lib/capistrano/tasks/deploy.rake

@@ -1,14 +1,3 @@
-include Capistrano::DSL
-
-
-stages.each do |stage|
-  Rake::Task.define_task(stage) do
-    load "config/deploy/#{stage}.rb"
-    set(:stage, stage.to_sym)
-    configure_ssh_kit
-  end
-end
-
 namespace :deploy do
 
   desc "starting"

lib/capistrano/templates/Capfile

@@ -1,4 +1,7 @@
 #!/usr/bin/env cap
+# loads dsl, sets up stages
+require 'capistrano/setup'
+
 # require bundled deployment tasks
 # remove to start from scratch
 require 'capistrano/deploy'

lib/capistrano/templates/Capfile.erb

@@ -1,15 +0,0 @@
-#!/usr/bin/env cap
-require 'capistrano/install'
-
-# Loads configuation
-load 'config/deploy.rb'
-
-# require bundled deployment tasks
-require 'capistrano/deploy'
-
-# require tasks from contrib gems
-# require 'capistrano/rails'
-# require 'capistrano/heroku'
-
-# Loads any rake tasks from lib/deploy/tasks
-Dir.glob('lib/deploy/tasks/*.rake').each { |r| import r }