mirror of
https://github.com/lukaszraczylo/filepuff-mcp.git
synced 2026-10-12 04:54:27 +00:00
V2/token optimization (#11)
* v2.0: token-optimization overhaul Additive (backward-compatible flags): - file_read: skeleton mode, strip (imports/license/block_comments), compact_line_numbers, 8-char etag with prefix-match compat - ast_query: format=verbose|compact|location, pagination cursor - file_search: cluster mode, pagination cursor - lsp_query (references): compact output Breaking (v2): - Preambles removed; opt-in verbose=true restores - edit_apply: response=count|diff|none, default count - ping tool removed - symbol_at/find_definition/find_references merged into lsp_query - Tool descriptions trimmed -83%, help moved to filepuff://help/<tool> - Batch file_read dedups by etag Protocol: - ResourceLink returned for file_read >64 KiB (force_inline override) - OnAfterInitialize hook reads capabilities.experimental.filepuff for session defaults (default_format, default_max_results, default_cluster, compact_refs, line_numbers, resource_link_threshold) * fix: drop --max-total-count from ripgrep args The flag does not exist in stable ripgrep (confirmed up to 15.1.0 -- "unrecognized flag --max-total-count, similar flags that are available: --max-count"). Every file_search call failed on hosts with stock rg. --max-count is per-file, not a drop-in replacement, so rely on the in-process truncation in parseOutput that was already the documented safety net.
This commit is contained in:
@@ -0,0 +1,299 @@
|
||||
package parser
|
||||
|
||||
import (
|
||||
"strings"
|
||||
|
||||
"github.com/lukaszraczylo/mcp-filepuff/pkg/protocol"
|
||||
)
|
||||
|
||||
// StripFlag names the categories of content to remove.
|
||||
type StripFlag string
|
||||
|
||||
const (
|
||||
StripImports StripFlag = "imports"
|
||||
StripLicense StripFlag = "license"
|
||||
StripBlockComments StripFlag = "block_comments"
|
||||
)
|
||||
|
||||
// StripResult holds the stripped content and which flags actually removed content.
|
||||
type StripResult struct {
|
||||
Content string
|
||||
Stripped []StripFlag
|
||||
}
|
||||
|
||||
// StripContent applies requested strip operations to content, in order:
|
||||
// license → imports → block_comments.
|
||||
// lang is used to pick language-specific heuristics.
|
||||
func StripContent(content string, flags []StripFlag, lang protocol.Language) StripResult {
|
||||
flagSet := make(map[StripFlag]bool, len(flags))
|
||||
for _, f := range flags {
|
||||
flagSet[f] = true
|
||||
}
|
||||
|
||||
var stripped []StripFlag
|
||||
|
||||
if flagSet[StripLicense] {
|
||||
next, removed := stripLicense(content)
|
||||
if removed {
|
||||
content = next
|
||||
stripped = append(stripped, StripLicense)
|
||||
}
|
||||
}
|
||||
|
||||
if flagSet[StripImports] {
|
||||
next, removed := stripImports(content, lang)
|
||||
if removed {
|
||||
content = next
|
||||
stripped = append(stripped, StripImports)
|
||||
}
|
||||
}
|
||||
|
||||
if flagSet[StripBlockComments] {
|
||||
next, removed := stripBlockComments(content, lang)
|
||||
if removed {
|
||||
content = next
|
||||
stripped = append(stripped, StripBlockComments)
|
||||
}
|
||||
}
|
||||
|
||||
return StripResult{Content: content, Stripped: stripped}
|
||||
}
|
||||
|
||||
// stripLicense removes a leading block comment that looks like a license header.
|
||||
// A comment qualifies if it contains "copyright", "license", or "spdx-license-identifier" (case-insensitive).
|
||||
func stripLicense(content string) (string, bool) {
|
||||
trimmed := strings.TrimLeft(content, " \t\n\r")
|
||||
|
||||
// C-style block comment at top
|
||||
if strings.HasPrefix(trimmed, "/*") {
|
||||
end := strings.Index(trimmed, "*/")
|
||||
if end >= 0 {
|
||||
candidate := trimmed[:end+2]
|
||||
lower := strings.ToLower(candidate)
|
||||
if strings.Contains(lower, "copyright") ||
|
||||
strings.Contains(lower, "license") ||
|
||||
strings.Contains(lower, "spdx-license-identifier") {
|
||||
rest := trimmed[end+2:]
|
||||
// Consume trailing newline(s)
|
||||
rest = strings.TrimLeft(rest, "\r\n")
|
||||
return rest, true
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Python/hash-style leading comment block
|
||||
if strings.HasPrefix(trimmed, "#") {
|
||||
lines := strings.Split(trimmed, "\n")
|
||||
var commentLines []string
|
||||
var rest []string
|
||||
inComment := true
|
||||
for i, l := range lines {
|
||||
if inComment && (strings.HasPrefix(l, "#") || strings.TrimSpace(l) == "") {
|
||||
commentLines = append(commentLines, l)
|
||||
} else {
|
||||
rest = lines[i:]
|
||||
break
|
||||
}
|
||||
}
|
||||
block := strings.Join(commentLines, "\n")
|
||||
lower := strings.ToLower(block)
|
||||
if strings.Contains(lower, "copyright") ||
|
||||
strings.Contains(lower, "license") ||
|
||||
strings.Contains(lower, "spdx-license-identifier") {
|
||||
return strings.Join(rest, "\n"), true
|
||||
}
|
||||
}
|
||||
|
||||
return content, false
|
||||
}
|
||||
|
||||
// stripImports removes top-of-file import blocks, language-specific.
|
||||
func stripImports(content string, lang protocol.Language) (string, bool) {
|
||||
switch lang {
|
||||
case protocol.LangGo:
|
||||
return stripGoImports(content)
|
||||
case protocol.LangTypeScript, protocol.LangJavaScript:
|
||||
return stripTSImports(content)
|
||||
case protocol.LangPython:
|
||||
return stripPythonImports(content)
|
||||
case protocol.LangRust:
|
||||
return stripRustImports(content)
|
||||
default:
|
||||
return content, false
|
||||
}
|
||||
}
|
||||
|
||||
// stripGoImports removes Go import(...) or single import "..." declarations.
|
||||
func stripGoImports(content string) (string, bool) {
|
||||
lines := strings.Split(content, "\n")
|
||||
var out []string
|
||||
removed := false
|
||||
i := 0
|
||||
for i < len(lines) {
|
||||
trimLine := strings.TrimSpace(lines[i])
|
||||
if strings.HasPrefix(trimLine, "import (") || trimLine == "import (" {
|
||||
// multi-line import block
|
||||
removed = true
|
||||
i++ // skip "import ("
|
||||
for i < len(lines) {
|
||||
if strings.TrimSpace(lines[i]) == ")" {
|
||||
i++ // skip closing ")"
|
||||
break
|
||||
}
|
||||
i++
|
||||
}
|
||||
// skip one blank line after
|
||||
if i < len(lines) && strings.TrimSpace(lines[i]) == "" {
|
||||
i++
|
||||
}
|
||||
continue
|
||||
}
|
||||
if strings.HasPrefix(trimLine, `import "`) || strings.HasPrefix(trimLine, "import `") {
|
||||
removed = true
|
||||
i++
|
||||
continue
|
||||
}
|
||||
out = append(out, lines[i])
|
||||
i++
|
||||
}
|
||||
if !removed {
|
||||
return content, false
|
||||
}
|
||||
return strings.Join(out, "\n"), true
|
||||
}
|
||||
|
||||
// stripTSImports removes TypeScript/JavaScript "import ... from ..." and "require(...)" lines.
|
||||
func stripTSImports(content string) (string, bool) {
|
||||
lines := strings.Split(content, "\n")
|
||||
var out []string
|
||||
removed := false
|
||||
for _, l := range lines {
|
||||
trimLine := strings.TrimSpace(l)
|
||||
if strings.HasPrefix(trimLine, "import ") || strings.HasPrefix(trimLine, "const {") && strings.Contains(trimLine, "require(") {
|
||||
removed = true
|
||||
continue
|
||||
}
|
||||
out = append(out, l)
|
||||
}
|
||||
if !removed {
|
||||
return content, false
|
||||
}
|
||||
return strings.Join(out, "\n"), true
|
||||
}
|
||||
|
||||
// stripPythonImports removes Python "import ..." and "from ... import ..." lines.
|
||||
func stripPythonImports(content string) (string, bool) {
|
||||
lines := strings.Split(content, "\n")
|
||||
var out []string
|
||||
removed := false
|
||||
for _, l := range lines {
|
||||
trimLine := strings.TrimSpace(l)
|
||||
if strings.HasPrefix(trimLine, "import ") || strings.HasPrefix(trimLine, "from ") {
|
||||
removed = true
|
||||
continue
|
||||
}
|
||||
out = append(out, l)
|
||||
}
|
||||
if !removed {
|
||||
return content, false
|
||||
}
|
||||
return strings.Join(out, "\n"), true
|
||||
}
|
||||
|
||||
// stripRustImports removes Rust "use ..." declarations.
|
||||
func stripRustImports(content string) (string, bool) {
|
||||
lines := strings.Split(content, "\n")
|
||||
var out []string
|
||||
removed := false
|
||||
inMulti := false
|
||||
for _, l := range lines {
|
||||
trimLine := strings.TrimSpace(l)
|
||||
if inMulti {
|
||||
// look for semicolon terminating multi-line use
|
||||
if strings.Contains(trimLine, ";") {
|
||||
inMulti = false
|
||||
}
|
||||
removed = true
|
||||
continue
|
||||
}
|
||||
if strings.HasPrefix(trimLine, "use ") {
|
||||
removed = true
|
||||
if !strings.HasSuffix(trimLine, ";") {
|
||||
inMulti = true
|
||||
}
|
||||
continue
|
||||
}
|
||||
out = append(out, l)
|
||||
}
|
||||
if !removed {
|
||||
return content, false
|
||||
}
|
||||
return strings.Join(out, "\n"), true
|
||||
}
|
||||
|
||||
// stripBlockComments removes /* ... */ block comments (Go/TS/C/Rust)
|
||||
// and Python triple-quoted docstrings.
|
||||
func stripBlockComments(content string, lang protocol.Language) (string, bool) {
|
||||
if lang == protocol.LangPython {
|
||||
return stripPythonDocstrings(content)
|
||||
}
|
||||
return stripCStyleBlockComments(content)
|
||||
}
|
||||
|
||||
// stripCStyleBlockComments removes /* ... */ from content.
|
||||
func stripCStyleBlockComments(content string) (string, bool) {
|
||||
removed := false
|
||||
var sb strings.Builder
|
||||
i := 0
|
||||
for i < len(content) {
|
||||
if i+1 < len(content) && content[i] == '/' && content[i+1] == '*' {
|
||||
// find closing */
|
||||
end := strings.Index(content[i+2:], "*/")
|
||||
if end >= 0 {
|
||||
removed = true
|
||||
// advance past */
|
||||
i = i + 2 + end + 2
|
||||
// consume trailing newline
|
||||
if i < len(content) && content[i] == '\n' {
|
||||
i++
|
||||
}
|
||||
continue
|
||||
}
|
||||
}
|
||||
sb.WriteByte(content[i])
|
||||
i++
|
||||
}
|
||||
if !removed {
|
||||
return content, false
|
||||
}
|
||||
return sb.String(), true
|
||||
}
|
||||
|
||||
// stripPythonDocstrings removes triple-quoted strings (""" and ”').
|
||||
func stripPythonDocstrings(content string) (string, bool) {
|
||||
removed := false
|
||||
var sb strings.Builder
|
||||
i := 0
|
||||
for i < len(content) {
|
||||
if i+2 < len(content) {
|
||||
triple := content[i : i+3]
|
||||
if triple == `"""` || triple == `'''` {
|
||||
end := strings.Index(content[i+3:], triple)
|
||||
if end >= 0 {
|
||||
removed = true
|
||||
i = i + 3 + end + 3
|
||||
if i < len(content) && content[i] == '\n' {
|
||||
i++
|
||||
}
|
||||
continue
|
||||
}
|
||||
}
|
||||
}
|
||||
sb.WriteByte(content[i])
|
||||
i++
|
||||
}
|
||||
if !removed {
|
||||
return content, false
|
||||
}
|
||||
return sb.String(), true
|
||||
}
|
||||
Reference in New Issue
Block a user