specter/kiln
1
0
Fork 0
mirror of https://git.sr.ht/~adnano/kiln synced 2024-10-30 01:13:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

frontmatter: Fix extraction algorithm

Browse source
This commit is contained in:
adnano 2021-04-24 13:24:31 -04:00
parent 2d543b9c1e
commit 81aac7692d
2 changed files with 11 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
frontmatter.go
View file

@ -11,13 +11,14 @@ func extractFrontmatter(b []byte) (frontmatter, content []byte) {
if !bytes.HasPrefix(b, frontmatterOpen) {
return nil, b
}
fm := b[len(frontmatterOpen):]
if len(fm) != 0 && fm[0] != '\n' {
if len(b) > len(frontmatterOpen) && b[len(frontmatterOpen)] != '\n' {
return nil, b
}
i := bytes.Index(fm, frontmatterClose)
b = b[len(frontmatterOpen):]
i := bytes.Index(b, frontmatterClose)
if i == -1 {
i = len(fm)
return b, nil
}
return fm[:i], b[len(frontmatterOpen)+len(fm):]
return b[:i], b[i+len(frontmatterClose):]
}

5
frontmatter_test.go
View file

@ -20,6 +20,11 @@ func TestExtractFrontmatter(t *testing.T) {
Frontmatter: "\na: b\nc: d",
Content: "",
},
{
Raw: "---\na: b\nc: d\n---\nHello, world!",
Frontmatter: "\na: b\nc: d",
Content: "\nHello, world!",
},
{
Raw: "# Hello, world!",
Frontmatter: "",