#!/usr/bin/env python3
import requests
from sys import stderr, stdout, argv
from os import environ
from os.path import basename

host = environ.get("PLEX_HOST", "localhost")
token = environ.get("PLEX_TOKEN", "")

args = argv[1:]
WGET = False
if args and args[0] == "-g":
    args = args[1:]
    WGET = True

def prompt(text, prelude=None):
    global args
    if not args:
        if prelude:
            print(prelude, file=stderr)
        print(text, file=stderr, end="")
    else:
        picked = args.pop(0)
        return picked
    return input()

def get(endpoint):
    url = host + ("/" if endpoint[0] != "/" else "") + endpoint
    r = requests.get(url, params={"X-Plex-Token": token}, headers={"Accept": "application/json"})
    return r.json()

def main():
    libraries = [d for d in get("/library/sections")["MediaContainer"]["Directory"]]
    _ = []
    for n,l in enumerate(libraries):
        _.append(f"[{l['key']}] {l['title']}\n")
    choice = prompt("Pick library: ", prelude="\n".join(_))
    library = [l for l in libraries if l["key"] == choice][0]
    lib_list = [i for i in get(f"/library/sections/{library['key']}/all")["MediaContainer"]["Metadata"]]
    _ = []
    for n,i in enumerate(lib_list):
        _.append(f"[{n}] {i['title']}")
    picked = lib_list[int(prompt("Pick item: ", prelude="\n".join(_)))]

    if picked["type"] == "artist":
        tracks = [t for t in get(picked["key"].replace("children", "allLeaves"))["MediaContainer"]["Metadata"]]
        _ = []
        for n,t in enumerate(tracks):
            _.append(f"[{n}] {t['title']}")
        picked = tracks[int(prompt("Pick track: ", prelude="\n".join(_)))]
    elif picked["type"] == "show":
        episodes = [e for e in get(picked["key"].replace("children", "allLeaves"))["MediaContainer"]["Metadata"]]
        _ = []
        for n,e in enumerate(episodes):
            _.append(f"[{n}] S{e['parentIndex']:02d}E{e['index']:02d} {e['title']}")
        picked = episodes[int(prompt("Pick episode: ", prelude="\n".join(_)))]

    if len(picked["Media"]) > 1:
        files = [f for f in picked["Media"]]
        _ = []
        for n,f in enumerate(files):
            _.append(f"[{n}] {f['title']}")
        picked = files[int(prompt("Pick file: ", prelude="\n".join(_)))]
    else:
        picked = picked["Media"][0]

    url = f"{host}{picked['Part'][0]['key']}?X-Plex-Token={token}"
    if WGET:
        fn = basename(picked["Part"][0]["file"].replace("\\", "/"))
        print(f"wget -c -O '{fn}' {url}")
    else:
        print(url)
    return 0

if __name__ == "__main__":
    from sys import exit
    exit(main())