import asyncio
import os

from api_client import get_phpsessid
from pixiv_api import headers, sanitize_filename, search

ROOT_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", ".."))
PAGES = 300 // 60

input_url = input("Enter the URL: ")

def main() -> int:
    phpsessid = get_phpsessid()
    pixiv_cookies = {"PHPSESSID": phpsessid}
    try:
        post_ids, filename = asyncio.run(search(
            input_url,
            PAGES,
            ai_only=False,
            real_only=True,
            cookies=pixiv_cookies,
            headers=headers,
        ))
    except Exception as exc:
        print(f"Error: {exc}")
        return 1
    filename = sanitize_filename(filename)
    with open(os.path.join(ROOT_DIR, f"{filename}.txt"), "w", encoding="utf-8") as f:
        f.write("\n".join(map(str, post_ids)))
    return 0

if __name__ == "__main__":
    raise SystemExit(main())