pry-theme 0.1.3 → 0.2.0

data/lib/pry-theme/color.rb

@@ -0,0 +1,431 @@
+module PryTheme
+  class Color
+
+    # Stores all possible colours for foregrounds and backgrounds, for 256, 16
+    # and 8 colour palettes.
+    @@colors = {
+      :foreground => {
+        256 => {
+          'black'                => [0, 16],
+          'maroon'               => 1,
+          'toad_in_love'         => 2,
+          'olive'                => 3,
+          'navy_blue'            => 4,
+          'violet_eggplant'      => 5,
+          'teal'                 => 6,
+          'silver'               => 7,
+          'gray'                 => [8, 244],
+          'red'                  => [9, 196],
+          'green'                => [10, 46],
+          'yellow'               => [11, 226],
+          'blue'                 => [12, 21],
+          'magenta'              => [13, 201],
+          'cyan'                 => [14, 51],
+          'white'                => [15, 231],
+          'dark_indigo'          => 17,
+          'ultramarine01'        => 18,
+          'ultramarine02'        => 19,
+          'persian_blue01'       => 20,
+          'dark_spring_green'    => 22,
+          'dark_turquoise'       => 23,
+          'cerulean_grey01'      => 24,
+          'denim01'              => 25,
+          'royal_blue01'         => 26,
+          'royal_blue02'         => 27,
+          'toad_in_love01'       => 28,
+          'sea_green01'          => 29,
+          'teal01'               => 30,
+          'cerulean_grey02'      => 31,
+          'klein_blue'           => 32,
+          'azure01'              => 33,
+          'vert_de_pomme01'      => 34,
+          'jade01'               => 35,
+          'jade02'               => 36,
+          'robin_egg_blue01'     => 37,
+          'bondi_blue'           => 38,
+          'light_blue01'         => 39,
+          'vert_de_pomme02'      => 40,
+          'malachite01'          => 41,
+          'emerald01'            => 42,
+          'turquoise'            => 43,
+          'robin_egg_blue02'     => 44,
+          'electric01'           => 45,
+          'spring_green01'       => 47,
+          'spring_green02'       => 48,
+          'mint_green01'         => 49,
+          'aquamarine01'         => 50,
+          'flea_belly'           => 52,
+          'plum'                 => 53,
+          'indigo'               => 54,
+          'purple01'             => 55,
+          'violet01'             => 56,
+          'persian_blue02'       => 57,
+          'khaki01'              => 58,
+          'wet_asphalt01'        => 59,
+          'seroburomalinovyj01'  => 60,
+          'denim02'              => 61,
+          'royal_blue03'         => 62,
+          'royal_blue04'         => 63,
+          'olive_drab'           => 64,
+          'fern_green'           => 65,
+          'slate_gray'           => 66,
+          'steel_blue'           => 67,
+          'cornflower_blue01'    => 68,
+          'azure02'              => 69,
+          'grass01'              => 70,
+          'emerald02'            => 71,
+          'sea_green02'          => 72,
+          'robin_egg_blue03'     => 73,
+          'bluish01'             => 74,
+          'light_blue02'         => 75,
+          'vert_de_pomme03'      => 76,
+          'pale_green01'         => 77,
+          'emerald03'            => 78,
+          'aquamarine02'         => 79,
+          'robin_egg_blue04'     => 80,
+          'sky01'                => 81,
+          'bright_green'         => 82,
+          'malachite02'          => 83,
+          'spring_green03'       => 84,
+          'chartreuse01'         => 85,
+          'aquamarine03'         => 86,
+          'cyan01'               => 87,
+          'maroon01'             => 88,
+          'eggplant01'           => 89,
+          'violet_eggplant01'    => 90,
+          'purple02'             => 91,
+          'violet02'             => 92,
+          'violet03'             => 93,
+          'khaki02'              => 94,
+          'pale_mauve01'         => 95,
+          'seroburomalinovyj02'  => 96,
+          'amethyst01'           => 97,
+          'amethyst02'           => 98,
+          'heliotrope01'         => 99,
+          'olive01'              => 100,
+          'dark_tan'             => 101,
+          'gray01'               => [102, 243],
+          'bluish02'             => 103,
+          'cornflower_blue02'    => 104,
+          'royal_blue05'         => 105,
+          'grass02'              => 106,
+          'asparagus'            => 107,
+          'swamp_green01'        => 108,
+          'light_grey01'         => 109,
+          'bluish03'             => 110,
+          'cornflower_blue03'    => 111,
+          'green_yellow01'       => 112,
+          'pale_green02'         => 113,
+          'emerald04'            => 114,
+          'celadon'              => 115,
+          'pale_blue01'          => 116,
+          'sky02'                => 117,
+          'viridian'             => 118,
+          'vert_de_pomme04'      => 119,
+          'mint_green02'         => 120,
+          'chartreuse02'         => 121,
+          'aquamarine04'         => 122,
+          'electric02'           => 123,
+          'bismarck_furious'     => 124,
+          'eggplant02'           => 125,
+          'red_violet01'         => 126,
+          'violet_eggplant02'    => 127,
+          'bright_violet'        => [128, 164],
+          'violet04'             => 129,
+          'ochre'                => 130,
+          'pale_mauve02'         => 131,
+          'pale_red_violet01'    => 132,
+          'orchid01'             => 133,
+          'amethyst03'           => 134,
+          'heliotrope02'         => 135,
+          'dark_goldenrod'       => 136,
+          'pale_brown'           => 137,
+          'mountbatten_pink'     => 138,
+          'lilac01'              => 139,
+          'wisteria01'           => 140,
+          'amethyst04'           => 141,
+          'old_gold'             => 142,
+          'brass01'              => 143,
+          'swamp_green02'        => 144,
+          'light_grey02'         => 145,
+          'niagara'              => 146,
+          'wisteria02'           => 147,
+          'lime01'               => 148,
+          'pistachio01'          => 149,
+          'moss_green'           => 150,
+          'dark_tea_green'       => 151,
+          'pale_blue02'          => 152,
+          'pale_cornflower_blue' => 153,
+          'green_yellow02'       => 154,
+          'green_yellow03'       => 155,
+          'pistachio02'          => 156,
+          'chartreuse03'         => 157,
+          'aquamarine05'         => 158,
+          'pale_blue03'          => 159,
+          'titian'               => 160,
+          'cerise01'             => 161,
+          'red_violet02'         => 162,
+          'hot_pink01'           => 163,
+          'magenta01'            => 165,
+          'tenne'                => 166,
+          'chestnut01'           => 167,
+          'pale_red_violet02'    => 168,
+          'orchid02'             => 169,
+          'orchid03'             => 170,
+          'heliotrope03'         => 171,
+          'siena'                => 172,
+          'dark_salmon'          => 173,
+          'puce01'               => 174,
+          'puce2'                => 175,
+          'violaceous01'         => 176,
+          'violaceous02'         => 177,
+          'dark_pear'            => 178,
+          'brass02'              => 179,
+          'tan'                  => 180,
+          'pale_chestnut'        => 181,
+          'lilac02'              => 182,
+          'wisteria03'           => 183,
+          'childs_surprise'      => 184,
+          'vert_de_peche'        => 185,
+          'flax'                 => 186,
+          'gray_tea_green'       => 187,
+          'abdel_kerims_beard01' => 188,
+          'periwinkle'           => 189,
+          'lime02'               => 190,
+          'pistachio03'          => 191,
+          'pistachio04'          => 192,
+          'tea_green01'          => 193,
+          'tea_green02'          => 194,
+          'pang'                 => 195,
+          'crimson'              => 197,
+          'cerise02'             => 198,
+          'hot_pink02'           => 199,
+          'hot_pink03'           => 200,
+          'international_orange' => 202,
+          'alizarin'             => 203,
+          'dark_pink01'          => 204,
+          'dark_pink02'          => 205,
+          'shocked_star'         => 206,
+          'fuchsia'              => 207,
+          'tangerine'            => 208,
+          'salmon'               => 209,
+          'chestnut02'           => 210,
+          'pale_red_violet03'    => 211,
+          'pale_magenta'         => 212,
+          'violaceous03'         => 213,
+          'orange'               => 214,
+          'pink_orange'          => 215,
+          'saumon'               => 216,
+          'pink01'               => 217,
+          'pink02'               => 218,
+          'violaceous04'         => 219,
+          'gold'                 => 220,
+          'mustard01'            => 221,
+          'mustard02'            => 222,
+          'dark_peach'           => 223,
+          'pale_pink'            => 224,
+          'thistle'              => 225,
+          'corn01'               => 227,
+          'corn02'               => 228,
+          'perhydor'             => 229,
+          'lemon_cream'          => 230,
+          'black01'              => 232,
+          'black02'              => 233,
+          'black03'              => 234,
+          'bistre'               => 235,
+          'anthracite'           => 236,
+          'wet_asphalt02'        => 237,
+          'wet_asphalt03'        => 238,
+          'wet_asphalt04'        => 239,
+          'wet_asphalt05'        => 240,
+          'wet_asphalt06'        => 241,
+          'wet_asphalt07'        => 242,
+          'gray02'               => 245,
+          'gray03'               => 246,
+          'gray04'               => 247,
+          'gray05'               => 248,
+          'light_grey03'         => 249,
+          'light_grey04'         => 250,
+          'silver01'             => 251,
+          'abdel_kerims_beard02' => 252,
+          'abdel_kerims_beard03' => 253,
+          'abdel_kerims_beard04' => 254,
+          'seashell'             => 255
+        },
+        16 => {
+          'black'          => 30,
+          'red'            => 31,
+          'green'          => 32,
+          'yellow'         => 33,
+          'blue'           => 34,
+          'magenta'        => 35,
+          'cyan'           => 36,
+          'white'          => 37,
+          'bright_black'   => [30, 1],
+          'bright_red'     => [31, 1],
+          'bright_green'   => [32, 1],
+          'bright_yellow'  => [33, 1],
+          'bright_blue'    => [34, 1],
+          'bright_magenta' => [35, 1],
+          'bright_cyan'    => [36, 1],
+          'bright_white'   => [37, 1],
+        },
+        8 => {
+          'black'   => 30,
+          'red'     => 31,
+          'green'   => 32,
+          'yellow'  => 33,
+          'blue'    => 34,
+          'magenta' => 35,
+          'cyan'    => 36,
+          'white'   => 37,
+        }
+      },
+      :background => {
+        16 => {
+          'black'          => 40,
+          'red'            => 41,
+          'green'          => 42,
+          'yellow'         => 43,
+          'blue'           => 44,
+          'magenta'        => 45,
+          'cyan'           => 46,
+          'white'          => 47,
+          'bright_black'   => 40,
+          'bright_red'     => 41,
+          'bright_green'   => 42,
+          'bright_yellow'  => 43,
+          'bright_blue'    => 44,
+          'bright_magenta' => 45,
+          'bright_cyan'    => 46,
+          'bright_white'   => 47,
+        },
+      }
+    }
+    @@colors[:background][256] = @@colors[:foreground][256]
+    @@colors[:background][8]   = @@colors[:background][16]
+
+    # The default colour options Hash.
+    OPTS = {
+      :from        => :readable,
+      :foreground  => false,
+      :background  => false,
+      :bold        => false,
+      :italic      => false,
+      :underline   => false,
+    }
+
+    attr_reader :color_model
+
+    # @return []
+    attr_reader :options
+
+    def initialize(color_model, options = {})
+      @options     = OPTS.merge(options)
+      @color_model = color_model
+      set_layers
+    end
+
+    def foreground(readable = false)
+      @readable_fg = find_color(:foreground)
+      readable ? @readable_fg : layer_color(colors[@readable_fg])
+    end
+
+    def background(readable = false)
+      @readable_bg = find_color(:background)
+      readable ? @readable_bg : layer_color(colors(:background)[@readable_bg])
+    end
+
+    def to_ansi
+      fg, bg = !!foreground, !!background
+      create_ansi_sequence(fg, bg)
+    end
+
+    private
+
+    def set_layers
+      foreground
+      background
+    end
+
+    def create_ansi_sequence(fg, bg, default_seq)
+      (if fg && bg
+        [[build_fg_sequence], [build_effects_sequence], [build_bg_sequence]].flatten
+      elsif fg && !bg
+        [[build_fg_sequence], [build_effects_sequence]].flatten
+      elsif !fg && bg
+        [[build_bg_sequence], [build_effects_sequence]].flatten
+      else
+        build_effects_sequence.tap { |sequence|
+          sequence << default_seq if sequence.empty?
+        }
+      end).join(';')
+    end
+
+    def build_fg_sequence
+      [foreground]
+    end
+
+    def build_effects_sequence
+      [bold, italic, underline].delete_if { |e| e == false }
+    end
+
+    def build_bg_sequence
+      [background]
+    end
+
+    def layer_color(layer)
+      if layer
+        layer.is_a?(Array) ? build_layer(layer) : layer
+      else
+        false
+      end
+    end
+
+    def build_layer(layer)
+      layer.first
+    end
+
+    def find_color(layer)
+      case (color_id = cast_color(layer))
+      when String
+        return color_id if colors.has_key?(color_id)
+      when Fixnum
+        color_id = find_from_fixnum(color_id)
+        return color_id if color_id
+      when false
+        return color_id
+      end
+
+      raise ArgumentError, %|invalid #{ layer.to_s } value "#{ color_id }"|
+    end
+
+    def cast_color(layer)
+      return false unless options[layer]
+
+      case options[:from]
+      when :readable
+        options[layer]
+      when :hex
+        HEX.new(options[layer]).to_term(color_model).to_i
+      when :rgb
+        RGB.new(options[layer]).to_term(color_model).to_i
+      when :term
+        TERM.new(options[layer], color_model).to_i
+      end
+    end
+
+    def colors(layer = :foreground)
+      @@colors[layer][color_model]
+    end
+
+    def sorted_colors
+      colors.sort_by { |k, v| v.is_a?(Array) ? v.first + 10 : v }
+    end
+
+  end
+end
+
+Dir[File.expand_path('../colors/*.rb', __FILE__)].each do |file|
+  require file
+end

data/lib/pry-theme/color_table.rb

@@ -0,0 +1,39 @@
+module PryTheme
+  class ColorTable
+
+    class << self
+      def t256
+        ColorTable.new(256).table
+      end
+
+      def t16
+        ColorTable.new(16).table
+      end
+
+      def t8
+        ColorTable.new(8).table
+      end
+
+      def build_color_string(color, fg = nil)
+        "\e[7;%sm%s\e[0m:\e[%sm%s\e[0m" %
+          [color.to_ansi, fg || color.foreground,
+           color.to_ansi, color.foreground(true)]
+      end
+    end
+
+    def initialize(color_model)
+      @color_model = color_model
+    end
+
+    def table
+      colors = []
+      0.upto(@color_model - 1) { |i|
+        color = PryTheme.const_get(:"Color#@color_model").new(
+          :from => :term, :foreground => i)
+        colors << self.class.build_color_string(color, i)
+      }
+      Pry::Helpers.tablify_or_one_line("Color model #@color_model", colors)
+    end
+
+  end
+end

data/lib/pry-theme/colors/color16.rb

@@ -0,0 +1,35 @@
+module PryTheme
+  class Color16 < Color
+
+    def initialize(options = {})
+      super(16, options)
+    end
+
+    private
+
+    def create_ansi_sequence(fg, bg)
+      super(fg, bg, '39;49')
+    end
+
+    def build_fg_sequence
+      [foreground]
+    end
+
+    def build_bg_sequence
+      [background]
+    end
+
+    def build_effects_sequence
+      []
+    end
+
+    def find_from_fixnum(color_id)
+      sorted_colors.each_with_index.to_a.rassoc(color_id).first.first
+    end
+
+    def build_layer(layer)
+      layer.join(';')
+    end
+
+  end
+end

