import argparse
from Downloader import Downloader
from Editer import Editer
import os
import shutil

def parse_args():
    """Parse input arguments."""
    parser = argparse.ArgumentParser(description='config')
    parser.add_argument('--comic_no', default='0000', type=str)
    parser.add_argument('--volume_no', default='1', type=int)
    parser.add_argument('--no_input', default=False, type=bool)
    args = parser.parse_args()
    return args


def query_chaps(comic_name, url_prev, is_gui=False, hang_signal=None, edit_line_hang=None):
    print('未输入卷号，将返回书籍目录信息......')
    editer = Downloader(comic_name=comic_name, root_path='./out', url_prev=url_prev)
    print('*******************************')
    editer.get_comic_msg(is_gui, hang_signal, edit_line_hang) 
    editer.get_comic_chaps()
    print(editer.comic_title, editer.comic_author)
    print('*******************************')
    
    for i, chap_name in enumerate(editer.chap_name_list):
        print(f'[{str(i+1)}]', chap_name)

    print('*******************************')
    print('请输入所需要的卷号进行下载（多卷可以用英文逗号分隔或直接使用连字符，详情见说明）')

def download_task(root_path,
                comic_name,
                chap_no_list,
                url_prev,
                high_quality,
                is_gui=False,
                multi_thread=False,
                hang_signal=None,
                progressring_signal=None,
                cover_signal=None,
                edit_line_hang=None):
    
    downloader = Downloader(comic_name=comic_name, root_path=root_path, url_prev=url_prev, high_quality=high_quality)
    print('正在积极地获取书籍信息....')
    downloader.get_comic_msg(is_gui, hang_signal, edit_line_hang) 
    downloader.get_comic_chaps()
    print(downloader.comic_title, downloader.comic_author)
    
    print('****************************')
    print('正在下载漫画....')
    for chap_no in chap_no_list:
        chap_name = downloader.chap_name_list[chap_no-1]
        chap_uuid = downloader.chap_uuid_list[chap_no-1]
        page_num = downloader.chap_pagenum_list[chap_no-1]
        downloader.download_single_chap(chap_name, chap_uuid, page_num, multithread=multi_thread, is_gui=is_gui, signal=progressring_signal)
        downloader.get_cover(chap_name=chap_name, is_gui=is_gui, signal=cover_signal)
    downloader.download_cover()
    print('漫画下载成功！', f'漫画路径【{downloader.comic_path}】')
    chap_list = [downloader.chap_name_list[chap_no-1] for chap_no in chap_no_list]
    editer = Editer(downloader.comic_title, downloader.comic_author, chap_list, downloader.comic_path, root_path, delete_comic=0)
    editer.pack_img()
    editer.typesetting()
    editer.get_epub()
    
    

def downloader_router(root_path,
                      comic_name,
                      chap_no,
                      url_prev,
                      high_quality, 
                      is_gui=False, 
                      multi_thread=False,
                      hang_signal=None,
                      progressring_signal=None,
                      cover_signal=None,
                      edit_line_hang=None):
    if len(comic_name)==0:
        print('请检查输入是否完整正确！')
        return
    elif chap_no == '':
        query_chaps(comic_name, url_prev, is_gui, hang_signal, edit_line_hang)
        return 
    elif chap_no.isdigit():
        chap_no = int(chap_no)
        chap_no_list = [chap_no]
        if chap_no<=0:
            print('请检查输入是否完整正确！') 
            return
    elif "-" in chap_no:
        start, end = map(str, chap_no.split("-"))
        if start.isdigit() and end.isdigit() and int(start)>0 and int(start)<int(end):
            chap_no_list = list(range(int(start), int(end) + 1))
        else:
            print('请检查输入是否完整正确！')
            return
    elif "," in chap_no:
        chap_no_list = [num for num in chap_no.split(",")]
        if all([num.isdigit() for num in chap_no_list]):
            chap_no_list = [int(num) for num in chap_no_list] 
        else:
            print('请检查输入是否完整正确！')
            return
    else:
            print('请检查输入是否完整正确！')
            return
    download_task(root_path, comic_name, chap_no_list, url_prev, high_quality, is_gui, multi_thread, hang_signal, progressring_signal, cover_signal, edit_line_hang)
    print('所有下载任务都已经完成！')
    
if __name__=='__main__':
    args = parse_args()
    download_path = os.path.join(os.path.expanduser('~'), 'Downloads')

    if args.no_input:
        downloader_router(root_path='out', comic_name=args.comic_no, chap_no=args.volume_no)
    else:
        while True:
            args.comic_name = input('请输入书籍号：')
            args.volume_no = input('请输入卷号(查看目录信息不输入直接按回车，下载多卷请使用逗号分隔或者连字符-)：')
            # args.comic_name = 'xinglingganying'
            # args.volume_no = '40'
            downloader_router(root_path='out', comic_name=args.comic_name, chap_no=args.volume_no, url_prev='.tv', high_quality=True, multi_thread=True)
            # exit(0)