import json from uuid import uuid4 from fastapi import APIRouter, Depends, HTTPException from pydantic import BaseModel from backend.app.api.dependencies import get_current_user from backend.app.database import get_connection router = APIRouter() class UserConfigUpdate(BaseModel): email: str | None = None bio: str | None = None avatar_url: str | None = None class UserPluginConfigUpdate(BaseModel): instance: str | None = None access_token: str | None = None post_prefix: str | None = None hashtag: str | None = None @router.get('/me') def get_current_user_profile(user: dict = Depends(get_current_user)): with get_connection() as conn: row = conn.execute( 'SELECT * FROM users WHERE id = ?', (user['id'],), ).fetchone() if row is None: raise HTTPException(status_code=404, detail='User not found') return dict(row) @router.put('/me') def update_current_user_profile( payload: UserConfigUpdate, user: dict = Depends(get_current_user), ): with get_connection() as conn: current = conn.execute('SELECT * FROM users WHERE id = ?', (user['id'],)).fetchone() if current is None: raise HTTPException(status_code=404, detail='User not found') email = payload.email or current['email'] bio = payload.bio if payload.bio is not None else current['bio'] avatar_url = payload.avatar_url if payload.avatar_url is not None else current['avatar_url'] conn.execute( ''' UPDATE users SET email = ?, bio = ?, avatar_url = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ? ''', (email, bio, avatar_url, user['id']), ) conn.commit() return {'status': 'updated'} @router.get('/plugins/{plugin_name}') def get_user_plugin_config(plugin_name: str, user: dict = Depends(get_current_user)): with get_connection() as conn: row = conn.execute( 'SELECT * FROM user_plugin_config WHERE user_id = ? AND plugin_name = ?', (user['id'], plugin_name), ).fetchone() if row is None: return {} config = json.loads(row['config']) if row['config'] else {} return config @router.put('/plugins/{plugin_name}') def update_user_plugin_config( plugin_name: str, payload: UserPluginConfigUpdate, user: dict = Depends(get_current_user), ): with get_connection() as conn: current = conn.execute( 'SELECT * FROM user_plugin_config WHERE user_id = ? AND plugin_name = ?', (user['id'], plugin_name), ).fetchone() current_config = json.loads(current['config']) if current and current['config'] else {} updates = payload.model_dump(exclude_none=True) merged = {**current_config, **updates} if current is None: conn.execute( ''' INSERT INTO user_plugin_config (id, user_id, plugin_name, config, created_at, updated_at) VALUES (?, ?, ?, ?, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP) ''', (str(uuid4()), user['id'], plugin_name, json.dumps(merged, ensure_ascii=False)), ) else: conn.execute( ''' UPDATE user_plugin_config SET config = ?, updated_at = CURRENT_TIMESTAMP WHERE user_id = ? AND plugin_name = ? ''', (json.dumps(merged, ensure_ascii=False), user['id'], plugin_name), ) conn.commit() return merged