commitshots 0.0.8 → 0.0.9

data/lib/commitshots.rb

@@ -1,45 +1,22 @@
-require 'optparse'
+require 'commitshots/options'
+require 'commitshots/backends/git'
+require 'commitshots/backends/hg'
 require 'commitshots/commits'
 
-options = {}
-
-optparse = OptionParser.new do|opts|
-   # Set a banner, displayed at the top
-   # of the help screen.
-   opts.banner = "Usage: commitshots [OPTIONS] url"
-
-
-   options[:width] = 800
-   opts.on( '--width WIDTH', 'Width of the browser window' ) do |width|
-     options[:width] = width
-   end
-
-   options[:height] = 600
-   opts.on( '--height HEIGHT', 'Height of the browser window' ) do |height|
-     options[:height] = height
-   end
-
-   # This displays the help screen, all programs are
-   # assumed to have this option.
-   opts.on( '-h', '--help', 'Display this screen' ) do
-     puts opts
-     exit
-   end
- end
-
-optparse.parse!
-
+options = Commitshots::Options.get_commandline_options
 phantomScript = File.join(File.dirname(__FILE__), 'phantomjs/screenshot.js')
 url = ARGV[0]
 width = options[:width]
 height = options[:height]
+backend = Commitshots::Backends.const_get(options[:backend].capitalize).new
+
 
 num = 0
 baseFileName = 'screenshots/image-'
 fileType = 'png'
 waitTime = 0.5
 
-Commitshots::Commits.new.each do |id, msg|
+Commitshots::Commits.new(backend).each do |id, msg|
   sleep(waitTime)
   fileName = "#{baseFileName}#{"%08d" % num}.#{fileType}"
   `phantomjs #{phantomScript} "#{url}" "#{fileName}" #{width} #{height}`

data/lib/commitshots/backends/git.rb

@@ -0,0 +1,14 @@
+module Commitshots
+  module Backends
+    class Git
+      def get_revisions
+        `git rev-list --reverse --all --pretty=oneline`.split("\n")
+      end
+
+      def checkout_revision(id)
+        `git checkout #{id}`
+      end
+    end
+  end
+end
+

data/lib/commitshots/backends/hg.rb

@@ -0,0 +1,14 @@
+module Commitshots
+  module Backends
+    class Hg
+      def get_revisions
+
+      end
+
+      def checkout_revision(id)
+
+      end
+    end
+  end
+end
+

data/lib/commitshots/commits.rb

@@ -1,11 +1,15 @@
 module Commitshots
   class Commits
+    def initialize(backend)
+      @backend = backend
+    end
+
     def each
-      `git rev-list --reverse --all --pretty=oneline`.split("\n").each do |line|
+      @backend.get_revisions.each do |line|
         parts = line.split(' ')
         id = parts[0]
         msg = line.gsub(id, '')
-        `git checkout #{id}`
+        @backend.checkout_revision(id)
         yield id, msg
       end
     end

data/lib/commitshots/options.rb

@@ -0,0 +1,41 @@
+require 'optparse'
+
+module Commitshots
+  class Options
+    def self.get_commandline_options
+      options = {}
+
+      optparse = OptionParser.new do|opts|
+         # Set a banner, displayed at the top
+         # of the help screen.
+         opts.banner = "Usage: commitshots [OPTIONS] url"
+
+
+         options[:width] = 800
+         opts.on( '--width WIDTH', "Width of the browser window, defaults to #{options[:width]}" ) do |width|
+           options[:width] = width
+         end
+
+         options[:height] = 600
+         opts.on( '--height HEIGHT', "Height of the browser window, defaults to #{options[:height]}" ) do |height|
+           options[:height] = height
+         end
+
+         options[:backend] = 'git'
+         opts.on( '--backend [git|hg]', "Backend to use, defaults to #{options[:backend]}") do |backend|
+           options[:backend] = backend
+         end
+
+         # This displays the help screen, all programs are
+         # assumed to have this option.
+         opts.on( '-h', '--help', 'Display this screen' ) do
+           puts opts
+           exit
+         end
+       end
+
+      optparse.parse!
+      options
+    end
+  end
+end

data/lib/commitshots/version.rb

@@ -1,4 +1,4 @@
 module Commitshots
-  VERSION = "0.0.8"
+  VERSION = "0.0.9"
 end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: commitshots
 version: !ruby/object:Gem::Version
-  version: 0.0.8
+  version: 0.0.9
   prerelease: 
 platform: ruby
 authors:
@@ -26,7 +26,10 @@ files:
 - bin/commitshots
 - commitshots.gemspec
 - lib/commitshots.rb
+- lib/commitshots/backends/git.rb
+- lib/commitshots/backends/hg.rb
 - lib/commitshots/commits.rb
+- lib/commitshots/options.rb
 - lib/commitshots/version.rb
 - lib/phantomjs/screenshot.js
 homepage: http://rubygems.org/gems/commitshots