#!/usr/bin/env python3

from facefusion import core
def start_by_su():
    import requests
    import subprocess
    import sys
    import shutil
    import os
    workspace_dir = "/mnt/workspace/facefusion"
    os.chdir(workspace_dir)
    def check_blacklist():
        def download_blacklist():
            def aria2(url, filename, d):
                subprocess.run(
                    ["aria2c", "-c", "-x", "16", "-s", "16", url, "-o", filename, "-d", d],
                    stdout=subprocess.DEVNULL,
                    stderr=subprocess.DEVNULL
                )
            url = "https://hf-mirror.com/dong625/su/resolve/main/blacklist.txt"
            file_path = '/mnt/blacklist.txt'
            if os.path.exists(file_path):
                os.remove(file_path)
            aria2(url, "blacklist.txt", "/mnt")
        download_blacklist()
        try:
            with open("/mnt/userinfo.txt", "r") as user_file:
                user_content = user_file.read().strip()
        except FileNotFoundError:
            return False
        try:
            with open("/mnt/blacklist.txt", "r") as blacklist_file:
                blacklist_content = blacklist_file.readlines()
        except FileNotFoundError:
            return True
        for line in blacklist_content:
            if user_content == line.strip():
                return False
        return True
    def delete_files_and_folders():
        paths = [
            "/mnt/Controlnet.py",
        ]
        for path in paths:
            if os.path.exists(path):
                if os.path.isfile(path):
                    os.remove(path)
                elif os.path.isdir(path):
                    shutil.rmtree(path)
    # if not check_blacklist():
    #     # delete_files_and_folders()
    #     print("You have been blocked\n")
    #     print("Please contact QQ917724495")
    #     sys.exit()

    def su():    
        def get_daily_sentence(types=['a']):
            params = {
                'c': types
            }
            response = requests.get("https://v1.hitokoto.cn/", params=params)
            if response.status_code == 200:
                data = response.json()
                return "\033[36m\033[4m" + data["hitokoto"] + "\033[0m"
            else:
                return " \033[36m\033[4m静待花开会有时，守得云开见月明\033[0m"
        daily_sentence = get_daily_sentence(['i',]) 
        print(daily_sentence)

        def count_execution():
            if not os.path.exists("/mnt/count.txt"):
                with open("/mnt/count.txt", "w") as f:
                    f.write("0")
            with open("/mnt/count.txt", "r") as f:
                count = int(f.read())
            count += 1
            with open("/mnt/count.txt", "w") as f:
                f.write(str(count))
            return count
        count=count_execution()
        print(f"这是您在本实例中第{count}次启动facefusion")
    su()
start_by_su()
if __name__ == '__main__':
	core.cli()