network-infra-utility 0.3.0 → 0.6.0

data/service/ssh/lib/network_infra_utility/ssh/terminal/logger.rb

@@ -0,0 +1,195 @@
+# frozen_string_literal: true
+
+require "fileutils"
+require "time"
+require "stringio"
+
+module NetworkInfraUtility
+  module SSH
+    module Terminal
+      # 会话日志记录与自动轮转。
+      # 对应需求 FR-TERM-006。
+      #
+      # LLD §10.1 可观测性：
+      #   路径模式  logs/<sess_id>/<ts>.log
+      #   格式      纯文本带时间戳；可选 HTML
+      #   轮转      单文件 100MB / 保留 10
+      #
+      # DoD（LLD §11.2）：自动轮转；HTML 格式带颜色还原
+      #
+      # 集成点：Emulator#feed 中 @logger&.write(data)
+      class Logger
+        DEFAULT_MAX_SIZE = 100 * 1024 * 1024 # 100MB
+        DEFAULT_ROTATE   = 10                 # 保留 10 个文件
+        TIMESTAMP_FORMAT = "%Y%m%d_%H%M%S"
+
+        attr_reader :path, :max_size, :rotate, :format, :current_size
+
+        # @param path [String] 日志文件路径（不含扩展名，自动加 .log 或 .html）
+        # @param max_size [Integer] 单文件最大字节数
+        # @param rotate [Integer] 保留轮转文件数
+        # @param format [Symbol] :text 或 :html
+        def initialize(path, max_size: DEFAULT_MAX_SIZE, rotate: DEFAULT_ROTATE, format: :text)
+          @base_path = path
+          @max_size = max_size
+          @rotate = rotate
+          @format = format
+          @current_size = 0
+          @write_mutex = Mutex.new
+          @io = nil
+          @html_header_written = false
+          @seq = 0
+          open_new_file
+        end
+
+        # 写入原始终端数据。
+        # 由 Emulator#feed 调用，每次收到 SSH 通道数据即写入。
+        # @param data [String] 原始字节
+        def write(data)
+          @write_mutex.synchronize do
+            return unless @io
+
+            if @format == :html
+              write_html(data)
+            else
+              write_text(data)
+            end
+
+            @current_size += data.bytesize
+            rotate_if_needed
+          end
+        end
+        alias << write
+
+        # 关闭当前日志文件
+        def close
+          @write_mutex.synchronize do
+            if @format == :html && @io
+              @io.write("</pre>\n</body>\n</html>\n")
+            end
+            @io&.close
+            @io = nil
+          end
+        end
+
+        # 切换格式（需重新打开文件）
+        def format=(fmt)
+          close
+          @format = fmt
+          @html_header_written = false
+          open_new_file
+        end
+
+        private
+
+        # 打开新的日志文件
+        def open_new_file
+          dir = File.dirname(@base_path)
+          FileUtils.mkdir_p(dir)
+
+          ext = @format == :html ? ".html" : ".log"
+          timestamp = Time.now.strftime(TIMESTAMP_FORMAT)
+          @seq += 1
+          @path = "#{@base_path}_#{timestamp}_#{@seq}#{ext}"
+
+          @io = File.open(@path, "wb")
+          @io.sync = true
+          @current_size = 0
+
+          if @format == :html
+            write_html_header
+          else
+            # 纯文本头部：时间戳
+            @io.write("# Session log started at #{Time.now.iso8601}\n")
+            @current_size += 30
+          end
+        end
+
+        # 纯文本写入：带时间戳前缀
+        def write_text(data)
+          # 每行加时间戳前缀
+          lines = data.split("\n", -1)
+          lines.each_with_index do |line, i|
+            next if line.empty? && i == lines.size - 1
+
+            ts = Time.now.strftime("[%H:%M:%S] ")
+            @io.write("#{ts}#{line}\n")
+          end
+        end
+
+        # HTML 写入：保留 ANSI 颜色信息，转义 HTML 实体
+        def write_html_header
+          @io.write(<<~HTML)
+            <!DOCTYPE html>
+            <html>
+            <head>
+            <meta charset="utf-8">
+            <title>SSH Session Log</title>
+            <style>
+              body { background: #1e1e1e; color: #d4d4d4; font-family: monospace; }
+              pre { white-space: pre-wrap; word-wrap: break-word; }
+            </style>
+            </head>
+            <body>
+            <pre>
+          HTML
+          @html_header_written = true
+        end
+
+        # HTML 写入：ANSI 转义转为 span 标签
+        def write_html(data)
+          # 转义 HTML 实体
+          escaped = data.gsub("&", "&amp;").gsub("<", "&lt;").gsub(">", "&gt;")
+
+          # ANSI SGR 转义序列 → <span style="color:...">
+          # 这里的实现是骨架，完整 ANSI→HTML 转换在具体实现阶段完善
+          # 覆盖常见 SGR：重置 \e[0m、前景色 \e[3Xm、加粗 \e[1m
+          escaped.gsub!(/\e\[0m/, "</span>")
+          escaped.gsub!(/\e\[1m/, '<span style="font-weight:bold">')
+          escaped.gsub!(/\e\[3(\d)m/) do
+            idx = Regexp.last_match(1).to_i
+            color = ansi_color_to_css(idx)
+            %(<span style="color:#{color}">)
+          end
+
+          @io.write(escaped)
+        end
+
+        # ANSI 8 色映射到 CSS 颜色
+        def ansi_color_to_css(idx)
+          %w[#000000 #cc0000 #00aa00 #aa5500 #0033cc #aa00aa #00aaaa #aaaaaa
+             #555555 #ff5555 #55ff55 #ffff55 #5555ff #ff55ff #55ffff #ffffff][idx] || "#ffffff"
+        end
+
+        # 检查是否需要轮转
+        def rotate_if_needed
+          return if @current_size < @max_size
+
+          @io.close
+          if @format == :html
+            # 需要给当前文件写尾部
+            File.open(@path, "a") { |f| f.write("</pre>\n</body>\n</html>\n") }
+          end
+
+          # 清理超出 rotate 数量的旧文件
+          cleanup_old_files
+          open_new_file
+        end
+
+        # 清理超出保留数量的旧日志文件
+        def cleanup_old_files
+          dir = File.dirname(@base_path)
+          base = File.basename(@base_path)
+          ext = @format == :html ? ".html" : ".log"
+
+          pattern = File.join(dir, "#{base}_*#{ext}")
+          files = Dir.glob(pattern).sort_by { |f| File.mtime(f) }
+          excess = files.size - @rotate
+          return if excess <= 0
+
+          files.first(excess).each { |f| File.delete(f) }
+        end
+      end
+    end
+  end
+end

data/service/ssh/lib/network_infra_utility/ssh/terminal/screen.rb

@@ -0,0 +1,212 @@
+# frozen_string_literal: true
+
+module NetworkInfraUtility
+  module SSH
+    module Terminal
+      # 屏幕状态：行数组 + 光标 + 滚动区 + alt buffer + 字符属性。
+      # 由 AnsiParser 经 Emulator 回调驱动。
+      class Screen
+        attr_reader :cols, :rows, :cursor_x, :cursor_y
+        attr_reader :scroll_top, :scroll_bottom
+        attr_reader :alt_mode
+
+        def initialize(cols, rows)
+          @cols = cols
+          @rows = rows
+          @cursor_x = 0
+          @cursor_y = 0
+          @scroll_top = 0
+          @scroll_bottom = rows - 1
+          @alt_mode = false
+          @main_lines = Array.new(rows) { Line.new(@cols) }
+          @alt_lines = Array.new(rows) { Line.new(@cols) }
+          @current_style = {}
+        end
+
+        def lines
+          @alt_mode ? @alt_lines : @main_lines
+        end
+
+        def put_char(ch, x, y, style = {})
+          line = lines[y]
+          return unless line
+
+          line.put_char(ch, x, style.empty? ? @current_style : style)
+          @cursor_x = x + 1
+          @cursor_y = y
+        end
+
+        def cursor_to(x, y)
+          @cursor_x = [[x, 0].max, @cols - 1].min
+          @cursor_y = [[y, 0].max, @rows - 1].min
+        end
+
+        def newline
+          if @cursor_y >= @scroll_bottom
+            scroll_up(1)
+          else
+            @cursor_y += 1
+          end
+          @cursor_x = 0
+        end
+
+        def backspace
+          @cursor_x = [@cursor_x - 1, 0].max
+        end
+
+        def tab
+          @cursor_x = ((@cursor_x / 8) + 1) * 8
+          @cursor_x = [@cursor_x, @cols - 1].min
+        end
+
+        def clear(mode = :full)
+          case mode
+          when :full
+            @main_lines.each { |l| l.clear(@cols) }
+            @alt_lines.each { |l| l.clear(@cols) }
+            @cursor_x = 0
+            @cursor_y = 0
+          when :above
+            clear_above_cursor
+          when :below
+            clear_below_cursor
+          end
+        end
+
+        def clear_line(mode = :full)
+          line = lines[@cursor_y]
+          return unless line
+
+          case mode
+          when :full then line.clear(@cols)
+          when :right then line.clear_from(@cursor_x, @cols)
+          when :left then line.clear_range(0, @cursor_x)
+          end
+        end
+
+        def scroll_up(n = 1)
+          n.times do
+            # 将滚动区首行移入回滚（由 Buffer 处理），后续行上移
+            @scroll_top.upto(@scroll_bottom - 1) do |i|
+              lines[i] = lines[i + 1]
+            end
+            lines[@scroll_bottom] = Line.new(@cols)
+          end
+        end
+
+        def scroll_down(n = 1)
+          n.times do
+            @scroll_bottom.downto(@scroll_top + 1) do |i|
+              lines[i] = lines[i - 1]
+            end
+            lines[@scroll_top] = Line.new(@cols)
+          end
+        end
+
+        def set_scroll_region(top, bottom)
+          @scroll_top = [[top, 0].max, @rows - 1].min
+          @scroll_bottom = [[bottom, 0].max, @rows - 1].min
+          @cursor_x = 0
+          @cursor_y = @scroll_top
+        end
+
+        def enter_alt
+          @alt_mode = true
+          @cursor_x = 0
+          @cursor_y = 0
+        end
+
+        def exit_alt
+          @alt_mode = false
+          @cursor_x = 0
+          @cursor_y = 0
+        end
+
+        def set_style(style)
+          @current_style = @current_style.merge(style)
+        end
+
+        def reset_style
+          @current_style = {}
+        end
+
+        def resize(cols, rows)
+          old_cols = @cols
+          @cols = cols
+          [@main_lines, @alt_lines].each do |ary|
+            if rows > ary.size
+              (rows - ary.size).times { ary << Line.new(cols) }
+            elsif rows < ary.size
+              ary.slice!(rows..)
+            end
+            ary.each { |l| l.resize(cols) } if cols != old_cols
+          end
+          @rows = rows
+          @scroll_bottom = rows - 1
+          @cursor_x = [@cursor_x, cols - 1].min
+          @cursor_y = [@cursor_y, rows - 1].min
+        end
+
+        def to_text
+          lines.map(&:to_s).join("\n")
+        end
+
+        private
+
+        def clear_above_cursor
+          0.upto(@cursor_y - 1) { |i| lines[i]&.clear(@cols) }
+          lines[@cursor_y]&.clear_range(0, @cursor_x)
+        end
+
+        def clear_below_cursor
+          lines[@cursor_y]&.clear_from(@cursor_x, @cols)
+          (@cursor_y + 1).upto(@rows - 1) { |i| lines[i]&.clear(@cols) }
+        end
+
+        # 一行字符
+        class Line
+          attr_reader :cells
+
+          def initialize(cols)
+            @cells = Array.new(cols) { Cell.new }
+          end
+
+          def put_char(ch, x, style = {})
+            @cells[x] = Cell.new(ch, style) if x < @cells.size
+          end
+
+          def clear(cols)
+            @cells = Array.new(cols) { Cell.new }
+          end
+
+          def clear_from(x, cols)
+            x.upto([cols - 1, @cells.size - 1].min) { |i| @cells[i] = Cell.new }
+          end
+
+          def clear_range(from, to)
+            from.upto([to, @cells.size - 1].min) { |i| @cells[i] = Cell.new }
+          end
+
+          def resize(cols)
+            if cols > @cells.size
+              @cells += Array.new(cols - @cells.size) { Cell.new }
+            elsif cols < @cells.size
+              @cells = @cells[0...cols]
+            end
+          end
+
+          def to_s
+            @cells.map(&:char).join.rstrip
+          end
+        end
+
+        # 单字符单元
+        Cell = Struct.new(:char, :style) do
+          def initialize(char = " ", style = {})
+            super
+          end
+        end
+      end
+    end
+  end
+end

data/service/ssh/lib/network_infra_utility/ssh/terminal/theme.rb

@@ -0,0 +1,127 @@
+# frozen_string_literal: true
+
+require "yaml"
+
+module NetworkInfraUtility
+  module SSH
+    module Terminal
+      # 配色方案加载/查询。
+      # 对应需求 FR-TERM-002：内置 ≥ 10 套配色，支持导入 iTerm2/VSCode。
+      class Theme
+        BUILTIN_THEMES = {
+          "default" => {
+            bg: "#1e1e1e", fg: "#d4d4d4", cursor: "#d4d4d4",
+            palette: %w[#000000 #cd3131 #0dbc79 #e5e510 #2472c8 #bc3fbc #11a8cd #e5e5e5
+                        #666666 #f14c4c #23d18b #f5f543 #3b8eea #d670d6 #29b8db #e5e5e5]
+          },
+          "solarized-dark" => {
+            bg: "#002b36", fg: "#839496", cursor: "#93a1a1",
+            palette: %w[#073642 #dc322f #859900 #b58900 #268bd2 #d33682 #2aa198 #eee8d5
+                        #002b36 #cb4b16 #586e75 #657b83 #839496 #6c71c4 #93a1a1 #fdf6e3]
+          },
+          "solarized-light" => {
+            bg: "#fdf6e3", fg: "#657b83", cursor: "#586e75",
+            palette: %w[#073642 #dc322f #859900 #b58900 #268bd2 #d33682 #2aa198 #eee8d5
+                        #002b36 #cb4b16 #586e75 #657b83 #839496 #6c71c4 #93a1a1 #fdf6e3]
+          },
+          "dracula" => {
+            bg: "#282a36", fg: "#f8f8f2", cursor: "#f8f8f2",
+            palette: %w[#000000 #ff5555 #50fa7b #f1fa8c #bd93f9 #ff79c6 #8be9fd #bfbfbf
+                        #4d4d4d #ff6e67 #5af78e #f4f99d #caa9fa #ff92d0 #9aedfe #e6e6e6]
+          },
+          "monokai" => {
+            bg: "#272822", fg: "#f8f8f2", cursor: "#f8f8f0",
+            palette: %w[#272822 #f92672 #a6e22e #fd971f #66d9ef #ae81ff #a1efe4 #f8f8f2
+                        #75715e #f92672 #a6e22e #fd971f #66d9ef #ae81ff #a1efe4 #f9f8f5]
+          },
+          "nord" => {
+            bg: "#2e3440", fg: "#d8dee9", cursor: "#d8dee9",
+            palette: %w[#3b4252 #bf616a #a3be8c #ebcb8b #81a1c1 #b48ead #88c0d0 #e5e9f0
+                        #4c566a #bf616a #a3be8c #ebcb8b #81a1c1 #b48ead #8fbcbb #eceff4]
+          },
+          "gruvbox-dark" => {
+            bg: "#282828", fg: "#ebdbb2", cursor: "#ebdbb2",
+            palette: %w[#282828 #cc241d #98971a #d79921 #458588 #b16286 #689d6a #a89984
+                        #928374 #fb4934 #b8bb26 #fabd2f #83a598 #d3869b #8ec07c #ebdbb2]
+          },
+          "one-dark" => {
+            bg: "#282c34", fg: "#abb2bf", cursor: "#abb2bf",
+            palette: %w[#282c34 #e06c75 #98c379 #e5c07b #61afef #c678dd #56b6c2 #abb2bf
+                        #5c6370 #e06c75 #98c379 #e5c07b #61afef #c678dd #56b6c2 #ffffff]
+          },
+          "tokyo-night" => {
+            bg: "#1a1b26", fg: "#a9b1d6", cursor: "#c0caf5",
+            palette: %w[#15161e #f7768e #9ece6a #e0af68 #7aa2f7 #bb9af7 #7dcfff #a9b1d6
+                        #414868 #f7768e #9ece6a #e0af68 #7aa2f7 #bb9af7 #7dcfff #c0caf5]
+          },
+          "catppuccin-mocha" => {
+            bg: "#1e1e2e", fg: "#cdd6f4", cursor: "#f5e0dc",
+            palette: %w[#45475a #f38ba8 #a6e3a1 #f9e2af #89b4fa #f5c2e7 #94e2d5 #bac2de
+                        #585b70 #f38ba8 #a6e3a1 #f9e2af #89b4fa #f5c2e7 #94e2d5 #a6adc8]
+          },
+          "github-dark" => {
+            bg: "#0d1117", fg: "#c9d1d9", cursor: "#6cae50",
+            palette: %w[#484f58 #ff7b72 #7ee787 #f2cc60 #79c0ff #d2a8ff #56d4dd #c9d1d9
+                        #6e7681 #ffa198 #56d364 #e3b341 #79c0ff #bc8cff #56d4dd #f0f6fc]
+          }
+        }.freeze
+
+        attr_reader :name, :data
+
+        class << self
+          def load(name)
+            data = BUILTIN_THEMES[name] || load_custom(name) || BUILTIN_THEMES["default"]
+            new(name, data)
+          end
+
+          def default
+            load("default")
+          end
+
+          def builtin_names
+            BUILTIN_THEMES.keys
+          end
+
+          # 从 iTerm2 .itermcolors 文件导入
+          def import_iterm(path)
+            # V2.0 功能，预留接口
+            raise NotImplementedError, "iTerm2 import is V2.0"
+          end
+
+          # 从 VSCode .json 主题文件导入
+          def import_vscode(path)
+            # V2.0 功能，预留接口
+            raise NotImplementedError, "VSCode import is V2.0"
+          end
+
+          private
+
+          def load_custom(name)
+            # 从 ~/.network-infra-utility/themes/<name>.yml 加载
+            themes_dir = File.expand_path("~/.network-infra-utility/themes")
+            path = File.join(themes_dir, "#{name}.yml")
+            return nil unless File.exist?(path)
+
+            YAML.safe_load(File.read(path), symbolize_names: true)
+          end
+        end
+
+        def initialize(name, data)
+          @name = name
+          @data = data.transform_keys(&:to_sym)
+        end
+
+        def bg;   @data[:bg] || "#000000"; end
+        def fg;   @data[:fg] || "#ffffff"; end
+        def cursor; @data[:cursor] || fg; end
+        def palette; @data[:palette] || []; end
+
+        def color(index)
+          return fg if index.nil? || index < 0 || index >= palette.size
+
+          palette[index]
+        end
+      end
+    end
+  end
+end

data/service/ssh/lib/network_infra_utility/ssh/version.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+module NetworkInfraUtility
+  module SSH
+    VERSION = "1.0.0"
+  end
+end

data/service/ssh/lib/network_infra_utility/ssh.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+require_relative "ssh/version"
+require_relative "ssh/client"
+require_relative "ssh/ipc/transport"
+require_relative "ssh/ipc/router"
+require_relative "ssh/ipc/coalesce"
+require_relative "ssh/ipc/errors"
+require_relative "ssh/session/manager"
+require_relative "ssh/session/session"
+require_relative "ssh/session/tree"
+require_relative "ssh/session/history"
+require_relative "ssh/terminal/emulator"
+require_relative "ssh/terminal/ansi_parser"
+require_relative "ssh/terminal/screen"
+require_relative "ssh/terminal/buffer"
+require_relative "ssh/terminal/theme"
+require_relative "ssh/terminal/logger"
+require_relative "ssh/security/vault"
+require_relative "ssh/security/host_key"
+require_relative "ssh/automation/macro_engine"
+require_relative "ssh/config/settings"
+require_relative "ssh/config/schema"
+require_relative "ssh/config/store"
+
+# NetworkInfraUtility::SSH — SSH 连接客户端
+#
+# 架构（LLD §1）：
+#   Erlang/OTP ssh 做协议核心，Ruby 做调度与终端，
+#   两者经 JSON-RPC 2.0 over Unix Socket/TCP 通信。
+#
+# 组合根：SSH::Client——构造时编排全部子系统，
+# 管理 Erlang 引擎生命周期与 IPC 连接。
+#
+# 用法：
+#   client = NetworkInfraUtility::SSH::Client.new
+#   client.start_engine
+#   session = client.connect(host: "10.0.0.1", user: "admin")
+#   terminal = session.open_terminal
+#   terminal.puts("show version")
+module NetworkInfraUtility
+  module SSH
+  end
+end