import json
from datetime import datetime


def _ts():
    return datetime.now().strftime('%H:%M:%S')


def execute_tool(tool_call, TOOL_HANDLERS):
    tname = tool_call['function']['name']
    targs = json.loads(tool_call['function']['arguments'])
    handler = TOOL_HANDLERS.get(tname)
    if not handler:
        return f'Tool {tname} not found'
    try:
        if tname == 'search_code':
            return handler(
                pattern=targs['pattern'],
                search_type=targs['search_type'],
                path=targs.get('path', '.'),
            )
        elif tname == 'git_operation':
            return handler(args=targs['args'])
        else:
            return handler(**targs)
    except Exception as e:
        return f'Error executing tool: {str(e)}'


def run_agent_loop(session, llm_client, TOOLS, TOOL_HANDLERS, max_iterations, on_tool_calls=None):
    for step in range(max_iterations):
        print(f'[{_ts()}] Step {step + 1} — calling LLM...')
        response = llm_client.chat(session.messages, tools=TOOLS)

        if response.tool_calls:
            amsg = {
                'role': 'assistant',
                'content': response.content,
                'tool_calls': response.tool_calls,
            }
            session.messages.append(amsg)

            tnames = [tc['function']['name'] for tc in response.tool_calls]
            print(f'[{_ts()}] Using tools: {", ".join(tnames)}')

            if on_tool_calls:
                on_tool_calls(tnames)

            for tc in response.tool_calls:
                result = execute_tool(tc, TOOL_HANDLERS)
                session.messages.append({
                    'role': 'tool',
                    'tool_call_id': tc['id'],
                    'content': str(result),
                })
        else:
            if response.content:
                print(f'[{_ts()}] Response generated ({len(response.content)} chars)')
                session.messages.append({'role': 'assistant', 'content': response.content})
                return response.content
            return None

    print(f'[{_ts()}] Max iterations ({max_iterations}) reached')
    session.messages.append({
        'role': 'assistant',
        'content': 'Max iterations reached without final answer.',
    })
    return None