locd 0.1.12 → 0.1.13

data/lib/locd/cli/command/agent/start.rb

@@ -0,0 +1,38 @@
+# encoding: UTF-8
+# frozen_string_literal: true
+
+require 'nrser/refinements/types'
+using NRSER::Types
+
+
+# Namespace
+# =======================================================================
+
+module  Locd
+module  CLI
+module  Command
+
+
+# Definitions
+# =======================================================================
+
+class Agent < Base
+  
+  desc  "start",    "Start an agent"
+
+  include_shared    t[ groups: t.HasAny( :pattern, :multi, :start ) ]
+  
+  def start
+    find_multi!( pattern ).
+      each { |agent| agent.start **option_kwds( groups: :start ) }
+  end
+
+end
+
+
+# /Namespace
+# =======================================================================
+
+end # module  Command
+end # module  CLI
+end # module  Locd

data/lib/locd/cli/command/agent/status.rb

@@ -0,0 +1,41 @@
+# encoding: UTF-8
+# frozen_string_literal: true
+
+require 'nrser/refinements/types'
+using NRSER::Types
+
+
+# Namespace
+# =======================================================================
+
+module  Locd
+module  CLI
+module  Command
+
+
+# Definitions
+# =======================================================================
+
+class Agent < Base
+  
+  desc            'status',
+                  "Print agent status"
+  
+  include_shared  t[ groups: :pattern ]
+  
+  def status
+    agent = find_only! pattern
+    respond \
+      label:  agent.label,
+      status: agent.status.to_h( compact: false )
+  end
+
+end
+
+
+# /Namespace
+# =======================================================================
+
+end # module  Command
+end # module  CLI
+end # module  Locd

data/lib/locd/cli/command/agent/stop.rb

@@ -0,0 +1,39 @@
+# encoding: UTF-8
+# frozen_string_literal: true
+
+require 'nrser/refinements/types'
+using NRSER::Types
+
+
+# Namespace
+# =======================================================================
+
+module  Locd
+module  CLI
+module  Command
+
+
+# Definitions
+# =======================================================================
+
+class Agent < Base
+  
+  desc            "stop",
+                  "Stop an agent"
+  
+  include_shared  t[ groups: t.HasAny( :pattern, :multi, :stop ) ]
+  
+  def stop
+    kwds = option_kwds groups: :stop
+    find_multi!( pattern ).each { |agent| agent.stop **kwds }
+  end
+
+end
+
+
+# /Namespace
+# =======================================================================
+
+end # module  Command
+end # module  CLI
+end # module  Locd

data/lib/locd/cli/command/agent/tail.rb

@@ -0,0 +1,82 @@
+# encoding: UTF-8
+# frozen_string_literal: true
+
+require 'nrser/refinements/types'
+using NRSER::Types
+
+
+# Namespace
+# =======================================================================
+
+module  Locd
+module  CLI
+module  Command
+
+
+# Definitions
+# =======================================================================
+
+class Agent < Base
+
+  desc              'tail [OPTIONS] PATTERN [-- TAIL_OPTIONS]',
+                    "Tail agent logs"
+  
+  include_shared    t[ groups: :pattern ]
+  
+  option            :stream,
+                    desc: "Stream to tail. May omit if uses single log file.",
+                    aliases: ['-s'],
+                    type: :string,
+                    enum: ['out', 'err']
+  
+  option            :follow,
+                    desc: "Run with `tail -F`",
+                    aliases: [ 'f', 'F' ],
+                    type: :boolean
+  
+  def tail          *tail_options
+    agent = find_only! pattern
+    
+    path = case options[:stream]
+    when nil
+      paths = agent.log_paths
+      
+      unless paths.length == 1
+        raise Thor::RequiredArgumentMissingError.new binding.erb <<~END
+          Agent `<%= agent.label %>` has multiple log files.
+          
+          out: <%= agent.out_path.to_s %>
+          err: <%= agent.err_path.to_s %>
+          
+          Must specify one via the `--stream` option.
+          
+        END
+      end
+      
+      paths[0]
+      
+    when 'out'
+      agent.out_path
+      
+    when 'err'
+      agent.err_path
+      
+    else
+      raise "WTF"
+    end
+    
+    cmd = ['tail']
+    cmd += ['-F'] if options[:follow]
+    
+    exec *cmd, *tail_options, path.to_s
+  end # #tail
+
+end
+
+
+# /Namespace
+# =======================================================================
+
+end # module  Command
+end # module  CLI
+end # module  Locd

data/lib/locd/cli/command/agent/truncate_logs.rb

@@ -0,0 +1,65 @@
+# encoding: UTF-8
+# frozen_string_literal: true
+
+require 'nrser/refinements/types'
+using NRSER::Types
+
+
+# Namespace
+# =======================================================================
+
+module  Locd
+module  CLI
+module  Command
+
+
+# Definitions
+# =======================================================================
+
+class Agent < Base
+
+  desc              'truncate-logs',
+                    "Truncate agent log file(s)."
+  
+  include_shared    t[ groups: t.has_any( :pattern, :multi ) ]
+  
+  option            :restart,
+                    desc: "Restart the agent after truncation",
+                    type: :boolean,
+                    default: true
+  
+  def truncate_logs
+    find_multi!( pattern ).each do |agent|
+      log_paths = [agent.out_path, agent.err_path].compact.uniq
+      
+      unless log_paths.empty?
+        restart = options[:restart] && agent.running?
+        
+        agent.stop if restart
+        
+        log_paths.each do |log_path|
+          begin
+            log_path.open( 'w' ) { |f| f.truncate 0 }
+          rescue Exception => error
+            logger.error "Failed to truncate #{ log_path }", error
+          else
+            logger.info "Truncated",
+              'file' => log_path.to_s,
+              'agent.label' => agent.label
+          end
+        end # each log_path
+        
+        agent.start if restart
+      end # unless log_paths.empty?
+    end # each agent
+  end # #truncate_logs
+
+end
+
+
+# /Namespace
+# =======================================================================
+
+end # module  Command
+end # module  CLI
+end # module  Locd

