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

use serde::{Deserialize, Serialize};

use crate::{
    core::{
        api_req::ApiRequest,
        api_resp::{ApiResponseTrait, BaseResponse, ResponseFormat},
        constants::AccessTokenType,
        req_option::RequestOption,
        SDKResult,
    },
    service::sheets::v3::{spreadsheet_sheet::Sheet, SpreadsheetSheetService},
};

/// 获取工作表请求
#[derive(Serialize, Deserialize, Debug, Default)]
pub struct QuerySpreadsheetSheetRequest {
    #[serde(skip)]
    api_request: ApiRequest,
    #[serde(skip)]
    spreadsheet_token: String,
}

impl SpreadsheetSheetService {
    /// 获取工作表
    /// 根据电子表格 token 获取表格中所有工作表及其属性信息，包括工作表
    /// ID、标题、索引位置、是否被隐藏等。
    pub async fn query(
        &self,
        request: QuerySpreadsheetSheetRequest,
        option: Option<RequestOption>,
    ) -> SDKResult<BaseResponse<QuerySpreadsheetSheetResponse>> {
        let mut api_req = request.api_request;
        api_req.api_path = format!(
            "/open-apis/sheets/v3/spreadsheets/{spreadsheet_token}/sheets/query",
            spreadsheet_token = request.spreadsheet_token
        );
        api_req.http_method = reqwest::Method::GET;
        api_req.supported_access_token_types = vec![AccessTokenType::Tenant, AccessTokenType::User];

        let api_resp = crate::core::http::Transport::request(api_req, &self.config, option).await?;

        Ok(api_resp)
    }
}

impl QuerySpreadsheetSheetRequest {
    pub fn builder() -> QuerySpreadsheetSheetRequestBuilder {
        QuerySpreadsheetSheetRequestBuilder::default()
    }
}

#[derive(Default)]
pub struct QuerySpreadsheetSheetRequestBuilder {
    request: QuerySpreadsheetSheetRequest,
}

impl QuerySpreadsheetSheetRequestBuilder {
    /// 表格的token
    pub fn spreadsheet_token(mut self, spreadsheet_token: impl ToString) -> Self {
        self.request.spreadsheet_token = spreadsheet_token.to_string();
        self
    }

    pub fn build(mut self) -> QuerySpreadsheetSheetRequest {
        self.request.api_request.body = serde_json::to_vec(&self.request).unwrap();
        self.request
    }
}

#[derive(Deserialize, Debug)]
pub struct QuerySpreadsheetSheetResponse {
    pub sheets: Vec<Sheet>,
}

impl ApiResponseTrait for QuerySpreadsheetSheetResponse {
    fn data_format() -> ResponseFormat {
        ResponseFormat::Data
    }
}