wrangler 1.21.0

Command-line interface for all things Cloudflare Workers
Documentation 
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

use cloudflare::endpoints::workerskv::list_namespaces::ListNamespaces;
use cloudflare::endpoints::workerskv::list_namespaces::ListNamespacesParams;
use cloudflare::endpoints::workerskv::WorkersKvNamespace;
use cloudflare::framework::apiclient::ApiClient;
use cloudflare::framework::response::ApiSuccess;

use anyhow::Result;
use serde::Deserialize;

use crate::commands::kv;
use crate::settings::toml::Target;

const MAX_NAMESPACES_PER_PAGE: u32 = 1000;

pub fn list(client: &impl ApiClient, target: &Target) -> Result<Vec<WorkersKvNamespace>> {
    let mut namespaces: Vec<WorkersKvNamespace> = Vec::new();
    let mut all_namespaces_added = false;
    let mut page_number = 1;
    while !all_namespaces_added {
        let params = ListNamespacesParams {
            page: Some(page_number),
            per_page: Some(MAX_NAMESPACES_PER_PAGE),
        };

        match client.request(&ListNamespaces {
            account_identifier: target.account_id.load()?,
            params,
        }) {
            Ok(response) => {
                namespaces.append(&mut response.result.clone());
                page_number += 1;
                all_namespaces_added = namespaces.len() >= get_total(&response)?;
            }
            Err(e) => anyhow::bail!("{}", kv::format_error(e)),
        }
    }
    Ok(namespaces)
}

fn get_total(list_response: &ApiSuccess<Vec<WorkersKvNamespace>>) -> Result<usize> {
    match list_response.result_info.clone() {
        Some(r) => {
            let result_info: ListResponseResultInfo = serde_json::from_value(r)?;
            Ok(result_info.total_count)
        }
        None => anyhow::bail!("KV list response lacks result_info field"),
    }
}

#[derive(Deserialize)]
struct ListResponseResultInfo {
    total_count: usize,
}