# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========

from enum import IntEnum
from typing import TYPE_CHECKING

from pydantic import BaseModel
from sqlalchemy import Column, SmallInteger, String
from sqlalchemy.orm import Mapped
from sqlalchemy_utils import ChoiceType
from sqlmodel import JSON, Field, Relationship

from app.model.abstract.model import AbstractModel, DefaultTimes
from app.model.mcp.category import Category, CategoryOut
from app.model.mcp.mcp_env import McpEnv
from app.shared.types.pydantic import HttpUrlStr

if TYPE_CHECKING:
    from app.model.mcp.mcp_user import McpUser


class Status(IntEnum):
    Online = 1
    Offline = -1


class McpType(IntEnum):
    Local = 1
    Remote = 2


class Mcp(AbstractModel, DefaultTimes, table=True):
    id: int = Field(default=None, primary_key=True)
    category_id: int = Field(foreign_key="category.id")
    name: str
    key: str = Field(sa_column=Column(String(128)))
    description: str = ""
    home_page: str = Field(default="", sa_column=Column(String(1024)))
    type: McpType = Field(default=McpType.Local, sa_column=Column(ChoiceType(McpType, SmallInteger())))
    status: Status = Field(default=Status.Online, sa_column=Column(ChoiceType(Status, SmallInteger())))
    sort: int = Field(default=0, sa_column=Column(SmallInteger))
    server_name: str = Field(default="", sa_column=Column(String(128)))
    install_command: dict = Field(default="{}", sa_column=Column(JSON))
    """{
        "command": "uvx",
        "args": ["mcp-server-everything-search"],
        "env": {
        "EVERYTHING_SDK_PATH": "path/to/Everything-SDK/dll/Everything64.dll"
        }
    }"""

    category: Mapped[Category] = Relationship()
    envs: Mapped[list[McpEnv]] = Relationship()
    # user_env: Mapped[McpUser] = Relationship()

    mcp_user: list["McpUser"] = Relationship(back_populates="mcp")


class McpIn(BaseModel):
    category_id: int
    name: str
    key: str
    description: str
    home_page: HttpUrlStr
    type: McpType
    status: Status
    install_command: dict


class McpOut(McpIn):
    id: int
    category: CategoryOut | None = None
    # envs: list[McpEnvOut] = []


class McpInfo(BaseModel):
    id: int
    name: str
    key: str