#!/usr/bin/env python3
"""Script for updating data. """

import argparse
import json
import os
import subprocess
from datetime import datetime, timedelta, tzinfo

from tqdm import tqdm

from fetch_holidays import CustomJSONEncoder, fetch_holiday


class ChinaTimezone(tzinfo):
    """Timezone of china.  """

    def tzname(self, dt):
        return 'UTC+8'

    def utcoffset(self, dt):
        return timedelta(hours=8)

    def dst(self, dt):
        return timedelta()


__dirname__ = os.path.abspath(os.path.dirname(__file__))


def _file_path(*other):

    return os.path.join(__dirname__, *other)


def update_data(year: int) -> str:
    """Update and store data for a year.

    Args:
        year (int): Year

    Returns:
        str: Stored data path
    """

    filename = _file_path(f'{year}.json')
    with open(filename, 'w', encoding='utf-8', newline='\n') as f:
        json.dump(fetch_holiday(year), f,
                  indent=4,
                  ensure_ascii=False,
                  cls=CustomJSONEncoder)
    return filename


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--all', action='store_true')
    args = parser.parse_args()

    now = datetime.now(ChinaTimezone())

    filenames = []
    progress = tqdm(range(2014 if args.all else now.year, now.year + 2))
    for i in progress:
        progress.set_description(f'Updating {i} data')
        filename = update_data(i)
        filenames.append(filename)

    subprocess.run(['git', 'add', *filenames], check=True)
    diff = subprocess.run(['git', 'diff', '--stat', '--cached'],
                          check=True,
                          stdout=subprocess.PIPE,
                          encoding='utf-8').stdout
    if not diff:
        print('Already up to date.')
        return

    subprocess.run(
        ['git', 'commit', '-m', 'Update data [skip ci]\n\n' + diff], check=True)
    subprocess.run(['git', 'tag', now.strftime('%Y.%m.%d')], check=True)
    subprocess.run(['git', 'push'], check=True)
    subprocess.run(['git', 'push', '--tags'], check=True)


if __name__ == '__main__':
    main()