chocomint 1.0.0

data/lib/chocomint/tools/make_dir.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+require_relative "base"
+
+module Chocomint
+  module Tools
+    # 許可ディレクトリ内にディレクトリを作成する (親も含め mkdir_p)。
+    class MakeDir < Base
+      def initialize(path_guard:)
+        @path_guard = path_guard
+      end
+
+      def name = "make_dir"
+
+      def description
+        "指定パスにディレクトリを作成する (親ディレクトリも作成)。path は許可ディレクトリ内であること。"
+      end
+
+      def schema
+        {
+          "type" => "object",
+          "properties" => {
+            "path" => { "type" => "string" }
+          },
+          "required" => %w[path],
+          "additionalProperties" => false
+        }
+      end
+
+      def call(args)
+        measure do
+          abs = @path_guard.resolve(args["path"])
+          if File.file?(abs)
+            next fail(stderr: "a file already exists at #{args['path']}")
+          end
+
+          require "fileutils"
+          FileUtils.mkdir_p(abs)
+          ok(stdout: "created directory #{args['path']}")
+        end
+      end
+    end
+  end
+end

data/lib/chocomint/tools/read_file.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+require_relative "base"
+
+module Chocomint
+  module Tools
+    # 許可ディレクトリ内のファイルを読み、内容を stdout に返す。
+    class ReadFile < Base
+      def initialize(path_guard:, max_file_bytes:)
+        @path_guard = path_guard
+        @max_file_bytes = max_file_bytes
+      end
+
+      def name = "read_file"
+
+      def description
+        "指定パスのファイル内容を読み取る。path は許可ディレクトリ内であること。"
+      end
+
+      def schema
+        {
+          "type" => "object",
+          "properties" => {
+            "path" => { "type" => "string" }
+          },
+          "required" => %w[path],
+          "additionalProperties" => false
+        }
+      end
+
+      def call(args)
+        measure do
+          abs = @path_guard.resolve(args["path"])
+          next fail(stderr: "no such file: #{args['path']}") unless File.file?(abs)
+
+          if File.size(abs) > @max_file_bytes
+            next fail(stderr: "file exceeds max_file_bytes (#{@max_file_bytes})")
+          end
+
+          ok(stdout: File.read(abs))
+        end
+      end
+    end
+  end
+end

data/lib/chocomint/tools/registry.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+require_relative "../errors"
+
+module Chocomint
+  module Tools
+    # ツール名 -> インスタンスの解決 (DESIGN §14: 未知ツールは FAIL)。
+    class Registry
+      def initialize
+        @tools = {}
+      end
+
+      def register(tool)
+        @tools[tool.name] = tool
+        self
+      end
+
+      def fetch(name)
+        @tools[name] or raise UnknownToolError, "unknown tool: #{name}"
+      end
+
+      def include?(name) = @tools.key?(name)
+
+      def all = @tools.values
+
+      def tool_definitions = all.map(&:to_tool_definition)
+    end
+  end
+end

data/lib/chocomint/tools/run_command.rb

