grit-ext 0.0.1 → 0.0.2

data/README.md

@@ -1 +1,14 @@
 # grit-ext
+
+## Resources
+
+* [grit](https://github.com/mojombo/grit)
+* [grit-ext on RubyGems](https://rubygems.org/gems/grit-ext)
+
+## How to Use
+
+To use grit-ext, after installing, just require 'grit-ext':
+
+    require 'grit-ext'
+
+All of the extensions, additionally to normal grit classes, will be available to you.

data/lib/grit/ext/diff.rb

@@ -4,7 +4,7 @@ module Grit
       lines = diff.split("\n")
       lines.each_with_index
            .find_all { |line, index| line.start_with?('@@') }
-           .map { |line, index| [DiffHunkHeader.new(line), index] }
+           .map { |line, index| [DiffHeader.new(line), index] }
            .map { |header, index|
              DiffHunk.new(header, lines[index, header.count])
            }

data/lib/grit/ext/diff_header.rb

@@ -0,0 +1,26 @@
+module Grit
+  class DiffHeader
+    attr_reader :removed_start, :removed_count,
+                :added_start, :added_count
+
+    alias_method :start, :added_start
+    alias_method :count, :added_count
+
+    def initialize(hunk_header)
+      groups = hunk_header.split('-').last.split('+')
+
+      removed = groups.first.strip
+      added = groups.last.sub(' @@', '').strip
+
+      @removed_start, @removed_count = get_numbers(removed)
+      @added_start, @added_count = get_numbers(added)
+    end
+
+    private
+
+    def get_numbers(s)
+      start, count = s.split(',')
+      [start.to_i, count.to_i]
+    end
+  end
+end

data/lib/grit/ext/diff_hunk.rb

@@ -1,5 +1,7 @@
 module Grit
   class DiffHunk
+    attr_reader :lines
+
     def initialize(header, diff_hunk)
       @lines = diff_hunk.each_with_index.map do |line, index|
         content = line[1..line.length - 1]
@@ -8,34 +10,22 @@ module Grit
                  when '-' then :removed
                  when ' ' then :unchanged
                  end
-        {
-          content: content,
-          status: status,
-          position: header.start + index
-        }
-      end
-    end
+        position = header.start + index
 
-    def lines
-      @lines
+        DiffLine.new(content, status, position)
+      end
     end
 
     def added
-      lines_by_status(:added)
+      @lines.find_all { |line| line.added? }
     end
 
     def removed
-      lines_by_status(:removed)
+      @lines.find_all { |line| line.removed? }
     end
 
     def unchanged
-      lines_by_status(:unchanged)
-    end
-
-    private
-
-    def lines_by_status(status)
-      @lines.find_all { |line| line[:status] == status }
+      @lines.find_all { |line| line.unchanged? }
     end
   end
 end

data/lib/grit/ext/diff_line.rb

@@ -0,0 +1,15 @@
+module Grit
+  class DiffLine < Struct.new(:content, :status, :position)
+    def added?
+      status == :added
+    end
+
+    def removed?
+      status == :removed
+    end
+
+    def unchanged?
+      status == :unchanged
+    end
+  end
+end

data/lib/grit/ext/version.rb

@@ -1,5 +1,5 @@
 module Grit
   module Ext
-    VERSION = '0.0.1'
+    VERSION = '0.0.2'
   end
 end

data/lib/grit-ext.rb

@@ -1,4 +1,5 @@
 require 'grit'
 require 'grit/ext/diff'
 require 'grit/ext/diff_hunk'
-require 'grit/ext/diff_hunk_header'
+require 'grit/ext/diff_header'
+require 'grit/ext/diff_line'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: grit-ext
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -68,8 +68,9 @@ extra_rdoc_files: []
 files:
 - lib/grit/ext/commit.rb
 - lib/grit/ext/diff.rb
+- lib/grit/ext/diff_header.rb
 - lib/grit/ext/diff_hunk.rb
-- lib/grit/ext/diff_hunk_header.rb
+- lib/grit/ext/diff_line.rb
 - lib/grit/ext/version.rb
 - lib/grit-ext.rb
 - LICENSE

data/lib/grit/ext/diff_hunk_header.rb

@@ -1,11 +0,0 @@
-module Grit
-  class DiffHunkHeader
-    attr_reader :start, :count
-
-    def initialize(hunk_header)
-      start, count = hunk_header.split('+').last.sub(' @@', '').split(',')
-      @start = start.to_i
-      @count = count.to_i
-    end
-  end
-end