github-copywriter 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 15cd7808bbe3d197e2893024b7f32088e7a0cbab
-  data.tar.gz: 292896e21d05194b9a9309b8f40df956ebdf45dc
+  metadata.gz: 28edcf28a7ba50ed82f9abe685af12fc4da9da92
+  data.tar.gz: ab8737a3e5db04074640029084f557b706ad05b4
 SHA512:
-  metadata.gz: 4289d219d03440699343066c243d7b42649feac8529fcf2302d802b769de82477b5a942ef06bf3c55a8a227630b5af855105c8384f18ac536ae2fc2db9f87870
-  data.tar.gz: de1363e6c37fe08bc4c5d9308fecc4d39b935a670cc8ba55ede33b37b1a77b435e30278139d1368786acbbf65badacda5117b83ce5e548015093dc1e51db572b
+  metadata.gz: 92201c3e9cb8988113b394a17fbf96e8c40865fddb1961dce9451be45c0bf0358296d5ba8a3e6ad29197bb518bdedf162b8bfd084194965f05b703374e3ef332
+  data.tar.gz: a6f1efaee70f5451706d3d663daa74bf7f159b743650520b31a89ea37c424cd8a8d8109628ae75dc840100c87ed923cd135cdc948cfcd9440a3144680f357e7a

data/README.md

