alou 0.1.6

智能自动化工作流系统 - 基于Rust和Model Context Protocol (MCP)的智能体
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154

// ============================================
// User API Endpoints
// ============================================

use serde::{Deserialize, Serialize};
use sqlx::PgPool;
use uuid::Uuid;
use warp::{reject, reply, Rejection, Reply};

use crate::models::user::{PublicUser, UpdateUser, User};

// ============================================
// Request/Response Types
// ============================================

#[derive(Debug, Deserialize)]
pub struct UpdateProfileRequest {
    pub name: Option<String>,
    pub avatar_url: Option<String>,
}

#[derive(Debug, Serialize)]
pub struct UserResponse {
    pub user: PublicUser,
}

#[derive(Debug, Serialize)]
pub struct MessageResponse {
    pub message: String,
}

// ============================================
// Error Types
// ============================================

#[derive(Debug)]
pub struct ApiError {
    pub message: String,
}

impl reject::Reject for ApiError {}

// ============================================
// Handler Functions
// ============================================

/// GET /api/user/me
/// Get current user information
pub async fn get_me_handler(user_id: Uuid, pool: PgPool) -> Result<impl Reply, Rejection> {
    let user = User::find_by_id(&pool, user_id)
        .await
        .map_err(|e| {
            reject::custom(ApiError {
                message: format!("Database error: {}", e),
            })
        })?
        .ok_or_else(|| {
            reject::custom(ApiError {
                message: "User not found".to_string(),
            })
        })?;

    let response = UserResponse {
        user: user.into(),
    };

    Ok(reply::json(&response))
}

/// PUT /api/user/profile
/// Update user profile
pub async fn update_profile_handler(
    user_id: Uuid,
    body: UpdateProfileRequest,
    pool: PgPool,
) -> Result<impl Reply, Rejection> {
    let update = UpdateUser {
        name: body.name,
        avatar_url: body.avatar_url,
        did: None,
        did_document_cid: None,
        wallet_address: None,
        public_key: None,
    };

    let user = User::update(&pool, user_id, update).await.map_err(|e| {
        reject::custom(ApiError {
            message: format!("Failed to update profile: {}", e),
        })
    })?;

    let response = UserResponse {
        user: user.into(),
    };

    Ok(reply::json(&response))
}

/// PUT /api/user/web3
/// Update Web3/DID information
#[derive(Debug, Deserialize)]
pub struct UpdateWeb3Request {
    pub did: Option<String>,
    pub did_document_cid: Option<String>,
    pub wallet_address: Option<String>,
    pub public_key: Option<String>,
}

pub async fn update_web3_handler(
    user_id: Uuid,
    body: UpdateWeb3Request,
    pool: PgPool,
) -> Result<impl Reply, Rejection> {
    let update = UpdateUser {
        name: None,
        avatar_url: None,
        did: body.did,
        did_document_cid: body.did_document_cid,
        wallet_address: body.wallet_address,
        public_key: body.public_key,
    };

    let user = User::update(&pool, user_id, update).await.map_err(|e| {
        reject::custom(ApiError {
            message: format!("Failed to update Web3 info: {}", e),
        })
    })?;

    let response = UserResponse {
        user: user.into(),
    };

    Ok(reply::json(&response))
}

/// DELETE /api/user/account
/// Deactivate user account
pub async fn deactivate_account_handler(
    user_id: Uuid,
    pool: PgPool,
) -> Result<impl Reply, Rejection> {
    User::deactivate(&pool, user_id).await.map_err(|e| {
        reject::custom(ApiError {
            message: format!("Failed to deactivate account: {}", e),
        })
    })?;

    let response = MessageResponse {
        message: "Account deactivated successfully".to_string(),
    };

    Ok(reply::json(&response))
}