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

use async_stream::try_stream;
use futures::{
  stream::Stream,
  {self},
};
use log::trace;
use reqwest::{Method, RequestBuilder, StatusCode};
use serde::{Deserialize, Serialize};

use crate::{errors::Result, v2};

#[derive(Debug, Default, Deserialize, Serialize)]
struct Catalog {
  pub repositories: Vec<String>,
}

impl v2::Client {
  pub fn get_catalog<'a, 'b: 'a>(&'b self, paginate: Option<u32>) -> impl Stream<Item = Result<String>> + 'a {
    let url = {
      let suffix = if let Some(n) = paginate {
        format!("?n={}", n)
      } else {
        "".to_string()
      };
      let ep = format!("{}/v2/_catalog{}", self.base_url.clone(), suffix);

      reqwest::Url::parse(&ep).map_err(crate::Error::from)
    };

    try_stream! {
        let req = self.build_reqwest(Method::GET, url?);

        let catalog = fetch_catalog(req).await?;

        for repo in catalog.repositories {
            yield repo;
        }
    }
  }
}

async fn fetch_catalog(req: RequestBuilder) -> Result<Catalog> {
  let r = req.send().await?;
  let status = r.status();
  trace!("Got status: {:?}", status);
  match status {
    StatusCode::OK => r.json::<Catalog>().await.map_err(Into::into),
    _ => Err(crate::Error::UnexpectedHttpStatus(status)),
  }
}