feat: rax jsx runtime

packages/rax/jsx-dev-runtime.js

@@ -0,0 +1 @@
+export {Fragment, jsxDEV} from './index';

packages/rax/jsx-runtime.js

@@ -0,0 +1 @@
+export {Fragment, jsx, jsxs, jsxDEV} from './index';

packages/rax/package.json

@@ -4,6 +4,11 @@
   "description": "A universal React-compatible render engine.",
   "license": "BSD-3-Clause",
   "main": "index.js",
+  "exports": {
+    ".": "./index.js",
+    "./jsx-runtime": "./jsx-runtime.js",
+    "./jsx-dev-runtime": "./jsx-dev-runtime.js"
+  },
   "repository": {
     "type": "git",
     "url": "git+https://github.com/alibaba/rax.git"

packages/rax/src/__tests__/jsx-runtime.test.js

@@ -0,0 +1,92 @@
+import Component from '../vdom/component';
+import createElement from '../createElement';
+import createRef from '../createRef';
+import { jsx, jsxs, jsxDEV } from '../jsx-runtime';
+import Fragment from '../fragment';
+
+describe('Support JSX-Runtime', () => {
+  it('should export all modules needed by importSource', () => {
+    expect(typeof jsx).toBe('function');
+    expect(typeof jsxs).toBe('function');
+    expect(typeof jsxDEV).toBe('function');
+    expect(typeof Fragment).toBe('function');
+  });
+
+  it('should add key', () => {
+    const vnode = jsx('div', null, 'foo');
+    expect(vnode.key).toBe('foo');
+  });
+
+  it('should keep ref in props', () => {
+    let ref = () => null;
+    let vnode = jsx('div', { ref });
+    expect(vnode.ref).toBe(ref);
+
+    ref = 'fooRef';
+    vnode = jsx('div', { ref });
+    expect(vnode.ref).toBe('fooRef');
+  });
+
+  it('should apply defaultProps', () => {
+    class Foo extends Component {
+      render() {
+        return <div />;
+      }
+    }
+
+    Foo.defaultProps = {
+      foo: 'bar',
+      fake: 'bar1',
+    };
+
+    let vnode = jsx(Foo, {}, null);
+    expect(vnode.props.foo).toBe('bar');
+
+    vnode = jsx(Foo, {fake: 'bar2'}, null);
+    expect(vnode.props.fake).toBe('bar2');
+  });
+
+  it('should keep props over defaultProps', () => {
+    class Foo extends Component {
+      render() {
+        return <div />;
+      }
+    }
+
+    Foo.defaultProps = {
+      foo: 'bar'
+    };
+
+    const vnode = jsx(Foo, { foo: 'baz' }, null);
+    expect(vnode.props).toEqual({
+      foo: 'baz'
+    });
+  });
+
+  it('should set __source and __self', () => {
+    const vnode = jsxDEV('div', { class: 'foo' }, 'key', 'source', 'self');
+    expect(vnode.__source).toBe('source');
+    expect(vnode.__self).toBe('self');
+  });
+
+  it('should return a vnode like createElement', () => {
+    const elementVNode = createElement('div', {
+      class: 'foo',
+      key: 'key'
+    });
+    const jsxVNode = jsx('div', { class: 'foo' }, 'key');
+    expect(jsxVNode).toEqual(elementVNode);
+  });
+
+  it('should remove ref from props', () => {
+    const ref = createRef();
+    const vnode = jsx('div', { ref }, null);
+    expect(vnode.props).toEqual({});
+    expect(vnode.ref).toBe(ref);
+  });
+
+  it('should receive children props', () => {
+    const fooProps = {children: []};
+    expect(Fragment(fooProps)).toEqual([]);
+  });
+});

packages/rax/src/index.js

@@ -7,6 +7,7 @@ export memo from './memo';
 export Fragment from './fragment';
 export render from './render';
 export Component, { PureComponent } from './vdom/component';
+export { jsx, jsxs, jsxDEV } from './jsx-runtime';
 export version from './version';
 
 import Host from './vdom/host';

packages/rax/src/jsx-runtime.js

@@ -0,0 +1,105 @@
+import Host from './vdom/host';
+import Element from './vdom/element';
+import { warning, throwError, throwMinifiedWarn } from './error';
+import { isArray, NOOP } from './types';
+import validateChildKeys from './validateChildKeys';
+
+const __DEV__ = process.env.NODE_ENV !== 'production';
+
+/**
+ * This module is adapted to react's jsx-runtime module.
+ * @see：[introducing-the-new-jsx-transform](https://reactjs.org/blog/2020/09/22/introducing-the-new-jsx-transform.htm)
+ */
+function jsxRuntime(type, props, key, __source, __self) {
+  let normalizedProps = {};
+  let propName;
+
+  // The default value of key and ref of rax element is null
+  let ref = props && props.ref || null;
+  key = key || null;
+
+  for (propName in props) {
+    if (propName !== 'ref') {
+      normalizedProps[propName] = props[propName];
+    }
+  }
+
+  let defaults;
+  if (typeof type === 'function' && (defaults = type.defaultProps)) {
+    for (propName in defaults)
+      if (normalizedProps[propName] === undefined) {
+        normalizedProps[propName] = defaults[propName];
+      }
+  }
+
+  return new Element(
+    type,
+    key,
+    ref,
+    normalizedProps,
+    Host.owner,
+    __source,
+    __self
+  );
+}
+
+
+function _jsxWithValidation(type, props, key, isStaticChildren, __source, __self) {
+  const elem = jsxRuntime(type, props, key, __source, __self);
+
+  if (type == null) {
+    if (__DEV__) {
+      throwError(`Invalid element type, expected a string or a class/function component but got "${type}".`);
+    } else {
+      // A empty component replaced avoid break render in production
+      type = NOOP;
+      throwMinifiedWarn(0);
+    }
+  }
+
+  const children = props && props.children;
+  if (children !== undefined) {
+    if (isStaticChildren) {
+      if (isArray(children)) {
+        for (let i = 0; i < children.length; i++) {
+          validateChildKeys(children[i], type);
+        }
+
+        if (Object.freeze) {
+          Object.freeze(children);
+        }
+      } else {
+        warning(
+          'Rax.jsx: Static children should always be an array. ' +
+          'You are likely explicitly calling Rax.jsxs or Rax.jsxDEV. ' +
+          'Use the Babel transform instead.',
+        );
+      }
+    } else {
+      validateChildKeys(children, type);
+    }
+  }
+
+  return elem;
+}
+
+function jsxWithValidation(type, props, key) {
+  return _jsxWithValidation(type, props, key, false);
+}
+function jsxsWithValidation(type, props, key) {
+  return _jsxWithValidation(type, props, key, true);
+}
+function jsxDEVWithValidation(type, props, key, __source, __self) {
+  return _jsxWithValidation(type, props, key, true, __source, __self);
+}
+
+const jsx = __DEV__ ? jsxWithValidation : jsxRuntime;
+const jsxs = __DEV__ ? jsxsWithValidation : jsxRuntime;
+const jsxDEV = __DEV__ ? jsxDEVWithValidation : undefined;
+
+export {
+  jsx,
+  jsxs,
+  jsxDEV
+};
+

packages/rax/src/vdom/element.js

@@ -1,6 +1,6 @@
 import checkPropTypes from 'prop-types/checkPropTypes';
 
-export default function Element(type, key, ref, props, owner) {
+export default function Element(type, key, ref, props, owner, __source, __self) {
   let element = {
     // Built-in properties that belong on the element
     type,
@@ -33,10 +33,24 @@ export default function Element(type, key, ref, props, owner) {
       value: false
     });
 
-    // Props is immutable
-    if (Object.freeze) {
-      Object.freeze(props);
-    }
+    Object.defineProperty(element, '__source', {
+      configurable: false,
+      enumerable: false,
+      writable: true,
+      value: __source
+    });
+
+    Object.defineProperty(element, '__self', {
+      configurable: false,
+      enumerable: false,
+      writable: true,
+      value: __self
+    });
+  }
+
+  // Props is immutable
+  if (Object.freeze) {
+    Object.freeze(props);
   }
 
   return element;