fix: refactor for less redundant with same logic for code blocks

Author: benjaminrobinet

Diff for: src/Block.tsx

@@ -16,30 +16,23 @@ const voidTypes = ['image'];
  */
 const augmentProps = (content: Node) => {
   const { children: childrenNodes, type, ...props } = content;
-  const getPlainText = (children: typeof childrenNodes): string => {
-    return children.reduce((currentPlainText, node) => {
-      if (node.type === 'text') {
-        return currentPlainText.concat(node.text);
-      }
 
-      if (node.type === 'link') {
-        return currentPlainText.concat(getPlainText(node.children));
-      }
-
-      return currentPlainText;
-    }, '');
-  };
-
-  if (type === 'code') {
+  if (type === 'code' || type === 'heading') {
     // Builds a plain text string from an array of nodes, regardless of links or modifiers
+    const getPlainText = (children: typeof childrenNodes): string => {
+      return children.reduce((currentPlainText, node) => {
+        if (node.type === 'text') {
+          return currentPlainText.concat(node.text);
+        }
 
-    return {
-      ...props,
-      plainText: getPlainText(content.children),
+        if (node.type === 'link') {
+          return currentPlainText.concat(getPlainText(node.children));
+        }
+
+        return currentPlainText;
+      }, '');
     };
-  }
 
-  if (type === 'heading') {
     return {
       ...props,
       plainText: getPlainText(content.children),