lean_ctx/core/providers/
gitlab.rs

1use super::cache;
2use super::config::GitLabConfig;
3use super::provider_trait::{ContextProvider, ProviderParams};
4use super::{ProviderItem, ProviderResult};
5
6const DEFAULT_PER_PAGE: usize = 20;
7const CACHE_TTL_SECS: u64 = 120;
8
9pub fn list_issues(
10    config: &GitLabConfig,
11    state: Option<&str>,
12    labels: Option<&str>,
13    limit: Option<usize>,
14) -> Result<ProviderResult, String> {
15    let project = config
16        .project_path
17        .as_deref()
18        .ok_or("No project path configured. Set CI_PROJECT_PATH or configure git remote.")?;
19    let encoded = urlencoding::encode(project);
20    let per_page = limit.unwrap_or(DEFAULT_PER_PAGE).min(100);
21
22    let mut url =
23        format!("/projects/{encoded}/issues?per_page={per_page}&order_by=updated_at&sort=desc");
24    if let Some(s) = state {
25        url.push_str(&format!("&state={s}"));
26    }
27    if let Some(l) = labels {
28        url.push_str(&format!("&labels={l}"));
29    }
30
31    let cache_key = format!("gitlab:issues:{project}:{state:?}:{labels:?}:{per_page}");
32    if let Some(cached) = cache::get_cached(&cache_key) {
33        if let Ok(result) = serde_json::from_str::<ProviderResult>(&cached) {
34            return Ok(result);
35        }
36    }
37
38    let body = api_get(config, &url)?;
39    let items: Vec<serde_json::Value> =
40        serde_json::from_str(&body).map_err(|e| format!("JSON parse error: {e}"))?;
41
42    let result = ProviderResult {
43        provider: "gitlab".to_string(),
44        resource_type: "issues".to_string(),
45        total_count: None,
46        truncated: items.len() >= per_page,
47        items: items.iter().map(parse_issue).collect(),
48    };
49
50    if let Ok(json) = serde_json::to_string(&result) {
51        cache::set_cached(&cache_key, &json, CACHE_TTL_SECS);
52    }
53    Ok(result)
54}
55
56pub fn show_issue(config: &GitLabConfig, iid: u64) -> Result<ProviderResult, String> {
57    let project = config
58        .project_path
59        .as_deref()
60        .ok_or("No project path configured.")?;
61    let encoded = urlencoding::encode(project);
62    let url = format!("/projects/{encoded}/issues/{iid}");
63
64    let body = api_get(config, &url)?;
65    let issue: serde_json::Value =
66        serde_json::from_str(&body).map_err(|e| format!("JSON parse error: {e}"))?;
67
68    Ok(ProviderResult {
69        provider: "gitlab".to_string(),
70        resource_type: "issue".to_string(),
71        total_count: Some(1),
72        truncated: false,
73        items: vec![parse_issue(&issue)],
74    })
75}
76
77pub fn list_mrs(
78    config: &GitLabConfig,
79    state: Option<&str>,
80    limit: Option<usize>,
81) -> Result<ProviderResult, String> {
82    let project = config
83        .project_path
84        .as_deref()
85        .ok_or("No project path configured.")?;
86    let encoded = urlencoding::encode(project);
87    let per_page = limit.unwrap_or(DEFAULT_PER_PAGE).min(100);
88
89    let mut url = format!(
90        "/projects/{encoded}/merge_requests?per_page={per_page}&order_by=updated_at&sort=desc"
91    );
92    if let Some(s) = state {
93        url.push_str(&format!("&state={s}"));
94    }
95
96    let cache_key = format!("gitlab:mrs:{project}:{state:?}:{per_page}");
97    if let Some(cached) = cache::get_cached(&cache_key) {
98        if let Ok(result) = serde_json::from_str::<ProviderResult>(&cached) {
99            return Ok(result);
100        }
101    }
102
103    let body = api_get(config, &url)?;
104    let items: Vec<serde_json::Value> =
105        serde_json::from_str(&body).map_err(|e| format!("JSON parse error: {e}"))?;
106
107    let result = ProviderResult {
108        provider: "gitlab".to_string(),
109        resource_type: "merge_requests".to_string(),
110        total_count: None,
111        truncated: items.len() >= per_page,
112        items: items.iter().map(parse_mr).collect(),
113    };
114
115    if let Ok(json) = serde_json::to_string(&result) {
116        cache::set_cached(&cache_key, &json, CACHE_TTL_SECS);
117    }
118    Ok(result)
119}
120
121pub fn list_pipelines(
122    config: &GitLabConfig,
123    status: Option<&str>,
124    limit: Option<usize>,
125) -> Result<ProviderResult, String> {
126    let project = config
127        .project_path
128        .as_deref()
129        .ok_or("No project path configured.")?;
130    let encoded = urlencoding::encode(project);
131    let per_page = limit.unwrap_or(DEFAULT_PER_PAGE).min(100);
132
133    let mut url =
134        format!("/projects/{encoded}/pipelines?per_page={per_page}&order_by=updated_at&sort=desc");
135    if let Some(s) = status {
136        url.push_str(&format!("&status={s}"));
137    }
138
139    let body = api_get(config, &url)?;
140    let items: Vec<serde_json::Value> =
141        serde_json::from_str(&body).map_err(|e| format!("JSON parse error: {e}"))?;
142
143    Ok(ProviderResult {
144        provider: "gitlab".to_string(),
145        resource_type: "pipelines".to_string(),
146        total_count: None,
147        truncated: items.len() >= per_page,
148        items: items
149            .iter()
150            .map(|p| ProviderItem {
151                id: p["id"].as_u64().unwrap_or(0).to_string(),
152                title: p["ref"].as_str().unwrap_or("").to_string(),
153                state: p["status"].as_str().map(std::string::ToString::to_string),
154                author: None,
155                created_at: p["created_at"]
156                    .as_str()
157                    .map(std::string::ToString::to_string),
158                updated_at: p["updated_at"]
159                    .as_str()
160                    .map(std::string::ToString::to_string),
161                url: p["web_url"].as_str().map(std::string::ToString::to_string),
162                labels: Vec::new(),
163                body: None,
164            })
165            .collect(),
166    })
167}
168
169pub struct GitLabProvider {
170    config: Result<GitLabConfig, String>,
171}
172
173impl GitLabProvider {
174    pub fn new() -> Self {
175        Self {
176            config: GitLabConfig::from_env(),
177        }
178    }
179}
180
181impl Default for GitLabProvider {
182    fn default() -> Self {
183        Self::new()
184    }
185}
186
187impl ContextProvider for GitLabProvider {
188    fn id(&self) -> &'static str {
189        "gitlab"
190    }
191
192    fn display_name(&self) -> &'static str {
193        "GitLab"
194    }
195
196    fn supported_actions(&self) -> &[&str] {
197        &["issues", "merge_requests", "pipelines"]
198    }
199
200    fn execute(&self, action: &str, params: &ProviderParams) -> Result<ProviderResult, String> {
201        let config = self.config.as_ref().map_err(std::clone::Clone::clone)?;
202        match action {
203            "issues" => list_issues(config, params.state.as_deref(), None, params.limit),
204            "merge_requests" | "mrs" => list_mrs(config, params.state.as_deref(), params.limit),
205            "pipelines" => list_pipelines(config, params.state.as_deref(), params.limit),
206            _ => Err(format!("Unknown GitLab action: {action}")),
207        }
208    }
209
210    fn cache_ttl_secs(&self) -> u64 {
211        CACHE_TTL_SECS
212    }
213
214    fn is_available(&self) -> bool {
215        self.config.is_ok()
216    }
217}
218
219fn api_get(config: &GitLabConfig, endpoint: &str) -> Result<String, String> {
220    let url = config.api_url(endpoint);
221    let response = ureq::get(&url)
222        .header("PRIVATE-TOKEN", &config.token)
223        .call()
224        .map_err(|e| format!("GitLab API error: {e}"))?;
225
226    if response.status() != 200 {
227        return Err(format!("GitLab API returned status {}", response.status()));
228    }
229
230    response
231        .into_body()
232        .read_to_string()
233        .map_err(|e| format!("Failed to read response: {e}"))
234}
235
236fn parse_issue(v: &serde_json::Value) -> ProviderItem {
237    ProviderItem {
238        id: v["iid"].as_u64().unwrap_or(0).to_string(),
239        title: v["title"].as_str().unwrap_or("").to_string(),
240        state: v["state"].as_str().map(std::string::ToString::to_string),
241        author: v["author"]["username"]
242            .as_str()
243            .map(std::string::ToString::to_string),
244        created_at: v["created_at"]
245            .as_str()
246            .map(std::string::ToString::to_string),
247        updated_at: v["updated_at"]
248            .as_str()
249            .map(std::string::ToString::to_string),
250        url: v["web_url"].as_str().map(std::string::ToString::to_string),
251        labels: v["labels"]
252            .as_array()
253            .map(|arr| {
254                arr.iter()
255                    .filter_map(|l| l.as_str().map(std::string::ToString::to_string))
256                    .collect()
257            })
258            .unwrap_or_default(),
259        body: v["description"]
260            .as_str()
261            .map(std::string::ToString::to_string),
262    }
263}
264
265fn parse_mr(v: &serde_json::Value) -> ProviderItem {
266    ProviderItem {
267        id: v["iid"].as_u64().unwrap_or(0).to_string(),
268        title: v["title"].as_str().unwrap_or("").to_string(),
269        state: v["state"].as_str().map(std::string::ToString::to_string),
270        author: v["author"]["username"]
271            .as_str()
272            .map(std::string::ToString::to_string),
273        created_at: v["created_at"]
274            .as_str()
275            .map(std::string::ToString::to_string),
276        updated_at: v["updated_at"]
277            .as_str()
278            .map(std::string::ToString::to_string),
279        url: v["web_url"].as_str().map(std::string::ToString::to_string),
280        labels: v["labels"]
281            .as_array()
282            .map(|arr| {
283                arr.iter()
284                    .filter_map(|l| l.as_str().map(std::string::ToString::to_string))
285                    .collect()
286            })
287            .unwrap_or_default(),
288        body: v["description"]
289            .as_str()
290            .map(std::string::ToString::to_string),
291    }
292}
lean_ctx/core/providers/gitlab.rs

lean_ctx/core/providers/
gitlab.rs
