pry-theme 0.1.3 → 0.2.0

data/lib/pry-theme/commands.rb

@@ -1,328 +1,290 @@
-module PryTheme
-  Commands = Pry::CommandSet.new do
-
-    create_command "pry-theme", "Manage your Pry themes." do
-      include PryTheme::Helper
-
-      banner <<-BANNER
-        Usage: pry-theme [OPTIONS] [--help]
-
-        Change your theme on the fly (for one session).
-
-          pry-theme pry-modern
-
-        Show all themes from Pry Theme Collection.
-
-          pry-theme -r
-
-        Install a theme from Pry Theme Collection.
-
-          pry-theme -i pry-classic
+# coding: utf-8
 
-        Test your current color theme.
-
-          pry-theme -t
-
-        Wiki: https://github.com/kyrylo/pry-theme/wiki/Pry-Theme-CLI
-        Cheatsheet: https://github.com/kyrylo/pry-theme/wiki/Pry-Theme-Cheatsheet
-      BANNER
-
-      def options(opt)
-        opt.on :a, "all-colors",  "Show all available 8/256 colors."
-        opt.on :c, "color",       "Show information about a specific color (256)."
-        opt.on :t, "test",        "Test your current theme", :argument => false
-        opt.on :e, "edit",        "Edit or create a theme"
-        opt.on :l, "list",        "Show a list of installed themes", :argument => false
-        opt.on :r, "remote-list", "Show a list of themes from Pry Theme Collection", :argument => false
-        opt.on :i, "install",     "Install a theme from Pry Theme Collection"
-      end
-
-      def process
-        if opts.a?
-          show_palette_colors
-        elsif opts.c?
-          show_specific_color
-        elsif opts.t?
-          test_theme
-        elsif opts.e?
-          edit_theme
-        elsif opts.l?
-          show_list
-        elsif opts.r?
-          show_remote_list
-        elsif opts.i?
-          install_theme
-        elsif args[0] =~ /\A\w+-?\w+\z/
-          switch_to_theme
-        elsif args.empty?
-          output.puts "Current theme: #{PryTheme.current_theme}"
+module PryTheme
+  Command = Class.new
+
+  class Command::PryTheme < Pry::ClassCommand
+    include Pry::Helpers::BaseHelpers
+    include Pry::Helpers::CommandHelpers
+
+    match 'pry-theme'
+    description 'Manage your Pry themes.'
+
+    banner <<-'BANNER'
+      Usage: pry-theme [OPTIONS] [--help]
+
+      The command comes from `pry-theme` plugin. It enbales color theme support for your
+      Pry. Set up your favorite theme in `~/.pryrc` config file. For example, if you
+      enjoy Zenburn theme, add the following line: `Pry.config.theme = 'zenburn'`.
+
+      You can create your own themes with help of `edit` subcommand. More information
+      can be found in the appropriate arcticle: http://is.gd/YPuTjM
+      Tokens cheatsheet might come in handy, too: http://is.gd/D7GoVe
+
+      pry-theme try pry-modern-256 # changes theme (lasts for current session only)
+      pry-theme current            # shows currently active theme name
+      pry-theme current --colors   # tests colors from the current theme
+      pry-theme list               # shows all installed themes as colorful snippets
+      pry-theme list --remote      # shows all themes from Pry Theme Collection
+      pry-theme colors             # shows the list of all color names
+      pry-theme colors --model 8   # shows the list of colors according to 8 color model
+      pry-theme install autumn     # installs a theme from Pry Theme Collection
+      pry-theme uninstall monokai  # uninstalls a theme
+      pry-theme convert -m 16 -t 3 # converts a single color to a term color
+      pry-theme edit zenburn       # opens Zenburn theme in `Pry.editor`
+    BANNER
+
+    def def_list(cmd)
+      cmd.command :list do |opt|
+        opt.description 'Show a list of all installed themes'
+
+        opt.on :r, :remote, 'Show a list of all themes from Pry Theme Collection'
+
+        opt.run do |opts, args|
+          opts.present?(:r) ? show_remote_list : show_local_list
         end
