teems 0.1.0

data/lib/teems/commands/base.rb

@@ -0,0 +1,146 @@
+# frozen_string_literal: true
+
+module Teems
+  module Commands
+    # Output helpers for CLI commands: logging, printing, JSON
+    module CommandOutput
+      protected
+
+      def success(msg) = @options[:quiet] || output.success(msg)
+      def info(msg) = @options[:quiet] || output.info(msg)
+      def warn(message) = output.warn(message)
+      def error(message) = output.error(message) || 1
+      def debug(msg) = @options[:verbose] && output.debug(msg)
+      def puts(message = '') = @options[:quiet] || output.puts(message)
+      def print(msg) = @options[:quiet] || output.print(msg)
+      def output_json(data) = output.puts(JSON.pretty_generate(data))
+    end
+
+    # Option validation helpers for CLI commands
+    module CommandValidation
+      protected
+
+      def check_unknown_options
+        return nil if @unknown_options.empty?
+
+        error("Unknown option: #{@unknown_options.first}")
+        error('Run with --help for available options.')
+        1
+      end
+
+      def unknown_options? = @unknown_options.any?
+      def show_help? = @options[:help]
+
+      def show_help
+        output.puts help_text
+        0
+      end
+
+      def validate_options
+        return show_help if show_help?
+        return check_unknown_options if unknown_options?
+
+        nil
+      end
+    end
+
+    # Base class for all CLI commands with option parsing and output helpers
+    class Base
+      include CommandOutput
+      include CommandValidation
+
+      attr_reader :runner, :options, :positional_args
+
+      def initialize(args, runner:)
+        @runner = runner
+        @options = default_options
+        @unknown_options = []
+        @positional_args = []
+        parse_options(args)
+      end
+
+      def execute
+        raise NotImplementedError, 'Subclass must implement #execute'
+      end
+
+      protected
+
+      # Convenience accessors
+      def output = runner.output
+      def config = runner.config
+      def cache_store = runner.cache_store
+      def token_store = runner.token_store
+      def api_client = runner.api_client
+
+      def default_options
+        { verbose: false, quiet: false, json: false, limit: 20 }
+      end
+
+      def parse_options(args)
+        pending = args.dup
+        while pending.any?
+          arg = pending.shift
+          if arg.start_with?('-')
+            parse_single_option(arg, pending)
+          else
+            @positional_args << arg
+          end
+        end
+        @positional_args
+      end
+
+      COMMON_OPTIONS = {
+        '-v' => :verbose, '--verbose' => :verbose,
+        '-q' => :quiet, '--quiet' => :quiet,
+        '--json' => :json, '-h' => :help, '--help' => :help
+      }.freeze
+
+      LIMIT_FLAGS = %w[-n --limit].freeze
+
+      private
+
+      def parse_single_option(arg, pending)
+        return @options[:limit] = pending.shift.to_i if LIMIT_FLAGS.include?(arg)
+        return @options[COMMON_OPTIONS[arg]] = true if COMMON_OPTIONS.key?(arg)
+
+        handle_option(arg, pending)
+      end
+
+      protected
+
+      # Override in subclass to handle command-specific options
+      def handle_option(arg, _pending)
+        @unknown_options << arg
+      end
+
+      def help_text
+        'No help available for this command.'
+      end
+
+      def require_auth
+        return nil if runner.configured?
+
+        error('Not authenticated. Run: teems auth login')
+        1
+      end
+
+      # Execute a block with automatic token refresh on 401
+      # Usage: with_token_refresh { runner.messages_api.chat_messages(chat_id: id) }
+      def with_token_refresh
+        yield
+      rescue ApiError => e
+        raise unless e.unauthorized? || e.message.include?('expired')
+
+        attempt_token_refresh
+        yield
+      end
+
+      def attempt_token_refresh
+        debug('Token expired, attempting refresh...')
+        raise ApiError, 'Token refresh failed' unless runner.refresh_tokens
+
+        debug('Token refreshed, retrying request...')
+      end
+    end
+  end
+end