from models.request import GetTokenRequest
from utils.response_builder import ResponseBuilder
from config.settings import settings
from typing import Dict


class TokenService:
    """Token管理服务"""

    def __init__(self):
        self.project_counter = settings.INITIAL_PROJECT_ID
        self.tokens = {}  # projectId -> token_data

    def create_token(self, request: GetTokenRequest) -> Dict:
        """创建Token

        Args:
            request: 获取Token请求

        Returns:
            Token响应字典
        """
        # 生成唯一项目ID
        self.project_counter += 1
        project_id = self.project_counter

        # 构建响应
        response = ResponseBuilder.build_success_response(
            "token",
            project_id=project_id,
            project_no=request.projectNo,
            entity_name=request.entityName
        )

        # 存储token信息
        self.tokens[project_id] = response.get("data")

        return response

    def get_project(self, project_id: int) -> Dict:
        """获取项目信息

        Args:
            project_id: 项目ID

        Returns:
            项目信息字典
        """
        return self.tokens.get(project_id)