Add helper for text color extention

remirror · Jul 21, 2022 · 4d04910 · 4d04910
1 parent bb33bd8
commit 4d04910
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/packages/remirror__extension-text-color/src/text-color-extension.ts b/packages/remirror__extension-text-color/src/text-color-extension.ts
@@ -115,6 +115,18 @@ export class TextColorExtension extends MarkExtension<TextColorOptions> {
     return this.store.commands.removeMark.original({ type: this.type, ...options, expand: true });
   }
 
+  /**
+   * Get the color for the selection
+   */
+  @helper()
+  getTextColorForSelection(selection?: Selection): Helper<string> {
+    if (!selection) return this.options.defaultColor;
+
+    const state = this.store.getState();
+    const textMarkInfo = getMarkRange(selection.$from, this.type);
+    if (!textMarkInfo) return  this.options.defaultColor;
+    return textMarkInfo.mark.attrs.color;
+  }
   /**
    * Get the color from the provided string. The string can be a computed
    * property as well.