import os
import json
import requests


COMMENTS_LIMIT = 30
DEFAULT_ENDPOINT = "https://q6-p.hf.space"


def read_dotenv_value(path, key):
    try:
        with open(path, "r") as env_file:
            for line in env_file:
                line = line.strip()
                if not line or line.startswith("#") or "=" not in line:
                    continue
                k, v = line.split("=", 1)
                if k == key:
                    return v
    except FileNotFoundError:
        return None
    return None


def get_phpsessid():
    phpsessid = os.getenv("PHPSESSID")
    if phpsessid:
        return phpsessid
    env_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", ".env"))
    phpsessid = read_dotenv_value(env_path, "PHPSESSID")
    if phpsessid:
        return phpsessid
    raise RuntimeError("PHPSESSID is not set in the environment or .env")


def get_endpoint():
    endpoint = os.getenv("PIXIF_ENDPOINT")
    if endpoint:
        return endpoint.rstrip("/")
    env_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", ".env"))
    endpoint = read_dotenv_value(env_path, "PIXIF_ENDPOINT")
    if endpoint:
        return endpoint.rstrip("/")
    return DEFAULT_ENDPOINT


def fetch_comments(post_ids, phpsessid, endpoint, limit=COMMENTS_LIMIT):
    payload = {"post_ids": post_ids, "phpsessid": phpsessid, "limit": limit}
    response = requests.post(f"{endpoint}/comments", json=payload, timeout=300)
    response.raise_for_status()
    data = response.json()
    comments = data.get("comments", data)
    if not isinstance(comments, dict):
        raise RuntimeError("Unexpected response from comments endpoint.")
    return comments


def parse_indexes(inputs):
    indexes = []
    for inp in inputs:
        if "-" in inp:
            start, end = map(int, inp.split("-"))
            indexes.extend(range(start - 1, end))
        elif inp.isdigit():
            indexes.append(int(inp) - 1)
    return indexes


def main():
    os.chdir(os.path.dirname(os.path.abspath(__file__)))
    os.makedirs("comments", exist_ok=True)

    valid = [f for f in os.listdir() if f.endswith(".txt")]
    for idx, file in enumerate(valid):
        print(f"{idx + 1}: {file}")

    if not valid:
        print("No .txt files found in the Client directory.")
        return

    inputs = input("Enter the index of the file: ").split()
    indexes = parse_indexes(inputs)

    phpsessid = get_phpsessid()
    endpoint = get_endpoint()

    for index in indexes:
        if index < 0 or index >= len(valid):
            continue
        group_name = valid[index].rsplit(".", 1)[0]
        with open(valid[index], "r") as f:
            raw_ids = f.read().split()

        post_ids = [int(post_id) for post_id in raw_ids if post_id.isdigit()]
        if not post_ids:
            continue

        try:
            fetched = fetch_comments(post_ids, phpsessid, endpoint)
        except Exception as exc:
            print(f"Failed to fetch comments for {group_name}: {exc}")
            fetched = {}

        comments_by_post = {str(post_id): fetched.get(str(post_id), []) for post_id in post_ids}

        output_path = os.path.join("comments", f"{group_name}.json")
        with open(output_path, "w", encoding="utf-8") as f:
            json.dump(comments_by_post, f, indent=2, ensure_ascii=False)

        print(f"Saved comments to {output_path}")


if __name__ == "__main__":
    main()