fix(runtime): fix #590, use an init set to check first time render

This patch use a more solid way to check init state setup, instead of render count.

packages/runtime/src/App.tsx

@@ -17,7 +17,7 @@ export const App: React.FC<AppProps> = props => {
   const {
     options,
     services,
-    debugStore = false,
+    debugStore = true,
     debugEvent = false,
     hooks,
     isInEditor = false,

packages/runtime/src/components/_internal/ImplWrapper/ImplWrapperMain.tsx

@@ -7,7 +7,7 @@ import { useRuntimeFunctions } from './hooks/useRuntimeFunctions';
 import { getSlotElements } from './hooks/useSlotChildren';
 import { useGlobalHandlerMap } from './hooks/useGlobalHandlerMap';
 import { useEleRef } from './hooks/useEleMap';
-import { initStateAndMethod } from '../../../utils/initStateAndMethod';
+import { initSingleComponentState } from '../../../utils/initStateAndMethod';
 import ComponentErrorBoundary from '../ComponentErrorBoundary';
 
 export const ImplWrapperMain = React.forwardRef<HTMLDivElement, ImplWrapperProps>(
@@ -23,7 +23,7 @@ export const ImplWrapperMain = React.forwardRef<HTMLDivElement, ImplWrapperProps
     // This code is to init dynamic generated components
     // because they have not be initialized before
     if (!stateManager.store[c.id]) {
-      initStateAndMethod(registry, stateManager, [c]);
+      initSingleComponentState(registry, stateManager, c);
     }
 
     const { eleRef, onRef, onRecoverFromError } = useEleRef(props);

packages/runtime/src/components/_internal/ImplWrapper/UnmountImplWrapper.tsx

@@ -37,9 +37,13 @@ export const UnmountImplWrapper = React.forwardRef<HTMLDivElement, ImplWrapperPr
     const traitChangeCallback = useCallback(
       (trait: RuntimeTraitSchema, properties: Record<string, unknown>) => {
         const result = executeTrait(trait, properties);
+
         const prevIsHidden = isHidden;
         setIsHidden(!!result.unmount);
-        if (result.unmount) {
+
+        const isLastRender = slotContext ? slotContext.renderSet.size === 0 : true;
+
+        if (result.unmount && isLastRender) {
           // Every component's state is initialized in initStateAnd Method.
           // So if if it should not render, we should remove it from store.
 
@@ -52,12 +56,12 @@ export const UnmountImplWrapper = React.forwardRef<HTMLDivElement, ImplWrapperPr
            * forever, it still need to teardown at the first time it rendered.
            */
           if (!prevIsHidden || stateManager.initSet.has(c.id)) {
-            stateManager.initSet.delete(c.id);
             delete stateManager.store[c.id];
+            stateManager.initSet.delete(c.id);
           }
         }
       },
-      [c.id, executeTrait, stateManager, isHidden]
+      [executeTrait, isHidden, stateManager, c.id, slotContext]
     );
 
     useEffect(() => {

packages/runtime/src/utils/initStateAndMethod.ts

@@ -13,12 +13,11 @@ import { JSONSchema7Object } from 'json-schema';
 export function initStateAndMethod(
   registry: RegistryInterface,
   stateManager: StateManagerInterface,
-  components: RuntimeComponentSchema[],
-  mark?: true
+  components: RuntimeComponentSchema[]
 ) {
   components.forEach(c => {
-    initSingleComponentState(registry, stateManager, c);
-    if (mark) {
+    const inited = initSingleComponentState(registry, stateManager, c);
+    if (inited) {
       stateManager.initSet.add(c.id);
     }
   });
@@ -28,7 +27,7 @@ export function initSingleComponentState(
   registry: RegistryInterface,
   stateManager: StateManagerInterface,
   c: RuntimeComponentSchema
-) {
+): boolean {
   if (stateManager.store[c.id]) {
     return false;
   }
@@ -60,4 +59,6 @@ export function initSingleComponentState(
       stateManager.store[moduleSchema.id] = moduleInitState;
     } catch {}
   }
+
+  return true;
 }