mroberts/shush
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

fix(processor): implement universal string-aware comment parsing

Browse Source 
- Fix block comment start detection to use string-aware parsing
- Fix block comment end detection to use string-aware parsing
- Fix lineHasComment to use string-aware parsing instead of strings.Contains
- Prevents removal of comment markers inside string literals
- Protects all supported languages: Go, JS, Python, Lua, etc.
This commit is contained in:
carraes
2025-07-06 12:01:15 -03:00
parent f941ac23cb
commit 64231c18fc
1 changed files with 11 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/processor/git_processor.go
+4 -13
View File
@@ -58,7 +58,6 @@ func (p *Processor) processGitChanges() error {
return nil return nil
} }
supportedChanges := make([]git.FileChange, 0, len(changes)) supportedChanges := make([]git.FileChange, 0, len(changes))
for _, change := range changes { for _, change := range changes {
if IsSupportedFile(change.Path) { if IsSupportedFile(change.Path) {
@@ -112,7 +111,6 @@ func (p *Processor) processFileWithLineRanges(filename string, lineRanges []git.
return err return err
} }
cfg, _, err := config.Load() cfg, _, err := config.Load()
if err != nil && p.cli.Verbose { if err != nil && p.cli.Verbose {
fmt.Printf("Warning: failed to load config, using defaults: %v\n", err) fmt.Printf("Warning: failed to load config, using defaults: %v\n", err)
@@ -212,7 +210,6 @@ func (p *Processor) removeCommentsFromLine(line string, language types.Language,
comment := strings.TrimSpace(result[idx:]) comment := strings.TrimSpace(result[idx:])
if cfg.ShouldPreserveComment(comment) { if cfg.ShouldPreserveComment(comment) {
return originalLine return originalLine
} }
@@ -228,12 +225,12 @@ func (p *Processor) removeCommentsFromLine(line string, language types.Language,
endComment := language.BlockComment.End endComment := language.BlockComment.End
for { for {
startIdx := strings.Index(result, startComment) startIdx := p.findCommentIndex(result, startComment)
if startIdx == -1 { if startIdx == -1 {
break break
} }
endIdx := strings.Index(result[startIdx:], endComment) endIdx := p.findCommentIndex(result[startIdx:], endComment)
if endIdx == -1 { if endIdx == -1 {
comment := strings.TrimSpace(result[startIdx:]) comment := strings.TrimSpace(result[startIdx:])
@@ -246,8 +243,7 @@ func (p *Processor) removeCommentsFromLine(line string, language types.Language,
break break
} }
blockComment := strings.TrimSpace(result[startIdx : startIdx+endIdx+len(endComment)])
blockComment := strings.TrimSpace(result[startIdx:startIdx+endIdx+len(endComment)])
if cfg.ShouldPreserveComment(blockComment) { if cfg.ShouldPreserveComment(blockComment) {
return originalLine return originalLine
} }
@@ -287,7 +283,6 @@ func (p *Processor) showGitPreviewWithTotals(filename string, lineRanges []git.L
return err return err
} }
cfg, _, err := config.Load() cfg, _, err := config.Load()
if err != nil { if err != nil {
cfg = config.Default() cfg = config.Default()
@@ -374,7 +369,6 @@ func (p *Processor) showGitPreviewWithTotals(filename string, lineRanges []git.L
return nil return nil
} }
func (p *Processor) findCommentIndex(line, commentMarker string) int { func (p *Processor) findCommentIndex(line, commentMarker string) int {
inString := false inString := false
var stringChar byte var stringChar byte
@@ -382,7 +376,6 @@ func (p *Processor) findCommentIndex(line, commentMarker string) int {
for i := 0; i < len(line); i++ { for i := 0; i < len(line); i++ {
char := line[i] char := line[i]
if !inString && (char == '"' || char == '\'' || char == '`') { if !inString && (char == '"' || char == '\'' || char == '`') {
inString = true inString = true
stringChar = char stringChar = char
@@ -397,7 +390,6 @@ func (p *Processor) findCommentIndex(line, commentMarker string) int {
continue continue
} }
if !inString { if !inString {
if i+len(commentMarker) <= len(line) && line[i:i+len(commentMarker)] == commentMarker { if i+len(commentMarker) <= len(line) && line[i:i+len(commentMarker)] == commentMarker {
return i return i
@@ -415,9 +407,8 @@ func (p *Processor) lineHasComment(line string, language types.Language) bool {
return true return true
} }
if language.BlockComment != nil { if language.BlockComment != nil {
if strings.Contains(line, language.BlockComment.Start) || strings.Contains(line, language.BlockComment.End) { if p.findCommentIndex(line, language.BlockComment.Start) != -1 || p.findCommentIndex(line, language.BlockComment.End) != -1 {
return true return true
} }
} }