add optional spaces in regexps

kovetskiy · Aug 8, 2019 · 4cfda3a · 4cfda3a
1 parent de2c876
commit 4cfda3a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pkg/mark/includes/templates.go b/pkg/mark/includes/templates.go
@@ -16,7 +16,9 @@ import (
 )
 
 var (
-	reIncludeDirective = regexp.MustCompile(`(?s)<!-- Include: (\S+)(.*?)-->`)
+	reIncludeDirective = regexp.MustCompile(
+		`(?s)<!--\s*Include:\s*(\S+)(.*?)-->`,
+	)
 )
 
 func LoadTemplate(

diff --git a/pkg/mark/macro/macro.go b/pkg/mark/macro/macro.go
@@ -14,7 +14,7 @@ import (
 )
 
 var reMacroDirective = regexp.MustCompile(
-	`(?s)<!-- Macro: ([^\n]+)\n\s*Template: (\S+)\n(.*?)-->`,
+	`(?s)<!--\s*Macro:\s*([^\n]+)\n\s*Template:\s*(\S+)\n(.*?)-->`,
 )
 
 type Macro struct {