chocomint 1.0.0

data/lib/chocomint/tools/append_file.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+require_relative "base"
+
+module Chocomint
+  module Tools
+    # 許可ディレクトリ内のファイルに追記する。ファイルが無ければ新規作成。
+    class AppendFile < Base
+      def initialize(path_guard:, max_file_bytes:)
+        @path_guard = path_guard
+        @max_file_bytes = max_file_bytes
+      end
+
+      def name = "append_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
+          abs = @path_guard.resolve(args["path"])
+
+          current = File.exist?(abs) ? File.size(abs) : 0
+          if current + content.bytesize > @max_file_bytes
+            next fail(stderr: "resulting file exceeds max_file_bytes (#{@max_file_bytes})")
+          end
+
+          require "fileutils"
+          FileUtils.mkdir_p(File.dirname(abs))
+          File.open(abs, "a") { |f| f.write(content) }
+          ok(stdout: "appended #{content.bytesize} bytes to #{args['path']}")
+        end
+      end
+    end
+  end
+end

data/lib/chocomint/tools/base.rb

@@ -0,0 +1,88 @@
+# frozen_string_literal: true
+
+module Chocomint
+  module Tools
+    # ツールの共通インターフェース (DESIGN §7)。
+    # call は必ず { exit_code:, stdout:, stderr:, duration_ms: } を返す。
+    class Base
+      def name
+        raise NotImplementedError
+      end
+
+      # json_schemer 用の JSON Schema (arguments の検証に使う)。
+      def schema
+        raise NotImplementedError
+      end
+
+      def call(_args)
+        raise NotImplementedError
+      end
+
+      # Anthropic Messages API の tools 配列に渡す形式。
+      def to_tool_definition
+        {
+          "name" => name,
+          "description" => description,
+          "input_schema" => schema
+        }
+      end
+
+      def description
+        ""
+      end
+
+      protected
+
+      # 経過時間を測りつつ結果ハッシュを組み立てるヘルパ。
+      def measure
+        started = now_ms
+        body = yield
+        body.merge(duration_ms: now_ms - started)
+      rescue StandardError => e
+        {
+          exit_code: 1,
+          stdout: "",
+          stderr: e.message,
+          duration_ms: now_ms - started
+        }
+      end
+
+      def ok(stdout: "", stderr: "")
+        { exit_code: 0, stdout: stdout, stderr: stderr }
+      end
+
+      def fail(stderr:, stdout: "")
+        { exit_code: 1, stdout: stdout, stderr: stderr }
+      end
+
+      def now_ms
+        Process.clock_gettime(Process::CLOCK_MONOTONIC, :millisecond)
+      end
+
+      # 外部プロセスの出力 (ASCII-8BIT) を妥当な UTF-8 文字列に正規化する。
+      # Windows の python 等は UTF-8 ではなく OS ロケール (日本語なら CP932) で
+      # 日本語を出力するため、UTF-8 として不正なら CP932 からの変換を試みる。
+      # それも失敗した場合のみ、無効バイトを置換文字にフォールバックする。
+      # こうしないと CP932 の日本語が UTF-8 誤認で全て "□" 化けしてしまう。
+      def to_utf8(text)
+        return text if text.nil?
+
+        s = text.dup.force_encoding("UTF-8")
+        return s if s.valid_encoding?
+
+        # OS の既定エンコーディング (Windows-日本語なら Windows-31J=CP932) を優先し、
+        # 無ければ CP932 を試す。undef/invalid は置換しつつも大半の日本語を救う。
+        [Encoding.default_external, Encoding::CP932].uniq.each do |enc|
+          next if enc == Encoding::UTF_8
+
+          begin
+            return text.dup.force_encoding(enc).encode("UTF-8", invalid: :replace, undef: :replace)
+          rescue EncodingError, ArgumentError
+            next
+          end
+        end
+        s.scrub("�")
+      end
+    end
+  end
+end

