From 1bc75751b3b3e7f758c8af186b34101f51113026 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Philipp=20Ku=CC=88hn?= <kontakt@philipp-kuehn.com>
Date: Mon, 30 Mar 2020 14:42:10 +0200
Subject: [PATCH] check active nodes

---
 packages/core/src/Editor.ts          | 53 ++++++++++++++++++++++------
 src/demos/HandleExtensions/index.vue |  2 +-
 src/layouts/Default.vue              |  5 +++
 3 files changed, 49 insertions(+), 11 deletions(-)

diff --git a/packages/core/src/Editor.ts b/packages/core/src/Editor.ts
index 17066e20..67d25b35 100644
--- a/packages/core/src/Editor.ts
+++ b/packages/core/src/Editor.ts
@@ -13,6 +13,8 @@ import elementFromString from './utils/elementFromString'
 import injectCSS from './utils/injectCSS'
 import getAllMethodNames from './utils/getAllMethodNames'
 import nodeIsActive from './utils/nodeIsActive'
+import markIsActive from './utils/markIsActive'
+import getMarkAttrs from './utils/getMarkAttrs'
 import ExtensionManager from './ExtensionManager'
 import Extension from './Extension'
 import Node from './Node'
@@ -40,6 +42,9 @@ export class Editor extends EventEmitter {
     extensions: [],
   }
   commands: { [key: string]: any } = {}
+  activeNodes: { [key: string]: any } = {}
+  activeMarks: { [key: string]: any } = {}
+  activeMarkAttrs: { [key: string]: any } = {}
   
   private lastCommand = Promise.resolve()
   
@@ -51,6 +56,7 @@ export class Editor extends EventEmitter {
     this.createExtensionManager()
     this.createSchema()
     this.createView()
+    this.setActiveNodesAndMarks()
     this.registerCommand('focus', require('./commands/focus').default)
     this.registerCommand('insertText', require('./commands/insertText').default)
     this.registerCommand('insertHTML', require('./commands/insertHTML').default)
@@ -166,7 +172,6 @@ export class Editor extends EventEmitter {
     this.view.updateState(state)
     this.storeSelection()
     this.setActiveNodesAndMarks()
-
     this.emit('transaction', { transaction })
     
     if (!transaction.docChanged || transaction.getMeta('preventUpdate')) {
@@ -176,16 +181,44 @@ export class Editor extends EventEmitter {
     this.emit('update', { transaction })
   }
 
-  public setActiveNodesAndMarks() {
-    // TODO
-    // const activeNodes = Object
-    // .entries(this.schema.nodes)
-    // .reduce((nodes, [name, node]) => ({
-    //   ...nodes,
-    //   [name]: (attrs = {}) => nodeIsActive(this.state, node, attrs),
-    // }), {})
+  setActiveNodesAndMarks() {
+    this.activeMarks = Object
+      .entries(this.schema.marks)
+      .reduce((marks, [name, mark]) => ({
+        ...marks,
+        // [name]: (attrs = {}) => markIsActive(this.state, mark, attrs),
+        [name]: () => markIsActive(this.state, mark),
+      }), {})
 
-    // console.log({activeNodes})
+    this.activeMarkAttrs = Object
+      .entries(this.schema.marks)
+      .reduce((marks, [name, mark]) => ({
+        ...marks,
+        [name]: getMarkAttrs(this.state, mark),
+      }), {})
+
+    this.activeNodes = Object
+      .entries(this.schema.nodes)
+      .reduce((nodes, [name, node]) => ({
+        ...nodes,
+        [name]: (attrs = {}) => nodeIsActive(this.state, node, attrs),
+      }), {})
+  }
+
+  getMarkAttrs(name: string) {
+    return this.activeMarkAttrs[name]
+  }
+
+  get isActive() {
+    return Object
+      .entries({
+        ...this.activeMarks,
+        ...this.activeNodes,
+      })
+      .reduce((types, [name, value]) => ({
+        ...types,
+        [name]: (attrs = {}) => value(attrs),
+      }), {})
   }
 
   // public setParentComponent(component = null) {
diff --git a/src/demos/HandleExtensions/index.vue b/src/demos/HandleExtensions/index.vue
index 14bd0c23..9b7ddd80 100644
--- a/src/demos/HandleExtensions/index.vue
+++ b/src/demos/HandleExtensions/index.vue
@@ -7,7 +7,7 @@
       <button @click="editor.redo().focus()">
         redo
       </button>
-      <button @click="editor.bold().focus()">
+      <button @click="editor.bold().focus()" :class="{ 'is-active': editor.isActive.bold() }">
         bold
       </button>
     </div>
diff --git a/src/layouts/Default.vue b/src/layouts/Default.vue
index b3138402..96fe0559 100644
--- a/src/layouts/Default.vue
+++ b/src/layouts/Default.vue
@@ -49,6 +49,11 @@ body {
   outline: none;
 }
 
+.is-active {
+  background: black;
+  color: white;
+}
+
 // :not(pre) > code[class*="language-"], pre[class*="language-"] {
 //   background-color: $colorBlack;
 // }