diff --git a/blog.go b/blog.go index 4673c18..8cc7277 100644 --- a/blog.go +++ b/blog.go @@ -67,12 +67,10 @@ func extractTitleAndContents(raw []byte) (title string, contents []byte, err err for i, c := range raw { // We are assuming that each file has one title as a H1 header if c == '\n' { - original := string(raw[:i]) - clean := strings.Replace(string(raw[:i]), "# ", "", 1) - if clean == original { - return "", nil, fmt.Errorf("could not find title") + if raw[0] != '#' { + return "", nil, fmt.Errorf("could not find '#', file seems to be missing a H1 header") } - title = clean + title = string(bytes.TrimSpace(raw[1:i])) contents = bytes.TrimSpace(raw[i:]) break }