-      rescue NoPaletteError => no_palette_error
-        warn no_palette_error
-      rescue NoColorError => no_color_error
-        warn no_color_error
-      end
-
-      private
-
-      def switch_to_theme
-        PryTheme.set_theme(args[0].strip) and
-        output.puts "Using #{ args[0] } theme"
       end
+    end
 
-      def show_palette_colors
-        lputs in_columns(Palette.new(args[0]).to_a), output
-      end
+    def def_colors(cmd)
+      cmd.command :colors do |opt|
+        opt.description 'Show all available colors'
 
-      def in_columns(list, cols=3)
-        color_table = []
+        opt.on :m, :model=, 'Display colors according to the given color model'
 
-        list.each_slice(cols) do |slice|
-          slice.each do |c|
-            color = c.scan(/(\e\[[\d;]+m)(\w+)(\e\[0m)(:)(\e\[[\d;]+m)(\w+)(\e\[0m)/).flatten
-            color[1] = color[1].ljust(3)
-            color[-2] = color[-2].ljust(22)
-            color_table << color
+        opt.run do |opts, args|
+          if opts.present?(:m)
+            display_colors(opts[:m].to_i)
+          else
+            display_colors(PryTheme.tput_colors)
           end
-          color_table << "\n"
         end
-        color_table.join
       end
+    end
 
-      def show_specific_color
-        c = args[0]
+    def def_try(cmd)
+      cmd.command :try do |opt|
+        opt.description 'Change theme on the fly (for the current session only)'
 
-        unless c
-          output.puts "This option requires an argument. Specify a color. Examples:"
-          output.puts "Pry Theme: black; RGB: 0,0,0; HEX: #000000, ANSI: 0."
-          return
+        opt.run do |opts, args|
+          if PryTheme::ThemeList.activate_theme(args.first)
+            output.puts %|Using "#{ args.first }" theme|
+          elsif args.first
+            output.puts %|Cannot find "#{args.first}" amongst themes in #{USER_THEMES_DIR}|
+          end
         end
+      end
+    end
 
-        color = if ansi?(c)
-                  find_color(c)
-                elsif hex?(c)
-                  find_color(ColorConverter.hex_to_ansi(c))
-                elsif rgb?(c)
-                  find_color(ColorConverter.rgb_to_ansi(c))
-                else
-                  h = ColorConverter::COLORS.select do |color|
-                    (color.human).to_s[c]
-                  end
-
-                  # Return human-readable colors if the
-                  # select selected any. If not, return nil.
-                  h if h.any?
-                end
-
-        case color
-        when Array
-          lputs color.map { |c| c.to_term(TermNotation::COLOR256) }.join("\n"), output
-        when nil
-          output.puts "Invalid color: #{ c }"
-        else
-          output.puts color.to_term(TermNotation::COLOR256)
+    def def_convert(cmd)
+      cmd.command :convert do |opt|
+        opt.description 'Convert the given color to proper terminal equivalent'
+
+        opt.on :t, :term=,  'Show a terminal color'
+        opt.on :h, :hex=,   'Convert from HEX'
+        opt.on :r, :rgb=,   'Convert from RGB'
+        opt.on :m, :model=, 'Convert accordingly to the given color model'
+
+        opt.run do |opts, args|
+          if color_model_option_only?(opts)
+            output.puts 'Provide a color value to be converted'
+          elsif color_model_option_and_other_one?(opts) || without_color_model?(opts)
+            convert_color(opts, args)
+          else
+            output.puts 'You must provide the `-m` and one of the rest switches.'
+            examples = [
+              'pry-theme convert --model 8 --term 2',
+              'pry-theme convert --model 8 --rgb 103,104,0',
+              'pry-theme convert --model 16 --hex #EEAA00',
+              'pry-theme convert --model 16 --term 0',
+              'pry-theme convert --model 256 --term 255',
+              'pry-theme convert --model 256 --hex #EEAA00',
+            ]
+            output.puts "Example: #{ examples[rand(examples.size)] }"
+          end
         end
       end
+    end
 
-      def find_color(color)
-        ColorConverter::COLORS.find { |c| c.term == color.to_i }
+    def def_uninstall(cmd)
+      cmd.command :uninstall do |opt|
+        opt.description 'Uninstall a theme'
+
+        opt.run do |opts, args|
+          args.each { |theme|
+            begin
+              FileUtils.rm(File.join(USER_THEMES_DIR, "#{ theme }.prytheme.rb"))
+              output.puts %|Successfully uninstalled "#{ theme }"!|
+            rescue
+              output.puts %|Cannot find theme "#{ theme }"|
+            end
+          }
+        end
       end
+    end
 
-      def test_theme
-        example = <<-TEST
-# "#{ PryTheme.current_theme }" theme.
-class PryTheme::ThisIsAClass
-  def this_is_a_method
-    THIS_IS_A_CONSTANT  = :this_is_a_symbol
-    this_is_a_local_var = "\#{this} \#@is a string.\\n"
-    this_is_a_float     = 10_000.00
-    this_is_an_integer  = 10_000
-
-    # TRUE and FALSE are predefined constants.
-    $this_is_a_global_variable = TRUE or FALSE
+    def def_install(cmd)
+      cmd.command :install do |opt|
+        opt.description 'Install a theme from Pry Theme Collection'
 
-    @this_is_an_instance_variable = `echo '\#@hi \#{system} call\\n'`
-    @@this_is_a_class_variable    = @@@\\\\$ # An error.
+        opt.run do |opts, args|
+          install_theme(args)
+        end
+      end
+    end
 
-    /[0-9]{1,3}this \#{is} a regexp\\w+/xi
-  end
-end
-        TEST
+    def def_current(cmd)
+      cmd.command :current do |opt|
+        opt.description 'Show information about currently active theme'
 
-        lputs colorize_code(example), output
-      end
+        opt.on :c, :colors, 'Display a painted code snippet'
 
-      def edit_theme
-        theme = args.first || PryTheme.current_theme
-        theme.tr!("+", "") if new_theme_flag = (theme[0] == "+")
-
-        if not new_theme_flag and not installed?(theme)
-          output.puts %(Can't find "#{ theme }" theme in `#{ THEME_DIR }`.)
-          output.puts "To create a new theme, prepend a `+` sign to its name."
-          output.puts "Example: `pry-theme -e +#{ theme }`."
-          return
-        elsif new_theme_flag and installed?(theme)
-          output.puts "Can't create a theme with the given name."
-          output.puts %(The "#{ theme }" theme is already exist in your system.)
-          output.puts "You can edit it with `pry-theme -e #{ theme }` command."
-          return
-        end
+        opt.run do |opts, args|
 
-        theme_path = pathify_theme(theme)
-
-        if new_theme_flag
-          template = <<-TEMPLATE
----
-meta:
-  theme-name  : #{ theme }
-  version     : 1
-  color-depth : 256 # Supports 8 or 256 colors.
-  description : # Should be less than 80 characters.
-  author      : # John Doe <johndoe@example.com>
-
-theme:
-  class               : # blue on yellow
-  class_variable      : # red
-  comment             : # on green
-  constant            : # (bu)
-  error               : # black (b) on white
-  float               : # (i)
-  global_variable     :
-  inline_delimiter    :
-  instance_variable   :
-  integer             :
-  keyword             :
-  method              :
-  predefined_constant :
-  regexp:
-    self              :
-    char              :
-    content           :
-    delimiter         :
-    modifier          :
-    escape            :
-  shell:
-    self              :
-    char              :
-    content           :
-    delimiter         :
-    escape            :
-  string:
-    self              :
-    char              :
-    content           :
-    delimiter         :
-    escape            :
-  symbol              :
-          TEMPLATE
-
-          # Create a template in themes directory.
-          File.open(theme_path, "w") { |f| f.puts template }
-
-          output.puts %(Created "#{ theme }" theme in `#{ THEME_DIR }`.)
-          output.puts "Opened it in #{ Pry.config.editor } for editing."
-        else
-          output.puts "Opened #{ theme } theme in #{ Pry.config.editor } for editing."
-          output.puts "Don't forget to increment a version number of theme!"
+          if opts.present?(:c)
+            stagger_output Preview.new(ThemeList.current_theme).long
+          elsif args.empty?
+            output.puts ThemeList.current_theme.name
+          end
         end
+      end
+    end
 
-        begin
-          old_theme = PryTheme.current_theme
+    def def_edit(cmd)
+      cmd.command :edit do |opt|
+        opt.description 'Edit a theme definition'
 
-          display_preview!(theme, %<Current "#{ theme }">)
-          invoke_editor(theme_path, line=1, reloading=true)
-          display_preview!(theme, %<Edited "#{ theme }">)
-        ensure
-          PryTheme.set_theme(old_theme)
+        opt.run do |opts, args|
+          BasicEditor.edit(args.first || ThemeList.current_theme.name)
         end
       end
+    end
 
-      # Please, pay attention to the fact that this method changes current
-      # theme to the given theme.
-      def display_preview!(theme, header_text)
-        PryTheme.set_theme(theme)
-        display_header(header_text, output)
-        test_theme
-      end
+    def subcommands(cmd)
+      [:def_list, :def_colors, :def_try, :def_edit,
+       :def_uninstall, :def_install, :def_current, :def_convert,
+      ].each { |m| __send__(m, cmd) }
 
-      def show_list
-        old_theme = PryTheme.current_theme.dup
+      cmd.add_callback(:empty) do
+        stagger_output opts.help
+      end
+    end
 
-        all_themes = installed_themes.map do |theme|
-          theme = File.basename(theme, ".prytheme")
-          meta = Theme.new(theme)
-          PryTheme.set_theme(theme)
+    def process
+      # "There's no emptiness in the life of a warrior. Everything is filled to
+      # the brim. Everything is filled to the brim, and everything is equal."
+    end
 
-          chunk = <<-CHUNK
-class Theme
-  def method
-    @ivar, @@cvar, lvar = 10_000, 400.00, "string"
-  end
-end
-          CHUNK
+    def complete(so_far)
+      PryTheme::ThemeList.themes.map(&:name)
+    end
 
-          mark_current = "* " if theme == old_theme
-          header = make_bold("#{mark_current}[#{theme}]")
-          snippet = colorize_code(chunk)
-          [header, meta.description, "---", snippet].compact.join("\n")
-        end
+    private
 
-        lputs all_themes.join("\n"), output
-      ensure
-        PryTheme.set_theme(old_theme)
+    def display_colors(color_model)
+      case color_model
+      when 256 then (stagger_output ColorTable.t256)
+      when 16  then (stagger_output ColorTable.t16)
+      when 8   then (stagger_output ColorTable.t8)
       end
+    end
 
-      def show_remote_list
-        require 'net/https'
-        require 'json'
+    def show_local_list
+      previews = ThemeList.themes.map { |theme| Preview.new(theme).short }
+      stagger_output previews.join("\n")
+    end
 
-        body = {}
-        fetch_collection("/") do |http, uri|
-          output.puts "Fetching list of themes..."
-          response = http.request(Net::HTTP::Get.new(uri.request_uri))
-          body = JSON.parse(response.body)
+    def show_remote_list
+      output.puts 'Fetching the list of themes from Pry Theme Collection...'
+      output.puts "#{windows?? '->':'→'} https://github.com/kyrylo/pry-theme-collection/"
+      body = json_body(PryTheme::PTC)
+
+      themes = body.map { |theme|
+        unless installed?(theme)
+          [theme['name'], theme['html_url']]
         end
+      }.compact
 
-        i = 0
-        remote_themes = body.map do |theme|
-          if (name = theme["name"]) =~ /\A(?:[a-z]|[0-9]|-)+\z/
-            "#{i+=1}. #{installed?(name) ? make_bold(name) : name}"
-          end
-        end.compact
+      out = "--\n"
+      themes.each.with_index(1) { |theme, i|
+        out += "#{ i }. "
+        out += theme.first + "\n"
 
-        lputs remote_themes.join("\n"), output
-      end
+        uri = URI.parse(PryTheme::SHORTENER + theme.last)
+        http = Net::HTTP.new(uri.host, uri.port)
+        response = http.request(Net::HTTP::Get.new(uri.request_uri))
 
-      def install_theme
-        return unless args[0]
+        out += response.body + "\n\n"
+      }
+      stagger_output out.chomp
+    end
 
-        require 'net/https'
-        require 'json'
-        require 'base64'
+    def json_body(address)
+      require 'net/https'
+      require 'json'
+      require 'base64'
+
+      uri = URI.parse(address)
+      http = Net::HTTP.new(uri.host, uri.port)
+      http.use_ssl = true
+      http.verify_mode = OpenSSL::SSL::VERIFY_NONE if windows?
+      response = http.request(Net::HTTP::Get.new(uri.request_uri))
+      JSON.parse(response.body)
+    end
 
-        body = {}
-        fetch_collection("/#{args[0]}/#{args[0]}.prytheme") do |http, uri|
-          output.puts "Fetching theme from the collection..."
-          response = http.request(Net::HTTP::Get.new(uri.request_uri))
-          body = JSON.parse(response.body)
-        end
+    def install_theme(args)
+      args.each { |theme|
+        output.puts %|Installing "#{ theme }" from Pry Theme Collection...|
+        body = json_body(PTC + "%s/%s#{ PT_EXT }" % [theme, theme])
 
-        if body["message"]
-          output.puts "Cannot find theme: #{args[0]}"
-          return
+        if body['message']
+          output.puts %|Cannot find theme "#{ theme }"...|
+          next
         end
 
-        theme = Base64.decode64(body["content"])
-
-        File.open(local_theme("#{args[0]}.prytheme"), "w") do |f|
-          f.puts theme
+        File.open(File.join(USER_THEMES_DIR, "#{theme}.prytheme.rb"), 'w') do |f|
+          f.puts Base64.decode64(body['content'])
         end
+        require File.join(USER_THEMES_DIR, "#{theme}.prytheme.rb")
+        output.puts %|Successfully installed "#{ theme }"!|
+      }
+    end
 
-        output.puts "Successfully installed #{args[0]}!"
-      rescue
-        output.puts "An error occurred!"
+    def convert_color(opts, args)
+      color_model = (opts.present?(:m) ? opts[:m] : PryTheme.tput_colors)
+
+      if opts.present?(:t)
+        color = PryTheme.color_const(color_model).new(
+          :from => :term, :foreground => opts[:t].to_i)
+      elsif opts.present?(:h)
+        color = PryTheme.color_const(color_model).new(
+          :from => :hex, :foreground => opts[:h])
+      elsif opts.present?(:r)
+        color = PryTheme.color_const(color_model).new(
+          :from => :rgb, :foreground => opts[:r])
       end
+      output.puts ColorTable.build_color_string(color)
+    rescue NameError
+      output.puts %|Unknown color model "#{ opts[:m] }". Try 8, 16 or 256|
     end
 
+    def color_model_option_only?(opts)
+      opts[:m] && !(opts[:h] || opts[:r] || opts[:t])
+    end
+
+    def color_model_option_and_other_one?(opts)
+      opts[:m] && (opts[:h] || opts[:r] || opts[:t])
+    end
+
+    def without_color_model?(opts)
+      !opts[:m] && (opts[:h] || opts[:r] || opts[:t])
+    end
+
+    def installed?(theme)
+      theme['name'] == 'README.md' ||
+        ThemeList.themes.map(&:name).include?(theme['name'])
+    end
   end
+
+  Pry::Commands.add_command(PryTheme::Command::PryTheme)
 end