Fix (grimoire): check for opening and closing brackets in source

yunginnanet · Jun 26, 2024 · e0b4c0a · e0b4c0a
1 parent 5724075
commit e0b4c0a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/internal/http/router.go b/internal/http/router.go
@@ -134,6 +134,11 @@ func SetupHeffalump(config *config.Parameters) error {
 			return fmt.Errorf("%w: grimoire file '%s' appears to be empty", io.EOF, config.Bespoke.Grimoire)
 		}
 
+		if !strings.Contains(src, "<") || !strings.Contains(src, ">") {
+			return fmt.Errorf("%w: grimoire file '%s' does not appear to be a valid source text, missing brackets",
+				io.ErrUnexpectedEOF, config.Bespoke.Grimoire)
+		}
+
 		markovMap := heffalump.MakeMarkovMap(strings.NewReader(src))
 		hellpotHeffalump = heffalump.NewHeffalump(markovMap, heffalump.DefaultBuffSize)
 	default: