import json
from pydantic import BaseModel
from browser_use import (
    Agent,
    Controller,
)
from lib.agents.run_agent import run_agent
from lib.utils.logger import logger
from lib.browser_use_utils.clean_resources import clean_agent_resources
from lib.browser_use_utils.create_google_ai import create_google_ai
from lib.config import GOOGLE_MODEL, GOOGLE_PLANNER_MODEL

NOT_FOUND_LOGIN_PAGE = 0
FOUND_LOGIN_PAGE = 1

class IsFound(BaseModel):
    status: int

async def find_login_page(target_url, session):
    initial_actions = [{"open_tab": {"url": target_url}}]
    task = "Navigate to the login page, and stop"
    extend_planner_system_message = "You are an expert in finding login pages. Your task is to navigate to the login page of the given URL and stop there."

    controller = Controller(output_model=IsFound, exclude_actions=['search_google'])
    agent = Agent(
        browser_session=session,
        initial_actions=initial_actions,
        task=task,
        llm=create_google_ai(GOOGLE_MODEL),
        planner_llm=create_google_ai(GOOGLE_PLANNER_MODEL),
        controller=controller,
        extend_planner_system_message=extend_planner_system_message,
    )

    status, final_result = await run_agent(agent)
    if status:
        logger(f"⚠️ 스캔 실패: {target_url} | {final_result}")
        print(f"⚠️ 스캔 실패: {target_url} | {final_result}")
        return False, None; 
    
    data = json.loads(final_result)
    try:
        is_found = IsFound(**data)
        if is_found.status == NOT_FOUND_LOGIN_PAGE:
            return False, "로그인 페이지를 찾을 수 없습니다."
        else:
            return True, "로그인 페이지를 찾았습니다."
    except Exception as e:
        logger(f"⚠️ 결과 파싱 실패: {target_url} | {e}\n원본 결과: {final_result}")
        print(f"⚠️ 결과 파싱 실패: {target_url} | {e}\n원본 결과: {final_result}")
        return False, "결과 파싱 실패"