import argparse

from converter.converter import WebtoonConverter
from data.kakao_cookie import COOKIE
from data.special_list import WEBTOON_NOT_PROCESSED
from data.path_constant import DOWNLOAD_DIR, DOWNLOAD_LIST_TXT, TEMP_DOWNLOAD_DIR, TEMP_DOWNLOAD_LIST_TXT
from downloaders.kakao_webtoon import KakaoWebtoon
from helper.get_kakao_list import get_kakao_id_list
from helper.prerequisite import get_download_list
from downloaders.webtoon_com import Webtoon

DOWNLOAD_WEBTOON = False
CONVERT_ALL = False

task = 'dc'
kakao_list = ""

kakao_ids = get_kakao_id_list(kakao_list)

valid_cookies = []
new_webtoons = []

def download():
    url_list = get_download_list(DOWNLOAD_LIST_TXT)
    
    for url in url_list:
        webtoon = None
        if 'webtoon.kakao.com' in url: 
            webtoon_id = url.split('/')[-1]
            if webtoon_id in kakao_ids:
                webtoon = KakaoWebtoon(webtoon_id, COOKIE)
                webtoon.download_webtoon(url, DOWNLOAD_DIR)    
        elif DOWNLOAD_WEBTOON and 'www.webtoons.com' in url:
            webtoon_id = url.split('=')[1]
            webtoon = Webtoon(webtoon_id)
            webtoon.download_webtoon(url, DOWNLOAD_DIR)
        if webtoon is not None and webtoon.new_webtoon != "":
            new_webtoons.append(webtoon.new_webtoon)
    print(new_webtoons)

    # temp_url_list = get_download_list(TEMP_DOWNLOAD_LIST_TXT)
    # for temp_url in temp_url_list:
    #     if 'webtoon.kakao.com' in temp_url: 
    #         webtoon_id = temp_url.split('/')[-1]
    #         webtoon = KakaoWebtoon(webtoon_id, COOKIE)
    #         webtoon.download_webtoon(temp_url, TEMP_DOWNLOAD_DIR)  

def convert():
    for webtoon_path in DOWNLOAD_DIR.iterdir():
        if len(new_webtoons) > 0:
            if webtoon_path.is_dir() and webtoon_path.name in new_webtoons:
                print(webtoon_path)
                converter = WebtoonConverter(webtoon_path)
                converter.do_convert()
        elif webtoon_path.is_dir() and CONVERT_ALL and webtoon_path.name not in WEBTOON_NOT_PROCESSED:
            print(webtoon_path)
            converter = WebtoonConverter(webtoon_path)
            converter.do_convert()

def main():
    parser = argparse.ArgumentParser(description="Run download or convert")
    parser.add_argument('function', nargs='?', choices=['download', 'convert'], help="Function to run")

    args = parser.parse_args()

    if args.function == 'download':
        download()
    elif args.function == 'convert':
        convert()
    else:
        download()
        convert()
    
if __name__ == "__main__":
    if 'd' in task:
        download()
    if 'c' in task:
        # new_webtoons.append('1995青春報告')
        # new_webtoons.append('Unsleep')
        # new_webtoons.append('鬼夜曲')
        # new_webtoons.append('Backlight')
        # new_webtoons.append('鄰居是公會成員')
        # new_webtoons.append('No Moral')
        # new_webtoons.append('披薩外送員與黃金宮')
        # new_webtoons.append('PAYBACK')
        # new_webtoons.append('融冰曲線')
        # new_webtoons.append('夢龍傳')
        # new_webtoons.append('棋子的世界')
        # new_webtoons.append('監禁倉庫')
        # new_webtoons.append('易地思之')
        convert()
    print('MyWebtoon')