from requests import get
import os
from tqdm import tqdm
from concurrent.futures import ThreadPoolExecutor


def save_file(src, path):
    responses = get(src)
    if not responses.ok:
        return

    os.makedirs(os.path.dirname(path), exist_ok=True)
    with open(path, 'wb') as file:
        file.write(responses.content)


def load_file_multiple(iterable):
    src, path = iterable
    responses = get(src)
    if not responses.ok:
        return

    return path, responses.content


def save_file_from_data(path, data):
    os.makedirs(os.path.dirname(path), exist_ok=True)
    with open(path, 'wb') as file:
        file.write(data)


def download(src, download_path):
    print('프로젝트 설정 파일 다운로드 시작')
    save_file(f'{src}/dist/platform.json', f'{download_path}/platform.json')
    save_file(f'{src}/dist/imageSource.json', f'{download_path}/imageSource.json')
    save_file(f'{src}/dist/nodes/list.json', f'{download_path}/nodes/list.json')

    lineSettingList = get(f'{src}/dist/nodes/list.json').json()

    print('라인 다운로드 시작')
    for name in tqdm(lineSettingList):
        save_file(f'{src}/dist/nodes/{name}', f'{download_path}/nodes/{name}')

    print('라인 다운로드 완료. 이미지 다운로드 시작')
    image_list: list[str] = get(f'{src}/dist/imageSource.json').json().keys()
    image_list_tmp = set()
    for image in image_list:
        image_list_tmp.add(image)
        if image.endswith('.webp'):
            continue
        pos = image.rindex('.')
        image_list_tmp.add(image[:pos].strip() + '.webp')
    image_list = list(image_list_tmp)
    image_list: list[(str, str)] = list(map(lambda x: (f'{src}/dist/images/{x}', f'{download_path}/images/{x}'), image_list))

    with ThreadPoolExecutor(max_workers=10) as executor:
        results = list(tqdm(executor.map(load_file_multiple, image_list), total=len(image_list)))
        for result in results:
            if result is None:
                continue
            path, data = result
            save_file_from_data(path, data)


if __name__ == '__main__':
    print('다운로드 할 프로젝트 주소 입력')
    src = input().strip()
    print('컴퓨터의 다운로드 위치')
    download_path = input().strip()
    download(src, download_path)
    print('완료')