data/lib/pry-theme/colors/color256.rb

@@ -0,0 +1,30 @@
+module PryTheme
+  class Color256 < PryTheme::Color
+
+    include Formattable
+
+    def initialize(options = {})
+      super(256, options)
+    end
+
+    private
+
+    def create_ansi_sequence(fg, bg)
+      super(fg, bg, '0')
+    end
+
+    def build_fg_sequence
+      ['38', '5', foreground]
+    end
+
+    def build_bg_sequence
+      ['48', '5', background]
+    end
+
+    def find_from_fixnum(color_id)
+      pair = colors.find { |*term| term.first.flatten.include?(color_id) }
+      pair.first if pair
+    end
+
+  end
+end

data/lib/pry-theme/colors/color8.rb

@@ -0,0 +1,31 @@
+module PryTheme
+  class Color8 < Color
+
+    def initialize(options = {})
+      super(8, options)
+    end
+
+    private
+
+    def create_ansi_sequence(fg, bg)
+      super(fg, bg, '39;49')
+    end
+
+    def build_fg_sequence
+      [foreground]
+    end
+
+    def build_bg_sequence
+      [background]
+    end
+
+    def find_from_fixnum(color_id)
+      sorted_colors.each_with_index.to_a.rassoc(color_id).first.first
+    end
+
+    def build_effects_sequence
+      []
+    end
+
+  end
+end