mtncloud-cli 1.0.0

data/lib/mtncloud/cli/commands/coloring_command.rb

@@ -0,0 +1,82 @@
+require 'mtncloud/cli/cli_command'
+
+# This is for use in dotfile scripts
+# It allows you to turn colors on or off globally
+class MtnCloud::Cli::ColoringCommand
+  include MtnCloud::Cli::CliCommand
+  set_command_name :coloring
+  set_command_hidden
+
+  def handle(args)
+    append_newline = true
+    options = {}
+    optparse = MtnCloud::Cli::OptionParser.new do|opts|
+      opts.banner = "Usage: #{prog_name} #{command_name} [on|off]"
+      #build_common_options(opts, options, [])
+      opts.on('-q','--quiet', "No Output, do not print to stdout") do
+        options[:quiet] = true
+      end
+      opts.on('-h', '--help', "Print this help" ) do
+        puts opts
+        exit
+      end
+      opts.footer = "Enable [on] or Disable [off] ANSI Colors for all output.\n" + 
+                    "Use [on?] or [off?] to print the current value and exit accordingly." + "\n" +
+                    "Passing no arguments is the same as `#{command_name} on`"
+    end
+    optparse.parse!(args)
+    if args.count > 1
+      raise_command_error "wrong number of arguments, expected 0-1 and got (#{args.count}) #{args.join(' ')}\n#{optparse}"
+    end
+    
+    coloring_was_enabled = Term::ANSIColor::coloring?
+    exit_code = 0
+
+    if args.count == 0
+      # no args means turn it on.
+      Term::ANSIColor::coloring = true
+    else
+      subcmd = args[0].to_s.strip
+      if subcmd == "on"
+        Term::ANSIColor::coloring = true
+      elsif subcmd == "off"
+        Term::ANSIColor::coloring = false
+        
+      elsif subcmd == "on?"
+        exit_code = Term::ANSIColor::coloring? ? 0 : 1
+      elsif subcmd == "off?"
+        exit_code = Term::ANSIColor::coloring? ? 1 : 0
+      else
+        puts optparse
+        return 127
+      end
+    end
+    unless options[:quiet]
+      if Term::ANSIColor::coloring?
+        if coloring_was_enabled == false
+          MtnCloud::Logging::DarkPrinter.puts "coloring enabled" if MtnCloud::Logging.debug?
+          recalculate_after_color_change()
+        end
+        puts "#{cyan}coloring: #{bold}#{green}on#{reset}"
+      else
+        if coloring_was_enabled == true
+          MtnCloud::Logging::DarkPrinter.puts "coloring disabled" if MtnCloud::Logging.debug?
+          recalculate_after_color_change()
+        end
+        puts "coloring: off"
+      end
+    end
+    return exit_code
+  end
+
+  protected
+
+  def recalculate_after_color_change()
+    # recalculate shell prompt after this change
+    MtnCloud::Cli::Echo.recalculate_variable_map()
+    if MtnCloud::Cli::Shell.has_instance?
+      MtnCloud::Cli::Shell.instance.recalculate_prompt()
+    end
+  end
+
+end