data/lib/locd/cli/command/agent/update.rb

@@ -0,0 +1,48 @@
+# encoding: UTF-8
+# frozen_string_literal: true
+
+require 'nrser/refinements/types'
+using NRSER::Types
+
+
+# Namespace
+# =======================================================================
+
+module  Locd
+module  CLI
+module  Command
+
+
+# Definitions
+# =======================================================================
+
+class Agent < Base
+  
+  desc              "update PATTERN [OPTIONS] [-- CMD_TEMPLATE...]",
+                    "Update an existing agent"
+  
+  include_shared  t[  groups:
+                        t.has_any(  :write,
+                                    :add,
+                                    :respond_with_agents  ) ]
+  
+  def update        *cmd_template
+    agent = find_only! pattern
+    
+    new_agent = agent.update \
+      cmd_template: cmd_template,
+      **option_kwds( groups: :write )
+    
+    logger.info "Agent `#{ agent.label }` updated"
+    
+    respond agent
+  end
+
+end
+
+# /Namespace
+# =======================================================================
+
+end # module  Command
+end # module  CLI
+end # module  Locd

data/lib/locd/cli/command/base.rb

@@ -11,6 +11,7 @@ require 'shellwords'
 # -----------------------------------------------------------------------
 require 'thor'
 require 'nrser/labs/i8'
+require 'rouge'
 
 # Project / Package
 # -----------------------------------------------------------------------
@@ -155,11 +156,27 @@ class Base < Thor
         end
       end
     end # #render_text
+
+
+    def render_color?
+      if options.key? :color
+        options[:color]
+      else
+        $stdout.isatty
+      end
+    end
+
+
+    def render_color src, lexer_name
+      formatter = Rouge::Formatters::Terminal256.new
+      lexer = Rouge::Lexers.const_get( lexer_name ).new
+      formatter.format( lexer.lex( src ) )
+    end
     
     
     def respond message, title: nil
       formatted = if options[:json]
-        begin
+        json = begin
           JSON.pretty_generate message
         rescue JSON::GeneratorError => error
           logger.debug "Error generating 'pretty' JSON, falling back to dump",
@@ -167,10 +184,25 @@ class Base < Thor
           
           JSON.dump message
         end
+
+        if render_color?
+          render_color json, :JSON
+        else
+          json
+        end
         
       elsif options[:yaml] || NRSER.hash_like?( message )
         # TODO  This sucks, but it's the easiest way to get "nice" YAML :/
-        YAML.dump JSON.load( JSON.dump message )
+        yaml = YAML.dump JSON.load( JSON.dump message )
+
+        if render_color?
+          # formatter = Rouge::Formatters::Terminal256.new
+          # lexer = Rouge::Lexers::YAML.new
+          # formatter.format( lexer.lex( yaml ) )
+          render_color yaml, :YAML
+        else
+          yaml
+        end
         
       else
         render_text message
@@ -202,7 +234,7 @@ class Base < Thor
         if error.count == 0
           logger.error "No results"
         else
-          logger.error "Too many results:\n\n#{ render_text error.subject }\n"
+          logger.error "Too many results:\n\n#{ render_text error.value }\n"
         end
       
         # If the command supports `--all`, let them know that they can use it

data/lib/locd/cli/command/job.rb

@@ -2,6 +2,14 @@
 # frozen_string_literal: true
 
 
+# Requirements
+# ============================================================================
+
+### Project / Package ###
+
+require_relative './agent'
+
+
 # Refinements
 # =======================================================================
 
@@ -9,12 +17,19 @@ require 'nrser/refinements/types'
 using NRSER::Types
 
 
+# Namespace
+# =======================================================================
+
+module  Locd
+module  CLI
+module  Command
+
 # Definitions
 # =======================================================================
 
 # TODO  Doc me pls
 # 
-class Locd::CLI::Command::Job < Locd::CLI::Command::Agent
+class Job < Agent
   
   # Helpers
   # ============================================================================
@@ -80,39 +95,17 @@ class Locd::CLI::Command::Job < Locd::CLI::Command::Agent
   public
   
   
-  # Write Commands
-  # ----------------------------------------------------------------------------
-  
-  desc              "add [OPTIONS] -- CMD_TEMPLATE...",
-                    "Add job that runs in the current directory"
-  include_options   groups: [:write, :add, :respond_with_agents]
-  option            :every,
-                    desc: "How often to start the job",
-                    type: :string
-  option            :at,
-                    desc: "Day/time to start the job",
-                    type: :hash
-  def add           *cmd_template, **kwds
-    logger.trace "#{ self.class.name }##{ __method__ }",
-      options: options,
-      cmd_template: cmd_template,
-      kwds: kwds,
-      shared: self.class.shared_method_options
-    
-    kwds[:start_interval] ||= get_start_interval
-    
-    super *cmd_template, **kwds
-    
-    # job = agent_class.add \
-    #   cmd_template: cmd_template,
-    #   start_interval: get_start_interval,
-    #   **option_kwds( groups: [:write, :add] )
-    # 
-    # logger.info "`#{ job.label }` job added."
-    # 
-    # job.reload if options[:load]
-    # 
-    # respond job
-  end
+  # Commands
+  # ============================================================================
+
+  require_relative './job/add'
   
-end # class Locd::CLI::Command::Job
+end # class Job
+
+
+# /Namespace
+# =======================================================================
+
+end # module  Command
+end # module  CLI
+end # module  Locd