Merge pull request #143 from dfinity/igor/diff-custom-domain

Add diff custom domain provider

Author: blind-oracle

Cargo.lock

@@ -48,7 +48,7 @@ http = "1.3.1"
 http-body = "1.0.1"
 http-body-util = "0.1.2"
 humantime = "2.2.0"
-ic-bn-lib = { git = "https://github.com/dfinity/ic-bn-lib", rev = "7f15e91888f76a6bfff9a59d478fdd11d76a5ff6", features = [
+ic-bn-lib = { git = "https://github.com/dfinity/ic-bn-lib", rev = "e23f4a8cf3bd7c06b8ef329d3c1e6117ea71b832", features = [
     "vector",
     "cert-providers",
     "clients-hyper",

Cargo.toml

@@ -262,6 +262,11 @@ pub struct Domain {
     #[clap(env, long, value_delimiter = ',')]
     pub domain_custom_provider_timestamped: Vec<Url>,
 
+    /// List of generic differential custom domain provider URLs.
+    /// It first downloads the full seed and then only applies incremental updates to it using a timestamp.
+    #[clap(env, long, value_delimiter = ',')]
+    pub domain_custom_provider_diff: Vec<Url>,
+
     /// How frequently to poll custom domain providers for updates
     #[clap(env, long, default_value = "30s", value_parser = parse_duration)]
     pub domain_custom_provider_poll_interval: Duration,

src/cli.rs

@@ -216,6 +216,16 @@ pub async fn main(
             }),
     );
 
+    custom_domain_providers.extend(cli.domain.domain_custom_provider_diff.iter().map(|x| {
+        warn!("Adding diff custom domain provider: {x}");
+
+        Arc::new(custom_domains::GenericProviderDiff::new(
+            http_client.clone(),
+            x.clone(),
+            cli.domain.domain_custom_provider_timeout,
+        )) as Arc<dyn ProvidesCustomDomains>
+    }));
+
     // Setup WAF
     let waf_layer = if cli.waf.waf_enable {
         let v = WafLayer::new_from_cli(&cli.waf, Some(http_client.clone()))