import { logger, SnakAgentInterface } from '@snakagent/core';
import { fixCairoCodeSchema } from '../schema/schema.js';
import { callCairoGenerationAPI, extractCairoCode } from '../utils/utils.js';
import { z } from 'zod';
import { scarb } from '@snakagent/database/queries';
/**
* Fix Cairo code using AI via API and update it in the database
* @param {SnakAgentInterface} agent - The Starknet agent interface
* @param {FixCairoCodeSchema} params - The parameters for code fixing
* @returns {Promise<string>} JSON string with the fixed code or error
*/
export const fixCairoCode = async (
_angent: SnakAgentInterface,
params: z.infer<typeof fixCairoCodeSchema>
): Promise<string> => {
try {
logger.debug('\n Fixing Cairo code');
logger.debug(JSON.stringify(params, null, 2));
if (!params?.programName || !params.programName.endsWith('.cairo')) {
throw new Error('Program name is required and must end with .cairo');
}
if (!params?.error) {
throw new Error('Error description is required for fixing Cairo code');
}
const projectData = await scarb.retrieveProjectData(params.projectName);
if (!projectData) {
throw new Error(`project ${params.projectName} does not exist`);
}
const program = projectData.programs.find(
(p) => p.name === params.programName
);
if (!program) {
throw new Error(
`Program ${params.programName} not found in the database`
);
}
let errorText = params.error;
if (
errorText.includes('[EXACT_ERROR_BEGIN]') &&
errorText.includes('[EXACT_ERROR_END]')
) {
errorText = errorText
.split('[EXACT_ERROR_BEGIN]')[1]
.split('[EXACT_ERROR_END]')[0]
.trim();
}
const fixPrompt = `I have the following Cairo code that has an error:
\`\`\`cairo
${program.source_code}
\`\`\`
The error is: ${errorText}
Can you fix the compilation errors?`;
const generatedContent = await callCairoGenerationAPI(fixPrompt);
const fixedCairoCode = extractCairoCode(generatedContent);
await scarb.insertProgram({
project_id: projectData.id,
name: params.programName,
source_code: fixedCairoCode,
});
return JSON.stringify({
status: 'success',
message: `Cairo code fixed and updated in database as ${params.programName}`,
programName: params.programName,
originalCode: program.source_code,
fixedCode: fixedCairoCode,
});
} catch (error) {
logger.error('Error fixing Cairo code:', error);
return JSON.stringify({
status: 'failure',
error: error instanceof Error ? error.message : 'Unknown error',
});
}
};
