import json
import os

COOKIE_PATH = os.path.join(os.path.dirname(__file__), '..', 'conf', 'cookies.json')

def save_cookies(driver):
    cookies = driver.get_cookies()
    with open(COOKIE_PATH, 'w', encoding='utf-8') as f:
        json.dump(cookies, f, indent=2)

def load_cookies(driver, url='https://asp.upsolution.co.kr/'):
    if not os.path.exists(COOKIE_PATH):
        raise FileNotFoundError("쿠키 파일이 존재하지 않습니다. 먼저 수동 로그인 후 쿠키를 저장해주세요.")
    
    driver.get(url)
    with open(COOKIE_PATH, 'r', encoding='utf-8') as f:
        cookies = json.load(f)
    for cookie in cookies:
        # domain 필드는 문제가 될 수 있으므로 제거
        cookie.pop('domain', None)
        driver.add_cookie(cookie)