@@ -0,0 +1,96 @@
+# frozen_string_literal: true
+
+require "open3"
+require_relative "base"
+require_relative "../errors"
+
+module Chocomint
+  module Tools
+    # whitelist に登録されたコマンドのみを実行する (DESIGN §13)。
+    # shell を介さず引数配列で spawn するため、shell 展開・インジェクションは起きない。
+    class RunCommand < Base
+      def initialize(allowed_commands:, timeout_sec:, max_output_bytes:, cwd: Dir.pwd,
+                     path_guard: nil, allow_all: false)
+        @allowed_commands = Array(allowed_commands)
+        @timeout_sec = timeout_sec
+        @max_output_bytes = max_output_bytes
+        @cwd = cwd
+        @path_guard = path_guard
+        @allow_all = allow_all
+      end
+
+      def name = "run_command"
+
+      def description
+        if @allow_all
+          "任意のコマンドを引数付きで実行する (whitelist 無効化 - allow_all_commands: true)。" \
+            "shell 展開は行われない (args は個別のトークンとして渡す)。"
+        else
+          "許可されたコマンドを引数付きで実行する。許可コマンド: #{@allowed_commands.join(', ')}。" \
+            "shell 展開は行われない (args は個別のトークンとして渡す)。"
+        end
+      end
+
+      def schema
+        {
+          "type" => "object",
+          "properties" => {
+            "command" => { "type" => "string" },
+            "args" => {
+              "type" => "array",
+              "items" => { "type" => "string" }
+            }
+          },
+          "required" => %w[command],
+          "additionalProperties" => false
+        }
+      end
+
+      def call(args)
+        measure do
+          command = args["command"].to_s
+          cmd_args = Array(args["args"]).map(&:to_s)
+
+          unless @allow_all || @allowed_commands.include?(command)
+            next fail(stderr: "command not allowed: #{command}")
+          end
+
+          # 外部プロセスは cwd 配下で自由に書き込めるため、引数に現れるパスが
+          # 許可ディレクトリ内に収まるか事前に検証する (allowed_dirs の隔離を守る)。
+          if @path_guard
+            begin
+              @path_guard.validate_args!(cmd_args)
+            rescue PathAccessError => e
+              next fail(stderr: e.message)
+            end
+          end
+
+          run(command, cmd_args)
+        end
+      end
+
+      private
+
+      def run(command, cmd_args)
+        stdout, stderr, status =
+          Open3.capture3([command, command], *cmd_args, chdir: @cwd)
+
+        {
+          exit_code: status.exitstatus || 1,
+          stdout: sanitize(stdout),
+          stderr: sanitize(stderr)
+        }
+      rescue Errno::ENOENT
+        fail(stderr: "command not found: #{command}")
+      end
+
+      # 外部プロセスの出力を「上限まで切り詰め + 妥当な UTF-8 に正規化」する。
+      # Open3 の出力は ASCII-8BIT で、Windows の python 等は UTF-8 ではなく CP932 で
+      # 日本語を出力するため、UTF-8 誤認で化けないよう to_utf8 で CP932 変換を試みる。
+      def sanitize(text)
+        text = text.byteslice(0, @max_output_bytes) if text.bytesize > @max_output_bytes
+        to_utf8(text)
+      end
+    end
+  end
+end

data/lib/chocomint/tools/shell.rb

