lib/microstatic/cli.rb in microstatic-0.4.0 vs lib/microstatic/cli.rb in microstatic-0.4.1

@@ -1,20 +1,23 @@
 require 'thor'
 require 'rainbow'
 
 module Microstatic
   class CLI < Thor
+    include Thor::Actions
+
+    source_root( File.expand_path( "../../../templates", __FILE__ ) )
+
     desc "setup <APP_NAME>", "do the needful to get started hosting your static site. S3, Route 53, whatever it takes."
     def setup( app_name = false )
       app_name ||= inferred_app_name
       bucket_name = subdomain_for(app_name)
 
       bucket = bucket( bucket_name )
       # TODO: pass bucket through to dns setup so we don't have to look it up again
       dns( app_name )
-
-      # TODO: add Rakefile, Gemfile, etc for doing a rake deploy using this gem
+      rakefile( bucket_name )
     end
 
     desc "bucket <BUCKET_NAME>", "create an S3 bucket which can host your static site"
     def bucket( bucket_name = false )
       bucket_name ||= guess_bucket_name
@@ -24,29 +27,46 @@
       # TODO: fail gracefully if aws creds not available
       
       describe_operation( "create S3 bucket '#{bucket_name}'" ) do
         S3BucketCreator.new( config.aws_creds ).create( bucket_name )
       end
+    rescue Microstatic::Config::MissingEnvVar
+      raise Thor::Error.new($!)
     end
 
     desc "dns <APP_NAME>", "create a Route 53 entry pointing to the S3 bucket containing your static site"
     def dns( app_name = false )
       app_name ||= inferred_app_name
       bucket_name = subdomain_for(app_name)
+      #
+      # TODO: handle the subdomain DNS record already existing
 
       describe_operation( "create Route 53 entry '#{bucket_name}'" ) do
         Route53Dns.new( config.aws_creds ).add_s3_record_for_bucket( bucket_name )
       end
+
+    rescue Microstatic::Config::MissingEnvVar
+      raise Thor::Error.new($!)
     end
 
+    desc "rakefile <BUCKET_NAME>", "create a simple Rakefile that will deploy to the specified bucket"
+    def rakefile( bucket_name = false )
+      bucket_name ||= guess_bucket_name
+      # TODO: check if Rakefile already exists
+      template( "Rakefile.erb", "Rakefile", :bucket_name => bucket_name )
+    end
+
+
     private 
 
     def guess_bucket_name
       subdomain_for( inferred_app_name )
     end
 
     def subdomain_for( app_name )
       "#{app_name}.#{config.site_name}"
+    rescue Microstatic::Config::MissingEnvVar
+      raise Thor::Error.new($!)
     end
 
     def inferred_app_name
       File.basename(Dir.pwd)
     end