lib/jeweler/generator.rb in namelessjon-jeweler-0.6.1 vs lib/jeweler/generator.rb in namelessjon-jeweler-0.6.2

@@ -1,8 +1,11 @@
 require 'git'
 require 'erb'
 
+require 'net/http'
+require 'uri'
+
 class Jeweler
   class NoGitUserName < StandardError
   end
   class NoGitUserEmail < StandardError
   end
@@ -10,18 +13,20 @@
   end
   class NoGitHubRepoNameGiven < StandardError
   end
   class NoGitHubUser < StandardError
   end
-  class GitInitFailed < StandardError
+  class NoGitHubToken < StandardError
   end
-    
+  class GitInitFailed < StandardError
+  end    
 
-  class Generator
+  class Generator    
     attr_accessor :target_dir, :user_name, :user_email,
-                  :github_repo_name, :github_remote, :github_url, :github_username,
-                  :lib_dir, :constant_name, :file_name_prefix, :config, :test_style
+                  :github_repo_name, :github_remote, :github_url, :github_username, :github_token,
+                  :lib_dir, :constant_name, :file_name_prefix, :config, :test_style,
+                  :repo, :should_create_repo
 
     def initialize(github_repo_name, options = {})
       check_user_git_config()
       
       if github_repo_name.nil?
@@ -35,15 +40,21 @@
       self.test_style = options[:test_style] || :shoulda
       self.target_dir = options[:directory] || self.github_repo_name
       self.lib_dir = File.join(target_dir, 'lib')
       self.constant_name = self.github_repo_name.split(/[-_]/).collect{|each| each.capitalize }.join
       self.file_name_prefix = self.github_repo_name.gsub('-', '_')
+      self.should_create_repo = options[:create_repo]
     end
 
     def run
       create_files
       gitify
+      puts "Jeweler has prepared your gem in #{github_repo_name}"
+      if should_create_repo
+        create_and_push_repo
+        puts "Jeweler has pushed your repo to #{github_url}"
+      end
     end
 
     def testspec
       case test_style
       when :shoulda
@@ -71,12 +82,12 @@
 
       output_template_in_target('.gitignore')
       output_template_in_target('Rakefile')
       output_template_in_target('LICENSE')
       output_template_in_target('README')
-      output_template_in_target("#{testspec}/#{testspec}_helper.rb")
-      output_template_in_target("#{testspec}/flunking_#{testspec}.rb", "#{testspec}/#{file_name_prefix}_#{testspec}.rb")
+      output_template_in_target("#{test_style}/#{testspec}_helper.rb", "#{testspec}/#{testspec}_helper.rb")
+      output_template_in_target("#{test_style}/flunking_#{testspec}.rb", "#{testspec}/#{file_name_prefix}_#{testspec}.rb")
 
       FileUtils.touch File.join(lib_dir, "#{file_name_prefix}.rb")
     end
 
     def check_user_git_config
@@ -91,14 +102,19 @@
       end
       
       unless config.has_key? 'github.user'
         raise NoGitHubUser, %Q{No github.user set in ~/.gitconfig. Set it with: git config --global github.user 'Your username here'}
       end
+      
+      unless config.has_key? 'github.token'
+        raise NoGitHubToken, %Q{No github.token set in ~/.gitconfig. Set it with: git config --global github.token 'Your token here'}
+      end
 
       self.user_name = config['user.name']
       self.user_email = config['user.email']
       self.github_username = config['github.user']
+      self.github_token = config['github.token']
     end
 
     def output_template_in_target(source, destination = source)
       template = ERB.new(File.read(File.join(File.dirname(__FILE__), 'templates', source)))
 
@@ -108,36 +124,45 @@
     def gitify
       saved_pwd = Dir.pwd
       Dir.chdir(target_dir)
       begin
         begin
-          repo = Git.init()
+          @repo = Git.init()
         rescue Git::GitExecuteError => e
           raise GitInitFailed, "Encountered an error during gitification. Maybe the repo already exists, or has already been pushed to?"
         end
 
         begin
-          repo.add('.')
+          @repo.add('.')
         rescue Git::GitExecuteError => e
           #raise GitAddFailed, "There was some problem adding this directory to the git changeset"
           raise
         end
 
         begin
-          repo.commit "Initial commit to #{github_repo_name}."
+          @repo.commit "Initial commit to #{github_repo_name}."
         rescue Git::GitExecuteError => e
           raise
         end
 
         begin
-          repo.add_remote('origin', github_remote)
+          @repo.add_remote('origin', github_remote)
         rescue Git::GitExecuteError => e
           puts "Encountered an error while adding origin remote. Maybe you have some weird settings in ~/.gitconfig?"
           raise
         end
       ensure
         Dir.chdir(saved_pwd)
       end
+    end
+    
+    def create_and_push_repo
+      Net::HTTP.post_form URI.parse('http://github.com/repositories'),
+                                'login' => github_username,
+                                'token' => github_token,
+                                'repository[name]' => github_repo_name
+      sleep 2
+      @repo.push('origin')
     end
 
     def read_git_config
       # we could just use Git::Base's .config, but that relies on a repo being around already
       # ... which we don't have yet, since this is part of a sanity check