import os
from google.genai import types

def write_file(working_directory, file_path, content):
    abs_path = os.path.abspath(working_directory)
    new_file_path = os.path.normpath(os.path.join(abs_path, file_path))
    valid_target_file = os.path.commonpath([abs_path, new_file_path]) == abs_path
    if not valid_target_file:
        return f'Error: Cannot write to "{file_path}" as it is outside the permitted working directory'
    if os.path.isdir(new_file_path):
       return f'Error: Cannot write to "{file_path}" as it is a directory'
    
    os.makedirs(os.path.dirname(new_file_path), exist_ok=True)
    try:
        with open(new_file_path, "w") as f:
            f.write(content)
    except Exception as e:
        return f"Error: {e}"
    return f'Successfully wrote to "{file_path}" ({len(content)} characters written)'
    
schema_write_file = types.FunctionDeclaration(
    name="write_file",
    description="Open file and write content into",
    parameters=types.Schema(
        type=types.Type.OBJECT,
        properties={
            "working_directory": types.Schema(
                type=types.Type.STRING,
                description="Directory path to list files from, relative to the working directory (default is the working directory itself)",
            ),
            "file_path": types.Schema(
                type=types.Type.STRING,
                description="File path",
            ),
            "content": types.Schema(
                type=types.Type.STRING,
                description="Content what should have been written to the file",
            ),
        },
    ),
)