feat: shape registry

packages/x6/src/research/registry/index.ts

@@ -0,0 +1 @@
+export { registerNode } from './node'

packages/x6/src/research/registry/node.ts

@@ -0,0 +1,52 @@
+import { Node as NodeRaw } from '../core/node'
+import { registerEntity } from './util'
+
+export interface Options extends NodeRaw.Defaults {
+  init?: () => void
+}
+
+export type NodeClass = typeof NodeRaw
+
+const nodes: { [name: string]: NodeClass } = {}
+
+function register(name: string, node: NodeClass, force: boolean) {
+  registerEntity(nodes, name, node, force, () => {
+    throw new Error(`Shape with name '${name}' already registered.`)
+  })
+
+  return node
+}
+
+export function registerNode(
+  name: string,
+  shape: NodeClass,
+  force?: boolean,
+): NodeClass
+export function registerNode(
+  name: string,
+  options: Options,
+  force?: boolean,
+): NodeClass
+export function registerNode(
+  name: string,
+  options: Options | NodeClass,
+  force: boolean = false,
+): NodeClass {
+  if (typeof options === 'function') {
+    return register(name, options, force)
+  }
+
+  const { init, ...defaults } = options
+  // tslint:disable-next-line
+  class Node extends NodeRaw {
+    init() {
+      if (typeof init === 'function') {
+        init.call(this)
+      }
+    }
+  }
+
+  Node.setDefaults(defaults)
+
+  return register(name, Node as any, force)
+}

packages/x6/src/research/registry/util.ts

@@ -0,0 +1,32 @@
+function getHMRStatus() {
+  const mod = module as any
+  if (mod != null && mod.hot != null && mod.hot.status != null) {
+    return mod.hot.status()
+  }
+  return 'unkonwn'
+}
+
+function isApplyingHMR() {
+  return getHMRStatus() === 'apply'
+}
+
+export function registerEntity<T>(
+  registry: { [name: string]: T },
+  name: string,
+  entity: T,
+  force: boolean,
+  onError: () => void,
+) {
+  if (registry[name] && !force && !isApplyingHMR()) {
+    onError()
+  }
+  registry[name] = entity
+}
+
+export function getEntityFromRegistry<T>(
+  registry: { [name: string]: T },
+  name: string,
+): T | null {
+  const ret = registry[name]
+  return typeof ret === 'function' ? ret : null
+}