serverChannel updates preview.tsx - main.ts update currently not working

src/index.ts

@@ -3,7 +3,7 @@ import { readConfig, writeConfig } from "@storybook/csf-tools";
 // eslint-disable-next-line import/no-unresolved
 import { run } from "chromatic/node";
 
-import { BUILD_STARTED, START_BUILD, UPDATE_PROJECT_ID } from "./constants";
+import { BUILD_STARTED, START_BUILD, UPDATE_PROJECT } from "./constants";
 import { findConfig } from "./utils/storybook.config.utils";
 
 /**
@@ -35,17 +35,33 @@ async function serverChannel(channel: Channel, { projectToken }: { projectToken:
     });
   });
 
-  channel.on(UPDATE_PROJECT_ID, async (id) => {
+  channel.on(UPDATE_PROJECT, async (id, token) => {
+
+    // update project id
     // find config file path
     const previewPath = await findConfig("preview");
     // Find config file
     const PreviewConfig = await readConfig(previewPath);
 
     // Add parameters to config file
-    PreviewConfig.setFieldValue(["projectId"], id);
+    const previousProjectId = PreviewConfig.getFieldNode(["parameters", "projectId"]);
+    console.log("updating project id", previousProjectId, id);
+    PreviewConfig.setFieldValue(["parameters", "projectId"], id);
 
     // Write to config file
     await writeConfig(PreviewConfig);
+
+
+
+    // update project token in main.ts - this is not currently working, does not select the main.ts section correctly
+    const mainPath = await findConfig("main");
+    const MainConfig = await readConfig(mainPath);
+
+    // TODO: Get the correct field node
+    const addonsConfig = MainConfig.getFieldNode(["addons"]);
+    console.log(addonsConfig);
+
+    MainConfig.setFieldValue(["addons", "projectToken"], token);
   });
 
   return channel;