dsc/api/
categories.rs

1use super::client::DiscourseClient;
2use super::error::http_error;
3use super::models::{CategoriesResponse, CategoryInfo, CategoryResponse, CreateCategoryResponse};
4use anyhow::{Context, Result, anyhow};
5use reqwest::StatusCode;
6use serde_json::Value;
7use std::collections::HashMap;
8
9impl DiscourseClient {
10    /// Fetch a category by ID (topics list included).
11    pub fn fetch_category(&self, category_id: u64) -> Result<CategoryResponse> {
12        let path = format!("/c/{}.json", category_id);
13        let response = self.get(&path)?;
14        let status = response.status();
15        let text = response.text().context("reading category response body")?;
16        if !status.is_success() {
17            if status == StatusCode::NOT_FOUND {
18                return Err(anyhow!("category not found: {}", category_id));
19            }
20            return Err(http_error("category request", status, &text));
21        }
22        let body: CategoryResponse =
23            serde_json::from_str(&text).context("reading category json")?;
24        Ok(body)
25    }
26
27    /// Fetch all categories.
28    pub fn fetch_categories(&self) -> Result<Vec<CategoryInfo>> {
29        let response = self.get("/categories.json?include_subcategories=true")?;
30        let status = response.status();
31        let text = response
32            .text()
33            .context("reading categories response body")?;
34        if !status.is_success() {
35            return Err(http_error("categories request", status, &text));
36        }
37        let body: CategoriesResponse =
38            serde_json::from_str(&text).context("reading categories json")?;
39        let mut categories = body.category_list.categories;
40        if let Ok(site_categories) = self.fetch_site_categories() {
41            let mut seen = HashMap::new();
42            for (idx, cat) in categories.iter().enumerate() {
43                if let Some(id) = cat.id {
44                    seen.insert(id, idx);
45                }
46            }
47            for cat in site_categories {
48                if let Some(id) = cat.id {
49                    if !seen.contains_key(&id) {
50                        categories.push(cat);
51                    }
52                }
53            }
54        }
55        Ok(categories)
56    }
57
58    /// Create a category with basic fields copied from a source category.
59    pub fn create_category(&self, category: &CategoryInfo) -> Result<u64> {
60        let mut payload = vec![("name", category.name.clone())];
61        if !category.slug.is_empty() {
62            payload.push(("slug", category.slug.clone()));
63        }
64        if let Some(color) = category.color.clone() {
65            payload.push(("color", color));
66        }
67        if let Some(text_color) = category.text_color.clone() {
68            payload.push(("text_color", text_color));
69        }
70        let response = self.send_retrying(|| Ok(self.post("/categories")?.form(&payload)))?;
71        let status = response.status();
72        let text = response.text().context("reading category response body")?;
73        if !status.is_success() {
74            return Err(http_error("create category request", status, &text));
75        }
76        let body: CreateCategoryResponse =
77            serde_json::from_str(&text).context("reading category response")?;
78        Ok(body.category.id)
79    }
80
81    fn fetch_site_categories(&self) -> Result<Vec<CategoryInfo>> {
82        let response = self.get("/site.json")?;
83        let status = response.status();
84        let text = response.text().context("reading site.json response body")?;
85        if !status.is_success() {
86            return Err(http_error("site.json request", status, &text));
87        }
88        let value: Value = serde_json::from_str(&text).context("parsing site.json")?;
89        let array = value
90            .get("categories")
91            .and_then(|v| v.as_array())
92            .or_else(|| {
93                value
94                    .get("site")
95                    .and_then(|v| v.get("categories"))
96                    .and_then(|v| v.as_array())
97            })
98            .ok_or_else(|| anyhow!("site.json missing categories list"))?;
99        let mut categories = Vec::new();
100        for item in array {
101            if let Ok(cat) = serde_json::from_value::<CategoryInfo>(item.clone()) {
102                categories.push(cat);
103            }
104        }
105        Ok(categories)
106    }
107}
dsc/api/categories.rs

dsc/api/
categories.rs
