#!/opt/homebrew/bin/python3
# -*- coding:utf-8 -*-

import json
import os
from datetime import datetime
from PyQt6.QtCore import *
from models import Project, Session
from logs import log
from PyQt6.QtGui import *

class ProjectBackend(QObject):
    def __init__(self):
        super().__init__()
        self.basePath = ''
        self.projectName = ''
        self.projectPath = ''
        self.projectPath_log = ''
        self.projectPath_screenshot = ''
        self.projectPath_tsdb = ''

    def setConfig(self,conf):
        self.basePath = conf['base_path']
        if self.projectName:
            self.createProjectPath(self.projectName)

    def setProjectPath(self, name):
        self.projectName = name
        self.projectPath = os.path.join(self.basePath, self.projectName)
        self.projectPath_log = os.path.join(self.projectPath, 'log')
        self.projectPath_screenshot = os.path.join(self.projectPath, 'screenshot')
        self.projectPath_tsdb = os.path.join(self.projectPath, 'tsdb')
        self.projectLog = f'{os.path.join(self.projectPath_log, self.projectName)}.txt'
        return self.projectPath

    def createProjectPath(self, name):
        path = self.setProjectPath(name)
        os.makedirs(self.projectPath_log, 0o755, True)
        os.makedirs(self.projectPath_screenshot, 0o755, True)
        os.makedirs(self.projectPath_tsdb, 0o755, True)
        return path

    def writeLogFile(self, t, data):
        try:
            with open(self.projectLog, 'a') as file:
                file.write(f"[{t}] : {data}{os.linesep}")
            return True, None
        except Exception as e:
            log.error(e)
            return False, None

    # 打开日志文件夹
    def open_log_folder(self):
        try:
            url = QUrl.fromLocalFile(self.projectPath_log)
            QDesktopServices.openUrl(url)
            return True, None
        except Exception as e:
            log.error(e)
            return False, None

    # 添加工程
    def add_project(self, json_str):
        json_dict = json.loads(json_str)
        name = json_dict['name']
        ok, data = Session.queryByName(Project, name)
        if ok:
            return False, None

        json_dict['path'] = self.createProjectPath(name)
        json_dict['date'] = str(datetime.now())[:-7]
        return Session.addByClass(Project, json_dict)

    # 更新工程str
    def update_project(self, json_str):
        json_dict = json.loads(json_str)
        json_dict['last_date'] = str(datetime.now())[:-7]
        id = json_dict.get('id')
        return Session.updateById(Project, id, json_dict)

    # 删除工程
    def delete_project(self, id):
        return Session.deleteById(Project, id)

    # 获取所有工程
    def get_projects(self):
        return Session.queryByAll(Project)

    # 获取某个工程
    def get_project(self, id):
        return Session.queryById(Project, id)

    # 切换工程
    def selected_project(self, name):
        try:
            return True, self.createProjectPath(name)
        except Exception as e:
            log.error(e)
            return False, None

projectBackend = ProjectBackend()