safe evaluate code refactor

lib/code-runner/run-prompt.tsx

@@ -49,11 +49,10 @@ export const runInitialPrompt = async (
     syntaxError,
     circuitErrors,
     typescriptErrors,
-  } = safeEvaluateCode(
-    codefence,
-    context.outputType,
-    context.preSuppliedImports,
-  )
+  } = safeEvaluateCode(codefence, {
+    outputType: context.outputType,
+    preSuppliedImports: context.preSuppliedImports,
+  })
 
   if (success) {
     return {

lib/code-runner/safe-evaluate-code.tsx

@@ -3,13 +3,13 @@ import * as React from "react"
 import { Circuit } from "@tscircuit/core"
 import { safeTranspileCode } from "./transpile-code"
 import Debug from "debug"
+import type { CodeRunnerContext } from "./code-runner-context"
 
 const debug = Debug("tscircuit:prompt")
 
 export const safeEvaluateCode = (
   code: string,
-  type: "board" | "footprint" | "package" | "model" = "board",
-  preSuppliedImports: Record<string, any> = {},
+  context: CodeRunnerContext,
 ):
   | {
       success: true
@@ -35,6 +35,8 @@ export const safeEvaluateCode = (
     } => {
   globalThis.React = React
 
+  const { preSuppliedImports = {}, outputType } = context
+
   // Add pre-supplied imports to the global scope
   for (const [key, value] of Object.entries(preSuppliedImports)) {
     ;(globalThis as any)[key] = value
@@ -118,21 +120,21 @@ export const safeEvaluateCode = (
 
   const circuit = new Circuit()
   try {
-    if (type === "board") {
+    if (outputType === "board") {
       circuit.add(<UserElm />)
-    } else if (type === "package") {
+    } else if (outputType === "package") {
       circuit.add(
         <board width="10mm" height="10mm">
           <UserElm name="U1" />
         </board>,
       )
-    } else if (type === "footprint") {
+    } else if (outputType === "footprint") {
       circuit.add(
         <board width="10mm" height="10mm">
           <chip name="U1" footprint={<UserElm />} />
         </board>,
       )
-    } else if (type === "model") {
+    } else if (outputType === "model") {
       // Note: This part might need adjustment as we don't have access to jscad-related imports
       // const jscadGeoms: any[] = []
       // const { createJSCADRoot } = createJSCADRenderer(jscadPlanner as any)

tests/lib/safe-evaluate-code.test.tsx

@@ -15,8 +15,11 @@ export default function TestComponent() {
 }
   `
 
-  const result = safeEvaluateCode(testCode, "board", {
-    "@tsci/seveibar.custom-component": MyCustomComponent,
+  const result = safeEvaluateCode(testCode, {
+    outputType: "board",
+    preSuppliedImports: {
+      "@tsci/seveibar.custom-component": MyCustomComponent,
+    },
   })
 
   expect(result.success).toBe(true)