import json
import threading
from datetime import datetime
import config
from scripts import ntro


def _add_msg(app, role, content, **kwargs):
    msg = {"role": role, "content": content}
    msg.update(kwargs)
    app.messages.append(msg)
    if app.current_session:
        app.session_mgr.add_message(
            app.current_session.doc_id, role, content, **kwargs
        )

WELCOME_ART = """
 __   __  _______  __    _  ______   ______    ___   ___   _ 
|  | |  ||       ||  |  | ||      | |    _ |  |   | |   | | |
|  |_|  ||    ___||   |_| ||  _    ||   | ||  |   | |   |_| |
|       ||   |___ |       || | |   ||   |_||_ |   | |      _|
|       ||    ___||  _    || |_|   ||    __  ||   | |     |_ 
|   _   ||   |___ | | |   ||       ||   |  | ||   | |    _  |
|__| |__||_______||_|  |__||______| |___|  |_||___| |___| |_|
"""
"""
╔══════════════════════════════════════════╗
║                                          ║
║     /\\_/\\                                ║
║    ( o.o )      HENDRIK                  ║
║     > ^ <                                ║
║    (     )      AI Agent                 ║
║     (___)                                ║
║                                          ║
╚══════════════════════════════════════════╝
"""


def log(app, role, text):
    app.log.append({
        "role": role,
        "text": text,
        "time": datetime.now().strftime("%H:%M"),
    })


def submit(app, stdscr):
    query = "\n".join(app.input_buffer).strip()
    if not query:
        return

    log(app, "user", query)
    model_info = (
        config.resolve_provider(app.llm.base_url, app.llm.model),
        app.llm.model
    )
    if app.log:
        app.log[-1]["model_info"] = model_info
    app.input_buffer = [""]
    app.input_line = 0
    app.input_col = 0
    app.scroll = 999999
    app.processing = True

    _add_msg(app, "user", query)

    app.agent_done.clear()
    app.agent_thread = threading.Thread(
        target=_agent_loop,
        args=(app,),
        daemon=True,
    )
    app.agent_thread.start()


def _agent_loop(app):
    stamp = ntro.start()

    for step in range(app.agent_max_iterations):
        stamp_step = ntro.start()
        log(app, "system", f"  step {step + 1} \u2014 Thinking...")
        app.scroll = 999999

        response = app.llm.chat(app.messages, tools=app.TOOLS)

        app.log.pop()

        if response.warning:
            log(app, "system", f"  {response.warning}")

        if response.tool_calls:
            _add_msg(app, "assistant", response.content, tool_calls=response.tool_calls)
            if response.content and response.content.strip():
                log(app, "ai", response.content)
                app.scroll = 999999
            for tc in response.tool_calls:
                tname = tc["function"]["name"]
                targs = tc["function"]["arguments"]
                log(app, "tool_call", json.dumps({
                    "name": tname,
                    "arguments": targs,
                }))
                app.scroll = 999999
                execute_tool(app, tc)
        else:
            if response.content:
                _add_msg(app, "assistant", response.content)
                log(app, "ai", response.content)
            log(app, "sep", "")
            ntro.end(stamp)
            app.agent_done.set()
            return
        ntro.end(stamp_step)

    log(app, "error", "Max iterations reached without final answer.")
    _add_msg(app, "assistant", "Max iterations reached without final answer.")
    ntro.end(stamp)
    app.agent_done.set()


def execute_tool(app, tool_call):
    tname = tool_call["function"]["name"]
    targs = json.loads(tool_call["function"]["arguments"])
    handler = app.TOOL_HANDLERS.get(tname)

    if not handler:
        result = f"Tool {tname} not found"
    else:
        try:
            if tname == "search_code":
                result = handler(
                    pattern=targs["pattern"],
                    search_type=targs["search_type"],
                    path=targs.get("path", "."),
                )
            elif tname == "git_operation":
                result = handler(args=targs["args"])
            else:
                result = handler(**targs)
        except Exception as e:
            result = f"Error executing tool: {str(e)}"

    _add_msg(app, "tool", str(result), tool_call_id=tool_call["id"])