Function createProgramForCode

src/node/services/ptc/typeValidator.ts:78–152

(
  wrappedCode: string,
  muxTypes: string,
  compilerOptions: ts.CompilerOptions
)

};

function createProgramForCode(
  wrappedCode: string,
  muxTypes: string,
  compilerOptions: ts.CompilerOptions
): {
  program: ts.Program;
  host: ts.CompilerHost;
  getSourceFile: () => ts.SourceFile;
  setSourceFile: (newWrappedCode: string) => void;
} {
  const scriptTarget = compilerOptions.target ?? ts.ScriptTarget.ES2020;
  let sourceFile = ts.createSourceFile("agent.ts", wrappedCode, scriptTarget, true);
  const muxSourceFile = getCachedMuxSourceFile(muxTypes, scriptTarget);
  const setSourceFile = (newWrappedCode: string) => {
    sourceFile = ts.createSourceFile("agent.ts", newWrappedCode, scriptTarget, true);
  };
  const host = ts.createCompilerHost(compilerOptions);

  // Override to read lib files from our bundled directory
  host.getDefaultLibLocation = () => LIB_DIR;
  host.getDefaultLibFileName = (options) => path.join(LIB_DIR, ts.getDefaultLibFileName(options));

  const originalGetSourceFile = host.getSourceFile.bind(host);
  const originalFileExists = host.fileExists.bind(host);
  const originalReadFile = host.readFile.bind(host);

  host.getSourceFile = (fileName, languageVersionOrOptions, onError, shouldCreateNewSourceFile) => {
    // languageVersionOrOptions can be ScriptTarget or CreateSourceFileOptions
    const target =
      typeof languageVersionOrOptions === "number" ? languageVersionOrOptions : scriptTarget;
    if (fileName === "agent.ts") return sourceFile;
    if (fileName === MUX_TYPES_FILE) return muxSourceFile;

    const isLibFile = fileName.includes("lib.") && fileName.endsWith(".d.ts");
    if (isLibFile) {
      const cached = getCachedLibSourceFile(fileName, target, () => {
        if (IS_PRODUCTION) {
          const libPath = resolveLibPath(fileName);
          return fs.existsSync(libPath) ? fs.readFileSync(libPath, "utf-8") : undefined;
        }
        return originalReadFile(fileName) ?? undefined;
      });
      if (cached) return cached;
    }

    return originalGetSourceFile(
      fileName,
      languageVersionOrOptions,
      onError,
      shouldCreateNewSourceFile
    );
  };
  host.fileExists = (fileName) => {
    if (fileName === "agent.ts" || fileName === MUX_TYPES_FILE) return true;
    // In production, check bundled lib directory for lib files
    if (IS_PRODUCTION && fileName.includes("lib.") && fileName.endsWith(".d.ts")) {
      return fs.existsSync(resolveLibPath(fileName));
    }