#!/usr/bin/env python
# License: GPLv3 Copyright: 2016, Kovid Goyal <kovid at kovidgoyal.net>

import argparse
import glob
import hashlib
import json
import os
import subprocess

duplicates = {
    'character-set': ['languages'],
    'calibre': ['library', 'lt', 'favicon-192', 'favicon-512'],
    'format-text-color': ['lookfeel'],
    'books_in_series': ['series'],
    'plugins.svg': ['plugins/plugin_upgrade_ok'],
}

sizes = {
    'lt': '256',
    'library': '1024',
    'default_cover': 'original',
    'viewer': '256',
    'tweak': '256',
    'favicon-192': '192',
    'favicon-512': '512',
}

skip = {'calibre'}

j = os.path.join
base = os.path.dirname(os.path.abspath(__file__))
output_base = j(os.path.dirname(base), 'resources', 'images')
hash_path = j(os.path.dirname(base), '.build-cache', 'imgsrc-gen.json')
if os.path.exists(hash_path):
    with open(hash_path, 'rb') as f:
        hashes = json.load(f)
else:
    hashes = {}
src_hashes = {}


def iterfiles(only=()):
    for src in glob.glob(j(base, '*.svg')) + glob.glob(j(base, 'plugins/*.svg')):
        name = os.path.relpath(src, base).rpartition('.')[0]
        if only and name not in only:
            continue
        src_hashes[name] = h = hashlib.sha1(open(src, 'rb').read()).hexdigest()
        if not only and h == hashes.get(name):
            continue
        output_names = [n for n in [name] + duplicates.get(name, []) if n not in skip]
        output_files = [j(output_base, n) + '.png' for n in output_names]
        if output_files:
            yield src, output_files


def rsvg(src, size, dest):
    cmd = ['rsvg-convert', '-d', '96', '-p', '96']
    if size != 'original':
        cmd += ['--width', size, '--height', size]
    subprocess.check_call(cmd + ['-o', dest, src])
    cmd = 'zopflipng -m -y --lossy_8bit --lossy_transparent'.split()
    subprocess.check_call(cmd + [dest, dest])


def render(src, output_files):
    for dest in output_files:
        oname = os.path.basename(dest).rpartition('.')[0]
        size = sizes.get(oname, '128')
        print('Rendering', oname, 'at size:', size)
        rsvg(src, size, dest)
        name = os.path.relpath(src, base).rpartition('.')[0]
        hashes[name] = src_hashes[name]


def main():
    p = argparse.ArgumentParser()
    p.add_argument('only', nargs='*', default=[], help='Only render the specified icons')
    args = p.parse_args()
    for src, ofiles in iterfiles(args.only):
        render(src, ofiles)
    with open(hash_path, 'w') as f:
        json.dump(hashes, f, indent=2, sort_keys=True)
    subprocess.check_call(['inkscape', os.path.join(base, 'calibre.svg'),
                           '--export-type=svg', '--export-plain-svg', '--export-filename',
                           os.path.join(output_base, 'calibre.svg')])


if __name__ == '__main__':
    main()
