vtcode_auth/
config.rs

1use serde::{Deserialize, Serialize};
2
3#[cfg(feature = "schema")]
4use schemars::JsonSchema;
5
6use crate::openrouter_oauth::OpenRouterOAuthConfig;
7
8#[derive(Debug, Clone, Default, Serialize, Deserialize)]
9#[cfg_attr(feature = "schema", derive(JsonSchema))]
10pub struct AuthConfig {
11    #[serde(default)]
12    pub openrouter: OpenRouterOAuthConfig,
13    #[serde(default)]
14    pub openai: OpenAIAuthConfig,
15    #[serde(default)]
16    pub copilot: CopilotAuthConfig,
17}
18
19#[derive(Debug, Clone, Serialize, Deserialize)]
20#[cfg_attr(feature = "schema", derive(JsonSchema))]
21#[serde(default)]
22pub struct CopilotAuthConfig {
23    pub command: Option<String>,
24    pub host: Option<String>,
25    #[serde(default = "default_copilot_available_tools")]
26    pub available_tools: Vec<String>,
27    #[serde(default)]
28    pub excluded_tools: Vec<String>,
29    #[serde(default = "default_vtcode_tool_allowlist")]
30    pub vtcode_tool_allowlist: Vec<String>,
31    pub startup_timeout_secs: u64,
32    pub auth_timeout_secs: u64,
33}
34
35impl Default for CopilotAuthConfig {
36    fn default() -> Self {
37        Self {
38            command: None,
39            host: None,
40            available_tools: default_copilot_available_tools(),
41            excluded_tools: Vec::new(),
42            vtcode_tool_allowlist: default_vtcode_tool_allowlist(),
43            startup_timeout_secs: 20,
44            auth_timeout_secs: 300,
45        }
46    }
47}
48
49fn default_copilot_available_tools() -> Vec<String> {
50    ["view", "glob", "grep"].into_iter().map(str::to_string).collect()
51}
52
53fn default_vtcode_tool_allowlist() -> Vec<String> {
54    ["exec_command", "write_stdin", "apply_patch", "code_search"]
55        .into_iter()
56        .map(str::to_string)
57        .collect()
58}
59
60#[derive(Debug, Clone, Copy, Default, PartialEq, Eq, Serialize, Deserialize)]
61#[cfg_attr(feature = "schema", derive(JsonSchema))]
62#[serde(rename_all = "snake_case")]
63pub enum OpenAIPreferredMethod {
64    #[default]
65    Auto,
66    ApiKey,
67    Chatgpt,
68}
69
70#[derive(Debug, Clone, Serialize, Deserialize)]
71#[cfg_attr(feature = "schema", derive(JsonSchema))]
72#[serde(default)]
73pub struct OpenAIAuthConfig {
74    pub preferred_method: OpenAIPreferredMethod,
75    pub callback_port: u16,
76    pub auto_refresh: bool,
77    pub flow_timeout_secs: u64,
78}
79
80impl Default for OpenAIAuthConfig {
81    fn default() -> Self {
82        Self {
83            preferred_method: OpenAIPreferredMethod::Auto,
84            callback_port: 1455,
85            auto_refresh: true,
86            flow_timeout_secs: 300,
87        }
88    }
89}
90
91impl OpenAIPreferredMethod {
92    #[must_use]
93    pub fn as_str(self) -> &'static str {
94        match self {
95            Self::Auto => "auto",
96            Self::ApiKey => "api_key",
97            Self::Chatgpt => "chatgpt",
98        }
99    }
100}
101
102impl OpenAIAuthConfig {
103    #[must_use]
104    pub fn callback_url(&self) -> String {
105        format!("http://localhost:{}/callback", self.callback_port)
106    }
107}
108
109#[cfg(test)]
110mod tests {
111    use super::*;
112
113    #[test]
114    fn openai_defaults_match_expected_values() {
115        let config = OpenAIAuthConfig::default();
116        assert_eq!(config.preferred_method, OpenAIPreferredMethod::Auto);
117        assert_eq!(config.callback_port, 1455);
118        assert!(config.auto_refresh);
119        assert_eq!(config.flow_timeout_secs, 300);
120    }
121
122    #[test]
123    fn copilot_defaults_match_expected_values() {
124        let config = CopilotAuthConfig::default();
125        assert!(config.command.is_none());
126        assert!(config.host.is_none());
127        assert_eq!(config.available_tools, vec!["view", "glob", "grep"]);
128        assert!(config.excluded_tools.is_empty());
129        assert_eq!(
130            config.vtcode_tool_allowlist,
131            vec!["exec_command", "write_stdin", "apply_patch", "code_search"]
132        );
133        assert_eq!(config.startup_timeout_secs, 20);
134        assert_eq!(config.auth_timeout_secs, 300);
135    }
136}
vtcode_auth/config.rs

vtcode_auth/
config.rs
