#!/usr/bin/env python3
"""
bsky_post.py — Post text + optional image or video to a Bluesky instance.

Usage examples:
  python3 bsky_post.py "DIVENDRES!!!!" --video friday.mp4
  python3 bsky_post.py "Dijous!!!!"    --image thursday.jpg
  python3 bsky_post.py "Bon dia!"
  python3 bsky_post.py "Bon dia!"      --image photo.jpg --lang ca
"""

import argparse
import logging
import mimetypes
import os
import sys
import time
import random
import re
import base64
import json

from dataclasses import dataclass
from atproto import Client, client_utils, models


# ============================================================
# Config
# ============================================================
@dataclass(frozen=True)
class RetryConfig:
    login_max_attempts: int = 5
    login_base_delay_seconds: float = 10.0
    login_max_delay_seconds: float = 600.0
    login_jitter_seconds: float = 3


# ============================================================
# Logging
# ============================================================
def setup_logging() -> None:
    logging.basicConfig(
        format="%(asctime)s %(message)s",
        level=logging.INFO,
        stream=sys.stdout,
    )


# ============================================================
# Text builder
# ============================================================
def make_rich(content: str):
    text_builder = client_utils.TextBuilder()
    content = content.strip()
    lines = content.splitlines()

    for line_idx, line in enumerate(lines):
        if not line.strip():
            if line_idx < len(lines) - 1:
                text_builder.text("\n")
            continue

        words = line.split(" ")
        for i, word in enumerate(words):
            if not word:
                if i < len(words) - 1:
                    text_builder.text(" ")
                continue

            if word.startswith("http://") or word.startswith("https://"):
                text_builder.link(word, word)
            elif word.startswith("#") and len(word) > 1:
                tag_name = word[1:].rstrip(".,;:!?)'\"")
                if tag_name:
                    text_builder.tag(word, tag_name)
                else:
                    text_builder.text(word)
            else:
                text_builder.text(word)

            if i < len(words) - 1:
                text_builder.text(" ")

        if line_idx < len(lines) - 1:
            text_builder.text("\n")

    return text_builder


# ============================================================
# Error helpers
# ============================================================
def is_rate_limited_error(error_obj) -> bool:
    text = repr(error_obj).lower()
    return (
        "429" in text
        or "ratelimitexceeded" in text
        or "too many requests" in text
        or "rate limit" in text
    )


def is_auth_error(error_obj) -> bool:
    text = repr(error_obj).lower()
    return (
        "401" in text
        or "403" in text
        or "invalid identifier or password" in text
        or "authenticationrequired" in text
        or "invalidtoken" in text
    )


def is_network_error(error_obj) -> bool:
    text = repr(error_obj)
    signals = [
        "ConnectError",
        "RemoteProtocolError",
        "ReadTimeout",
        "WriteTimeout",
        "TimeoutException",
        "503",
        "502",
        "504",
        "ConnectionResetError",
    ]
    return any(sig in text for sig in signals)


def is_transient_error(error_obj) -> bool:
    error_text = repr(error_obj)
    transient_signals = [
        "InvokeTimeoutError",
        "ReadTimeout",
        "WriteTimeout",
        "TimeoutException",
        "RemoteProtocolError",
        "ConnectError",
        "503",
        "502",
        "504",
    ]
    return any(signal in error_text for signal in transient_signals)


def get_rate_limit_wait_seconds(error_obj, default_delay: float, max_delay: float) -> float:
    """
    Parse common rate-limit headers and return a bounded wait time in seconds.
    """
    try:
        now_ts = int(time.time())

        # Direct headers on exception
        headers = getattr(error_obj, "headers", None) or {}
        retry_after = headers.get("retry-after") or headers.get("Retry-After")
        if retry_after:
            return min(max(float(retry_after), 1.0), max_delay)

        x_after = headers.get("x-ratelimit-after") or headers.get("X-RateLimit-After")
        if x_after:
            return min(max(float(x_after), 1.0), max_delay)

        reset_value = headers.get("ratelimit-reset") or headers.get("RateLimit-Reset")
        if reset_value:
            wait_seconds = max(float(reset_value) - now_ts + 1.0, default_delay)
            return min(wait_seconds, max_delay)
    except Exception:
        pass

    try:
        # Nested response headers
        response = getattr(error_obj, "response", None)
        headers = getattr(response, "headers", None) or {}
        now_ts = int(time.time())

        retry_after = headers.get("retry-after") or headers.get("Retry-After")
        if retry_after:
            return min(max(float(retry_after), 1.0), max_delay)

        x_after = headers.get("x-ratelimit-after") or headers.get("X-RateLimit-After")
        if x_after:
            return min(max(float(x_after), 1.0), max_delay)

        reset_value = headers.get("ratelimit-reset") or headers.get("RateLimit-Reset")
        if reset_value:
            wait_seconds = max(float(reset_value) - now_ts + 1.0, default_delay)
            return min(wait_seconds, max_delay)
    except Exception:
        pass

    # repr fallback parsing
    text = repr(error_obj)
    m = re.search(r"'retry-after': '(\d+)'", text, re.IGNORECASE)
    if m:
        return min(max(float(m.group(1)), 1.0), max_delay)

    m = re.search(r"'x-ratelimit-after': '(\d+)'", text, re.IGNORECASE)
    if m:
        return min(max(float(m.group(1)), 1.0), max_delay)

    m = re.search(r"'ratelimit-reset': '(\d+)'", text, re.IGNORECASE)
    if m:
        now_ts = int(time.time())
        wait_seconds = max(float(m.group(1)) - now_ts + 1.0, default_delay)
        return min(wait_seconds, max_delay)

    return default_delay


