omcp 0.1.11

Utility functions
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

use std::str::FromStr;

use reqwest::header::{HeaderMap, HeaderName, HeaderValue};

use crate::{
    client::{io::OMcpClientTrait, sse::SseClient, types::OMcpServerType},
    error::Result,
};

pub struct OMcpClientBuilder {
    pub url: String,
    pub server_type: OMcpServerType,
    pub headers: HeaderMap,
}

impl OMcpClientBuilder {
    pub fn new(server_type: OMcpServerType) -> Self {
        Self {
            url: "".into(),
            server_type,
            headers: HeaderMap::new(),
        }
    }

    pub fn with_sse_url<S>(mut self, url: S) -> Self
    where
        S: AsRef<str>,
    {
        self.url = url.as_ref().into();
        self
    }

    pub fn with_sse_bearer<S>(self, bearer: S) -> Result<Self>
    where
        S: AsRef<str>,
    {
        let bearer_value = format!("Bearer {}", bearer.as_ref());
        self.with_sse_header("Authorization", bearer_value)
    }

    pub fn with_sse_header<K, V>(mut self, key: K, value: V) -> Result<Self>
    where
        K: AsRef<str>,
        V: AsRef<str>,
    {
        let key = HeaderName::from_str(key.as_ref())?;
        let value = HeaderValue::from_str(value.as_ref())?;
        self.headers.insert(key, value);
        Ok(self)
    }

    pub fn build(self) -> Box<dyn OMcpClientTrait> {
        match self.server_type {
            OMcpServerType::Sse => {
                let sse = SseClient::from_builder(self);
                Box::new(sse)
            }
            OMcpServerType::Baked => {
                todo!()
            }
        }
    }
}