from const.function_calls import GET_FILES, DEV_STEPS, IMPLEMENT_CHANGES, CODE_CHANGES
from database.models.files import File
from helpers.files import update_file
from helpers.AgentConvo import AgentConvo
from helpers.Agent import Agent

class CodeMonkey(Agent):
    def __init__(self, project, developer):
        super().__init__('code_monkey', project)
        self.developer = developer

    def implement_code_changes(self, convo, code_changes_description, step_index=0):
        if convo == None:
            convo = AgentConvo(self)

        files_needed = convo.send_message('development/task/request_files_for_code_changes.prompt', {
            "step_description": code_changes_description,
            "directory_tree": self.project.get_directory_tree(True),
            "step_index": step_index,
            "finished_steps": ', '.join(f"#{j}" for j in range(step_index))
        }, GET_FILES)


        changes = convo.send_message('development/implement_changes.prompt', {
            "instructions": code_changes_description,
            "directory_tree": self.project.get_directory_tree(True),
            "files": self.project.get_files(files_needed),
        }, IMPLEMENT_CHANGES)

        for file_data in changes:
            file_data['full_path'] = self.project.get_full_file_path(file_data['path'], file_data['name'])

            if file_data['description'] != '':
                    (File.insert(app=self.project.app, path=file_data['path'], name=file_data['name'], description=file_data['description'])
                        .on_conflict(
                            conflict_target=[File.app, File.name, File.path],
                            preserve=[],
                            update={'description': file_data['description']})
                        .execute())

            update_file(file_data['full_path'], file_data['content'])

        return convo