# ============================================================
# Login with backoff
# ============================================================
def login_with_backoff(
    client: Client,
    username: str,
    password: str,
    service_url: str,
    max_attempts: int = 5,
    base_delay: float = 10.0,
    max_delay: float = 600.0,
    jitter: float = 1.5,
) -> bool:
    for attempt in range(1, max_attempts + 1):
        try:
            logging.info(
                f"🔑 Login attempt {attempt}/{max_attempts} → {service_url} as {username}"
            )
            client.login(username, password)
            logging.info("✅ Login successful.")
            return True

        except Exception as e:
            logging.exception("❌ Login exception")

            # Fail fast on invalid credentials
            if is_auth_error(e):
                logging.error("❌ Bad credentials. Check handle/password.")
                return False

            # Respect explicit rate-limit timing
            if is_rate_limited_error(e):
                if attempt < max_attempts:
                    wait = get_rate_limit_wait_seconds(e, default_delay=base_delay, max_delay=max_delay)
                    wait = wait + random.uniform(0, jitter)
                    logging.warning(
                        f"⏳ Rate-limited on login (attempt {attempt}/{max_attempts}). "
                        f"Retrying in {wait:.1f}s..."
                    )
                    time.sleep(wait)
                    continue

                logging.error("❌ Exhausted login retries due to rate limiting.")
                return False

            # Retry transient/network problems
            if is_network_error(e) or is_transient_error(e):
                if attempt < max_attempts:
                    wait = min(base_delay * attempt, max_delay) + random.uniform(0, jitter)
                    logging.warning(
                        f"⏳ Transient login error (attempt {attempt}/{max_attempts}). "
                        f"Retrying in {wait:.1f}s..."
                    )
                    time.sleep(wait)
                    continue

                logging.error("❌ Exhausted login retries after transient/network errors.")
                return False

            # Unknown errors: bounded retry anyway
            if attempt < max_attempts:
                wait = min(base_delay * attempt, max_delay) + random.uniform(0, jitter)
                logging.warning(
                    f"⏳ Unknown login error (attempt {attempt}/{max_attempts}). "
                    f"Retrying in {wait:.1f}s..."
                )
                time.sleep(wait)
                continue

    return False


# ============================================================
# Media upload
# ============================================================
def detect_mime_type(path: str) -> str:
    mime, _ = mimetypes.guess_type(path)
    if mime:
        return mime
    ext = os.path.splitext(path)[1].lower()
    fallbacks = {
        ".jpg": "image/jpeg", ".jpeg": "image/jpeg",
        ".png": "image/png",  ".gif": "image/gif",
        ".webp": "image/webp",
        ".mp4": "video/mp4",  ".mov": "video/quicktime",
        ".webm": "video/webm",
    }
    return fallbacks.get(ext, "application/octet-stream")


def upload_image(
    client: Client,
    image_path: str,
    alt_text: str = "",
) -> models.AppBskyEmbedImages.Image | None:
    try:
        mime = detect_mime_type(image_path)
        with open(image_path, "rb") as f:
            data = f.read()

        logging.info(f"🖼️  Uploading image: {image_path} ({len(data)/1024:.1f} KB, {mime})")
        response = client.upload_blob(data)
        logging.info("✅ Image uploaded successfully.")

        return models.AppBskyEmbedImages.Image(
            image=response.blob,
            alt=alt_text,
        )

    except Exception as e:
        logging.error(f"❌ Failed to upload image: {repr(e)}")
        return None

def upload_video_and_wait(
    client: Client, 
    video_data: bytes, 
    alt_text: str = ""
) -> models.AppBskyEmbedVideo.Main | None:
    
    # --- Helper classes to mock the SDK's internal Session object ---
    class MockPayload:
        def __init__(self, exp):
            self.exp = exp

    class VideoSession:
        def __init__(self, token, handle, did):
            self.access_jwt = token
            self.refresh_jwt = ""
            self.handle = handle
            self.did = did
            
            # Default to 1 hour from now if decoding fails
            exp_time = int(time.time()) + 3600 
            
            try:
                parts = token.split('.')
                if len(parts) == 3:
                    payload = parts[1]
                    # Pad base64 string if necessary
                    payload += '=' * (-len(payload) % 4)
                    decoded = json.loads(base64.urlsafe_b64decode(payload).decode('utf-8'))
                    if 'exp' in decoded:
                        exp_time = decoded['exp']
            except Exception:
                pass
                
            # Assign as an object so the SDK can call .exp
            self.access_jwt_payload = MockPayload(exp_time)

    try:
        logging.info("🎬 Requesting Service Auth for Video Upload...")
        
        # 1. Get a Service Auth token from your home PDS
        VIDEO_SERVICE_DID = "did:web:video.bsky.app"
        auth_response = client.com.atproto.server.get_service_auth({
            'aud': VIDEO_SERVICE_DID
        })
        service_auth_token = auth_response.token

        logging.info("🎬 Uploading video to Bluesky Video Service...")
        
        # 2. Create a temporary client pointing to the dedicated video service
        video_client = Client(base_url="https://video.bsky.app")
        
        # 3. Inject the Service Auth token using our mock session
        video_client._session = VideoSession(
            token=service_auth_token, 
            handle=client.me.handle, 
            did=client.me.did
        )

        # 4. Upload the video to the official service
        response = video_client.app.bsky.video.upload_video(video_data)
        job_id = response.job_id
        
        logging.info(f"⏳ Video uploaded! Job ID: {job_id}. Waiting for processing...")
        
        # 5. Poll the job status using the video client
        while True:
            status_resp = video_client.app.bsky.video.get_job_status({'job_id': job_id})
            state = status_resp.job_status.state
            
            if state == 'JOB_STATE_COMPLETED':
                logging.info("✅ Processing complete! Waiting 10s for CDN propagation...")
                time.sleep(10) 
                
                return models.AppBskyEmbedVideo.Main(
                    video=status_resp.job_status.blob,
                    alt=alt_text
                )
            elif state == 'JOB_STATE_FAILED':
                logging.error("❌ Video processing failed on Bluesky's servers.")
                return None
                
            logging.info("   ...still processing...")
            time.sleep(3) 

    except Exception as e:
        logging.error(f"❌ Failed to upload/process video: {repr(e)}")
        return None

# ============================================================
# Post
# ============================================================
def post_to_bsky(
    client: Client,
    text: str,
    langs: list[str],
    image_path: str | None = None,
    video_path: str | None = None,
    alt_text: str = "",
    password: str = "",
) -> bool:
    rich_text = make_rich(text)

    try:
        # --- VIDEO POSTING ---
        if video_path:
            logging.info(f"🎬 Preparing video upload: {video_path}")
            with open(video_path, "rb") as f:
                video_data = f.read()
            
            # Pass the password to our custom polling function
            # Use our custom polling function (no password needed)
            video_embed = upload_video_and_wait(client, video_data, alt_text)

            
            if not video_embed:
                logging.error("❌ Aborting post: video upload/processing failed.")
                return False
                
            logging.info(f"🚀 Sending video post...")
            result = client.send_post(
                text=rich_text, 
                embed=video_embed, 
                langs=langs
            )

        # --- IMAGE POSTING ---
        elif image_path:
            image = upload_image(client, image_path, alt_text=alt_text)
            if not image:
                logging.error("❌ Aborting post: image upload failed.")
                return False
            
            embed = models.AppBskyEmbedImages.Main(images=[image])
            logging.info(f"🚀 Sending image post...")
            result = client.send_post(text=rich_text, embed=embed, langs=langs)

        # --- TEXT ONLY POSTING ---
        else:
            logging.info(f"🚀 Sending text post...")
            result = client.send_post(text=rich_text, langs=langs)

        uri = getattr(result, "uri", None)
        logging.info(f"✅ Post published! URI: {uri}")
        return True

    except Exception as e:
        logging.error(f"❌ Failed to send post: {repr(e)}")
        return False


# ============================================================
# CLI
# ============================================================
def main():
    setup_logging()

    parser = argparse.ArgumentParser(
        description="Post text + optional image or video to a Bluesky instance."
    )
    parser.add_argument("text",        help="Post text content")
    parser.add_argument("--username",  required=True, help="Bluesky handle or email")
    parser.add_argument("--password",  required=True, help="Bluesky app password")
    parser.add_argument("--service",   default="https://bsky.social", help="Bluesky PDS URL")
    parser.add_argument("--lang",      default="ca", help="Comma-separated language codes (e.g. ca,es)")
    parser.add_argument("--image",     default=None, help="Path to image file to attach")
    parser.add_argument("--video",     default=None, help="Path to video file to attach")
    parser.add_argument("--alt",       default="", help="Alt text for media")

    args = parser.parse_args()

    client = Client(base_url=args.service)
    success = login_with_backoff(
        client,
        args.username,
        args.password,
        args.service,
        max_attempts=RetryConfig.login_max_attempts,
        base_delay=RetryConfig.login_base_delay_seconds,
        max_delay=RetryConfig.login_max_delay_seconds,
        jitter=RetryConfig.login_jitter_seconds,
    )
    if not success:
        sys.exit(1)

    langs = [l.strip() for l in args.lang.split(",") if l.strip()]
    post_success = post_to_bsky(
        client,
        text=args.text,
        langs=langs,
        image_path=args.image,
        video_path=args.video,
        alt_text=args.alt,
        password=args.password,
    )

    if not post_success:
        sys.exit(1)


if __name__ == "__main__":
    main()