agentswap 0.7.0

AgentSwap CLI — AI-agent-optimized DEX aggregator
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

// HTTP transport helpers backing the AgentSwap client.
// Exports: internal Client get/post helpers.
// Deps: super::Client, reqwest, serde_json, eyre.

use super::Client;
use eyre::{eyre, Result};
use serde::de::DeserializeOwned;

impl Client {
    pub async fn get<T: DeserializeOwned>(&self, path: &str) -> Result<T> {
        let url = format!("{}{path}", self.base_url);
        self.get_raw(&url).await
    }

    pub(super) async fn get_raw<T: DeserializeOwned>(&self, url: &str) -> Result<T> {
        let mut req = self.http.get(url);
        if let Some(key) = &self.api_key {
            req = req.header("x-api-key", key);
        }
        let resp = req.send().await.map_err(|e| eyre!("request failed: {e}"))?;
        let resp = if resp.status() == reqwest::StatusCode::PAYMENT_REQUIRED {
            self.retry_get_with_payment(url, resp).await?
        } else {
            resp
        };
        let status = resp.status();
        if !status.is_success() {
            let body = resp.text().await.unwrap_or_default();
            if status == reqwest::StatusCode::UNAUTHORIZED {
                return Err(eyre!("HTTP 401 Unauthorized — set SR_API_KEY env var or register with: agentswap register"));
            }
            return Err(eyre!("HTTP {status}: {body}"));
        }
        resp.json().await.map_err(|e| eyre!("parse failed: {e}"))
    }

    pub(super) async fn post<T: DeserializeOwned>(
        &self,
        path: &str,
        body: &serde_json::Value,
    ) -> Result<T> {
        let url = format!("{}{path}", self.base_url);
        let mut req = self.http.post(&url).json(body);
        if let Some(key) = &self.api_key {
            req = req.header("x-api-key", key);
        }
        let resp = req.send().await.map_err(|e| eyre!("request failed: {e}"))?;
        let resp = if resp.status() == reqwest::StatusCode::PAYMENT_REQUIRED {
            self.retry_post_with_payment(&url, body, resp).await?
        } else {
            resp
        };
        let status = resp.status();
        if !status.is_success() {
            let body = resp.text().await.unwrap_or_default();
            if status == reqwest::StatusCode::UNAUTHORIZED {
                return Err(eyre!("HTTP 401 Unauthorized — set SR_API_KEY env var or register with: agentswap register"));
            }
            return Err(eyre!("HTTP {status}: {body}"));
        }
        resp.json().await.map_err(|e| eyre!("parse failed: {e}"))
    }

    async fn retry_get_with_payment(
        &self,
        url: &str,
        resp: reqwest::Response,
    ) -> Result<reqwest::Response> {
        let payment = self.payment_header(resp).await?;
        self.http
            .get(url)
            .header("X-PAYMENT", payment)
            .send()
            .await
            .map_err(|e| eyre!("x402 retry failed: {e}"))
    }

    async fn retry_post_with_payment(
        &self,
        url: &str,
        body: &serde_json::Value,
        resp: reqwest::Response,
    ) -> Result<reqwest::Response> {
        let payment = self.payment_header(resp).await?;
        self.http
            .post(url)
            .header("X-PAYMENT", payment)
            .json(body)
            .send()
            .await
            .map_err(|e| eyre!("x402 retry failed: {e}"))
    }

    async fn payment_header(&self, resp: reqwest::Response) -> Result<String> {
        let body = resp.text().await.unwrap_or_default();
        if !self.x402.enabled || (self.api_key.is_some() && !self.x402.prefer_x402) {
            return Err(eyre!("HTTP 402 Payment Required: {body}"));
        }
        let signer = self
            .x402_signer
            .clone()
            .ok_or_else(|| eyre!("HTTP 402 Payment Required but no x402 signer configured"))?;
        crate::x402::payment_header(&body, &self.x402, signer).await
    }
}