#!/usr/bin/env python3
"""
Turns a YouTube URL with timestamp into a Markdown link
"""

import argparse
import sys
import re
import json
from urllib.request import Request, urlopen
from urllib.error import URLError, HTTPError
import urllib.parse

def parse_timestamp(url: str) -> str:
    parsed = urllib.parse.urlparse(url)
    qs = urllib.parse.parse_qs(parsed.query)
    t_str = qs.get("t", [None])[0] or re.search(r"[?&]t=(\d+)s?", url)
    if t_str is None:
        return ""

    if isinstance(t_str, str):
        # came from query string
        seconds = int(t_str)
    else:
        # came from regex
        seconds = int(t_str.group(1))

    hh, mm = divmod(seconds, 3600)
    mm, ss = divmod(mm, 60)
    marker = f" ({hh:02d}:{mm:02d}:{ss:02d})"
    return marker

def get_title(url: str) -> str:
    params = urllib.parse.urlencode({"url": url, "format": "json"})
    oembed_url = f"https://www.youtube.com/oembed?{params}"
    req = Request(oembed_url, headers={"User-Agent": "Mozilla/5.0"})
    with urlopen(req, timeout=10) as resp:
        data = resp.read().decode("utf-8")
    info = json.loads(data)
    title = info.get("title")
    if not title:
        raise RuntimeError("oEmbed response missing title")
    return title

def get_video_id(url: str) -> str:
      video_id_match = re.search(r"(?:v=|youtu\.be/)([a-zA-Z0-9_-]{11})", url)
      if not video_id_match:
        sys.exit("Error: Could not extract video ID from URL")
      video_id = video_id_match.group(1)
      return video_id

def main() -> None:
    parser = argparse.ArgumentParser(prog="yt-md-link.py", description="Turn a YouTube URL with timestamp into a Markdown link or image.")
    parser.add_argument("url", help="YouTube URL with timestamp")
    parser.add_argument("-t", "--title", help="Override fetched title", metavar="STRING")
    parser.add_argument("-i", "--image", action="store_true", help="Output image markdown instead of a link")
    args = parser.parse_args()

    video_id = get_video_id(args.url)
    time_marker = parse_timestamp(args.url)
    title = args.title if args.title is not None else get_title(args.url)

    if args.image:
      print(f"[![{title}{time_marker}](https://img.youtube.com/vi/{video_id}/0.jpg)]({args.url})")
    else:
      print(f"[{title}{time_marker}]({args.url})")

if __name__ == "__main__":
    main()