#!/usr/bin/env python3

import os
import re
import json
import sys
from io import open
from time import time
from shutil import rmtree
from collections import OrderedDict

if len(sys.argv) == 1 or not sys.argv[1]:
    raise SystemExit('Build dir missing.')


def mkdirs(path):
    try:
        os.makedirs(path)
    finally:
        return os.path.exists(path)

pj = os.path.join
build_dir = os.path.abspath(sys.argv[1])

description = ''

# locales
locale_dir = pj(build_dir, '_locales')

for alpha2 in sorted(os.listdir(locale_dir)):
    locale_path = pj(locale_dir, alpha2, 'messages.json')
    with open(locale_path, encoding='utf-8') as f:
        string_data = json.load(f, object_pairs_hook=OrderedDict)

    if alpha2 == 'en':
        description = string_data['extShortDesc']['message']

    for string_name in string_data:
        string_data[string_name] = string_data[string_name]['message']

    rmtree(pj(locale_dir, alpha2))

    alpha2 = alpha2.replace('_', '-')
    locale_path = pj(locale_dir, alpha2 + '.json')

    mkdirs(pj(locale_dir))

    with open(locale_path, 'wb') as f:
        f.write(json.dumps(string_data, ensure_ascii=False).encode('utf8'))


# update Info.plist
proj_dir = pj(os.path.split(os.path.abspath(__file__))[0], '..')
chromium_manifest = pj(proj_dir, 'platform', 'chromium', 'manifest.json')

with open(chromium_manifest, encoding='utf-8') as m:
    manifest = json.load(m)

manifest['buildNumber'] = int(time())
manifest['description'] = description

# Fetch extension version
with open(os.path.join(proj_dir, 'dist', 'version')) as f:
    version = f.read().strip()
manifest['version'] = version

# Development build? If so, modify name accordingly.
match = re.search('^\d+\.\d+\.\d+\.\d+$', version)
if match:
    dev_build = ' dev build'
    manifest['name'] += dev_build
    manifest['short_name'] += dev_build
    manifest['browser_action']['default_title'] += dev_build

info_plist = pj(build_dir, 'Info.plist')

with open(info_plist, 'r+t', encoding='utf-8', newline='\n') as f:
    info_plist = f.read()
    f.seek(0)

    f.write(info_plist.format(**manifest))

# update Update.plist
update_plist = pj(proj_dir, 'platform', 'safari', 'Update.plist')
update_plist_build = pj(build_dir, '..', os.path.basename(update_plist))

with open(update_plist_build, 'wt', encoding='utf-8', newline='\n') as f:
    with open(update_plist, encoding='utf-8') as u:
        update_plist = u.read()

    f.write(update_plist.format(**manifest))
