allow regex replace

telekom · Sep 20, 2024 · 73fdf33 · 73fdf33
1 parent c8f1de6
commit 73fdf33
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/pkg/config/config.go b/pkg/config/config.go
@@ -28,8 +28,9 @@ type VRFConfig struct {
 }
 
 type Replacement struct {
-	Old string `yaml:"old"`
-	New string `yaml:"new"`
+	Old   string `yaml:"old"`
+	New   string `yaml:"new"`
+	Regex bool   `yaml:"regex"`
 }
 
 func LoadConfig() (*Config, error) {

diff --git a/pkg/frr/configure.go b/pkg/frr/configure.go
@@ -178,7 +178,12 @@ func fixRouteTargetReload(frrConfig []byte) []byte {
 // fixCfgReplacements replaces placeholders in the configuration with the actual values.
 func fixCfgReplacements(frrConfig []byte, replacements []config.Replacement) []byte {
 	for _, replacement := range replacements {
-		frrConfig = bytes.ReplaceAll(frrConfig, []byte(replacement.Old), []byte(replacement.New))
+		if !replacement.Regex {
+			frrConfig = bytes.ReplaceAll(frrConfig, []byte(replacement.Old), []byte(replacement.New))
+		} else {
+			re := regexp.MustCompile(replacement.Old)
+			frrConfig = re.ReplaceAll(frrConfig, []byte(replacement.New))
+		}
 	}
 	return frrConfig
 }