#!/usr/bin/env python3
import os
import subprocess
import sys
from pathlib import Path

import click


RUNS_DIR = Path('/data/ofm/http_host/runs')


@click.command()
def cli():
    if not Path('/etc/fstab').exists():
        sys.exit('Needs to be run on Linux')

    if os.geteuid() != 0:
        sys.exit('Needs sudo')

    if not Path('/mnt/ofm').exists():
        sys.exit('mounter.py needs to be run first')

    with open(Path(__file__).parent / 'nginx_template_cf.conf') as fp:
        nginx_template = fp.read()

    location_block_str = ''
    curl_text = ''

    for subdir in Path('/mnt/ofm').iterdir():
        if not subdir.is_dir():
            continue

        area, version = subdir.name.split('-')

        run_dir = RUNS_DIR / area / version
        if not run_dir.is_dir():
            print(f"{run_dir} doesn't exists, skipping")
            continue

        tilejson_path = run_dir / 'tilejson-tiles-com.json'

        metadata_path = subdir / 'metadata.json'
        if not metadata_path.is_file():
            print(f"{metadata_path} doesn't exists, skipping")
            continue

        url_prefix = f'https://tiles.openfreemap.com/{area}/{version}'

        subprocess.run(
            [
                sys.executable,
                Path(__file__).parent.parent / 'metadata_to_tilejson.py',
                '--minify',
                metadata_path,
                tilejson_path,
                url_prefix,
            ],
            check=True,
        )

        # TODO raise expires if everything is stable
        version_str = f"""
            location /{area}/{version} {{    # no trailing hash
                alias {tilejson_path};       # no trailing hash
                default_type application/json;

                add_header 'Access-Control-Allow-Origin' '*' always;
                add_header Cache-Control public;
                expires 1d;
            }}

            location /{area}/{version}/ {{    # trailing hash
                alias {subdir}/tiles/;        # trailing hash
                try_files $uri @empty;

                add_header Content-Encoding gzip;
                add_header 'Access-Control-Allow-Origin' '*' always;
                add_header Cache-Control public;
                expires 1d;  # target 10y
            }}
            """

        location_block_str += version_str

        if not curl_text:
            curl_text = (
                '\ntest with:\n'
                f'curl -H "Host: ofm" -I http://localhost/{area}/{version}/14/8529/5975.pbf\n'
                f'curl -I https://tiles.openfreemap.com/{area}/{version}/14/8529/5975.pbf'
            )

    nginx_template = nginx_template.replace('___LOCATION_BLOCKS___', location_block_str)

    with open('/data/nginx/sites/ofm-tiles-com.conf', 'w') as fp:
        fp.write(nginx_template)
        print('nginx config written')

    subprocess.run(['nginx', '-t'], check=True)
    subprocess.run(['systemctl', 'reload', 'nginx'], check=True)

    print(curl_text)


if __name__ == '__main__':
    cli()