From e1f61bb156f2e63acc8cec567bf4be87b8b97819 Mon Sep 17 00:00:00 2001
From: Ashton Meuser <ashtonmeuser@gmail.com>
Date: Thu, 4 Mar 2021 09:37:44 -0800
Subject: [PATCH] Add task class to all task list items

---
 extension/_test/tasklist.txt | 12 ++++++------
 extension/tasklist.go        |  4 +++-
 2 files changed, 9 insertions(+), 7 deletions(-)

diff --git a/extension/_test/tasklist.txt b/extension/_test/tasklist.txt
index dad65c6..dfe65f1 100644
--- a/extension/_test/tasklist.txt
+++ b/extension/_test/tasklist.txt
@@ -4,8 +4,8 @@
 - [x] bar
 //- - - - - - - - -//
 <ul>
-<li><input disabled="" type="checkbox"> foo</li>
-<li><input checked="" disabled="" type="checkbox"> bar</li>
+<li class="task"><input disabled="" type="checkbox"> foo</li>
+<li class="task"><input checked="" disabled="" type="checkbox"> bar</li>
 </ul>
 //= = = = = = = = = = = = = = = = = = = = = = = =//
 
@@ -19,12 +19,12 @@
 - [ ] bim
 //- - - - - - - - -//
 <ul>
-<li><input checked="" disabled="" type="checkbox"> foo
+<li class="task"><input checked="" disabled="" type="checkbox"> foo
 <ul>
-<li><input disabled="" type="checkbox"> bar</li>
-<li><input checked="" disabled="" type="checkbox"> baz</li>
+<li class="task"><input disabled="" type="checkbox"> bar</li>
+<li class="task"><input checked="" disabled="" type="checkbox"> baz</li>
 </ul>
 </li>
-<li><input disabled="" type="checkbox"> bim</li>
+<li class="task"><input disabled="" type="checkbox"> bim</li>
 </ul>
 //= = = = = = = = = = = = = = = = = = = = = = = =//
diff --git a/extension/tasklist.go b/extension/tasklist.go
index 1f3e52c..358a9eb 100644
--- a/extension/tasklist.go
+++ b/extension/tasklist.go
@@ -40,7 +40,8 @@ func (s *taskCheckBoxParser) Parse(parent gast.Node, block text.Reader, pc parse
 		return nil
 	}
 
-	if _, ok := parent.Parent().(*gast.ListItem); !ok {
+	listItem, ok := parent.Parent().(*gast.ListItem)
+	if !ok {
 		return nil
 	}
 	line, _ := block.PeekLine()
@@ -48,6 +49,7 @@ func (s *taskCheckBoxParser) Parse(parent gast.Node, block text.Reader, pc parse
 	if m == nil {
 		return nil
 	}
+	listItem.SetAttributeString("class", []byte("task"))
 	value := line[m[2]:m[3]][0]
 	block.Advance(m[1])
 	checked := value == 'x' || value == 'X'