@@ -24,7 +24,7 @@ Questions and comments are welcome as well. Checkout
 [CONTRIBUTING.md](//github.com/ryanmjacobs/github-copywriter/blob/master/CONTRIBUTING.md)
 for more info.
 
-Feel free to [fork](//github.com/ryanmjacobs/github-copywriter) this project, and:
+Feel free to [fork](//github.com/ryanmjacobs/github-copywriter/fork) this project, and:
 
 * fix bugs
 * add some sweet features

data/bin/github-copywriter

@@ -24,7 +24,7 @@ Usage: #{executable_name} [repos...]
 
     EOS
 
-    opts.on("--all", "Update copyrights on ALL repositories.") do
+    opts.on("-a", "--all", "Update copyrights on ALL repositories.") do
         options[:all] = true
     end
 
@@ -33,7 +33,20 @@ Usage: #{executable_name} [repos...]
     end
 
     opts.on("-y", "--year YEAR", "Update copyrights to a give year.") do |year|
-        options[:year] = year
+        if year.to_i >= 1000 && year.to_i <= 9999 then
+            options[:year] = year
+        else
+            puts "error: year must be four digits".red
+            exit
+        end
+    end
+
+    opts.on("-b", "--branches branch1,branch2", "Update specific branches.") do |comma_separated_branches|
+        options[:branches] = comma_separated_branches.split(",").map(&:strip)
+    end
+
+    opts.on("--all-branches", "Update all branches.") do
+        options[:all_branches] = true
     end
 
     opts.on_tail("-h","--help", "Show this message.") do
@@ -49,7 +62,7 @@ end.parse!
 
 # Must have at least one repo!
 if ARGV.empty? and not options[:all]
-    puts "error: No repositories supplied!"
+    puts "error: No repositories supplied!".red
     exit
 end
 

data/lib/github-copywriter/version.rb

@@ -8,6 +8,6 @@ class Copywriter
 
     # Current version
     # @return [String]
-    VERSION = "0.1.0".freeze
+    VERSION = "0.1.1".freeze
 
 end

data/lib/github-copywriter.rb

@@ -32,7 +32,13 @@ class Copywriter
 
     def run!(options={})
         # Default options
-        options = {all: false, skip_forks: false, year: nil}.merge(options)
+        options = {
+            all: false,
+            skip_forks: false,
+            year: nil,
+            branches: nil,
+            all_branches: false
+        }.merge(options)
 
         if options[:all] then
             repos = @client.repositories()
@@ -58,56 +64,78 @@ class Copywriter
             # Skip if repo is a fork and --skip-forks is on
             next if options[:skip_forks] and repo[:fork]
 
-            # Get repo info
-            repo_name     = repo[:full_name]
-            ref           = "heads/#{repo[:default_branch]}"
-            latest_commit = @client.ref(repo_name, ref).object
-            root_tree     = @client.commit(repo_name, latest_commit.sha).commit.tree
-            puts "\n"+repo_name+":"
-
-            # Grab the *whole* tree
-            tree = @client.tree(repo_name, root_tree.sha, :recursive => true)
-
-            # warn user about truncation
-            if tree[:truncated] then
-                puts "  warning: tree truncated because number of items exceeded limit.".yellow
-                puts "           If you feel like fixing this, see issue #xx".yellow
-                puts "           http://github.com/ryanmjacobs/github-copywriter/xx".yellow
-            end
+            # Repo name
+            repo_name = repo[:full_name]
+            puts "\n#{repo_name}:".cyan
 
-            # Stores updated files until we commit
-            # @modified_files is a hash {:path, :content}
-            @modified_files = Array.new
+            # Repo refs (branches) to loop through
+            if options[:branches] != nil then
+                refs = options[:branches].map { |branch| "heads/#{branch}" }
+            else
+                refs = "heads/#{repo[:default_branch]}"
+            end
 
-            # Loop through all of the tree's blobs
-            tree[:tree].each do |file|
-                next if file[:type] != "blob"
+            # Use all branches if we have the flag
+            if options[:all_branches] then
+                # Get all refs of type "heads"; remove leading "refs/"
+                refs = @client.refs(repo_name, "heads").map { |r| r[:ref].sub("refs/", "") }
+            end
 
-                if Copywriter::Regex.accepted_name?(file[:path]) then
-                    # Grab file from repo
-                    file = @client.contents(repo_name, :path => file[:path])
-                    if file[:type] != "file" then raise "error: #{file_path} on #{repo} is not a file!" end
+            # Loop through each ref
+            refs.each do |ref|
+                puts "  #{ref}:".yellow
+
+                # Grab the *whole* tree
+                begin
+                    latest_commit = @client.ref(repo_name, ref).object
+                    root_tree     = @client.commit(repo_name, latest_commit.sha).commit.tree
+                    whole_tree    = @client.tree(repo_name, root_tree.sha, :recursive => true)
+                rescue Octokit::NotFound
+                    puts "error: ref #{ref} not found.".red
+                    exit
+                end
 
-                    # Update the copyright
-                    new_content = Copywriter::Regex.update_copyright(cur_year, Base64.decode64(file[:content]))
+                # Warn the user about truncation
+                if whole_tree[:truncated] then
+                    puts "    warning: tree truncated because number of items exceeded limit.".yellow
+                    puts "             If you feel like fixing this, see issue #xx".yellow
+                    puts "             http://github.com/ryanmjacobs/github-copywriter/xx".yellow
+                end
 
-                    # Add to list of files to commit, only if the file has changed
-                    if new_content != nil then
-                        @modified_files << {:path => file[:path], :content => new_content}
-                        puts "  #{file[:path]} is now up-to-date.".green
-                    else
-                        puts "  #{file[:path]} is already up-to-date."
+                # Stores updated files until we commit
+                # @modified_files is a hash {:path, :content}
+                @modified_files = Array.new
+
+                # Loop through all of whole_tree's blobs
+                whole_tree[:tree].each do |file|
+                    next if file[:type] != "blob"
+
+                    if Copywriter::Regex.accepted_name?(file[:path]) then
+                        # Grab file from repo
+                        file = @client.contents(repo_name, :path => file[:path], :ref => ref)
+                        if file[:type] != "file" then raise "error: #{file_path} on #{repo} is not a file!" end
+
+                        # Update the copyright
+                        new_content = Copywriter::Regex.update_copyright(cur_year, Base64.decode64(file[:content]))
+
+                        # Add to list of files to commit, only if the file has changed
+                        if new_content != nil then
+                            @modified_files << {:path => file[:path], :content => new_content}
+                            puts "    #{file[:path]} is now up-to-date.".green
+                        else
+                            puts "    #{file[:path]} is already up-to-date."
+                        end
                     end
                 end
-            end
 
-            # Commit stored up files
-            if @modified_files.size > 0 then
-                print "  Committing #{@modified_files.size} files..."
-                commit_files(repo_name, ref, "100644", @modified_files, COMMIT_MSG)
-                puts " done"
-            else
-                puts "  No files needed to be commited."
+                # Commit stored up files
+                if @modified_files.size > 0 then
+                    print "    Committing #{@modified_files.size} files..."
+                    commit_files(repo_name, ref, "100644", @modified_files, COMMIT_MSG)
+                    puts " done"
+                else
+                    puts "    No files needed to be commited."
+                end
             end
         end
     end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: github-copywriter
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Ryan Jacobs
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-12-15 00:00:00.000000000 Z
+date: 2014-12-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler