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

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

args = argv[1:]

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"] == "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]
    print(f"{host}{picked['Part'][0]['key']}?X-Plex-Token={token}")
    return 0

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