From d27b86d29614e1aead825f4a7e50a1d89f970665 Mon Sep 17 00:00:00 2001
From: Gubarz <1037896+Gubarz@users.noreply.github.com>
Date: Wed, 13 May 2026 03:19:41 -0600
Subject: [PATCH] fix single line cheat comment

this resolves a bug where a single line cheat comment was not being recognised
---
 internal/parser/lex.go         |  5 ++++-
 internal/parser/parser_test.go | 32 ++++++++++++++++++++++++++++++++
 2 files changed, 36 insertions(+), 1 deletion(-)

diff --git a/internal/parser/lex.go b/internal/parser/lex.go
index 6a74012..aeae0d3 100644
--- a/internal/parser/lex.go
+++ b/internal/parser/lex.go
@@ -141,7 +141,7 @@ func parseCodeBlockStart(line []byte) (lang, desc string, ok bool) {
 
 // parseCheatSingleLine parses <!-- cheat ... --> and returns the content.
 func parseCheatSingleLine(line []byte) (string, bool) {
-	if len(line) < 15 {
+	if len(line) < len("<!--cheat-->") {
 		return "", false
 	}
 	if !bytes.HasPrefix(line, []byte("<!--")) {
@@ -158,6 +158,9 @@ func parseCheatSingleLine(line []byte) (string, bool) {
 	if !bytes.EqualFold(inner[:5], []byte("cheat")) {
 		return "", false
 	}
+	if len(inner) > 5 && inner[5] != ' ' && inner[5] != '\t' {
+		return "", false
+	}
 
 	return string(bytes.TrimSpace(inner[5:])), true
 }
diff --git a/internal/parser/parser_test.go b/internal/parser/parser_test.go
index f985897..81f2085 100644
--- a/internal/parser/parser_test.go
+++ b/internal/parser/parser_test.go
@@ -157,6 +157,38 @@ func TestParseCheatDSL_Comments(t *testing.T) {
 	}
 }
 
+func TestParseSingleLineEmptyCheatComment(t *testing.T) {
+	dir := t.TempDir()
+	path := filepath.Join(dir, "single.md")
+	content := "# Cheats\n\n## Empty metadata\n\n```sh\necho hi\n```\n<!-- cheat -->\n"
+	if err := os.WriteFile(path, []byte(content), 0o644); err != nil {
+		t.Fatal(err)
+	}
+
+	p := NewParser()
+	index, err := p.ParseSingleFile(path)
+	if err != nil {
+		t.Fatalf("ParseSingleFile() error: %v", err)
+	}
+
+	if len(index.Cheats) != 1 {
+		t.Fatalf("ParseSingleFile() cheats = %d, want 1", len(index.Cheats))
+	}
+	cheat := index.Cheats[0]
+	if !cheat.HasCheatBlock {
+		t.Fatal("single-line <!-- cheat --> did not mark cheat as having a cheat block")
+	}
+	if cheat.Command != "echo hi" {
+		t.Fatalf("command = %q, want %q", cheat.Command, "echo hi")
+	}
+}
+
+func TestParseCheatSingleLineRequiresKeywordBoundary(t *testing.T) {
+	if _, ok := parseCheatSingleLine([]byte("<!-- cheating -->")); ok {
+		t.Fatal("parseCheatSingleLine accepted non-cheat keyword")
+	}
+}
+
 func TestNewCheatIndex(t *testing.T) {
 	idx := NewCheatIndex()
 	if idx == nil {