kscript 1.0.2 → 1.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 812d19779e0fadd4fc72f94999c3d8776a2391147c467c14067d203050c94c01
-  data.tar.gz: adad5fb5f74798030650f6595f8e90fc86eacfb490a4b7bb5c2fefca1d4552d7
+  metadata.gz: a38a099a45b7be84d12e6b0122dae45eb9891dab7ec2b24804b0c2628111de9a
+  data.tar.gz: 7b60a6c3aa2d320fe4f92e9bc23affcc6cd208aa6bf4ee68ccb154d92008269a
 SHA512:
-  metadata.gz: e2fff6ee0202f526fb11452d3848ce5d2c0f872bb6c457ebbf8dec4f55e45c0532fbd1b8daddc0d58133814c83553ef6a9f8e063c1ef4812bdf9bd4c869ec364
-  data.tar.gz: 6866406cb4c5fd9c74b079c636d778171099db330309f38e13dcfd85f2adbdcd3aa2d96a453b8ad46f46fc505353d4c175b0c40a41732743da7073ce386f2c43
+  metadata.gz: 41ce72c538b84c8d27ced7099f4072cea6ec794a4ee758f568a61461263a5bf38d40c6ff7874a15361c9ead95834dd1c028fa5efe1c1dee78832a919b76e2554
+  data.tar.gz: de04332a98285246ea14f7dc583d2a89248a90594ca7ac00cee9c628d8181bacea464f88a2e3be368ca22e8667430302fff8ae1947856941fd67d069346131b5

data/README.md

@@ -25,6 +25,7 @@ _  ,<  _(__  )/ /__ _  /   _  / __  /_/ / /_
 - **参数健壮**：所有插件兼容多余参数，支持全局参数过滤
 - **CI/CD 自动发布**：GitHub Actions 自动构建并推送 RubyGem
 - **多平台支持**：macOS、Linux、Windows（部分工具）
+- **自动生成配置**：首次安装自动生成 `~/.kscript/.env` 配置示例
 
 ---
 
@@ -70,6 +71,7 @@ kscript apnic CN
 kscript portscan 192.168.1.1
 kscript sh 'ls -l'
 kscript projscan ~/projects
+kscript aws_s3 --file local.txt --bucket my-bucket --key test.txt --region ap-northeast-1 --access_key xxx --secret_key yyy
 ```
 
 ### 结构化日志模式
@@ -106,6 +108,8 @@ kscript portscan 192.168.1.1 --log-level=debug
   - `kibana`：Kibana 空间/索引/用户/角色自动化
 - **ci**
   - `jenkins`：Jenkins Job 导入导出
+- **cloud**
+  - `aws_s3`：AWS S3 文件上传测试
 - **其它**
   - `usd`：美元汇率工具
   - `lvm`：LVM 卷管理
@@ -113,13 +117,15 @@ kscript portscan 192.168.1.1 --log-level=debug
 
 ---
 
-## ⚡ Shell 自动补全
+## ⚡ Shell 自动补全 & 配置示例
 
-- **首次运行自动为 zsh/bash 部署补全脚本**，无需手动操作
+- **首次安装/升级自动为 zsh/bash 部署补全脚本，并生成 `~/.kscript/.env` 配置示例**
 - 补全脚本路径：
   - zsh: `~/.zsh/completions/_kscript`
   - bash: `~/.bash_completion.d/kscript`
-- 手动生成：
+- 配置文件路径：
+  - `~/.kscript/.env`（自动生成，支持 ENV 变量注释说明）
+- 手动生成补全：
   ```bash
   kscript completion zsh > ~/.zsh/completions/_kscript
   kscript completion bash > ~/.bash_completion.d/kscript
@@ -127,12 +133,20 @@ kscript portscan 192.168.1.1 --log-level=debug
 
 ---
 
-## ⚙️ 全局配置
+## ⚙️ 全局配置（.env 格式）
 
-在 `~/.kscriptrc` (YAML) 配置全局参数：
+所有全局参数均通过 `~/.kscript/.env` 文件（自动生成，标准 .env 格式）或环境变量注入。例如：
 
-```yaml
-log_level: debug
+```env
+# AWS S3 upload config
+AWS_BUCKET=my-bucket
+AWS_REGION=ap-northeast-1
+AWS_ACCESS_KEY_ID=xxx
+AWS_SECRET_ACCESS_KEY=yyy
+
+# Logging config
+KSCRIPT_LOG_LEVEL=info
+LOG=1
 ```
 
 ---
@@ -143,6 +157,7 @@ log_level: debug
 - 支持 `self.description`、`self.usage`、`self.arguments`、`self.group`、`self.author`
 - 输出统一用 `logger.kinfo`/`logger.kerror`，支持结构化日志
 - 兼容多余参数，避免 ArgumentError
+- 依赖统一在主入口 require，插件只需 require 'kscript'
 
 ---
 
@@ -154,6 +169,14 @@ log_level: debug
 
 ---
 
+## 📦 依赖与兼容性
+
+- Ruby >= 3.0
+- 依赖：bcrypt, http, nokogiri, thor, aws-sdk-s3 等
+- 支持 macOS、Linux，部分工具支持 Windows
+
+---
+
 ## 📄 许可证
 
 MIT License. 详见 [LICENSE](LICENSE)。

data/ext/mkrf_conf.rb

@@ -0,0 +1,28 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require 'fileutils'
+
+# Create Rakefile for gem install hook
+dir = File.dirname(__FILE__)
+File.write(File.join(dir, 'Rakefile'), <<~RAKEFILE)
+  task :default do
+    # Add lib directory to load path
+    lib_path = File.expand_path('../../lib', __FILE__)
+    $LOAD_PATH.unshift(lib_path) unless $LOAD_PATH.include?(lib_path)
+    require 'kscript/post_install'
+    Kscript::PostInstall.run
+  end
+RAKEFILE
+
+# Run post-install script immediately for local development
+def local_dev?
+  ENV['GEM_ENV'] != 'production'
+end
+
+if local_dev?
+  lib_path = File.expand_path('../lib', __dir__)
+  $LOAD_PATH.unshift(lib_path) unless $LOAD_PATH.include?(lib_path)
+  require 'kscript/post_install'
+  Kscript::PostInstall.run
+end

data/lib/completions/kscript.bash

@@ -0,0 +1,20 @@
+# kscript bash completion
+# Auto-generated command completion for kscript CLI
+
+_kscript_completions() {
+    local cur
+    COMPREPLY=()
+    cur="${COMP_WORDS[COMP_CWORD]}"
+
+    # Main command list as array
+    local opts=(
+        version help list apnic cleaner es_fingerprint ffmpeg ip jenkins kibana lvm optimize portscan projscan rename sh syscheck top usd wg_acl wg_pass
+    )
+
+    if [[ ${COMP_CWORD} == 1 ]]; then
+        COMPREPLY=( $(compgen -W "${opts[*]}" -- "${cur}") )
+        return 0
+    fi
+}
+
+complete -F _kscript_completions kscript

data/lib/completions/kscript.zsh

@@ -0,0 +1,29 @@
+#compdef kscript
+_kscript() {
+  local -a commands
+  commands=(
+    'version:kscript command'
+    'help:kscript command'
+    'list:kscript command'
+    'apnic:kscript command'
+    'cleaner:kscript command'
+    'es_fingerprint:kscript command'
+    'ffmpeg:kscript command'
+    'ip:kscript command'
+    'jenkins:kscript command'
+    'kibana:kscript command'
+    'lvm:kscript command'
+    'optimize:kscript command'
+    'portscan:kscript command'
+    'projscan:kscript command'
+    'rename:kscript command'
+    'sh:kscript command'
+    'syscheck:kscript command'
+    'top:kscript command'
+    'usd:kscript command'
+    'wg_acl:kscript command'
+    'wg_pass:kscript command'
+  )
+  _describe 'command' commands
+}
+compdef _kscript kscript

data/lib/kscript/base.rb

@@ -7,11 +7,12 @@ module Kscript
   class Base
     attr_reader :logger
 
-    def initialize(service: nil, log_level: nil)
-      config = defined?(Kscript::Utils::Config) ? Kscript::Utils::Config.load : nil
-      log_level ||= config&.log_level || ENV['KSCRIPT_LOG_LEVEL'] || :info
-      @logger = Kscript::Logger.new(service: service || self.class.name, level: log_level)
+    def initialize(**opts)
+      service = opts.delete(:service) || self.class.name
+      log_level = opts.delete(:log_level) || ENV['KSCRIPT_LOG_LEVEL'] || :info
+      @logger = Kscript::Logger.new(service: service, level: log_level)
       @logger.set_human_output(human_output?)
+      # 其余 opts 可由子类使用
     end
 
     # 通用工具方法可在此扩展

data/lib/kscript/cli.rb

@@ -5,6 +5,8 @@ require 'kscript/banner'
 
 module Kscript
   class CLI < Thor
+    Kscript::Config.load!
+
     class_option :log_level, type: :string, desc: 'Set log level (debug, info, warn, error, fatal)',
                              aliases: '--log-level'
     class_option :log, type: :boolean, desc: 'Enable structured log output', default: false
@@ -72,12 +74,27 @@ module Kscript
       klass = info[:class]
       desc command,
            (info[:description] || 'No description') + (reserved.include?(orig_command) ? " (original: #{orig_command})" : '')
+
+      # 动态注册 option
+      if klass.respond_to?(:arguments) && klass.arguments.is_a?(String)
+        # 支持 '--file FILE --bucket BUCKET ...' 或 '<file> <bucket>'
+        arg_str = klass.arguments
+        arg_names = arg_str.scan(/--([a-zA-Z0-9_-]+)/).flatten
+        # 兼容 <file> <bucket> 形式
+        arg_names += arg_str.scan(/<([a-zA-Z0-9_-]+)>/).flatten
+        arg_names.uniq.each do |param|
+          next if %w[help version].include?(param)
+
+          option param.to_sym, type: :string, desc: param
+        end
+      end
+
       define_method(command) do |*args|
         puts Kscript::BANNER
         puts color('─' * 80, 90)
         begin
-          instance = klass.new(*args)
-          instance.run
+          instance = klass.new(**options)
+          instance.run(*args)
         rescue ArgumentError => e
           warn "Argument error: #{e.message}"
           puts "Usage: kscript #{command} #{klass.respond_to?(:arguments) ? klass.arguments : '[args...]'}"
@@ -86,38 +103,20 @@ module Kscript
       end
     end
 
-    desc 'completion SHELL', 'Generate shell completion script (zsh or bash)'
-    def completion(shell = 'zsh', capture: false)
-      commands = %w[version help
-                    list] + Kscript::PluginLoader.plugin_infos.map do |info|
-                              cmd = info[:name].to_s.sub(/^kk_/, '')
-                              cmd = 'sh' if cmd == 'shell'
-                              cmd
-                            end
-      output = StringIO.new
-      case shell
-      when 'zsh'
-        output.puts "#compdef kscript\n"
-        output.puts '_kscript() {'
-        output.puts '  local -a commands'
-        output.puts '  commands=('
-        commands.each do |cmd|
-          output.puts "    '#{cmd}:kscript command'"
-        end
-        output.puts '  )'
-        output.puts "  _describe 'command' commands"
-        output.puts '}'
-        output.puts 'compdef _kscript kscript'
-      when 'bash'
-        output.puts '_kscript_completions() {'
-        output.puts "  COMPREPLY=($(compgen -W \"#{commands.join(' ')}\" -- \"${COMP_WORDS[1]}\"))"
-        output.puts '}'
-        output.puts 'complete -F _kscript_completions kscript'
+    desc 'env', 'Show current environment variables relevant to kscript'
+    def env
+      puts Kscript::BANNER
+      puts color('─' * 80, 90)
+      keys = ENV.keys.grep(/^(KSCRIPT_|AWS_|LOG$|SHELL$|HOME$)/)
+      if keys.empty?
+        puts gray('No relevant environment variables found.')
       else
-        output.puts "Unsupported shell: #{shell}. Only 'zsh' and 'bash' are supported."
-        return capture ? output.string : (puts output.string)
+        puts bold('Loaded environment variables:')
+        keys.sort.each do |k|
+          v = ENV[k]
+          puts green(k.ljust(28)) + gray('=') + cyan(v)
+        end
       end
-      capture ? output.string : (puts output.string)
     end
 
     no_commands do
@@ -172,11 +171,3 @@ module Kscript
     end
   end
 end
-
-# CLI 启动时自动检测并安装 shell 补全脚本
-begin
-  require_relative 'utils'
-  Kscript::Utils::Config.ensure_completion_installed
-rescue StandardError => e
-  warn "[kscript] Shell completion auto-install failed: #{e.message}"
-end

data/lib/kscript/config.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+# Kscript::Config
+# Responsible for loading ~/.kscript/.env and providing config access
+# Usage:
+#   Kscript::Config.load!
+#   Kscript::Config.get('AWS_ACCESS_KEY_ID')
+
+require 'dotenv'
+
+module Kscript
+  module Config
+    CONFIG_PATH = File.expand_path('~/.kscript/.env')
+    @loaded = false
+
+    # Load ~/.kscript/.env using dotenv
+    def self.load!
+      return if @loaded
+
+      Dotenv.load(CONFIG_PATH)
+      @loaded = true
+    end
+
+    # Get config value from ENV, with optional default
+    # @param key [String] ENV key
+    # @param default [Object] default value if not set
+    def self.get(key, default = nil)
+      ENV[key] || default
+    end
+  end
+end

data/lib/kscript/plugins/kk_apnic_utils.rb

@@ -12,9 +12,9 @@ module Kscript
     attr_reader :country_sn, :cache_file
 
     # Initialize class instance, set country code and cache file path
-    def initialize(country_sn = 'CN', *_args, **opts)
-      super(**opts.merge(service: 'kk_apnic'))
-      @country_sn = country_sn
+    def initialize(country_sn = 'CN', *args, **opts)
+      super(*args, **opts)
+      @country_sn = country_sn || 'CN'
       @cache_file = RUBY_PLATFORM.match?(/(linux|darwin)/) ? '/tmp/apnic.txt' : 'apnic.txt'
     end
 

data/lib/kscript/plugins/kk_aws_s3_utils.rb

@@ -0,0 +1,99 @@
+# frozen_string_literal: true
+
+# Copyright (c) 2025 Kk
+#
+# This software is released under the MIT License.
+# https://opensource.org/licenses/MIT
+
+require 'kscript'
+
+# AWS S3 文件上传测试工具
+# 用法示例：
+#   kscript aws_s3 --file local.txt --bucket my-bucket --key test.txt --region ap-northeast-1 --access_key xxx --secret_key yyy
+#
+# 依赖：aws-sdk-s3（已在主入口 require）
+
+module Kscript
+  class KkAwsS3Utils < Base
+    # 初始化，支持所有参数通过 CLI 传递
+    def initialize(*args, **opts)
+      super(*args, **opts)
+      # 兼容 --file test.txt 以及 --file=test.txt
+      @file = opts[:file] || args[0]
+      @bucket = opts[:bucket] || args[1] || ENV['AWS_BUCKET']
+      @key = opts[:key] || args[2]
+      @region = opts[:region] || ENV['AWS_REGION']
+      @access_key = opts[:access_key] || ENV['AWS_ACCESS_KEY_ID']
+      @secret_key = opts[:secret_key] || ENV['AWS_SECRET_ACCESS_KEY']
+    end
+
+    # 主入口
+    def run(*_args, **_opts)
+      with_error_handling do
+        upload_file_to_s3
+      end
+    end
+
+    # 上传文件到 S3
+    def upload_file_to_s3
+      validate_params!
+      logger.kinfo('Uploading file to S3...', file: @file, bucket: @bucket, key: @key, region: @region)
+      s3 = Aws::S3::Resource.new(
+        region: @region,
+        access_key_id: @access_key,
+        secret_access_key: @secret_key
+      )
+      obj = s3.bucket(@bucket).object(@key)
+      obj.upload_file(@file)
+      url = obj.public_url
+      logger.kinfo('Upload success', url: url)
+    end
+
+    # 参数声明
+    def self.arguments
+      '--file FILE --bucket BUCKET --key KEY --region REGION --access_key AK --secret_key SK'
+    end
+
+    # 用法
+    def self.usage
+      'kscript aws_s3 --file local.txt --bucket my-bucket --key test.txt --region ap-northeast-1 --access_key xxx --secret_key yyy'
+    end
+
+    # 分组
+    def self.group
+      'cloud'
+    end
+
+    # 作者
+    def self.author
+      'kk'
+    end
+
+    # 描述
+    def self.description
+      'Upload a file to AWS S3 for testing.'
+    end
+
+    private
+
+    # 校验参数
+    def validate_params!
+      missing = []
+      missing << 'file' unless @file
+      missing << 'bucket' unless @bucket
+      missing << 'key' unless @key
+      missing << 'region' unless @region
+      missing << 'access_key' unless @access_key
+      missing << 'secret_key' unless @secret_key
+      unless missing.empty?
+        logger.kerror('Missing required parameters', missing: missing)
+        logger.kinfo(self.class.usage)
+        raise ArgumentError, "Missing: #{missing.join(', ')}"
+      end
+      return if File.exist?(@file)
+
+      logger.kerror('File not found', file: @file)
+      raise ArgumentError, "File not found: #{@file}"
+    end
+  end
+end

data/lib/kscript/plugins/kk_cleaner_utils.rb

@@ -16,10 +16,10 @@ module Kscript
     # Initialize the cleaner with path and retention settings
     # @param source_path [String] path to source code directory
     # @param retain_count [Integer] number of versions to keep
-    def initialize(source_path = '/data/sources/*/**', retain_count = DEFAULT_RETAIN_VERSIONS, *_args, **opts)
-      super(**opts.merge(service: 'kk_source_cleaner'))
-      @source_path = source_path
-      @retain_count = retain_count
+    def initialize(*args, **opts)
+      super(*args, **opts)
+      @source_path = args[0] || '/data/sources/*/**'
+      @retain_count = args[1] || DEFAULT_RETAIN_VERSIONS
     end
 
     def run

data/lib/kscript/plugins/kk_es_fingerprint_utils.rb

@@ -15,8 +15,9 @@ module Kscript
 
     # Initialize with certificate path
     # @param cert_path [String] path to the certificate file
-    def initialize(cert_path = DEFAULT_CERT_PATH)
-      @cert_path = cert_path
+    def initialize(*args, **opts)
+      super(*args, **opts)
+      @cert_path = opts[:cert_path] || self.class::DEFAULT_CERT_PATH
     end
 
     def run

data/lib/kscript/plugins/kk_ffmpeg_utils.rb

@@ -9,6 +9,10 @@ require 'kscript'
 
 module Kscript
   class KkFfmpegUtils < Base
+    def initialize(*args, **opts)
+      super(*args, **opts)
+    end
+
     def run
       with_error_handling do
         install

data/lib/kscript/plugins/kk_ip_utils.rb

@@ -14,9 +14,9 @@ module Kscript
 
     attr_reader :ip_address
 
-    def initialize(ip_address = nil, *_args, **opts)
-      super(**opts.merge(service: 'kk_ip_api'))
-      @ip_address = ip_address || fetch_public_ip
+    def initialize(*args, **opts)
+      super(*args, **opts)
+      @ip_address = args.first || fetch_public_ip
     end
 
     def run

data/lib/kscript/plugins/kk_jenkins_utils.rb

@@ -9,13 +9,9 @@ require 'kscript'
 
 module Kscript
   class KkJenkinsUtils < Base
-    def run
-      with_error_handling do
-        logger.kinfo('Jenkins job manager executed.')
-      end
-    end
-
-    def initialize(jenkins_url, user, password)
+    def initialize(*args, **opts)
+      super(*args, **opts)
+      jenkins_url, user, password = args
       @jenkins_url = jenkins_url
       @user = user
       @password = password
@@ -24,6 +20,12 @@ module Kscript
       @output.sync = true
     end
 
+    def run
+      with_error_handling do
+        logger.kinfo('Jenkins job manager executed.')
+      end
+    end
+
     def export_all_jobs
       FileUtils.mkdir_p('jobs')
 

data/lib/kscript/plugins/kk_kibana_utils.rb

@@ -9,13 +9,9 @@ require 'kscript'
 
 module Kscript
   class KkKibanaUtils < Base
-    def run
-      with_error_handling do
-        logger.kinfo('Kibana utils executed.')
-      end
-    end
-
-    def initialize(project_name, project_env, base_url, username, password)
+    def initialize(*args, **opts)
+      super(*args, **opts)
+      project_name, project_env, base_url, username, password = args
       @base_url = base_url
       @username = username
       @password = password
@@ -27,6 +23,12 @@ module Kscript
       create_space unless space_exists?
     end
 
+    def run
+      with_error_handling do
+        logger.kinfo('Kibana utils executed.')
+      end
+    end
+
     # Check if the space exists
     def space_exists?
       url = "#{@base_url}/api/spaces/space/#{@space_name}"

data/lib/kscript/plugins/kk_lvm_utils.rb

@@ -20,8 +20,9 @@ module Kscript
 
     # Initialize the LVM mounter with configuration
     # @param config [Hash] configuration options
-    def initialize(config = {}, *_args, **_opts)
-      @config = DEFAULT_CONFIG.merge(config)
+    def initialize(*args, **opts)
+      super(*args, **opts)
+      @config = DEFAULT_CONFIG.merge(opts)
     end
 
     def run

data/lib/kscript/plugins/kk_optimize_utils.rb

@@ -9,6 +9,10 @@ require 'kscript'
 
 module Kscript
   class KkOptimizeUtils < Base
+    def initialize(*args, **opts)
+      super(*args, **opts)
+    end
+
     def run
       with_error_handling do
         optimize

data/lib/kscript/plugins/kk_portscan_utils.rb

@@ -15,11 +15,11 @@ module Kscript
     # @param host [String] target host to scan
     # @param ports [Array<Integer>] list of ports to scan
     # @param thread_count [Integer] number of concurrent threads
-    def initialize(target = nil, ports = (1..1024), thread_count = 50, *_args, **opts)
-      super(**opts.merge(service: 'kk_port_scanner'))
-      @target = target
-      @ports = parse_ports(ports)
-      @thread_count = (opts[:thread_count] || thread_count || 50).to_i
+    def initialize(*args, **opts)
+      super(*args, **opts)
+      @target = args[0]
+      @ports = parse_ports(args[1] || (1..1024))
+      @thread_count = (opts[:thread_count] || 50).to_i
     end
 
     def run

data/lib/kscript/plugins/kk_projscan_utils.rb

@@ -9,9 +9,8 @@ require 'kscript'
 
 module Kscript
   class KkProjscanUtils < Base
-    def initialize(src_path = nil, *_args, **opts)
-      super(**opts.merge(service: 'kk_project_scanner'))
-      @src_path = src_path || File.expand_path('~/projects/src')
+    def initialize(*args, **opts)
+      super(*args, **opts)
     end
 
     def run

data/lib/kscript/plugins/kk_rename_utils.rb

@@ -11,11 +11,11 @@ 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
+    def initialize(*args, **opts)
+      super(*args, **opts)
+      @source_pattern = args[0]
+      @target_pattern = args[1]
+      @directory = args[2] || Dir.pwd
     end
 
     def run

data/lib/kscript/plugins/kk_sh_utils.rb

@@ -15,9 +15,9 @@ module Kscript
 
     # 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
+    def initialize(*args, **opts)
+      super(*args, **opts)
+      @command = args.first
     end
 
     def run

data/lib/kscript/plugins/kk_syscheck_utils.rb

@@ -9,6 +9,10 @@ require 'kscript'
 
 module Kscript
   class KkSyscheckUtils < Base
+    def initialize(*_args, **opts)
+      super(**opts.merge(service: 'kk_syscheck'))
+    end
+
     def run
       with_error_handling do
         check
@@ -20,20 +24,33 @@ module Kscript
       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)
-
+      # CPU Usage
+      logger.kinfo('===============================')
+      logger.kinfo(' CPU Usage (Top 10)')
+      logger.kinfo('===============================')
+      cpu_output = `ps aux | sort -nrk 3 | head -n 10`
+      logger.kinfo(cpu_output)
+
+      # Memory Usage
+      logger.kinfo('===============================')
+      logger.kinfo(' Memory Usage (Top 10)')
+      logger.kinfo('===============================')
+      mem_output = `ps aux | sort -nrk 4 | head -n 10`
+      logger.kinfo(mem_output)
+
+      # GPU Usage
       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)
+        logger.kinfo('===============================')
+        logger.kinfo(' GPU Usage')
+        logger.kinfo('===============================')
+        gpu_output = `sudo powermetrics --samplers gpu_power -n 1 2>/dev/null`
+        logger.kinfo(gpu_output)
       end
 
+      # Network Connections
       logger.kinfo("\n👉 Top 10 Processes by Network Connections:")
       lsof_output = `lsof -i -nP | grep ESTABLISHED`
       counts = Hash.new(0)
@@ -45,6 +62,7 @@ module Kscript
         logger.kinfo("#{proc}: #{count} connections")
       end
 
+      # System Overview
       logger.kinfo("\n👉 System Overview:")
       cpu_core = `sysctl -n hw.ncpu`.strip
       mem_size = `sysctl -n hw.memsize`.to_i / 1024 / 1024 / 1024

data/lib/kscript/plugins/kk_top_utils.rb

@@ -7,15 +7,12 @@
 
 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 initialize(*args, **opts)
+      super(*args, **opts)
+    end
+
     def run
       with_error_handling do
         print_report

data/lib/kscript/plugins/kk_usd_utils.rb

@@ -14,8 +14,8 @@ 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'))
+    def initialize(currency_code = 'CNY', *args, **opts)
+      super(*args, **opts)
       @currency_code = currency_code
     end
 
@@ -31,18 +31,9 @@ module Kscript
       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
+        logger.kinfo("1 USD = #{rate} #{@currency_code.upcase}")
       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

data/lib/kscript/plugins/kk_wg_acl_utils.rb

@@ -12,6 +12,10 @@ module Kscript
     WIREGUARD_PORT = 51_821
     ALLOWED_IPS = %w[127.0.0.1].freeze
 
+    def initialize(*args, **opts)
+      super(*args, **opts)
+    end
+
     def run
       with_error_handling do
         apply_rules

data/lib/kscript/plugins/kk_wg_pass_utils.rb

@@ -9,9 +9,8 @@ require 'kscript'
 
 module Kscript
   class KkWgPassUtils < Base
-    def initialize(length = 32, *_args, **opts)
-      super(**opts.merge(service: 'kk_wg_pass'))
-      @length = length.to_i
+    def initialize(*args, **opts)
+      super(*args, **opts)
     end
 
     def run

data/lib/kscript/post_install.rb

@@ -0,0 +1,128 @@
+# frozen_string_literal: true
+
+module Kscript
+  class PostInstall
+    class << self
+      def run
+        setup_env_example
+        setup_shell_completion
+      end
+
+      private
+
+      def setup_env_example
+        env_dir = File.expand_path('~/.kscript')
+        env_file = File.join(env_dir, '.env')
+        return if File.exist?(env_file)
+
+        FileUtils.mkdir_p(env_dir)
+        File.open(env_file, 'w') do |f|
+          f.puts "# kscript config: all variables are optional, remove '#' to enable"
+          f.puts '# AWS S3 upload config'
+          f.puts '# AWS_BUCKET=my-bucket            # S3 bucket name'
+          f.puts '# AWS_REGION=ap-northeast-1      # AWS region, e.g. ap-northeast-1'
+          f.puts '# AWS_ACCESS_KEY_ID=xxx         # AWS access key id'
+          f.puts '# AWS_SECRET_ACCESS_KEY=yyy     # AWS secret access key'
+          f.puts
+          f.puts '# Shell environment (auto detected, usually not needed)'
+          f.puts '# SHELL=/bin/zsh                # User shell'
+          f.puts '# HOME=/Users/yourname          # User home directory'
+          f.puts
+          f.puts '# Logging config'
+          f.puts '# KSCRIPT_LOG_LEVEL=info        # Log level: debug, info, warn, error, fatal'
+          f.puts '# LOG=1                         # Enable structured log output'
+        end
+        puts '[kscript] ~/.kscript/.env example created. Edit and uncomment as needed.'
+      end
+
+      def setup_shell_completion
+        setup_bash_completion
+        setup_zsh_completion
+      end
+
+      def setup_bash_completion
+        return unless File.exist?(bashrc_path)
+
+        completion_path = find_completion_file('kscript.bash')
+        return if completion_path.nil?
+
+        source_line = "source \"#{completion_path}\""
+        content = File.read(bashrc_path)
+        if content.match?(/source.*kscript\.bash/)
+          # Update existing config
+          new_content = content.gsub(/source.*kscript\.bash.*$/, source_line)
+          File.write(bashrc_path, new_content)
+        else
+          # Append new config
+          File.open(bashrc_path, 'a') do |f|
+            f.puts "\n# Kscript completion"
+            f.puts source_line
+          end
+        end
+        puts "✅ Bash completion configured in #{bashrc_path}"
+      rescue StandardError => e
+        puts "⚠️ Failed to configure Bash completion: #{e.message}"
+      end
+
+      def setup_zsh_completion
+        return unless File.exist?(zshrc_path)
+
+        completion_path = find_completion_file('kscript.zsh')
+        return if completion_path.nil?
+
+        source_lines = [
+          "source \"#{completion_path}\"",
+          'autoload -Uz compinit && compinit'
+        ]
+        content = File.read(zshrc_path)
+        if content.match?(/source.*kscript\.zsh/)
+          # Update existing config
+          new_content = content.gsub(/source.*kscript\.zsh.*$/, source_lines[0])
+          File.write(zshrc_path, new_content)
+        else
+          # Append new config
+          File.open(zshrc_path, 'a') do |f|
+            f.puts "\n# Kscript completion"
+            source_lines.each { |line| f.puts line unless content.include?(line) }
+          end
+        end
+        puts "✅ Zsh completion configured in #{zshrc_path}"
+      rescue StandardError => e
+        puts "⚠️ Failed to configure Zsh completion: #{e.message}"
+      end
+
+      def find_completion_file(filename)
+        # Try all possible paths
+        paths = [
+          # Dev path
+          File.expand_path("../completions/#{filename}", __FILE__),
+          # Gem install paths
+          *Gem.path.map { |path| File.join(path, "gems/kscript-*/lib/completions/#{filename}") },
+          # System paths
+          "/usr/local/share/kscript/completions/#{filename}",
+          "/usr/share/kscript/completions/#{filename}"
+        ]
+        paths.each do |path|
+          if path.include?('*')
+            matches = Dir.glob(path)
+            return matches.first if matches.any?
+          elsif File.exist?(path)
+            return path
+          end
+        end
+        puts "⚠️ Could not find completion file: #{filename}"
+        puts 'Searched paths:'
+        paths.each { |path| puts " - #{path}" }
+        nil
+      end
+
+      def bashrc_path
+        File.join(Dir.home, '.bashrc')
+      end
+
+      def zshrc_path
+        File.join(Dir.home, '.zshrc')
+      end
+    end
+  end
+end

data/lib/kscript/utils.rb

@@ -1,57 +1 @@
 # frozen_string_literal: true
-
-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
-
-      # 自动检测并安装 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

@@ -6,5 +6,5 @@
 # https://opensource.org/licenses/MIT
 
 module Kscript
-  VERSION = '1.0.2'
+  VERSION = '1.0.4'
 end

data/lib/kscript.rb

@@ -24,7 +24,10 @@ require 'http'
 require 'bcrypt'
 require 'nokogiri'
 require 'thor'
+require 'aws-sdk-s3'
+require 'dotenv'
 
+require 'kscript/config'
 require 'kscript/plugins'
 require 'kscript/base'
 require 'kscript/logger'

metadata

@@ -1,15 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: kscript
 version: !ruby/object:Gem::Version
-  version: 1.0.2
+  version: 1.0.4
 platform: ruby
 authors:
 - Kk
 autorequire:
-bindir: bin
+bindir: exe
 cert_chain: []
-date: 2025-07-05 00:00:00.000000000 Z
+date: 2025-07-06 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-s3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.130'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.130'
 - !ruby/object:Gem::Dependency
   name: bcrypt
   requirement: !ruby/object:Gem::Requirement
@@ -30,6 +44,20 @@ dependencies:
     - - "<"
       - !ruby/object:Gem::Version
         version: '4.0'
+- !ruby/object:Gem::Dependency
+  name: dotenv
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
 - !ruby/object:Gem::Dependency
   name: http
   requirement: !ruby/object:Gem::Requirement
@@ -70,6 +98,20 @@ dependencies:
     - - "<"
       - !ruby/object:Gem::Version
         version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
 - !ruby/object:Gem::Dependency
   name: thor
   requirement: !ruby/object:Gem::Requirement
@@ -90,36 +132,38 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.0'
+        version: '1.21'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.0'
+        version: '1.21'
 description: Kscript provides a set of handy Ruby scripts for system administration,
   development, and automation.
 email:
 - kevin197011@outlook.com
 executables:
 - kscript
-extensions: []
+extensions:
+- ext/mkrf_conf.rb
 extra_rdoc_files: []
 files:
-- Gemfile
-- LICENSE
 - README.md
-- Rakefile
-- bin/kscript
-- kscript.gemspec
+- exe/kscript
+- ext/mkrf_conf.rb
+- lib/completions/kscript.bash
+- lib/completions/kscript.zsh
 - lib/kscript.rb
 - lib/kscript/banner.rb
 - lib/kscript/base.rb
 - lib/kscript/cli.rb
+- lib/kscript/config.rb
 - lib/kscript/logger.rb
 - lib/kscript/plugins.rb
 - lib/kscript/plugins/kk_apnic_utils.rb
+- lib/kscript/plugins/kk_aws_s3_utils.rb
 - lib/kscript/plugins/kk_cleaner_utils.rb
 - lib/kscript/plugins/kk_es_fingerprint_utils.rb
 - lib/kscript/plugins/kk_ffmpeg_utils.rb
@@ -137,14 +181,21 @@ files:
 - lib/kscript/plugins/kk_usd_utils.rb
 - lib/kscript/plugins/kk_wg_acl_utils.rb
 - lib/kscript/plugins/kk_wg_pass_utils.rb
+- lib/kscript/post_install.rb
 - lib/kscript/utils.rb
 - lib/kscript/version.rb
 homepage: https://github.com/kevin197011/kscript
 licenses:
 - MIT
 metadata:
+  homepage_uri: https://github.com/kevin197011/kscript
+  source_code_uri: https://github.com/kevin197011/kscript.git
+  changelog_uri: https://github.com/kevin197011/kscript/blob/main/CHANGELOG.md
   rubygems_mfa_required: 'true'
-post_install_message:
+post_install_message: "\U0001F389 Thanks for installing kscript!\n\nShell completion
+  and config example will be configured automatically.\nYou may need to restart your
+  shell or run:\n  - For Bash: source ~/.bashrc\n  - For Zsh: source ~/.zshrc\n\nEdit
+  ~/.kscript/.env to customize your environment variables.\n\nHappy scripting! \U0001F680\n"
 rdoc_options: []
 require_paths:
 - lib

data/Gemfile

@@ -1,3 +0,0 @@
-# frozen_string_literal: true
-
-# See kscript.gemspec for gem dependencies

data/LICENSE

@@ -1,21 +0,0 @@
-MIT License
-
-Copyright (c) 2021 kevin197011
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.

data/Rakefile

@@ -1,42 +0,0 @@
-# frozen_string_literal: true
-
-# Copyright (c) 2025 kk
-#
-# This software is released under the MIT License.
-# https://opensource.org/licenses/MIT
-
-require 'time'
-require 'rake'
-require 'bundler/gem_tasks'
-
-task default: %w[push]
-
-task :push do
-  system <<-SHELL
-    rubocop -A
-    git update-index --chmod=+x push
-    git add .
-    git commit -m "Update #{Time.now}"
-    git pull
-    git push origin main
-  SHELL
-end
-
-# 其他自定义任务可在此添加
-task :dev do
-  sh <<-SHELL
-    gem uninstall kscript -aIx
-    gem build kscript.gemspec
-    gem install kscript-*.gem
-    kscript help
-    kscript list
-    kscript version
-  SHELL
-end
-
-task :install do
-  sh <<-SHELL
-    gem uninstall kscript -aIx
-    gem install kscript
-  SHELL
-end

data/kscript.gemspec

@@ -1,40 +0,0 @@
-# frozen_string_literal: true
-
-require_relative 'lib/kscript/version'
-
-Gem::Specification.new do |spec|
-  spec.name          = 'kscript'
-  spec.version       = Kscript::VERSION
-  spec.authors       = ['Kk']
-  spec.email         = ['kevin197011@outlook.com']
-
-  spec.summary       = 'A collection of Ruby utility scripts for sysadmin and development.'
-  spec.description   = 'Kscript provides a set of handy Ruby scripts for system administration, development, and automation.'
-  spec.homepage      = 'https://github.com/kevin197011/kscript'
-  spec.license       = 'MIT'
-
-  spec.required_ruby_version = '>= 3.0.0'
-
-  spec.files = Dir.chdir(__dir__) do
-    files = `git ls-files -z`.split("\x0")
-    files.select! do |f|
-      f =~ %r{^(lib/|bin/kscript|README|LICENSE|Rakefile|Gemfile|kscript\.gemspec)}
-    end
-    files
-  end
-  spec.bindir        = 'bin'
-  spec.executables   = ['kscript']
-  spec.require_paths = ['lib']
-
-  # Runtime dependencies
-  spec.add_dependency 'bcrypt', '>= 3.1', '< 4.0'
-  spec.add_dependency 'http', '>= 4.0', '< 6.0'
-  spec.add_dependency 'nokogiri', '>= 1.13', '< 2.0'
-  spec.add_dependency 'thor', '1.3.2'
-
-  # Development dependencies
-  spec.add_development_dependency 'rubocop', '~> 1.0'
-  spec.metadata['rubygems_mfa_required'] = 'true'
-
-  # spec.extensions = ['ext/install.rb'] # 已移除，防止 native extension build 错误
-end