#!/usr/bin/env python3

import os
import sys
import unicodedata

import ox
import ox.api


def get_extension(api, oshash):
    r = api.findMedia({'query': {
        'conditions': [{'key': 'oshash', 'value': oshash}]
    }, 'keys': ['extension']})['data']
    return r['items'][0]['extension'].lower()


COLON = '\uf022'
COLON = ':'

if __name__ == '__main__':
    if len(sys.argv) > 1:
        target = sys.argv[1]
    else:
        target = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'Films')

    if not target.endswith('/'):
        target += '/'

    site = 'pandora.cinemusespace.com'
    api = ox.api.signin('https://%s/api/' % site)

    keep = []
    r = api.find({'range': [0, 10000], 'keys': ['id']})
    for i in r['data']['items']:
        item = api.get(id=i['id'], keys=['id', 'streams', 'title', 'director', 'year', 'instances'])['data']
        director = item.get('director', ['Unknown Director'])
        path = os.path.join('; '.join(director), '%s' % (item['title']))
        if 'year' in item:
            path += ' (%s)' % item['year']
        path = ox.decode_html(path)
        prefix = ox.decode_html(item['title'])

        path = path.replace(':', COLON)
        prefix = prefix.replace(':', COLON)

        keep.append(unicodedata.normalize('NFD', os.path.join(target, path, 'annotations.json')).lower())

        parts = []
        if len(item['streams']) == 1:
            id = item['streams'][0]
            ext = get_extension(api, id)
            name = '%s.%s' % (item['title'], ext)
            name = ox.decode_html(name)
            name = name.replace(':', COLON)
            part = 1
            parts.append([os.path.join(path, name), item['id'], id, part])
        else:
            part = 1
            for id in item['streams']:
                ext = get_extension(api, id)
                name = '%s.Part %d.%s' % (prefix, part, ext)
                name = ox.decode_html(name)
                name = name.replace(':', COLON)
                parts.append([os.path.join(path, name), item['id'], id, part])
                part += 1

        for path, id, oshash, part in parts:
            abspath = os.path.join(target, path)
            if os.path.exists(abspath) and ox.oshash(abspath) != oshash:
                print('file changed', abspath[len(target):])
                os.unlink(abspath)
            if not os.path.exists(abspath):
                url = 'https://%s/%s/download/source/%s' % (site, id, part)
                print('downloading', abspath[len(target):])
                try:
                    api.save_url(url, abspath)
                except:
                    print('failed to download', url)
            keep.append(unicodedata.normalize('NFD', abspath).lower())

    for root, folders, files in os.walk(target):
        for f in files:
            path = os.path.join(root, f)
            if unicodedata.normalize('NFD', path).lower() not in keep:
                print('deleting', path)
                os.unlink(path)