53 lines
1.2 KiB
Rust
53 lines
1.2 KiB
Rust
|
use serde::{Deserialize, Serialize};
|
||
|
|
use std::collections::HashMap;
|
||
|
|
|
||
|
|
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
|
||
|
|
#[serde(rename_all = "snake_case")]
|
||
|
|
pub enum McpTransportType {
|
||
|
|
Stdio,
|
||
|
|
Http,
|
||
|
|
Sse,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Default for McpTransportType {
|
||
|
|
fn default() -> Self {
|
||
|
|
Self::Stdio
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||
|
|
pub struct McpServer {
|
||
|
|
pub id: String,
|
||
|
|
pub name: String,
|
||
|
|
#[serde(default)]
|
||
|
|
pub transport_type: McpTransportType,
|
||
|
|
pub command: Option<String>,
|
||
|
|
#[serde(default)]
|
||
|
|
pub args: Vec<String>,
|
||
|
|
#[serde(default)]
|
||
|
|
pub env: HashMap<String, String>,
|
||
|
|
pub url: Option<String>,
|
||
|
|
#[serde(default)]
|
||
|
|
pub headers: HashMap<String, String>,
|
||
|
|
pub created_at: String,
|
||
|
|
pub updated_at: String,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl McpServer {
|
||
|
|
pub fn new(name: String) -> Self {
|
||
|
|
let now = chrono::Utc::now().to_rfc3339();
|
||
|
|
Self {
|
||
|
|
id: uuid::Uuid::new_v4().to_string(),
|
||
|
|
name,
|
||
|
|
transport_type: McpTransportType::default(),
|
||
|
|
command: None,
|
||
|
|
args: Vec::new(),
|
||
|
|
env: HashMap::new(),
|
||
|
|
url: None,
|
||
|
|
headers: HashMap::new(),
|
||
|
|
created_at: now.clone(),
|
||
|
|
updated_at: now,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|