#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import argparse
import os
import shutil
import sys
import tempfile
import urllib.request
from pathlib import Path

################################################################################################
# todo :

################################################################################################


################################################################################################
# Constants
#
################################################################################################
BLOG_MASTER_BRANCH_README_URL = 'https://raw.githubusercontent.com/kenshin579/advenoh.pe.kr/master/README.md'
LOCAL_README_FILE = '../README.md'

################################################################################################
# Functions
#
################################################################################################

def save_file(url, download_folder):
    download_filename = 'blog_README.md'
    mem = urllib.request.urlopen(url).read()
    full_path = os.path.join(download_folder, download_filename)
    with open(full_path, mode="wb") as tmp_dir:
        tmp_dir.write(mem)
    print("saved file")

    return full_path

def update_readme():
    tmp_dir = tempfile.TemporaryDirectory(dir="/tmp")
    print('tmp dir', tmp_dir.name)
    remote_file = save_file(BLOG_MASTER_BRANCH_README_URL, tmp_dir.name)
    print('remote_file', remote_file)
    remote_file_size = Path(remote_file).stat().st_size
    local_file_size = Path(LOCAL_README_FILE).stat().st_size
    if remote_file_size > local_file_size:
        # shutil.copy(remote_file, LOCAL_README_FILE)
        with open(remote_file) as f:
            lines = f.readlines()
            lines[0] = "[![HitCount](http://hits.dwyl.com/kenshin579/tutorials-java.svg)](http://hits.dwyl.com/kenshin579/tutorials-java)\n"
        with open(LOCAL_README_FILE, "w") as f:
            f.writelines(lines)

################################################################################################
# Main function
#
################################################################################################

def main():
    parser = argparse.ArgumentParser(description="Maintenance script for tutorials-java")

    parser.add_argument("-u", "--update", action='store_true',
                        help="Update README file from blog branch")

    if len(sys.argv[1:]) == 0:
        parser.print_help()
        parser.exit()

    args = parser.parse_args()

    print('args', args)

    if args.update:
        update_readme()

if __name__ == "__main__":
    sys.exit(main())