prettys 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 5644d68aec8dcb4ef481fa2a2139f77485532258
-  data.tar.gz: 051970260a1f4ee9650e8ea63df6e16366aa3954
+  metadata.gz: 524b43c2fff5dfe3b7fb931b0a02f2fe56622b2b
+  data.tar.gz: 594bdff6580a5e1ccf8fa1918e6af238550895c2
 SHA512:
-  metadata.gz: 996274699b7f321c5badc5e345174a45bfcbfbe14d86c36e1be008959f18f74dce31da3f17e876208b2e759d256f98b7748aa0505df907d8a40b18d67e41d566
-  data.tar.gz: 6092d2f928608e664a43eeca79b15e351900f2b996115d9e516265f7f8738e02b04f4a7f46aa1237aad796678ccaf5a6819b94340c76eb7eb315073a4f4c95ff
+  metadata.gz: 510247031a3a15f42cfe1f58037ee2c7115649af7f53c323a2529dd5716825b95f41551ac02c70a90c2473d89045b3a3e30d52bbd177b7706a23fd7e6bc6e403
+  data.tar.gz: b3b1f461a08f66d5c94bf66e5f1f7dec3af4d7da24dcbd40c65e2905a220157bb49947415656cf6f31a68f86ea7bf0fce7c9a45dabdcf487686d4d7a206ffabe

data/lib/colorizer.rb

@@ -0,0 +1,71 @@
+require 'matcher'
+
+module Prettys
+  class Colorizer
+
+    COLOR_NAMES = [
+      :black, 
+      :red, 
+      :green, 
+      :yellow, 
+      :blue, 
+      :magenta, 
+      :cyan, 
+      :white
+    ]
+
+    CHROMATIC_COLOR_NAMES = COLOR_NAMES[1...-1]
+
+    COLORS = {}
+    COLOR_NAMES.each_with_index do |color_name, index|
+      COLORS[color_name] = index + 30
+    end
+
+    private_constant :COLOR_NAMES, :CHROMATIC_COLOR_NAMES, :COLORS
+
+    def escape_sequence(options)
+      options = add_default_options(options)
+      color_code = COLORS[options[:color]]
+      color_code += 10 if options[:type] == :background
+      "\e[#{color_code.to_s};#{(options[:bold] ? 1 : 2).to_s}m"
+    end
+
+    def end_of_escape_sequence
+      "\e[0m"
+    end
+
+    def escaped_string(options)
+      escape_sequence(options) + options[:string] + end_of_escape_sequence
+    end
+
+    def colorize(options = {})
+      options = add_default_options(options)
+      unless [:background, :foreground].include?(options[:type])
+        raise(ArgumentError, "Type must be a :background or :foreground")
+      end
+      marked_strings = Matcher.marked_strings(options[:string], options[:pattern])
+      marked_strings.map do |ms| 
+        if ms[:marked]
+          escaped_string({
+            string: ms[:string],
+            pattern: options[:pattern],
+            type: options[:type],
+            bold: options[:bold],
+            color: options[:color]
+          })
+        else
+          ms[:string]
+        end
+      end.join
+    end
+
+    private
+
+    def add_default_options(options)
+      options[:type] = :foreground unless options[:type]
+      options[:bold] = false unless options[:bold]
+      options[:color] = COLOR_NAMES[1] unless options[:color]
+      return options
+    end
+  end
+end

data/lib/matcher.rb

@@ -0,0 +1,15 @@
+module Prettys
+  module Matcher
+
+    def Matcher.marked_strings(string, pattern)
+      matched =  string.scan(pattern).map { |s| { string: s, marked: true } }
+      not_matched = string.split(pattern).map { |s| { string: s, marked: false } }
+      if (string.match(pattern).begin(0))
+        not_matched.zip(matched)
+      else
+        matched.zip(not_matched)
+      end.flatten.compact
+    end
+
+  end
+end

data/lib/prettys.rb

@@ -3,8 +3,10 @@ require 'yaml'
 
 module Prettys
   require 'converter'
+  require 'colorizer'
 
   @converter = Converter.new(:json)
+  @colorizer = Colorizer.new
 
   class << self
     def format
@@ -15,14 +17,16 @@ module Prettys
       @converter.format = format
     end
 
-    def prettys(object)
-      @converter.convert(object)
+    def prettys(object, options)
+      options[:string] = @converter.convert(object)
+      @colorizer.colorize(options)
     end
   end
 end
 
 class Object
-  def prettys 
-    Prettys.prettys(self)
+  def prettys(pattern, options = {})
+    options[:pattern] = pattern
+    Prettys.prettys(self, options)
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: prettys
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Maciej Ciemborowicz
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-10-20 00:00:00.000000000 Z
+date: 2013-11-08 00:00:00.000000000 Z
 dependencies: []
 description: Colorful inspector for Ruby objects.
 email: pub@ciemborowicz.pl
@@ -18,6 +18,8 @@ extra_rdoc_files: []
 files:
 - lib/prettys.rb
 - lib/converter.rb
+- lib/colorizer.rb
+- lib/matcher.rb
 homepage: http://rubygems.org/gems/prettys
 licenses:
 - MIT