@@ -0,0 +1,169 @@
+# frozen_string_literal: true
+
+require "open3"
+require "timeout"
+require_relative "base"
+
+module Chocomint
+  module Tools
+    # shell を明示的に介してスクリプトを実行する (bash / powershell)。
+    #
+    # DESIGN §13 の「shell 展開禁止」原則からは外れる強力なツールなので、
+    # cwd を base_dir に固定し、timeout と出力サイズ制限を必ず適用する。
+    # bash / powershell の 2 種類を launcher で切り替える。
+    class Shell < Base
+      # Windows では PATH 上の "bash" が WSL インタロップ (system32\bash.exe) に
+      # 解決されることがあり、これは -c に渡した複数文スクリプト中の変数展開が
+      # 壊れる既知の癖を持つ。Git Bash を優先的に探し、無ければ "bash" にフォールバックする。
+      GIT_BASH_CANDIDATES = [
+        "C:/Program Files/Git/bin/bash.exe",
+        "C:/Program Files (x86)/Git/bin/bash.exe"
+      ].freeze
+
+      def self.resolve_bash_exe
+        GIT_BASH_CANDIDATES.find { |path| File.exist?(path) } || "bash"
+      end
+
+      LAUNCHERS = {
+        "bash" => {
+          name: "bash",
+          exe: resolve_bash_exe,
+          argv: ->(script) { ["-c", script] }
+        },
+        "powershell" => {
+          name: "powershell",
+          exe: "pwsh",
+          argv: ->(script) { ["-NoProfile", "-NonInteractive", "-Command", script] }
+        }
+      }.freeze
+
+      def self.build(kind, **opts)
+        spec = LAUNCHERS.fetch(kind)
+        new(spec: spec, **opts)
+      end
+
+      def initialize(spec:, timeout_sec:, max_output_bytes:, cwd: Dir.pwd, path_guard: nil)
+        @spec = spec
+        @timeout_sec = timeout_sec
+        @max_output_bytes = max_output_bytes
+        @cwd = cwd
+        @path_guard = path_guard
+      end
+
+      def name = @spec[:name]
+
+      def description
+        "#{@spec[:name]} スクリプトを実行する。作業ディレクトリは固定され、" \
+          "timeout #{@timeout_sec}s と出力サイズ制限が適用される。"
+      end
+
+      def schema
+        {
+          "type" => "object",
+          "properties" => {
+            "script" => { "type" => "string" }
+          },
+          "required" => %w[script],
+          "additionalProperties" => false
+        }
+      end
+
+      def call(args)
+        measure do
+          script = args["script"].to_s
+          if (denied = sandbox_violation(script))
+            next fail(stderr: denied)
+          end
+
+          run(script)
+        end
+      end
+
+      private
+
+      # スクリプトは自由記述で完全なトークン分割はできないため、明白な脱出パターン
+      # (絶対パス・親ディレクトリ遡り) だけを弾く軽量ガード。run_command の引数検証と
+      # 違い網羅は狙わず、workspace 外への露骨な書き込みを防ぐ最低限の防波堤とする。
+      # cwd は base_dir (プロジェクトルート) 固定で workspace 外なので、絶対パスと
+      # ".." による遡りを疑わしいトークンとして弾く。
+
+      # scheme://... の URL。絶対パス判定の前に取り除き、http:// を誤検知しない。
+      URL = %r{\b[a-zA-Z][a-zA-Z0-9+.\-]*://\S*}
+      # Windows ドライブ (C:\ / C:/)、UNC (\\host)、Unix 絶対パス (区切り直後の / で
+      # 始まり、ルート単独 "/" も含む)。
+      ABSOLUTE_PATH = %r{[A-Za-z]:[\\/]|\\\\[^\s]|(?<=\A|[\s'"=(><|&;])/}
+      # "../" もしくは "..\" (先頭が単語構成文字でない = パス区切り直後)。
+      PARENT_ESCAPE = %r{(?<![\w.])\.\.[\\/]}
+
+      def sandbox_violation(script)
+        return nil if @path_guard.nil?
+
+        # URL を除去してから絶対パスを探す (スキームの // を絶対パスと混同しないため)。
+        stripped = script.gsub(URL, " ")
+
+        if stripped.match?(ABSOLUTE_PATH)
+          return "absolute paths are not allowed in shell scripts (stay within #{workspace_hint})"
+        end
+        if stripped.match?(PARENT_ESCAPE)
+          return "parent-directory escape (..) is not allowed in shell scripts"
+        end
+
+        nil
+      end
+
+      def workspace_hint
+        @path_guard.allowed_roots.first
+      end
+
+      # 色付き出力を抑止する環境変数。多くの CLI が尊重する慣習 (NO_COLOR / TERM=dumb)。
+      # これで大半の色コードは出なくなるが、残った ANSI は表示側でも除去する (二重の対策)。
+      NO_COLOR_ENV = { "NO_COLOR" => "1", "TERM" => "dumb", "CLICOLOR" => "0" }.freeze
+
+      def run(script)
+        argv = @spec[:argv].call(script)
+        out = +""
+        err = +""
+        status = nil
+
+        Open3.popen3(NO_COLOR_ENV, @spec[:exe], *argv, chdir: @cwd) do |stdin, stdout, stderr, wait|
+          stdin.close
+          Timeout.timeout(@timeout_sec) do
+            out << stdout.read.to_s
+            err << stderr.read.to_s
+            status = wait.value
+          end
+        rescue Timeout::Error
+          Process.kill("KILL", wait.pid) rescue nil
+          return fail(stderr: "timeout after #{@timeout_sec}s")
+        end
+
+        {
+          exit_code: status&.exitstatus || 1,
+          stdout: sanitize(out),
+          # strip_noise (gsub) は妥当な UTF-8 でないと落ちるため、先に sanitize する。
+          stderr: strip_noise(sanitize(err))
+        }
+      rescue Errno::ENOENT
+        fail(stderr: "shell not found: #{@spec[:exe]}")
+      end
+
+      # 外部プロセスの出力を「上限まで切り詰め + 妥当な UTF-8 に正規化」する。
+      # popen3 の出力は ASCII-8BIT で、Windows のコマンドは UTF-8 ではなく CP932 で
+      # 日本語を出力するため、UTF-8 誤認で化けないよう to_utf8 で CP932 変換を試みる。
+      def sanitize(text)
+        text = text.byteslice(0, @max_output_bytes) if text.bytesize > @max_output_bytes
+        to_utf8(text)
+      end
+
+      # WSL bash 実行時に毎回出る無害な systemd 警告。実行結果とは無関係だが、
+      # stderr に残ると verifier LLM が失敗と誤判定しやすいため取り除く。
+      WSL_SYSTEMD_NOISE = /^wsl: Failed to start the systemd user session.*\n?/
+
+      def strip_noise(text)
+        return text unless @spec[:name] == "bash"
+
+        text.gsub(WSL_SYSTEMD_NOISE, "")
+      end
+    end
+  end
+end

data/lib/chocomint/tools/write_file.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+require_relative "base"
+
+module Chocomint
+  module Tools
+    # 許可ディレクトリ内にファイルを書き込む (DESIGN §8)。
+    class WriteFile < Base
+      def initialize(path_guard:, max_file_bytes:)
+        @path_guard = path_guard
+        @max_file_bytes = max_file_bytes
+      end
+
+      def name = "write_file"
+
+      def description
+        "指定パスにテキストを書き込む。path は許可ディレクトリ内であること。"
+      end
+
+      def schema
+        {
+          "type" => "object",
+          "properties" => {
+            "path" => { "type" => "string" },
+            "content" => { "type" => "string" }
+          },
+          "required" => %w[path content],
+          "additionalProperties" => false
+        }
+      end
+
+      def call(args)
+        measure do
+          content = args["content"].to_s
+          if content.bytesize > @max_file_bytes
+            next fail(stderr: "content exceeds max_file_bytes (#{@max_file_bytes})")
+          end
+
+          abs = @path_guard.resolve(args["path"])
+          FileUtils_mkdir(File.dirname(abs))
+          bytes = File.write(abs, content)
+          ok(stdout: "wrote #{bytes} bytes to #{args['path']}")
+        end
+      end
+
+      private
+
+      def FileUtils_mkdir(dir)
+        require "fileutils"
+        FileUtils.mkdir_p(dir)
+      end
+    end
+  end
+end

data/lib/chocomint/validator/machine_validator.rb

@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+
+module Chocomint
+  module Validator
+    # ツール実行結果の機械的検証 (DESIGN §3)。
+    # LLM を呼ばずに Ruby だけで判定する。true=合格 / false=RETRY。
+    class MachineValidator
+      def initialize(path_guard:, max_output_bytes:)
+        @path_guard = path_guard
+        @max_output_bytes = max_output_bytes
+      end
+
+      # proposal: { "tool" => name, "arguments" => {...} }
+      # result:   { exit_code:, stdout:, stderr:, duration_ms: }
+      def valid?(proposal, result)
+        return false unless common_checks(result)
+
+        args = proposal["arguments"]
+        case proposal["tool"]
+        when "write_file"  then write_file_checks(args)
+        when "read_file"   then result[:stdout].to_s.bytesize >= 0
+        when "list_dir"    then true
+        when "append_file" then File.exist?(@path_guard.resolve(args["path"]))
+        when "delete_file" then !File.exist?(@path_guard.resolve(args["path"]))
+        when "make_dir"    then File.directory?(@path_guard.resolve(args["path"]))
+        when "file_info"   then json?(result[:stdout])
+        when "run_command" then true # exit_code は common_checks で 0 を確認済み
+        when "edit"        then File.exist?(@path_guard.resolve(args["path"]))
+        when "ls"          then json?(result[:stdout])
+        when "grep", "glob" then true
+        when "bash", "powershell" then true # exit_code は common_checks で確認済み
+        else true
+        end
+      rescue StandardError
+        false
+      end
+
+      private
+
+      def common_checks(result)
+        result[:exit_code] == 0 &&
+          result[:stdout].to_s.bytesize <= @max_output_bytes &&
+          result[:stderr].to_s.bytesize <= @max_output_bytes
+      end
+
+      # ファイル存在 + 内容一致 (DESIGN §3 File Tool Checks)。
+      def write_file_checks(args)
+        abs = @path_guard.resolve(args["path"])
+        return false unless File.exist?(abs)
+        return false unless File.size(abs) >= 0
+
+        File.read(abs) == args["content"].to_s
+      end
+
+      # file_info の stdout が JSON としてパース可能か (DESIGN §3 JSON Tool Checks)。
+      def json?(text)
+        require "json"
+        JSON.parse(text.to_s)
+        true
+      rescue JSON::ParserError
+        false
+      end
+    end
+  end
+end

data/lib/chocomint/validator/semantic_validator.rb

@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+
+module Chocomint
+  module Validator
+    # 副 LLM による意味的検証 (DESIGN §4)。
+    # verifier 無効時は常に合格。有効時は 1/0 のみを合否とし、
+    # 不正出力 (nil) は「不合格 = RETRY」として扱う (DESIGN §14)。
+    class SemanticValidator
+      # result にファイル内容を含まないツール (write_file 等は「N bytes 書いた」
+      # としか返さない) は、verifier が中身を確認できず不当に不合格判定しがちなので、
+      # 対象ファイルの中身を読んで result に付加してから渡す。
+      FILE_CONTENT_TOOLS = %w[write_file append_file edit].freeze
+
+      def initialize(verifier_client:, enabled:, path_guard: nil)
+        @verifier_client = verifier_client
+        @enabled = enabled
+        @path_guard = path_guard
+      end
+
+      # 戻り値: [pass(boolean), verifier_output(String|nil)]
+      def validate(request:, result:, expectations: nil, trace_id: nil, proposal: nil)
+        return [true, nil] unless @enabled
+
+        output = @verifier_client.verify(
+          request: request,
+          expectations: expectations,
+          result: augment_with_file_content(result, proposal),
+          trace_id: trace_id
+        )
+
+        [output == "1", output]
+      end
+
+      private
+
+      def augment_with_file_content(result, proposal)
+        return result unless @path_guard && proposal.is_a?(Hash)
+        return result unless FILE_CONTENT_TOOLS.include?(proposal["tool"])
+
+        path = proposal.dig("arguments", "path")
+        return result unless path
+
+        abs = @path_guard.resolve(path)
+        return result unless File.file?(abs)
+
+        result.merge(file_content: File.read(abs))
+      rescue StandardError
+        result
+      end
+    end
+  end
+end

data/lib/chocomint.rb

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+
+module Chocomint
+  VERSION = "1.0.0"
+end
+
+require_relative "chocomint/errors"
+require_relative "chocomint/config"
+require_relative "chocomint/security/path_guard"
+require_relative "chocomint/tools/base"
+require_relative "chocomint/tools/registry"
+require_relative "chocomint/tools/write_file"
+require_relative "chocomint/tools/read_file"
+require_relative "chocomint/tools/list_dir"
+require_relative "chocomint/tools/append_file"
+require_relative "chocomint/tools/delete_file"
+require_relative "chocomint/tools/file_info"
+require_relative "chocomint/tools/make_dir"
+require_relative "chocomint/tools/run_command"
+require_relative "chocomint/tools/edit"
+require_relative "chocomint/tools/grep"
+require_relative "chocomint/tools/glob"
+require_relative "chocomint/tools/ls"
+require_relative "chocomint/tools/shell"
+require_relative "chocomint/tools/fetch_url"
+require_relative "chocomint/validator/machine_validator"
+require_relative "chocomint/validator/semantic_validator"
+require_relative "chocomint/logger/sqlite_logger"
+require_relative "chocomint/llm/base_client"
+require_relative "chocomint/llm/primary_client"
+require_relative "chocomint/llm/verifier_client"
+require_relative "chocomint/llm/chat_client"
+require_relative "chocomint/orchestrator"
+require_relative "chocomint/planner"
+require_relative "chocomint/factory"
+require_relative "chocomint/server"
+
+# AI エディタのコンソール (ConPTY + WebSocket) は Windows + ffi + websocket-driver 依存。
+# 未対応環境や gem 未導入で chocomint 全体のロードが壊れないよう、読み込み失敗は
+# 握りつぶす (コンソール機能なしで動作する)。CONSOLE_AVAILABLE で可否を判定できる。
+module Chocomint
+  CONSOLE_AVAILABLE =
+    begin
+      require_relative "chocomint/console/conpty"
+      require_relative "chocomint/console/worker_process"
+      require_relative "chocomint/console/server"
+      true
+    rescue LoadError
+      false
+    end
+end