jamescook-pow 0.1.0

data/.gitignore

@@ -0,0 +1,3 @@
+*.swp
+pkg
+pkg/*

data/README

@@ -0,0 +1,17 @@
+P O W !  ( Not like prisoner of war, but from the movie Step Brothers where that guy from the Daily Show is saying "POW" repeatedly. )
+
+Override your 'puts' for fun shell coloring.
+
+Usage:
+  require "pow"
+
+Examples:
+  puts.yellow "Hello world in yellow"
+  puts.red    "Hello world in red" 
+  puts.red!   "Hello world in red, but with boldness"
+  puts.red_   "Hello world in red, but with strikethrough"
+  puts.rainbow "Hellow world, but more silly."
+
+
+TODO:
+ - Figure out if you can bold a 'dark' color

data/lib/pow.rb

@@ -0,0 +1,155 @@
+module Pow
+  # Override puts on include to allow coloring (but also retain existing function)
+  def self.included(base)
+    base.send(:define_method, :puts){ |*args| Puts.new(*args) }
+  end
+  CODES = {
+          :clear           => 0,
+          :reset           => 0, #clear
+          :bold            => 1,
+          :dark            => 2,
+          :italic          => 3,
+          :underline       => 4,
+          :underscore      => 4,
+          :blink           => 5,
+          :rapid_blink     => 6,
+          :negative        => 7,
+          :concealed       => 8,
+          :strikethrough   => 9,
+          :black           => 30,
+          :red             => 31,
+          :green           => 32,
+          :yellow          => 33,
+          :blue            => 34,
+          :magenta         => 35,
+          :purple          => 35,
+          :cyan            => 36,
+          :white           => 37,
+          :on_black        => 40,
+          :on_red          => 41,
+          :on_green        => 42,
+          :on_yellow       => 43,
+          :on_blue         => 44,
+          :on_magenta      => 45, 
+          :on_purple       => 45, 
+          :on_cyan         => 46,
+          :on_white        => 47,
+          :brown           => [:yellow, :dark],
+          :navy            => [:blue, :dark],
+          :dark_red        => [:red, :dark],
+          :dark_blue       => [:blue, :dark],
+          :dark_cyan       => [:cyan, :dark],
+          :dark_green      => [:green, :dark],
+          :dark_magenta    => [:magenta, :dark],
+          :dark_purple     => [:purple, :dark],
+          :gray            => [:white, :dark],
+          :grey            => [:white, :dark]
+          }.freeze
+
+  class Puts  
+    attr_accessor :writer
+    def initialize(*args)
+      options = args[0].is_a?(String) ? {:text => args[0] } : (args[0] || {})
+      CODES.keys.each do |key|
+        # Color
+        self.class.send(:define_method, key.to_sym)       { |*args| Puts.new({:color => key.to_sym, :text => args[0], :misc => args[1]}).out! }
+        # Bold
+        self.class.send(:define_method, "#{key}!".to_sym) { |*args| Puts.new({:color => key.to_sym, :text => args[0], :bold => true, :misc => args[1]}).out! }
+        # Strikethrough
+        self.class.send(:define_method, "#{key}_".to_sym) { |*args| Puts.new({:color => key.to_sym, :text => args[0], :strikethrough => true, :misc => args[1]}).out! }
+      end     
+      # FIXME make this part a method or less fug
+      if options[:misc] && options[:misc].is_a?(Hash)
+        options[:bold]       ||= options[:misc][:bold]
+        options[:underline]  ||= options[:misc][:underline]
+        options[:background] ||= (options[:misc][:background] || options[:misc][:on])
+        options[:strikethrough]  ||= options[:misc][:strikethrough]
+      end
+      @writer = options[:writer] || STDOUT
+      @formatted_text = format_string(options)
+      out!(@formatted_text)
+    end
+   
+    # Write the coloured text to IO
+    def out!(string=nil)
+      if string && string != ""
+        printf(writer, string)
+      end
+    end
+
+    def rainbow(*args)
+      text = args[0]
+      out!(text.to_s.split("").inject(""){|m, word| m+= format_string(:text => word, :bold => true, :newline => "", :color => rainbow_keys.sort_by{|k| rand}[0]) + " "  } + "\n")
+    end
+
+    # Feel the painbow
+    def painbow(*args)
+      text = args[0]
+      out!(text.to_s.split("").inject(""){|m, word| m+= format_string(:text => word, :bold => true, :newline => "", :color => painbow_keys.sort_by{|k| rand}[0]) + " "  } + "\n")
+    end
+
+    def inspect
+      nil
+    end
+
+    protected
+    def painbow_keys
+      [:red, :gray, :white].freeze
+    end
+
+    def rainbow_keys
+      [:red, :blue, :yellow, :green, :magenta, :cyan, :dark_purple].freeze
+    end
+
+    def format_string(options={})
+      newline        = options.fetch(:newline){ "\n" }
+      color          = options.fetch(:color){ :white }
+      text           = options.fetch(:text){ '' }
+      bold           = options.fetch(:bold){ false }
+      underline      = options.fetch(:underline){ false }
+      background     = options.fetch(:background){ false }
+      concealed      = options.fetch(:concealed){ false }
+      strikethrough  = options.fetch(:strikethrough){ false }
+
+      if text != ""
+        result = [escape_sequence(color), text, escape_sequence(:reset), newline].join
+      end
+
+      if bold
+        result.insert(0, escape_sequence(:bold))
+      end
+
+      if underline
+        result.insert(0, escape_sequence(:underline))
+      end
+
+      if background
+        result.insert(0, escape_sequence("on_#{background}".to_sym))
+      end
+
+      if concealed
+        result.insert(0, escape_sequence(:concealed))
+      end
+
+      if strikethrough
+        result.insert(0, escape_sequence(:strikethrough))
+      end
+
+      return result
+    end
+
+    def escape_sequence(code)
+      if CODES[code].is_a?(Array)
+        sequence = ""
+        CODES[code].each do |key|
+          sequence << escape_sequence(key)
+        end
+        return sequence
+      else
+        return "\e[#{CODES[code]}m"
+      end
+    end
+  end
+end
+
+include Pow

data/pow.rb

@@ -0,0 +1 @@
+require "lib/pow"

data/test/pow_test.rb

@@ -0,0 +1,41 @@
+require 'test/unit'
+require "stringio"
+require File.join( File.dirname(__FILE__), "..", "pow")
+
+class PowTest < Test::Unit::TestCase
+  include Pow
+  def setup
+    @puts = Pow::Puts
+    @writer = StringIO.new
+  end
+
+  def test_puts
+    @puts.new(:text => "TEST", :writer => @writer).out!
+    @writer.rewind
+    assert_equal "\e[37mTEST\e[0m\n", @writer.gets  # White text
+  end
+
+  def test_puts_with_red
+    @puts.new(:text => "TEST", :color => :red, :writer => @writer).out!
+    @writer.rewind
+    assert_equal "\e[31mTEST\e[0m\n", @writer.gets
+  end
+
+  def test_puts_with_dark_color  #brown =  yellow + dark
+    @puts.new(:text => "TEST", :color => :brown, :writer => @writer).out!
+    @writer.rewind
+    assert_equal "\e[33m\e[2mTEST\e[0m\n", @writer.gets
+  end
+
+  def test_puts_with_bold
+    @puts.new(:text => "TEST", :color => :bold, :writer => @writer).out!
+    @writer.rewind
+    assert_equal "\e[1mTEST\e[0m\n", @writer.gets
+  end
+
+  def test_puts_with_strikethrough
+    @puts.new(:text => "TEST", :strikethrough => :true, :writer => @writer).out!
+    @writer.rewind
+    assert_equal "\e[9m\e[37mTEST\e[0m\n", @writer.gets
+  end
+end

metadata

@@ -0,0 +1,59 @@
+--- !ruby/object:Gem::Specification 
+name: jamescook-pow
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+platform: ruby
+authors: 
+- James Cook
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-11-19 00:00:00 -06:00
+default_executable: 
+dependencies: []
+
+description: "'puts' with shell colors."
+email: jamecook@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README
+files: 
+- .gitignore
+- README
+- lib/pow.rb
+- pow.rb
+- test/pow_test.rb
+has_rdoc: true
+homepage: http://github.com/jamescook/pow
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: puts with colors
+test_files: 
+- test/pow_test.rb