ansi256 0.2.4 → 0.2.5

data/README.md

@@ -77,8 +77,8 @@ puts say_hello_world.plain.fg(27)
 
 ![say-hello-world](https://github.com/junegunn/ansi256/raw/master/say-hello-world.png)
 
-_"Just gimme the code!"_
------------------------
+Ansi256 methods
+---------------
 
 ```ruby
 Ansi256.fg(232)
@@ -88,8 +88,24 @@ Ansi256.on_green
 Ansi256.bold
 Ansi256.underline
 Ansi256.reset
+
+Ansi256.fg(232, 'Hello')
+Ansi256.bg(226, 'World')
+Ansi256.green('Hello World')
 ```
 
+Disabling extended String methods
+---------------------------------
+
+```ruby
+Ansi256.enabled?
+  # true
+
+Ansi256.enabled = false
+
+"Hello".fg(232)
+  # Hello
+```
 
 ansi256 executable
 ------------------

data/lib/ansi256.rb

@@ -133,7 +133,4 @@ module Ansi256
   end
 end
 
-class String
-  include Ansi256::Mixin
-end
-
+Ansi256.enabled = true

data/lib/ansi256/mixin.rb

@@ -1,21 +1,74 @@
 module Ansi256
-  module Mixin
-    def fg code
-      Ansi256.fg(code, self)
-    end
+  class << self
+    def enabled= bool
+      if bool
+        String.class_eval do
+          [:fg, :bg, :plain].each do |name|
+            undef_method(name) if method_defined?(name)
+          end
 
-    def bg code
-      Ansi256.bg(code, self)
-    end
+          def fg code
+            Ansi256.fg(code, self)
+          end
 
-    def plain
-      Ansi256.plain self
-    end
+          def bg code
+            Ansi256.bg(code, self)
+          end
+
+          def plain
+            Ansi256.plain self
+          end
+
+          Ansi256::CODE.each do |name, code|
+            undef_method(name) if method_defined?(name)
+            define_method name do
+              Ansi256.send name, self
+            end
+          end
+        end
 
-    Ansi256::CODE.each do |name, code|
-      define_method name do
-        Ansi256.send name, self
+        Ansi256.class_eval do
+          class << self
+            undef_method(:enabled?) if method_defined?(:enabled?)
+            def enabled?
+              true
+            end
+          end
+        end
+      else
+        String.class_eval do
+          [:fg, :bg, :plain].each do |name|
+            undef_method(name) if method_defined?(name)
+          end
+
+          def fg code
+            self
+          end
+          alias bg fg
+
+          def plain
+            Ansi256.plain self
+          end
+
+          Ansi256::CODE.each do |name, code|
+            undef_method(name) if method_defined?(name)
+            define_method name do
+              self
+            end
+          end
+        end
+
+        Ansi256.class_eval do
+          class << self
+            undef_method(:enabled?) if method_defined?(:enabled?)
+            def enabled?
+              false
+            end
+          end
+        end
       end
+
+      bool
     end
   end
 end

data/lib/ansi256/version.rb

@@ -1,3 +1,3 @@
 module Ansi256
-  VERSION = "0.2.4"
+  VERSION = "0.2.5"
 end

data/test/test_ansi256.rb

@@ -194,4 +194,23 @@ class TestAnsi256 < MiniTest::Unit::TestCase
     puts "RGB Color".fg('#ff9900').bg('#003366')
     puts "RGB Color (Monochrome)".fg('ef').bg('3a')
   end
+
+  def test_enabled
+    2.times do
+      [:fg, :bg].each do |m|
+        assert Ansi256.enabled?
+        hello = "hello".send(m, 'f90')
+        assert_equal 'hello', hello.plain
+
+        Ansi256.enabled = false
+        assert !Ansi256.enabled?
+        assert hello.length > "hello".send(m, 'f90').length
+        assert_equal 'hello', hello.plain
+
+        Ansi256.enabled = true
+        assert Ansi256.enabled?
+        assert_equal hello, "hello".send(m, 'f90')
+      end
+    end
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ansi256
 version: !ruby/object:Gem::Version
-  version: 0.2.4
+  version: 0.2.5
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-05-28 00:00:00.000000000 Z
+date: 2013-05-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler