kscript 0.1.0

data/lib/kscript/plugins/kk_rename_utils.rb

@@ -0,0 +1,82 @@
+# frozen_string_literal: true
+
+# Copyright (c) 2025 Kk
+#
+# This software is released under the MIT License.
+# https://opensource.org/licenses/MIT
+
+require 'kscript'
+
+module Kscript
+  class KkRenameUtils < Base
+    attr_reader :source_pattern, :target_pattern, :directory
+
+    def initialize(source_pattern = nil, target_pattern = nil, directory = Dir.pwd, *_args, **opts)
+      super(**opts.merge(service: 'kk_rename'))
+      @source_pattern = source_pattern
+      @target_pattern = target_pattern
+      @directory = directory
+    end
+
+    def run
+      with_error_handling do
+        rename
+      end
+    end
+
+    def rename
+      Dir.entries(@directory).each do |filename|
+        process_file(filename)
+      end
+    end
+
+    def self.arguments
+      '<pattern> <replacement> [path]'
+    end
+
+    def self.usage
+      "kscript rename foo bar ./src\nkscript rename 'test' 'prod' ~/projects"
+    end
+
+    def self.group
+      'project'
+    end
+
+    def self.author
+      'kk'
+    end
+
+    def self.description
+      'Batch rename files by pattern.'
+    end
+
+    private
+
+    def process_file(filename)
+      return unless should_process?(filename)
+
+      new_name = generate_new_name(filename)
+      return unless new_name
+
+      rename_file(filename, new_name)
+    end
+
+    def should_process?(filename)
+      File.file?(File.join(@directory, filename)) && filename =~ /#{@source_pattern}/
+    end
+
+    def generate_new_name(filename)
+      eval("\"#{filename}\"".gsub(/#{@source_pattern}/, @target_pattern))
+    rescue StandardError => e
+      logger.kerror("Error processing #{filename}: #{e.message}")
+      nil
+    end
+
+    def rename_file(old_name, new_name)
+      File.rename(File.join(@directory, old_name), File.join(@directory, new_name))
+      logger.kinfo("Renamed: #{old_name} -> #{new_name}")
+    rescue StandardError => e
+      logger.kerror("Error renaming #{old_name}: #{e.message}")
+    end
+  end
+end

data/lib/kscript/plugins/kk_sh_utils.rb

@@ -0,0 +1,112 @@
+# frozen_string_literal: true
+
+# Copyright (c) 2025 Kk
+#
+# This software is released under the MIT License.
+# https://opensource.org/licenses/MIT
+
+require 'kscript'
+require 'nokogiri'
+
+module Kscript
+  class KkShUtils < Base
+    CHT_SH_URL = 'https://cht.sh'
+
+    attr_reader :command
+
+    # Initialize with shell command to look up
+    # @param command [String] command to get help for
+    def initialize(command = nil, *_args, **opts)
+      super(**opts.merge(service: 'kk_sh'))
+      @command = command
+    end
+
+    def run
+      with_error_handling do
+        help
+      end
+    end
+
+    def help
+      if command
+        fetch_help
+      else
+        logger.kinfo("Usage: #{$PROGRAM_NAME} <command>")
+        exit 1
+      end
+    end
+
+    # Fetch and display command documentation
+    def fetch_help
+      response = make_request
+      display_result(response)
+    rescue HTTP::Error => e
+      display_error(e)
+    end
+
+    def self.arguments
+      '[subcommand] [args...]'
+    end
+
+    def self.usage
+      "kscript sh 'ls'\nkscript sh 'echo hello'"
+    end
+
+    def self.group
+      'system'
+    end
+
+    def self.author
+      'kk'
+    end
+
+    def self.description
+      'Query sh command usage and cheatsheets.'
+    end
+
+    private
+
+    # Make HTTP request to cht.sh
+    # @return [HTTP::Response] response from cht.sh
+    def make_request
+      HTTP.get("#{CHT_SH_URL}/#{command}")
+    end
+
+    # Display command documentation
+    # @param response [HTTP::Response] response from cht.sh
+    def display_result(response)
+      if response.status.success?
+        text = extract_plain_text(response.body)
+        logger.kinfo(text)
+      else
+        logger.kerror("Failed to retrieve data: #{response.status}")
+      end
+    end
+
+    # 提取纯文本内容
+    def extract_plain_text(body)
+      body = body.to_s
+      begin
+        doc = Nokogiri::HTML(body)
+        doc.search('script,style').remove
+        text = doc.text.lines.map(&:strip)
+        # 过滤掉包含广告/社交/Follow等内容的行
+        text = text.reject { |line| line =~ /Follow @|twitter|github|sponsor|donate|chubin|^!function/ }
+        text.reject!(&:empty?)
+        # 去除顶部多余空白行，只保留正文
+        text = text.drop_while(&:empty?)
+        text.join("\n")
+      rescue StandardError
+        body.gsub(/\e\[[\d;]*m/, '').lines.reject do |line|
+          line =~ /Follow @|twitter|github|sponsor|donate|chubin|^!function/
+        end.drop_while { |line| line.strip.empty? }.join
+      end
+    end
+
+    # Display error message
+    # @param error [StandardError] error to display
+    def display_error(error)
+      logger.kerror("An error occurred: #{error.message}")
+    end
+  end
+end

data/lib/kscript/plugins/kk_syscheck_utils.rb

@@ -0,0 +1,82 @@
+# frozen_string_literal: true
+
+# Copyright (c) 2025 Kk
+#
+# This software is released under the MIT License.
+# https://opensource.org/licenses/MIT
+
+require 'kscript'
+
+module Kscript
+  class KkSyscheckUtils < Base
+    def run
+      with_error_handling do
+        check
+      end
+    end
+
+    def check
+      logger.kinfo('======= 🍎 macOS System Resource Monitor Report =======')
+      logger.kinfo("📅 Date Time: #{Time.now}")
+      logger.kinfo('')
+
+      # macOS ps command for process monitoring
+      cpu_cmd = 'ps aux | sort -nrk 3 | head -n 10'
+      mem_cmd = 'ps aux | sort -nrk 4 | head -n 10'
+
+      run('CPU Usage (Top 10)', cpu_cmd)
+      run('Memory Usage (Top 10)', mem_cmd)
+
+      if `which powermetrics`.strip.empty?
+        logger.kinfo("\n👉 GPU Usage:")
+        logger.kwarn('⚠️ powermetrics not installed, please run: xcode-select --install')
+      else
+        run('GPU Usage', 'powermetrics --samplers gpu_power -n 1', sudo: true)
+      end
+
+      logger.kinfo("\n👉 Top 10 Processes by Network Connections:")
+      lsof_output = `lsof -i -nP | grep ESTABLISHED`
+      counts = Hash.new(0)
+      lsof_output.each_line do |line|
+        process = line.split.first
+        counts[process] += 1
+      end
+      counts.sort_by { |_, v| -v }.first(10).each do |proc, count|
+        logger.kinfo("#{proc}: #{count} connections")
+      end
+
+      logger.kinfo("\n👉 System Overview:")
+      cpu_core = `sysctl -n hw.ncpu`.strip
+      mem_size = `sysctl -n hw.memsize`.to_i / 1024 / 1024 / 1024
+      logger.kinfo("CPU Cores: #{cpu_core}")
+      logger.kinfo("Physical Memory: #{mem_size} GB")
+
+      vm_stats = `vm_stat`
+      vm_stats.each_line do |line|
+        logger.kinfo(line) if line =~ /Pages (active|wired down|free):/
+      end
+
+      logger.kinfo("\n✅ System resource check completed!")
+    end
+
+    def self.description
+      'Show macOS system resource monitor report.'
+    end
+
+    def self.arguments
+      ''
+    end
+
+    def self.usage
+      "kscript syscheck\nkscript syscheck --detail"
+    end
+
+    def self.group
+      'macos'
+    end
+
+    def self.author
+      'kk'
+    end
+  end
+end

data/lib/kscript/plugins/kk_top_utils.rb

@@ -0,0 +1,74 @@
+# frozen_string_literal: true
+
+# Copyright (c) 2025 Kk
+#
+# This software is released under the MIT License.
+# https://opensource.org/licenses/MIT
+
+require 'kscript'
+
+# 彩色输出定义
+RED    = "\e[1;31m"
+GREEN  = "\e[1;32m"
+YELLOW = "\e[1;33m"
+CYAN   = "\e[1;36m"
+NC     = "\e[0m" # No Color
+
+module Kscript
+  class KkTopUtils < Base
+    def run
+      with_error_handling do
+        print_report
+      end
+    end
+
+    def print_report
+      logger.kinfo("System Resource Top Report - #{Time.now.strftime('%Y-%m-%d %H:%M:%S')}")
+      print_header('Top 10 Processes by CPU Usage')
+      print_process_list(:cpu)
+      print_header('Top 10 Processes by Memory Usage')
+      print_process_list(:mem)
+    end
+
+    def print_header(title)
+      logger.kinfo('')
+      logger.kinfo('===============================')
+      logger.kinfo(" #{title}")
+      logger.kinfo('===============================')
+    end
+
+    def print_process_list(sort_field)
+      lines = `ps aux`.split("\n")
+      lines.shift
+      processes = lines.map { |line| line.split(/\s+/, 11) }
+      index = sort_field == :cpu ? 2 : 3
+      top = processes.sort_by { |p| -p[index].to_f }.first(10)
+      logger.kinfo(format('%-10s %-8s %-5s %-5s %-10s', 'USER', 'PID', '%CPU', '%MEM', 'COMMAND'))
+      top.each do |p|
+        logger.kinfo(format('%-10s %-8s %-5s %-5s %-10s', p[0], p[1], p[2], p[3], p[10][0..30]))
+      end
+    end
+
+    def self.description
+      'Show top 10 processes by CPU/memory on macOS.'
+    end
+
+    def self.arguments
+      ''
+    end
+
+    def self.usage
+      'kscript top'
+    end
+
+    def self.group
+      'macos'
+    end
+
+    def self.author
+      'kk'
+    end
+  end
+end
+
+Kscript::KkTopUtils.new.run if __FILE__ == $PROGRAM_NAME

data/lib/kscript/plugins/kk_usd_utils.rb

@@ -0,0 +1,71 @@
+# frozen_string_literal: true
+
+# Copyright (c) 2025 Kk
+#
+# This software is released under the MIT License.
+# https://opensource.org/licenses/MIT
+
+require 'kscript'
+
+require 'net/http'
+require 'json'
+
+module Kscript
+  class KkUsdUtils < Base
+    API_URL = 'https://api.exchangerate-api.com/v4/latest/USD'
+
+    def initialize(currency_code = nil, *_args, **opts)
+      super(**opts.merge(service: 'kk_usd'))
+      @currency_code = currency_code
+    end
+
+    def run
+      with_error_handling do
+        fetch_rates
+      end
+    end
+
+    def fetch_rates
+      uri = URI(API_URL)
+      response = Net::HTTP.get(uri)
+      data = JSON.parse(response)
+      if @currency_code && data['rates'][@currency_code.upcase]
+        rate = data['rates'][@currency_code.upcase]
+        if human_output?
+          logger.kinfo("1 USD = #{rate} #{@currency_code.upcase}")
+        else
+          logger.kinfo("USD -> #{@currency_code.upcase}", rate: rate)
+        end
+      elsif @currency_code
+        if human_output?
+        end
+        logger.kerror("Currency code not found: #{@currency_code}")
+      elsif human_output?
+        logger.kinfo('USD Exchange Rates:')
+        data['rates'].each { |k, v| logger.kinfo("  1 USD = #{v} #{k}") }
+      else
+        logger.kinfo('USD rates', rates: data['rates'])
+      end
+    end
+
+    def self.arguments
+      '[currency_code]'
+    end
+
+    def self.usage
+      "kscript usd CNY\nkscript usd EUR"
+    end
+
+    def self.group
+      'finance'
+    end
+
+    def self.author
+      'kk'
+    end
+
+    def self.description
+      'Get latest USD exchange rates.'
+    end
+  end
+end

data/lib/kscript/plugins/kk_wg_acl_utils.rb

@@ -0,0 +1,95 @@
+# frozen_string_literal: true
+
+# Copyright (c) 2025 Kk
+#
+# This software is released under the MIT License.
+# https://opensource.org/licenses/MIT
+
+require 'kscript'
+
+module Kscript
+  class KkWgAclUtils < Base
+    WIREGUARD_PORT = 51_821
+    ALLOWED_IPS = %w[127.0.0.1].freeze
+
+    def run
+      with_error_handling do
+        apply_rules
+      end
+    end
+
+    def apply_rules
+      add_whitelist_rules
+      ensure_firewall_enabled
+      display_current_rules
+    end
+
+    def self.arguments
+      '[subcommand] [options]'
+    end
+
+    def self.usage
+      "kscript wg_acl add --ip=10.0.0.2\nkscript wg_acl list"
+    end
+
+    def self.group
+      'network'
+    end
+
+    def self.author
+      'kk'
+    end
+
+    def self.description
+      'Manage WireGuard firewall ACL rules.'
+    end
+
+    private
+
+    # Fetch current UFW rules
+    # @return [Array<String>] list of current firewall rules
+    def fetch_current_rules
+      `sudo ufw status`.lines.map(&:strip)
+    end
+
+    # Check if a specific rule exists
+    # @param ip [String] IP address to check
+    # @param port [Integer] port number to check
+    # @return [Boolean] true if rule exists
+    def rule_exists?(ip, port)
+      @current_rules.any? do |line|
+        line.match?(/#{Regexp.escape(ip)}.*ALLOW.*#{port}/)
+      end
+    end
+
+    # Add whitelist rules for allowed IPs
+    def add_whitelist_rules
+      ALLOWED_IPS.each do |ip|
+        if rule_exists?(ip, WIREGUARD_PORT)
+          logger.kinfo("✅ Rule exists: #{ip} → #{WIREGUARD_PORT}, skipping.")
+        else
+          logger.kinfo("👉 Adding rule: allow #{ip} to access port #{WIREGUARD_PORT}")
+          system("sudo ufw allow from #{ip} to any port #{WIREGUARD_PORT}")
+        end
+      end
+    end
+
+    # Ensure UFW firewall is enabled
+    def ensure_firewall_enabled
+      ufw_status = `sudo ufw status`.strip
+      if ufw_status.start_with?('Status: inactive')
+        logger.kinfo('🔧 UFW is currently disabled, enabling...')
+        system('sudo ufw enable')
+      else
+        logger.kinfo('✅ UFW is enabled.')
+      end
+    end
+
+    # Display current firewall rules
+    def display_current_rules
+      logger.kinfo("\n📋 Current firewall rules:")
+      system('sudo ufw status verbose')
+      logger.kinfo("\n✅ Firewall rules update completed!")
+    end
+  end
+end

data/lib/kscript/plugins/kk_wg_pass_utils.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+# Copyright (c) 2025 Kk
+#
+# This software is released under the MIT License.
+# https://opensource.org/licenses/MIT
+
+require 'kscript'
+require 'bcrypt'
+
+module Kscript
+  class KkWgPassUtils < Base
+    def initialize(length = 32, *_args, **opts)
+      super(**opts.merge(service: 'kk_wg_pass'))
+      @length = length.to_i
+    end
+
+    def run
+      with_error_handling do
+        generate
+      end
+    end
+
+    def generate
+      password = Array.new(@length) { rand(33..126).chr }.join
+      logger.kinfo('Generated WireGuard password', password: password)
+      logger.kinfo(password)
+    end
+
+    def self.arguments
+      '[length]'
+    end
+
+    def self.usage
+      "kscript wg_pass [length]\nkscript wg_pass 32"
+    end
+
+    def self.group
+      'network'
+    end
+
+    def self.author
+      'kk'
+    end
+
+    def self.description
+      'Generate a random password for WireGuard.'
+    end
+  end
+end

data/lib/kscript/plugins.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module Kscript
+  module PluginLoader
+    PLUGIN_DIR = File.expand_path('plugins', __dir__)
+
+    def self.load_all
+      Dir.glob(File.join(PLUGIN_DIR, 'kk_*.rb')).sort.each do |file|
+        # 解析出类名
+        basename = File.basename(file, '.rb')
+        class_name = basename.split('_').map(&:capitalize).join
+        # 只在未定义时 require
+        require file unless Kscript.const_defined?(class_name, false)
+      end
+    end
+
+    # 返回所有已注册插件的元信息
+    def self.plugin_infos
+      Kscript::Plugin.all.map do |name, klass|
+        {
+          name: name,
+          class: klass,
+          description: klass.respond_to?(:description) ? klass.description : nil,
+          arguments: klass.respond_to?(:arguments) ? klass.arguments : nil,
+          usage: klass.respond_to?(:usage) ? klass.usage : nil,
+          group: klass.respond_to?(:group) ? klass.group : nil,
+          author: klass.respond_to?(:author) ? klass.author : nil
+        }
+      end
+    end
+  end
+end

data/lib/kscript/utils.rb

@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+
+require 'yaml'
+require 'fileutils'
+
+module Kscript
+  module Utils
+    class Config
+      DEFAULT_PATH = File.expand_path('~/.kscriptrc')
+
+      def self.load
+        @load ||= new
+      end
+
+      def initialize
+        @data = File.exist?(DEFAULT_PATH) ? YAML.load_file(DEFAULT_PATH) : {}
+      end
+
+      def [](key)
+        @data[key.to_s] || ENV["KSCRIPT_#{key.to_s.upcase}"]
+      end
+
+      def log_level
+        self['log_level']
+      end
+
+      def trace_id
+        self['trace_id']
+      end
+
+      # 自动检测并安装 shell 补全脚本
+      def self.ensure_completion_installed(shell = nil)
+        shell ||= ENV['SHELL']
+        home = Dir.respond_to?(:home) ? Dir.home : ENV['HOME']
+        return unless shell && home
+
+        if shell.include?('zsh')
+          completion_path = File.join(home, '.zsh/completions/_kscript')
+          shell_type = 'zsh'
+        elsif shell.include?('bash')
+          completion_path = File.join(home, '.bash_completion.d/kscript')
+          shell_type = 'bash'
+        else
+          return
+        end
+
+        # 已存在则跳过
+        return if File.exist?(completion_path)
+
+        # 生成补全脚本内容
+        require_relative 'cli'
+        script = case shell_type
+                 when 'zsh' then Kscript::CLI.new.completion('zsh', capture: true)
+                 when 'bash' then Kscript::CLI.new.completion('bash', capture: true)
+                 end
+        FileUtils.mkdir_p(File.dirname(completion_path))
+        File.write(completion_path, script)
+        puts "\e[32m[kscript] Shell completion installed to #{completion_path}\e[0m"
+      rescue StandardError => e
+        warn "[kscript] Failed to install shell completion: #{e.message}"
+      end
+    end
+  end
+end

data/lib/kscript/version.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+# Copyright (c) 2024 Kk
+#
+# This software is released under the MIT License.
+# https://opensource.org/licenses/MIT
+
+module Kscript
+  VERSION = '0.1.0'
+end