add rendering of marks from schema

examples/rich-text/index.js

@@ -23,30 +23,23 @@ const schema = {
     'heading-two': props => <h2 {...props.attributes}>{props.children}</h2>,
     'list-item': props => <li {...props.attributes}>{props.children}</li>,
     'numbered-list': props => <ol {...props.attributes}>{props.children}</ol>,
-  }
-}
-
-/**
- * Define a set of mark renderers.
- *
- * @type {Object}
- */
-
-const MARKS = {
-  bold: {
-    fontWeight: 'bold'
   },
-  code: {
-    fontFamily: 'monospace',
-    backgroundColor: '#eee',
-    padding: '3px',
-    borderRadius: '4px'
-  },
-  italic: {
-    fontStyle: 'italic'
-  },
-  underlined: {
-    textDecoration: 'underline'
+  marks: {
+    bold: {
+      fontWeight: 'bold'
+    },
+    code: {
+      fontFamily: 'monospace',
+      backgroundColor: '#eee',
+      padding: '3px',
+      borderRadius: '4px'
+    },
+    italic: {
+      fontStyle: 'italic'
+    },
+    underlined: {
+      textDecoration: 'underline'
+    }
   }
 }
 

lib/components/content.js

@@ -56,7 +56,6 @@ class Content extends React.Component {
     onSelect: React.PropTypes.func.isRequired,
     readOnly: React.PropTypes.bool.isRequired,
     renderDecorations: React.PropTypes.func.isRequired,
-    renderMark: React.PropTypes.func.isRequired,
     schema: React.PropTypes.object,
     spellCheck: React.PropTypes.bool.isRequired,
     state: React.PropTypes.object.isRequired,
@@ -684,7 +683,7 @@ class Content extends React.Component {
    */
 
   renderNode = (node) => {
-    const { editor, renderDecorations, renderMark, schema, state } = this.props
+    const { editor, renderDecorations, schema, state } = this.props
 
     return (
       <Node
@@ -694,7 +693,6 @@ class Content extends React.Component {
         state={state}
         editor={editor}
         renderDecorations={renderDecorations}
-        renderMark={renderMark}
       />
     )
   }

lib/components/editor.js

@@ -63,8 +63,6 @@ class Editor extends React.Component {
     plugins: React.PropTypes.array,
     readOnly: React.PropTypes.bool,
     renderDecorations: React.PropTypes.func,
-    renderMark: React.PropTypes.func,
-    renderNode: React.PropTypes.func,
     schema: React.PropTypes.object,
     spellCheck: React.PropTypes.bool,
     state: React.PropTypes.object.isRequired,
@@ -267,7 +265,6 @@ class Editor extends React.Component {
         onChange={this.onChange}
         readOnly={this.props.readOnly}
         renderDecorations={this.renderDecorations}
-        renderMark={this.renderMark}
         schema={this.state.schema}
         spellCheck={this.props.spellCheck}
         state={this.state.state}

lib/components/leaf.js

@@ -33,7 +33,7 @@ class Leaf extends React.Component {
     marks: React.PropTypes.object.isRequired,
     node: React.PropTypes.object.isRequired,
     ranges: React.PropTypes.object.isRequired,
-    renderMark: React.PropTypes.func.isRequired,
+    schema: React.PropTypes.object.isRequired,
     state: React.PropTypes.object.isRequired,
     text: React.PropTypes.string.isRequired
   };
@@ -83,7 +83,7 @@ class Leaf extends React.Component {
     if (
       props.index != this.props.index ||
       props.marks != this.props.marks ||
-      props.renderMark != this.props.renderMark ||
+      props.schema != this.props.schema ||
       props.text != this.props.text
     ) {
       return true
@@ -257,11 +257,11 @@ class Leaf extends React.Component {
    */
 
   renderMarks(props) {
-    const { marks, renderMark } = props
+    const { marks, schema } = props
     const text = this.renderText(props)
 
     return marks.reduce((children, mark) => {
-      const Component = renderMark(mark, marks)
+      const Component = mark.getComponent(schema)
       if (!Component) return children
       return <Component mark={mark} marks={marks}>{children}</Component>
     }, text)

lib/components/node.js

@@ -35,7 +35,6 @@ class Node extends React.Component {
     editor: React.PropTypes.object.isRequired,
     node: React.PropTypes.object.isRequired,
     renderDecorations: React.PropTypes.func.isRequired,
-    renderMark: React.PropTypes.func.isRequired,
     schema: React.PropTypes.object.isRequired,
     state: React.PropTypes.object.isRequired
   };
@@ -214,7 +213,7 @@ class Node extends React.Component {
    */
 
   renderNode = (child) => {
-    const { editor, node, renderDecorations, renderMark, schema, state } = this.props
+    const { editor, node, renderDecorations, schema, state } = this.props
     const block = node.kind == 'block' ? node : this.props.block
     return (
       <Node
@@ -225,7 +224,6 @@ class Node extends React.Component {
         state={state}
         editor={editor}
         renderDecorations={renderDecorations}
-        renderMark={renderMark}
       />
     )
   }
@@ -309,7 +307,7 @@ class Node extends React.Component {
    */
 
   renderLeaf = (ranges, range, index, offset) => {
-    const { node, renderMark, state } = this.props
+    const { node, schema, state } = this.props
     const text = range.text
     const marks = range.marks
 
@@ -317,12 +315,12 @@ class Node extends React.Component {
       <Leaf
         key={`${node.key}-${index}`}
         index={index}
-        state={state}
-        node={node}
-        text={text}
         marks={marks}
+        node={node}
         ranges={ranges}
-        renderMark={renderMark}
+        schema={schema}
+        state={state}
+        text={text}
       />
     )
   }

lib/models/mark.js

@@ -1,5 +1,6 @@
 
 import Data from './data'
+import memoize from '../utils/memoize'
 import { Map, Record, Set } from 'immutable'
 
 /**
@@ -66,6 +67,14 @@ class Mark extends new Record(DEFAULTS) {
 
 }
 
+/**
+ * Memoize read methods.
+ */
+
+memoize(Mark.prototype, [
+  'getComponent',
+])
+
 /**
  * Export.
  */

lib/models/schema.js

@@ -1,6 +1,8 @@
 
+import React from 'react'
 import RULES from '../constants/rules'
 import includes from 'lodash/includes'
+import isReactComponent from '../utils/is-react-component'
 import typeOf from 'type-of'
 import memoize from '../utils/memoize'
 import { Record } from 'immutable'
@@ -150,7 +152,24 @@ class Schema extends new Record(DEFAULTS) {
   __getComponent(object) {
     const match = this.rules.find(rule => rule.match(object) && rule.component)
     if (!match) return
-    return match.component
+
+    const { component } = match
+
+    // Marks need special handling to normalize components.
+    if (object.kind != 'mark') return match.component
+
+    // If it's a React component, we're good.
+    if (isReactComponent(component)) return component
+
+    // Handle all other types...
+    switch (typeOf(component)) {
+      case 'function':
+        return component
+      case 'object':
+        return props => <span style={component}>{props.children}</span>
+      case 'string':
+        return props => <span className={component}>{props.children}</span>
+    }
   }
 
   /**
@@ -331,18 +350,19 @@ function normalizeSpec(obj, giveReason) {
 
       const fail = CHECKS[key]
       const failure = fail(node, value)
+      if (failure === undefined) continue
 
-      if (!giveReason) {
-        return !failure
-      }
-
-      if (failure != undefined) {
+      if (giveReason) {
         return {
           type: key,
           value: failure
         }
+      } else {
+        return false
       }
     }
+
+    return giveReason ? undefined : true
   }
 }