Implement named capture groups

This commit is contained in:
2025-03-26 12:27:40 +01:00
parent a8c2257f20
commit c5fb20e96a

View File

@@ -145,31 +145,42 @@ func (p *RegexProcessor) ProcessContent(content string, pattern string, luaExpr
continue
}
namedCaptures := make(map[string]string)
groupNames := compiledPattern.SubexpNames()
for i, name := range groupNames {
if i == 0 {
continue
}
if i < len(match) {
namedCaptures[name] = string(match[i])
}
}
log.Printf("Named captures: %v", namedCaptures)
captures := make([]string, 0, len(groups)/2)
for j := 0; j < len(groups); j += 2 {
captures = append(captures, content[groups[j]:groups[j+1]])
}
log.Printf("Captured groups: %v", captures)
namedCaptures := make(map[string]string)
namedCaptureGroups := make(map[string][2]int)
groupNames := compiledPattern.SubexpNames()[1:]
for i, name := range groupNames {
if name == "" {
continue
}
namedCaptures[name] = captures[i]
namedCaptureGroups[name] = [2]int{groups[i*2], groups[i*2+1]}
}
log.Printf("Named captures: %v", namedCaptures)
if err := p.ToLua(L, captures); err != nil {
log.Printf("Error setting Lua variables: %v", err)
continue
}
log.Println("Lua variables set successfully")
for name, capture := range namedCaptures {
if name == "" {
continue
}
if val, err := strconv.ParseFloat(capture, 64); err == nil {
L.SetGlobal(name, lua.LNumber(val))
} else {
L.SetGlobal(name, lua.LString(capture))
}
}
if err := L.DoString(luaExpr); err != nil {
log.Printf("Error executing Lua code %s for group %s: %v", luaExpr, captures, err)
continue
@@ -202,6 +213,17 @@ func (p *RegexProcessor) ProcessContent(content string, pattern string, luaExpr
replacement = replacement[:groupStart] + newVal + replacement[groupEnd:]
}
for i := len(namedCaptures) - 1; i >= 0; i-- {
name := groupNames[i]
if name == "" {
continue
}
groupStart := namedCaptureGroups[name][0] - matchIndices[0]
groupEnd := namedCaptureGroups[name][1] - matchIndices[0]
luaValue := L.GetGlobal(name).String()
replacement = replacement[:groupStart] + luaValue + replacement[groupEnd:]
}
modificationCount++
result = result[:matchIndices[0]] + replacement + result[matchIndices[1]:]
log.Printf("Modification count updated: %d", modificationCount)