data/lib/chocomint/tools/delete_file.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require_relative "base"
+
+module Chocomint
+  module Tools
+    # 許可ディレクトリ内のファイルを削除する。ディレクトリは対象外。
+    class DeleteFile < Base
+      def initialize(path_guard:)
+        @path_guard = path_guard
+      end
+
+      def name = "delete_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.exist?(abs)
+          next fail(stderr: "not a file: #{args['path']}") unless File.file?(abs)
+
+          File.delete(abs)
+          ok(stdout: "deleted #{args['path']}")
+        end
+      end
+    end
+  end
+end

data/lib/chocomint/tools/edit.rb

@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+
+require_relative "base"
+
+module Chocomint
+  module Tools
+    # 許可ディレクトリ内のファイルで文字列置換を行う (Claude Code の Edit 相当)。
+    # old_string は一意でなければならない (replace_all=true で全置換)。
+    class Edit < Base
+      def initialize(path_guard:, max_file_bytes:)
+        @path_guard = path_guard
+        @max_file_bytes = max_file_bytes
+      end
+
+      def name = "edit"
+
+      def description
+        "ファイル内の old_string を new_string に置換する。既定では old_string が一意である" \
+          "必要があり、replace_all=true で全置換。path は許可ディレクトリ内であること。"
+      end
+
+      def schema
+        {
+          "type" => "object",
+          "properties" => {
+            "path" => { "type" => "string" },
+            "old_string" => { "type" => "string" },
+            "new_string" => { "type" => "string" },
+            "replace_all" => { "type" => "boolean" }
+          },
+          "required" => %w[path old_string new_string],
+          "additionalProperties" => false
+        }
+      end
+
+      def call(args)
+        measure do
+          old_str = args["old_string"].to_s
+          new_str = args["new_string"].to_s
+          replace_all = args["replace_all"] == true
+
+          abs = @path_guard.resolve(args["path"])
+          next fail(stderr: "no such file: #{args['path']}") unless File.file?(abs)
+          next fail(stderr: "old_string and new_string are identical") if old_str == new_str
+
+          content = File.read(abs)
+          count = content.scan(Regexp.new(Regexp.escape(old_str))).size
+          next fail(stderr: "old_string not found") if count.zero?
+          if count > 1 && !replace_all
+            next fail(stderr: "old_string is not unique (#{count} matches); use replace_all")
+          end
+
+          updated = replace_all ? content.gsub(old_str, new_str) : content.sub(old_str, new_str)
+          if updated.bytesize > @max_file_bytes
+            next fail(stderr: "resulting file exceeds max_file_bytes (#{@max_file_bytes})")
+          end
+
+          replaced = replace_all ? count : 1
+          File.write(abs, updated)
+          ok(stdout: "replaced #{replaced} occurrence(s) in #{args['path']}")
+        end
+      end
+    end
+  end
+end

data/lib/chocomint/tools/fetch_url.rb

