open-lark 0.14.0

use reqwest::Method;
use serde::{Deserialize, Serialize};

use crate::{
    core::{
        api_req::ApiRequest,
        api_resp::{ApiResponseTrait, BaseResponse, ResponseFormat},
        config::Config,
        constants::AccessTokenType,
        endpoints::report::*,
        http::Transport,
        req_option::RequestOption,
        SDKResult,
    },
    service::report::models::{PageResponse, ReportTask},
};

/// 任务管理服务
pub struct TaskService {
    pub config: Config,
}

impl TaskService {
    /// 创建任务管理服务实例
    pub fn new(config: Config) -> Self {
        Self { config }
    }

    /// 查询任务
    ///
    /// 查询汇报任务列表，支持分页和条件筛选。
    ///
    /// # Arguments
    ///
    /// * `request` - 任务查询请求
    /// * `option` - 请求选项，可选
    ///
    /// # Returns
    ///
    /// 返回任务列表
    pub async fn query(
        &self,
        request: TaskQueryRequest,
        option: Option<RequestOption>,
    ) -> SDKResult<BaseResponse<TaskQueryResponse>> {
        let mut api_req = ApiRequest {
            http_method: Method::GET,
            api_path: REPORT_V1_TASKS_QUERY.to_string(),
            supported_access_token_types: vec![AccessTokenType::Tenant, AccessTokenType::User],
            body: vec![],
            ..Default::default()
        };

        // 添加查询参数
        if let Some(page_token) = request.page_token {
            api_req.query_params.insert("page_token", page_token);
        }

        if let Some(page_size) = request.page_size {
            api_req
                .query_params
                .insert("page_size", page_size.to_string());
        }

        if let Some(rule_id) = request.rule_id {
            api_req.query_params.insert("rule_id", rule_id);
        }

        if let Some(status) = request.status {
            api_req.query_params.insert("status", status);
        }

        if let Some(task_type) = request.task_type {
            api_req.query_params.insert("task_type", task_type);
        }

        if let Some(reporter_id) = request.reporter_id {
            api_req.query_params.insert("reporter_id", reporter_id);
        }

        if let Some(start_time) = request.start_time {
            api_req
                .query_params
                .insert("start_time", start_time.to_string());
        }

        if let Some(end_time) = request.end_time {
            api_req
                .query_params
                .insert("end_time", end_time.to_string());
        }

        Transport::request(api_req, &self.config, option).await
    }
}

/// 任务查询请求
#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct TaskQueryRequest {
    /// 页码标记
    #[serde(skip_serializing_if = "Option::is_none")]
    pub page_token: Option<String>,
    /// 每页数量
    #[serde(skip_serializing_if = "Option::is_none")]
    pub page_size: Option<i32>,
    /// 规则ID筛选
    #[serde(skip_serializing_if = "Option::is_none")]
    pub rule_id: Option<String>,
    /// 任务状态筛选
    #[serde(skip_serializing_if = "Option::is_none")]
    pub status: Option<String>,
    /// 任务类型筛选
    #[serde(skip_serializing_if = "Option::is_none")]
    pub task_type: Option<String>,
    /// 汇报者ID筛选
    #[serde(skip_serializing_if = "Option::is_none")]
    pub reporter_id: Option<String>,
    /// 开始时间戳筛选
    #[serde(skip_serializing_if = "Option::is_none")]
    pub start_time: Option<i64>,
    /// 结束时间戳筛选
    #[serde(skip_serializing_if = "Option::is_none")]
    pub end_time: Option<i64>,
}

/// 任务查询响应
#[derive(Debug, Serialize, Deserialize)]
pub struct TaskQueryResponse {
    /// 任务列表
    #[serde(flatten)]
    pub tasks: PageResponse<ReportTask>,
}

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