add setNodeAttributes command

packages/core/src/extensions/index.ts

@@ -10,6 +10,7 @@ export { RemoveMarks } from './removeMarks'
 export { ScrollIntoView } from './scrollIntoView'
 export { SelectAll } from './selectAll'
 export { SelectParentNode } from './selectParentNode'
+export { SetNodeAttributes } from './setNodeAttributes'
 export { SetBlockType } from './setBlockType'
 export { SetContent } from './setContent'
 export { SinkListItem } from './sinkListItem'

packages/core/src/extensions/setNodeAttributes.ts

@@ -0,0 +1,27 @@
+import { Command } from '../Editor'
+import { createExtension } from '../Extension'
+
+export const SetNodeAttributes = createExtension({
+  addCommands() {
+    return {
+      setNodeAttributes: (attributes: {}): Command => ({ tr, state }) => {
+        const { selection } = tr
+        const { from, to } = selection
+
+        state.doc.nodesBetween(from, to, (node, pos) => {
+          if (!node.type.isText) {
+            tr.setNodeMarkup(pos, undefined, attributes)
+          }
+        })
+
+        return true
+      },
+    }
+  },
+})
+
+declare module '../Editor' {
+  interface AllExtensions {
+    SetNodeAttributes: typeof SetNodeAttributes,
+  }
+}

packages/extension-text-align/index.ts

@@ -1,15 +1,23 @@
 import { createExtension } from '@tiptap/core'
 
+type TextAlignOptions = {
+  types: string[],
+}
+
 const TextAlign = createExtension({
+  defaultOptions: <TextAlignOptions>{
+    types: ['heading', 'paragraph'],
+  },
+
   addGlobalAttributes() {
     return [
       {
-        types: ['paragraph'],
+        types: this.options.types,
         attributes: {
-          align: {
+          textAlign: {
             default: 'left',
             renderHTML: attributes => ({
-              style: `text-align: ${attributes.align}`,
+              style: `text-align: ${attributes.textAlign}`,
             }),
           },
         },