From 5010abb01b51a0c07c3c2f26999c8f126011ba43 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Philipp=20Ku=CC=88hn?= <kontakt@philipp-kuehn.com>
Date: Wed, 30 Sep 2020 19:19:51 +0200
Subject: [PATCH] refactoring

---
 packages/extension-code-block/index.ts | 14 +++++---------
 1 file changed, 5 insertions(+), 9 deletions(-)

diff --git a/packages/extension-code-block/index.ts b/packages/extension-code-block/index.ts
index 4ee60c57e..4119e7ec7 100644
--- a/packages/extension-code-block/index.ts
+++ b/packages/extension-code-block/index.ts
@@ -36,14 +36,8 @@ export default new Node<CodeBlockOptions>()
       {
         tag: 'pre',
         preserveWhitespace: 'full',
-        getAttrs(dom) {
-          const code = (dom as HTMLElement).firstChild
-
-          if (!code) {
-            return null
-          }
-
-          const classAttribute = ((code as HTMLElement).getAttribute('class') as string)
+        getAttrs(node) {
+          const classAttribute = (node as Element).firstElementChild?.getAttribute('class')
 
           if (!classAttribute) {
             return null
@@ -55,7 +49,9 @@ export default new Node<CodeBlockOptions>()
         },
       },
     ],
-    toDOM: node => ['pre', ['code', { class: node.attrs.language && options.languageClassPrefix + node.attrs.language }, 0]],
+    toDOM: node => ['pre', ['code', {
+      class: node.attrs.language && options.languageClassPrefix + node.attrs.language,
+    }, 0]],
   }))
   .commands(({ name }) => ({
     codeBlock: attrs => ({ commands }) => {