WIP

stylemistake · Nov 8, 2019 · d1b1385 · d1b1385
1 parent b5ce411
commit d1b1385
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 0 deletions.
diff --git a/tgui-next/packages/inferno/package.json b/tgui-next/packages/inferno/package.json
@@ -1,4 +1,5 @@
 {
+  "private": true,
   "name": "inferno",
   "version": "7.3.2",
   "license": "MIT",

diff --git a/tgui-next/packages/inferno/src/DOM/mounting.ts b/tgui-next/packages/inferno/src/DOM/mounting.ts
@@ -7,6 +7,7 @@ import { mountProps } from './props';
 import { createClassComponentInstance } from './utils/componentUtil';
 import { validateKeys } from '../core/validate';
 import { mountRef } from '../core/refs';
+import { options } from './utils/common';
 
 export function mount(vNode: VNode, parentDOM: Element | null, context: Object, isSVG: boolean, nextNode: Element | null, lifecycle: Function[]): void {
   const flags = (vNode.flags |= VNodeFlags.InUse);
@@ -186,5 +187,8 @@ export function mountFunctionalComponentCallbacks(vNode: VNode, lifecycle: Funct
     if (isFunction(ref.onComponentDidMount)) {
       lifecycle.push(createOnMountCallback(ref, vNode));
     }
+    if (options.beforeRender) {
+      options.beforeRender(vNode);
+    }
   }
 }
diff --git a/tgui-next/packages/inferno/src/DOM/patching.ts b/tgui-next/packages/inferno/src/DOM/patching.ts
@@ -458,6 +458,9 @@ function patchFunctionalComponent(lastVNode, nextVNode, parentDOM, context, isSV
     if (nextHooksDefined && isFunction(nextRef.onComponentWillUpdate)) {
       nextRef.onComponentWillUpdate(lastProps, nextProps);
     }
+    if (options.beforeUpdate) {
+      options.beforeUpdate(nextVNode);
+    }
     const type = nextVNode.type;
     const nextInput = normalizeRoot(nextVNode.flags & VNodeFlags.ForwardRef ? type.render(nextProps, nextRef, context) : type(nextProps, context));
 
@@ -466,6 +469,9 @@ function patchFunctionalComponent(lastVNode, nextVNode, parentDOM, context, isSV
     if (nextHooksDefined && isFunction(nextRef.onComponentDidUpdate)) {
       nextRef.onComponentDidUpdate(lastProps, nextProps);
     }
+    if (options.afterUpdate) {
+      options.afterUpdate(nextVNode);
+    }
   } else {
     nextVNode.children = lastInput;
   }

diff --git a/tgui-next/packages/inferno/src/DOM/unmounting.ts b/tgui-next/packages/inferno/src/DOM/unmounting.ts
@@ -4,6 +4,7 @@ import { VNode } from '../core/types';
 import { syntheticEvents, unmountSyntheticEvent } from './events/delegation';
 import { EMPTY_OBJ, findDOMfromVNode, removeVNodeDOM } from './utils/common';
 import { unmountRef } from '../core/refs';
+import { options } from './utils/common';
 
 export function remove(vNode: VNode, parentDOM: Element) {
   unmount(vNode);
@@ -53,6 +54,9 @@ export function unmount(vNode) {
       if (!isNullOrUndef(ref) && isFunction(ref.onComponentWillUnmount)) {
         ref.onComponentWillUnmount(findDOMfromVNode(vNode, true) as Element, vNode.props || EMPTY_OBJ);
       }
+      if (options.beforeUnmount) {
+        options.beforeUnmount(vNode);
+      }
 
       unmount(children);
     } else if (flags & VNodeFlags.Portal) {

diff --git a/tgui-next/packages/inferno/src/DOM/utils/common.ts b/tgui-next/packages/inferno/src/DOM/utils/common.ts
@@ -161,6 +161,11 @@ export const options: {
   createVNode: ((vNode: VNode) => void) | null;
   renderComplete: ((rootInput: VNode | InfernoNode, parentDOM: Element | SVGAElement | ShadowRoot | DocumentFragment | HTMLElement | Node) => void) | null;
   reactStyles?: boolean;
+  // Hooks
+  beforeRender?: any,
+  beforeUpdate?: any,
+  afterUpdate?: any,
+  beforeUnmount?: any,
 } = {
   componentComparator: null,
   createVNode: null,