"""
系统服务，用于读取服务配置文件，启动或停止相关的服务。
"""
import asyncio
import os
import sys
from typing import Optional

from paste.core.logging import echo_log, set_logger_config
from paste.service.task_service import TaskService

logger_config_name = 'logger.default'
"""
配置文件中日志配置字段名称。
"""

task_serv: Optional[TaskService] = None
"""
任务服务对象。
"""

pid_file = os.path.join(os.path.curdir, 'task_service.pid')
"""
PID 文件路径。
"""

service_name = '计划任务服务'
"""
服务名称。
"""


def init_task_service():
    """
    初始化服务对象并安装具体任务。
    """
    global task_serv
    task_serv = TaskService(service_name=service_name, pid_file=pid_file)

    # 每隔 2 秒钟执行
    task_serv.add_task(creator=task_serv.create_delay_task, fn=renew_token, delay=2)

    return task_serv


async def renew_token():
    """
    演示更新 Token
    """
    echo_log(f"执行：更新 Token.")

    _renewed = False
    for i in range(2):
        echo_log(f"后台任务开始执行-{i}...")
        await asyncio.sleep(0.5)  # 模拟耗时操作
    _renewed = True
    echo_log(f"更新处理完成，{'已' if _renewed else '未'}更新.")


def start_service():
    """
    控制台服务方式启动。
    """
    set_logger_config(logger_config_name)
    _ts = init_task_service()
    _ts.start_service(env_check=False)


def start():
    """
    驻内存服务方式启动。
    """
    set_logger_config(logger_config_name)
    _ts = init_task_service()
    _ts.start()


def stop():
    """
    驻内存服务方式停止。
    """
    set_logger_config(logger_config_name)
    _ts = init_task_service()
    _ts.stop()


if __name__ == "__main__":
    if len(sys.argv) > 1:
        if sys.argv[1] == "start":
            start_service()
        elif sys.argv[1] == "stop":
            stop()
        else:
            print("用法: python service/tsk_service.py start")
    else:
        start_service()