Function addCreateFile

src/functions/createFile.ts:32–57

(codepilot: Codepilot)

 * Adds the createFile function to the codepilot instance.
 */
export function addCreateFile(codepilot: Codepilot): void {
    codepilot.addFunction(createFileFunction, async (args: any) => {
        const { filePath, contents } = args;

        // Check if the file already exists
        if (await fs.access(path.join(process.cwd(), filePath)).then(() => true).catch(() => false)) {
            return `A file already exists at that path.\nGive the user detailed instructions for how they should modify that file instead.`;
        }

        try {
            // Create the directory path if it doesn't exist
            const directoryPath = path.dirname(filePath);
            await fs.mkdir(directoryPath, { recursive: true });

            // Write the code to the file
            await fs.writeFile(path.join(process.cwd(), filePath), contents);

            // Add the file to the code index
            await codepilot.index.upsertDocument(filePath);
            console.log(Colorize.highlight(`Created a new file: ${filePath}`));
            return `Successfully created file at ${filePath}`;
        } catch (error) {
            return `Failed to create file at ${filePath} due to the following error:\n${(error as Error).message}`;
        }
    });
}