netget-tool 0.3.5

Network protocol stubbing toolkit: ~280 protocol servers and clients whose responses come from an LLM, a script, a static rule or a person.
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

//! Client connect context
//!
//! Provides all the necessary context for connecting a protocol client.

use crate::llm::OllamaClient;
use crate::protocol::StartupParams;
use crate::state::app_state::AppState;
use crate::state::ClientId;
use std::sync::Arc;
use tokio::sync::mpsc;

/// Context passed to protocol clients during connection
///
/// Contains all the dependencies and configuration needed to connect to a remote server.
#[derive(Clone)]
pub struct ConnectContext {
    /// Remote server address (hostname:port or IP:port)
    /// Example: "example.com:80" or "192.168.1.1:6379"
    pub remote_addr: String,

    /// LLM client for generating requests and interpreting responses
    pub llm_client: OllamaClient,

    /// Application state
    pub state: Arc<AppState>,

    /// Channel for sending status updates to UI
    pub status_tx: mpsc::UnboundedSender<String>,

    /// Unique identifier for this client instance
    pub client_id: ClientId,

    /// Optional type-safe startup parameters specific to the protocol
    ///
    /// Parameters can only be accessed if they were declared in the protocol's
    /// `get_startup_parameters()` implementation. Attempting to access undeclared
    /// parameters will panic at runtime.
    ///
    /// For example:
    /// - HTTP Client: request_headers, user_agent, follow_redirects
    /// - SSH Client: username, password, private_key_path
    /// - MySQL Client: username, password, database
    pub startup_params: Option<StartupParams>,
}

impl ConnectContext {
    /// Create a new connect context
    pub fn new(
        remote_addr: String,
        llm_client: OllamaClient,
        state: Arc<AppState>,
        status_tx: mpsc::UnboundedSender<String>,
        client_id: ClientId,
    ) -> Self {
        Self {
            remote_addr,
            llm_client,
            state,
            status_tx,
            client_id,
            startup_params: None,
        }
    }

    /// Create connect context with startup parameters
    pub fn with_params(
        remote_addr: String,
        llm_client: OllamaClient,
        state: Arc<AppState>,
        status_tx: mpsc::UnboundedSender<String>,
        client_id: ClientId,
        startup_params: StartupParams,
    ) -> Self {
        Self {
            remote_addr,
            llm_client,
            state,
            status_tx,
            client_id,
            startup_params: Some(startup_params),
        }
    }
}