From 127d9f29ac7478d3191d8f2ef9506ecef56b2390 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Philipp=20Ku=CC=88hn?= <kontakt@philipp-kuehn.com>
Date: Wed, 2 Dec 2020 18:13:16 +0100
Subject: [PATCH] fix a bug for task item

---
 packages/extension-task-item/src/index.ts | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/packages/extension-task-item/src/index.ts b/packages/extension-task-item/src/index.ts
index 562e6132..3b43e125 100644
--- a/packages/extension-task-item/src/index.ts
+++ b/packages/extension-task-item/src/index.ts
@@ -72,7 +72,12 @@ const TaskItem = Node.create({
   },
 
   addNodeView() {
-    return ({ HTMLAttributes, getPos, editor }) => {
+    return ({
+      node,
+      HTMLAttributes,
+      getPos,
+      editor,
+    }) => {
       const { view } = editor
       const listItem = document.createElement('li')
       const checkbox = document.createElement('input')
@@ -91,7 +96,7 @@ const TaskItem = Node.create({
         }
       })
 
-      if (HTMLAttributes['data-checked'] === true) {
+      if (node.attrs.checked) {
         checkbox.setAttribute('checked', 'checked')
       }
 
@@ -104,11 +109,17 @@ const TaskItem = Node.create({
       return {
         dom: listItem,
         contentDOM: content,
-        update: node => {
-          if (node.type !== this.type) {
+        update: updatedNode => {
+          if (updatedNode.type !== this.type) {
             return false
           }
 
+          if (updatedNode.attrs.checked) {
+            checkbox.setAttribute('checked', 'checked')
+          } else {
+            checkbox.removeAttribute('checked')
+          }
+
           return true
         },
       }