mirror of
https://github.com/eigent-ai/eigent.git
synced 2026-04-28 11:40:25 +00:00
78 lines
3.1 KiB
Python
78 lines
3.1 KiB
Python
from typing import List, Optional
|
|
from fastapi import Depends, HTTPException, Query, Response, APIRouter
|
|
from sqlmodel import Session, select
|
|
from app.component.database import session
|
|
from app.component.auth import Auth, auth_must
|
|
from fastapi_babel import _
|
|
from app.model.mcp.mcp_user import McpUser, McpUserIn, McpUserOut, McpUserUpdate, Status
|
|
from loguru import logger
|
|
|
|
router = APIRouter(tags=["McpUser Management"])
|
|
|
|
|
|
@router.get("/mcp/users", name="list mcp users", response_model=List[McpUserOut])
|
|
async def list_mcp_users(
|
|
mcp_id: Optional[int] = None,
|
|
session: Session = Depends(session),
|
|
auth: Auth = Depends(auth_must),
|
|
):
|
|
user_id = auth.user.id
|
|
query = select(McpUser)
|
|
if mcp_id is not None:
|
|
query = query.where(McpUser.mcp_id == mcp_id)
|
|
if user_id is not None:
|
|
query = query.where(McpUser.user_id == user_id)
|
|
mcp_users = session.exec(query).all()
|
|
return mcp_users
|
|
|
|
|
|
@router.get("/mcp/users/{mcp_user_id}", name="get mcp user", response_model=McpUserOut)
|
|
async def get_mcp_user(mcp_user_id: int, session: Session = Depends(session), auth: Auth = Depends(auth_must)):
|
|
query = select(McpUser).where(McpUser.id == mcp_user_id)
|
|
mcp_user = session.exec(query).first()
|
|
if not mcp_user:
|
|
raise HTTPException(status_code=404, detail=_("McpUser not found"))
|
|
return mcp_user
|
|
|
|
|
|
@router.post("/mcp/users", name="create mcp user", response_model=McpUserOut)
|
|
async def create_mcp_user(mcp_user: McpUserIn, session: Session = Depends(session), auth: Auth = Depends(auth_must)):
|
|
exists = session.exec(
|
|
select(McpUser).where(McpUser.mcp_id == mcp_user.mcp_id, McpUser.user_id == auth.user.id)
|
|
).first()
|
|
if exists:
|
|
raise HTTPException(status_code=400, detail=_("mcp is installed"))
|
|
db_mcp_user = McpUser(mcp_id=mcp_user.mcp_id, user_id=auth.user.id, env=mcp_user.env)
|
|
session.add(db_mcp_user)
|
|
session.commit()
|
|
session.refresh(db_mcp_user)
|
|
return db_mcp_user
|
|
|
|
|
|
@router.put("/mcp/users/{id}", name="update mcp user")
|
|
async def update_mcp_user(
|
|
id: int,
|
|
update_item: McpUserUpdate,
|
|
session: Session = Depends(session),
|
|
auth: Auth = Depends(auth_must),
|
|
):
|
|
model = session.get(McpUser, id)
|
|
if not model:
|
|
raise HTTPException(status_code=404, detail=_("Mcp Info not found"))
|
|
if model.user_id != auth.user.id:
|
|
raise HTTPException(status_code=400, detail=_("current user have no permission to modify"))
|
|
update_data = update_item.model_dump(exclude_unset=True)
|
|
model.update_fields(update_data)
|
|
model.save(session)
|
|
session.refresh(model)
|
|
return model
|
|
|
|
|
|
@router.delete("/mcp/users/{mcp_user_id}", name="delete mcp user")
|
|
async def delete_mcp_user(mcp_user_id: int, session: Session = Depends(session), auth: Auth = Depends(auth_must)):
|
|
db_mcp_user = session.get(McpUser, mcp_user_id)
|
|
if not db_mcp_user:
|
|
raise HTTPException(status_code=404, detail=_("Mcp Info not found"))
|
|
session.delete(db_mcp_user)
|
|
session.commit()
|
|
return Response(status_code=204)
|