Further testing

frontend/src/framework/Module.tsx

@@ -1,5 +1,6 @@
 import React from "react";
 
+import { Atom } from "jotai";
 import { cloneDeep } from "lodash";
 
 import { BroadcastChannelsDef, InputBroadcastChannelDef } from "./Broadcaster";
@@ -46,6 +47,7 @@ export class Module<StateType extends StateBaseType> {
     private _drawPreviewFunc: DrawPreviewFunc | null;
     private _description: string | null;
     private _inputChannelDefs: InputBroadcastChannelDef[];
+    private _atoms: Atom<any>[];
 
     constructor(
         name: string,
@@ -54,7 +56,8 @@ export class Module<StateType extends StateBaseType> {
         broadcastChannelsDef: BroadcastChannelsDef = {},
         inputChannelDefs: InputBroadcastChannelDef[] = [],
         drawPreviewFunc: DrawPreviewFunc | null = null,
-        description: string | null = null
+        description: string | null = null,
+        atoms: Atom<any>[] = []
     ) {
         this._name = name;
         this._defaultTitle = defaultTitle;
@@ -69,6 +72,7 @@ export class Module<StateType extends StateBaseType> {
         this._inputChannelDefs = inputChannelDefs;
         this._drawPreviewFunc = drawPreviewFunc;
         this._description = description;
+        this._atoms = atoms;
     }
 
     getDrawPreviewFunc(): DrawPreviewFunc | null {
@@ -123,7 +127,8 @@ export class Module<StateType extends StateBaseType> {
             instanceNumber,
             this._channelsDef,
             this._workbench,
-            this._inputChannelDefs
+            this._inputChannelDefs,
+            this._atoms
         );
         this._moduleInstances.push(instance);
         this.maybeImportSelf();

frontend/src/framework/ModuleInstance.ts

@@ -1,6 +1,6 @@
 import { ErrorInfo } from "react";
 
-import { createStore } from "jotai";
+import { Atom, createStore } from "jotai";
 import { cloneDeep } from "lodash";
 
 import { BroadcastChannel, BroadcastChannelsDef, InputBroadcastChannelDef } from "./Broadcaster";
@@ -45,13 +45,15 @@ export class ModuleInstance<StateType extends StateBaseType> {
     private _workbench: Workbench;
 
     private _jotaiStore: ReturnType<typeof createStore> = createStore();
+    private _jotaiAtoms: Atom<unknown>[] = [];
 
     constructor(
         module: Module<StateType>,
         instanceNumber: number,
         broadcastChannelsDef: BroadcastChannelsDef,
         workbench: Workbench,
-        inputChannelDefs: InputBroadcastChannelDef[]
+        inputChannelDefs: InputBroadcastChannelDef[],
+        atoms: Atom<unknown>[]
     ) {
         this._id = `${module.getName()}-${instanceNumber}`;
         this._title = module.getDefaultTitle();
@@ -92,6 +94,8 @@ export class ModuleInstance<StateType extends StateBaseType> {
                     );
             });
         }
+
+        this._jotaiAtoms = atoms;
     }
 
     getJotaiStore(): ReturnType<typeof createStore> {

frontend/src/framework/ModuleRegistry.ts

@@ -1,3 +1,5 @@
+import { Atom } from "jotai";
+
 import { BroadcastChannelsDef, InputBroadcastChannelDef } from "./Broadcaster";
 import { Module } from "./Module";
 import { DrawPreviewFunc } from "./Preview";
@@ -13,6 +15,7 @@ export type RegisterModuleOptions = {
     inputChannelDefs?: InputBroadcastChannelDef[];
     preview?: DrawPreviewFunc;
     description?: string;
+    atoms?: Atom<unknown>[];
 };
 
 export class ModuleNotFoundError extends Error {
@@ -42,7 +45,8 @@ export class ModuleRegistry {
             options.broadcastChannelsDef,
             options.inputChannelDefs,
             options.preview ?? null,
-            options.description ?? null
+            options.description ?? null,
+            options.atoms ?? []
         );
         this._registeredModules[options.moduleName] = module;
         return module;

frontend/src/modules/MyModule2/atoms.ts

@@ -14,3 +14,5 @@ export const vectorsAtom = atomWithQuery((get) => ({
         ),
 }));
 export const atomBasedOnVectors = atom<boolean>((get) => get(vectorsAtom).isFetching);
+
+export const atoms = [selectedEnsembleAtom, vectorsAtom, atomBasedOnVectors];

frontend/src/modules/MyModule2/registerModule.ts

@@ -1,5 +1,6 @@
 import { ModuleRegistry } from "@framework/ModuleRegistry";
 
+import { atoms } from "./atoms";
 import { State } from "./state";
 
-ModuleRegistry.registerModule<State>({ moduleName: "MyModule2", defaultTitle: "My Module 2" });
+ModuleRegistry.registerModule<State>({ moduleName: "MyModule2", defaultTitle: "My Module 2", atoms: atoms });