import re
import time
import os

from handlers.pixiv.comb import extract_pixiv_info, get_prefix_name_on_entry
from handlers.pixiv.Webdav import Webdav
from handlers.pixiv.request import downloadPic
from handlers.pixiv.file import save_file, remove_file


webdav = Webdav({
    'webdav_hostname': os.getenv('webdav_hostname'),
    'webdav_login': os.getenv('webdav_login'),
    'webdav_password': os.getenv('webdav_password'),
    'disable_check': True,
    'default_upload_path': os.getenv('default_upload_path')
})

# 处理结果
def pixiv_result_handler(entries):
    list = extract_pixiv_info(entries)
    
    success_entries = []
    failed_entries = []
    
    for item in list:
        prefix_name = get_prefix_name_on_entry(item)
        
        result_flag = True
        
        need_sync_files = []

        for url in item['link']:
            file_name_pattern = re.compile(r'\/(\w*\.(?:jpg|png))$')
            file_name = ','.join(re.findall(file_name_pattern, url))
            
            if file_name:
                full_name = f'{prefix_name} {file_name}'
                # 替换不符合文件名规范的字符
                full_name = re.sub(r'[\/\\\:\*\?\"\<\>\|]', "_", full_name)
                (status, data) = downloadPic(url)
                if status:
                    save_file(full_name, data)
                    need_sync_files.append({ 'id': item['id'], 'file_name': full_name })
                else:
                    result_flag = False
                time.sleep(10)
        
        if result_flag:
            (success_ids, failed_ids) = webdav.batch_sync(need_sync_files)
            
            for entry in entries:
                if entry['id'] in success_ids:
                    success_entries.append(entry)
                elif entry['id'] in failed_ids:
                    failed_entries.append(entry)
        else:
            for entry in entries:
                if entry['id'] == item['id']:
                    failed_entries.append(entry)

    return (success_entries, failed_entries)