From c86d2996e33fa58ecc4df5d97bd7c222019c9316 Mon Sep 17 00:00:00 2001
From: cha0s <git@cha0s.io>
Date: Fri, 26 Jun 2020 03:22:40 -0500
Subject: [PATCH] fix: hot reloading

---
 src/client/resource.jsx                    | 11 +++++++++++
 src/common/resources/entity/trait-pane.jsx | 11 +++++++++++
 2 files changed, 22 insertions(+)

diff --git a/src/client/resource.jsx b/src/client/resource.jsx
index 03c2d9f..d8fad09 100644
--- a/src/client/resource.jsx
+++ b/src/client/resource.jsx
@@ -2,6 +2,7 @@ import './app.scss';
 
 import PropTypes from 'prop-types';
 import React from 'react';
+import {registerHooks} from 'scwp';
 
 import {all} from './resources.scwp';
 
@@ -29,3 +30,13 @@ Resource.propTypes = {
 };
 
 export default Resource;
+
+if (module.hot) {
+  registerHooks({
+    autoreg$accept: (type) => {
+      if ('resource' === type) {
+        module.hot.invalidate();
+      }
+    },
+  }, module.id);
+}
diff --git a/src/common/resources/entity/trait-pane.jsx b/src/common/resources/entity/trait-pane.jsx
index 61d9421..6b3a33b 100644
--- a/src/common/resources/entity/trait-pane.jsx
+++ b/src/common/resources/entity/trait-pane.jsx
@@ -4,6 +4,7 @@ import {all as allTraitComponents} from '@avocado/entity/trait/trait-components.
 import PropTypes from 'prop-types';
 import React, {useMemo} from 'react';
 import {useDispatch} from 'react-redux';
+import {registerHooks} from 'scwp';
 
 import Value from '~/client/value';
 
@@ -94,3 +95,13 @@ TraitPane.propTypes = {
 };
 
 export default TraitPane;
+
+if (module.hot) {
+  registerHooks({
+    autoreg$accept: (type, M) => {
+      if ('type-renderer' === type) {
+        module.hot.invalidate();
+      }
+    },
+  }, module.id);
+}