solagent-plugin-cookie 0.1.2

solagent plugin cookie
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

use serde_json::Value;
use solagent_core::SolanaAgentKit;
use std::error::Error;

/// Retrieve agent details in specified interval by one of its tokens contract address.
///
/// # Parameters
///
/// - `agent`: An instance of `SolanaAgentKit`.
/// - `contract_address`: Contract address of one of the tokens contracts (matches case insensitive)
/// - `interval`: An optional Interval for twitter stats and deltas (_3Days, _7Days). If not provided, returns the _7Days.
///
/// # Returns
///
/// A `Result` that agent details
pub async fn get_agent_by_ca(
    agent: &SolanaAgentKit,
    contract_address: &str,
    interval: Option<u32>,
) -> Result<Value, Box<dyn Error>> {
    // Get the Cookie API key from the agent's configuration
    let api_key = match agent.config.cookie_api_key.as_ref() {
        Some(key) => key,
        None => return Err("Missing Cookie API key in agent.config.cookie_api_key".into()),
    };

    let api_url = "https://api.cookie.fun/v2/agents/contractAddress";
    let url = format!("{}/{}?interval=_{}Days", api_url, contract_address, interval.unwrap_or(7));
    let client = reqwest::Client::new();

    let response = client.get(&url).header("x-api-key", api_key).send().await?;

    let json: Value = response.json().await?;
    Ok(json)
}

/// Retrieve agent details in specified interval by twitter username.
///
/// # Parameters
///
/// - `agent`: An instance of `SolanaAgentKit`.
/// - `twitter_name`: Twitter username of agent (matches case insensitive)
/// - `interval`: An optional Interval for twitter stats and deltas (_3Days, _7Days). If not provided, returns the _7Days.
///
/// # Returns
///
/// A `Result` that agent details
pub async fn get_agent_by_name(
    agent: &SolanaAgentKit,
    twitter_name: &str,
    interval: Option<u32>,
) -> Result<Value, Box<dyn Error>> {
    // Get the Cookie API key from the agent's configuration
    let api_key = match agent.config.cookie_api_key.as_ref() {
        Some(key) => key,
        None => return Err("Missing Cookie API key in agent.config.cookie_api_key".into()),
    };

    let api_url = "https://api.cookie.fun/v2/agents/twitterUsername";
    let url = format!("{}/{}?interval=_{}Days", api_url, twitter_name, interval.unwrap_or(7));
    let client = reqwest::Client::new();

    let response = client.get(&url).header("x-api-key", api_key).send().await?;

    let json: Value = response.json().await?;
    Ok(json)
}

/// Retrieve popular content matching search query, created in time range {from} - {to} (YYYY-MM-DD dates).
///
/// # Parameters
///
/// - `agent`: An instance of `SolanaAgentKit`.
/// - `search_query`: Word or phrase to be searched for in text
/// - `from`: Only consider content created after given date, eg. 2025-01-01
/// - `to`: Only consider content created before given date, eg. 2025-01-20
///
/// # Returns
///
/// A `Result` that tweets details
pub async fn search_tweets(
    agent: &SolanaAgentKit,
    tweets: &str,
    from: &str,
    to: &str,
) -> Result<Value, Box<dyn Error>> {
    // Get the Cookie API key from the agent's configuration
    let api_key = match agent.config.cookie_api_key.as_ref() {
        Some(key) => key,
        None => return Err("Missing Cookie API key in agent.config.cookie_api_key".into()),
    };

    let api_url = "https://api.cookie.fun/v1/hackathon/search";
    let url = format!("{}/{}?from={}&to={}", api_url, tweets, from, to);
    let client = reqwest::Client::new();

    let response = client.get(&url).header("x-api-key", api_key).send().await?;

    let json: Value = response.json().await?;
    Ok(json)
}