@@ -0,0 +1,156 @@
+# frozen_string_literal: true
+
+require "net/http"
+require "uri"
+require_relative "base"
+
+module Chocomint
+  module Tools
+    # 指定 URL のウェブページを取得し、本文テキストを抽出したうえで、
+    # 別途 LLM (ChatClient#extract) に「ユーザーの目的に沿った情報」を要約・抽出させる。
+    #
+    # 取得は標準ライブラリ net/http のみ (依存追加なし)。http/https に限定し、
+    # リダイレクトを数回だけ追い、応答サイズに上限を設ける。生 HTML をそのまま
+    # LLM に流すとノイズが多く読み取りにくいため、タグ除去で本文テキスト化してから渡す。
+    class FetchUrl < Base
+      MAX_REDIRECTS = 5
+      # 取得を許す本文サイズの上限 (バイト)。巨大ページで LLM 入力が溢れるのを防ぐ。
+      DEFAULT_MAX_BYTES = 2_000_000
+      # LLM に渡す抽出後テキストの上限 (文字)。小型ローカルモデルの文脈長に配慮する。
+      EXTRACT_TEXT_LIMIT = 12_000
+
+      # chat_client: ChatClient (extract を持つ)。timeout_sec: 取得タイムアウト。
+      def initialize(chat_client:, timeout_sec: 30, max_bytes: DEFAULT_MAX_BYTES)
+        @chat_client = chat_client
+        @timeout_sec = timeout_sec
+        @max_bytes = max_bytes
+      end
+
+      def name = "fetch_url"
+
+      def description
+        "指定した URL のウェブページを取得し、その内容から目的 (query) に沿った情報を" \
+          "抽出して返す。最新情報の調査や外部ドキュメントの参照に使う。" \
+          "url は http:// または https:// で始まること。"
+      end
+
+      def schema
+        {
+          "type" => "object",
+          "properties" => {
+            "url" => { "type" => "string", "description" => "参照する http/https の URL" },
+            "query" => {
+              "type" => "string",
+              "description" => "そのページから知りたいこと (任意。未指定ならページの要点をまとめる)"
+            }
+          },
+          "required" => %w[url],
+          "additionalProperties" => false
+        }
+      end
+
+      def call(args)
+        measure do
+          url = args["url"].to_s
+          uri = parse_http_uri(url)
+          next fail(stderr: "url must be an http/https URL: #{url}") if uri.nil?
+
+          body, err = fetch(uri)
+          next fail(stderr: err) if body.nil?
+
+          text = html_to_text(body)
+          if text.empty?
+            next fail(stderr: "no readable text extracted from #{url}")
+          end
+
+          summary = @chat_client.extract(args["query"], text[0, EXTRACT_TEXT_LIMIT], url: url)
+          ok(stdout: summary)
+        end
+      end
+
+      private
+
+      def parse_http_uri(url)
+        uri = URI.parse(url)
+        return uri if uri.is_a?(URI::HTTP) && uri.host
+
+        nil
+      rescue URI::InvalidURIError
+        nil
+      end
+
+      # リダイレクトを MAX_REDIRECTS 回まで追って本文を取得する。
+      # 戻り値: [body_or_nil, error_or_nil]。
+      def fetch(uri, redirects = 0)
+        return [nil, "too many redirects"] if redirects > MAX_REDIRECTS
+
+        response = request(uri)
+        case response
+        when Net::HTTPSuccess
+          [read_limited(response), nil]
+        when Net::HTTPRedirection
+          location = response["location"]
+          return [nil, "redirect without location"] if location.nil?
+
+          nxt = parse_http_uri(URI.join(uri.to_s, location).to_s)
+          return [nil, "redirect to non-http URL"] if nxt.nil?
+
+          fetch(nxt, redirects + 1)
+        else
+          [nil, "HTTP #{response.code} #{response.message}"]
+        end
+      rescue Net::OpenTimeout, Net::ReadTimeout, Timeout::Error
+        [nil, "timeout after #{@timeout_sec}s"]
+      rescue SocketError => e
+        [nil, "could not resolve or connect: #{e.message}"]
+      rescue StandardError => e
+        [nil, "fetch failed: #{e.message}"]
+      end
+
+      def request(uri)
+        http = Net::HTTP.new(uri.host, uri.port)
+        http.use_ssl = uri.is_a?(URI::HTTPS)
+        http.open_timeout = @timeout_sec
+        http.read_timeout = @timeout_sec
+
+        req = Net::HTTP::Get.new(uri)
+        req["User-Agent"] = "chocomint-fetch_url/1.0"
+        req["Accept"] = "text/html,application/xhtml+xml,text/plain;q=0.9,*/*;q=0.8"
+        http.request(req)
+      end
+
+      # 応答本文を max_bytes まで読み、妥当な UTF-8 に正規化する。
+      def read_limited(response)
+        body = response.body.to_s
+        body = body.byteslice(0, @max_bytes) if body.bytesize > @max_bytes
+        to_utf8(body)
+      end
+
+      # HTML から本文テキストを素朴に抽出する。script/style/コメントを落とし、
+      # タグを除去して実体参照を復元する。完全なパーサではないが、LLM に渡す
+      # 読み取り用テキストとしては十分。
+      def html_to_text(html)
+        text = html.dup
+        text.gsub!(%r{<!--.*?-->}m, " ")
+        text.gsub!(%r{<script\b[^>]*>.*?</script>}mi, " ")
+        text.gsub!(%r{<style\b[^>]*>.*?</style>}mi, " ")
+        text.gsub!(%r{<(br|/p|/div|/li|/h[1-6]|/tr)\b[^>]*>}i, "\n")
+        text.gsub!(/<[^>]+>/m, " ")
+        text = unescape_entities(text)
+        text.gsub(/[ \t ]+/, " ").gsub(/\n{3,}/, "\n\n").strip
+      end
+
+      ENTITIES = {
+        "&amp;" => "&", "&lt;" => "<", "&gt;" => ">",
+        "&quot;" => '"', "&#39;" => "'", "&apos;" => "'", "&nbsp;" => " "
+      }.freeze
+
+      def unescape_entities(text)
+        text = text.gsub(/&amp;|&lt;|&gt;|&quot;|&#39;|&apos;|&nbsp;/, ENTITIES)
+        # 数値文字参照 (&#123; / &#x1F600;) を復元する。
+        text = text.gsub(/&#(\d+);/) { [Regexp.last_match(1).to_i].pack("U") rescue " " }
+        text.gsub(/&#x([0-9a-fA-F]+);/) { [Regexp.last_match(1).to_i(16)].pack("U") rescue " " }
+      end
+    end
+  end
+end

data/lib/chocomint/tools/file_info.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+require "json"
+require_relative "base"
+
+module Chocomint
+  module Tools
+    # 許可ディレクトリ内のパスのメタ情報 (存在・種別・サイズ) を JSON で返す。
+    class FileInfo < Base
+      def initialize(path_guard:)
+        @path_guard = path_guard
+      end
+
+      def name = "file_info"
+
+      def description
+        "指定パスの存在・種別 (file/directory)・サイズを JSON で返す。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"])
+          info = {
+            "path" => args["path"],
+            "exists" => File.exist?(abs),
+            "type" => file_type(abs),
+            "size" => (File.size(abs) if File.exist?(abs))
+          }
+          ok(stdout: JSON.generate(info))
+        end
+      end
+
+      private
+
+      def file_type(abs)
+        return "file" if File.file?(abs)
+        return "directory" if File.directory?(abs)
+
+        nil
+      end
+    end
+  end
+end

data/lib/chocomint/tools/glob.rb

@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+require_relative "base"
+
+module Chocomint
+  module Tools
+    # 許可ディレクトリ内で glob パターンに一致するファイルを列挙する。
+    class Glob < Base
+      def initialize(path_guard:, base_dir: Dir.pwd)
+        @path_guard = path_guard
+        @base_dir = base_dir
+      end
+
+      def name = "glob"
+
+      def description
+        "glob パターン (例: **/*.rb) に一致するファイルを列挙する。base は許可ディレクトリ内であること。"
+      end
+
+      def schema
+        {
+          "type" => "object",
+          "properties" => {
+            "pattern" => { "type" => "string" },
+            "base" => { "type" => "string" }
+          },
+          "required" => %w[pattern],
+          "additionalProperties" => false
+        }
+      end
+
+      def call(args)
+        measure do
+          base_rel = args["base"] || allowed_root
+          base_abs = @path_guard.resolve(base_rel)
+          next fail(stderr: "no such directory: #{base_rel}") unless File.directory?(base_abs)
+
+          matches = Dir.glob(File.join(base_abs, args["pattern"].to_s), File::FNM_PATHNAME)
+                       .select { |p| File.file?(p) && @path_guard.allowed?(p) }
+                       .map { |p| relative(p) }
+                       .sort
+          ok(stdout: matches.join("\n"))
+        end
+      end
+
+      private
+
+      # 許可ルートの先頭 (base 未指定時のデフォルト)。
+      def allowed_root
+        @path_guard.allowed_roots.first
+      end
+
+      def relative(abs)
+        abs.delete_prefix(@base_dir.to_s + File::SEPARATOR)
+      end
+    end
+  end
+end

data/lib/chocomint/tools/grep.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+
+require_relative "base"
+
+module Chocomint
+  module Tools
+    # 許可ディレクトリ内のファイル内容を正規表現で検索する (Claude Code の Grep 相当)。
+    class Grep < Base
+      def initialize(path_guard:, max_output_bytes:, base_dir: Dir.pwd)
+        @path_guard = path_guard
+        @max_output_bytes = max_output_bytes
+        @base_dir = base_dir
+      end
+
+      def name = "grep"
+
+      def description
+        "許可ディレクトリ内のファイルを正規表現で検索し、一致行を file:line:text 形式で返す。" \
+          "glob (例: **/*.rb) で対象を絞れる。ignore_case で大文字小文字を無視。"
+      end
+
+      def schema
+        {
+          "type" => "object",
+          "properties" => {
+            "pattern" => { "type" => "string" },
+            "base" => { "type" => "string" },
+            "glob" => { "type" => "string" },
+            "ignore_case" => { "type" => "boolean" }
+          },
+          "required" => %w[pattern],
+          "additionalProperties" => false
+        }
+      end
+
+      def call(args)
+        measure do
+          base_rel = args["base"] || @path_guard.allowed_roots.first
+          base_abs = @path_guard.resolve(base_rel)
+          next fail(stderr: "no such directory: #{base_rel}") unless File.directory?(base_abs)
+
+          flags = args["ignore_case"] ? Regexp::IGNORECASE : 0
+          re = Regexp.new(args["pattern"].to_s, flags)
+          glob = args["glob"] || "**/*"
+
+          lines = search(base_abs, glob, re)
+          ok(stdout: truncate(lines.join("\n")))
+        end
+      end
+
+      private
+
+      def search(base_abs, glob, re)
+        Dir.glob(File.join(base_abs, glob), File::FNM_PATHNAME)
+           .select { |p| File.file?(p) && @path_guard.allowed?(p) }
+           .sort
+           .flat_map { |path| matches_in(path, re) }
+      end
+
+      def matches_in(path, re)
+        rel = path.delete_prefix(@base_dir.to_s + File::SEPARATOR)
+        out = []
+        File.foreach(path).with_index(1) do |line, no|
+          out << "#{rel}:#{no}:#{line.chomp}" if re.match?(line)
+        end
+        out
+      rescue ArgumentError
+        [] # バイナリ等でエンコーディング不能な行はスキップ
+      end
+
+      def truncate(text)
+        return text if text.bytesize <= @max_output_bytes
+
+        text.byteslice(0, @max_output_bytes)
+      end
+    end
+  end
+end

data/lib/chocomint/tools/list_dir.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+require_relative "base"
+
+module Chocomint
+  module Tools
+    # 許可ディレクトリ内のエントリ一覧を改行区切りで stdout に返す。
+    class ListDir < Base
+      def initialize(path_guard:)
+        @path_guard = path_guard
+      end
+
+      def name = "list_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"])
+          next fail(stderr: "no such directory: #{args['path']}") unless File.directory?(abs)
+
+          entries = Dir.children(abs).sort
+          ok(stdout: entries.join("\n"))
+        end
+      end
+    end
+  end
+end

data/lib/chocomint/tools/ls.rb

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+
+require "json"
+require_relative "base"
+
+module Chocomint
+  module Tools
+    # 許可ディレクトリ内のエントリを種別・サイズ付きで返す (Claude Code の LS 相当)。
+    # list_dir が名前のみを返すのに対し、こちらは各エントリの詳細を JSON 配列で返す。
+    class Ls < Base
+      def initialize(path_guard:)
+        @path_guard = path_guard
+      end
+
+      def name = "ls"
+
+      def description
+        "指定ディレクトリのエントリを name/type/size 付きの JSON 配列で返す。" \
+          "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 directory: #{args['path']}") unless File.directory?(abs)
+
+          entries = Dir.children(abs).sort.map do |name|
+            full = File.join(abs, name)
+            {
+              "name" => name,
+              "type" => File.directory?(full) ? "directory" : "file",
+              "size" => (File.size(full) if File.file?(full))
+            }
+          end
+          ok(stdout: JSON.generate(entries))
+        end
+      end
+    end
+  end
+end