tenant.openagents/omega

1//! An HTTP client.
2
3pub use crate::wit::zed::extension::http_client::{
4    HttpMethod, HttpRequest, HttpResponse, HttpResponseStream, RedirectPolicy, fetch, fetch_stream,
5};
6
7impl HttpRequest {
8    /// Returns a builder for an [`HttpRequest`].
9    pub fn builder() -> HttpRequestBuilder {
10        HttpRequestBuilder::new()
11    }
12
13    /// Executes the [`HttpRequest`] with [`fetch`].
14    pub fn fetch(&self) -> Result<HttpResponse, String> {
15        fetch(self)
16    }
17
18    /// Executes the [`HttpRequest`] with [`fetch_stream`].
19    pub fn fetch_stream(&self) -> Result<HttpResponseStream, String> {
20        fetch_stream(self)
21    }
22}
23
24/// A builder for an [`HttpRequest`].
25#[derive(Clone)]
26pub struct HttpRequestBuilder {
27    method: Option<HttpMethod>,
28    url: Option<String>,
29    headers: Vec<(String, String)>,
30    body: Option<Vec<u8>>,
31    redirect_policy: RedirectPolicy,
32}
33
34impl Default for HttpRequestBuilder {
35    fn default() -> Self {
36        Self::new()
37    }
38}
39
40impl HttpRequestBuilder {
41    /// Returns a new [`HttpRequestBuilder`].
42    pub fn new() -> Self {
43        HttpRequestBuilder {
44            method: None,
45            url: None,
46            headers: Vec::new(),
47            body: None,
48            redirect_policy: RedirectPolicy::NoFollow,
49        }
50    }
51
52    /// Sets the HTTP method for the request.
53    pub fn method(mut self, method: HttpMethod) -> Self {
54        self.method = Some(method);
55        self
56    }
57
58    /// Sets the URL for the request.
59    pub fn url(mut self, url: impl Into<String>) -> Self {
60        self.url = Some(url.into());
61        self
62    }
63
64    /// Adds a header to the request.
65    pub fn header(mut self, name: impl Into<String>, value: impl Into<String>) -> Self {
66        self.headers.push((name.into(), value.into()));
67        self
68    }
69
70    /// Adds the specified headers to the request.
71    pub fn headers(mut self, headers: impl IntoIterator<Item = (String, String)>) -> Self {
72        self.headers.extend(headers);
73        self
74    }
75
76    /// Sets the body of the request.
77    pub fn body(mut self, body: impl Into<Vec<u8>>) -> Self {
78        self.body = Some(body.into());
79        self
80    }
81
82    /// Sets the redirect policy for the request.
83    pub fn redirect_policy(mut self, policy: RedirectPolicy) -> Self {
84        self.redirect_policy = policy;
85        self
86    }
87
88    /// Builds the [`HttpRequest`].
89    pub fn build(self) -> Result<HttpRequest, String> {
90        let method = self.method.ok_or_else(|| "Method not set".to_string())?;
91        let url = self.url.ok_or_else(|| "URL not set".to_string())?;
92
93        Ok(HttpRequest {
94            method,
95            url,
96            headers: self.headers,
97            body: self.body,
98            redirect_policy: self.redirect_policy,
99        })
100    }
101}
102