dns01_client.rs

// SPDX-FileCopyrightText: © 2024 Phala Network <dstack@phala.network>
//
// SPDX-License-Identifier: Apache-2.0

use anyhow::Result;
use cloudflare::CloudflareClient;
use enum_dispatch::enum_dispatch;
use serde::{Deserialize, Serialize};
use tracing::debug;

mod cloudflare;

#[derive(Debug, Deserialize, Serialize)]
/// Represents a DNS record
pub(crate) struct Record {
    /// Unique identifier for the record
    pub id: String,
    /// The name of the DNS record (e.g., "_acme-challenge.example.com")
    pub name: String,
    /// The content of the DNS record (e.g., the TXT value for ACME challenges)
    pub content: String,
    /// The type of DNS record (e.g., "TXT" for ACME challenges)
    pub r#type: String,
}

#[enum_dispatch]
pub(crate) trait Dns01Api {
    /// Creates a TXT DNS record with the given domain and content.
    ///
    /// Returns the ID of the created record.
    /// The `ttl` parameter specifies the time-to-live in seconds (1 = auto, min 60 for Cloudflare).
    async fn add_txt_record(&self, domain: &str, content: &str, ttl: u32) -> Result<String>;

    /// Add a CAA record for the given domain.
    async fn add_caa_record(
        &self,
        domain: &str,
        flags: u8,
        tag: &str,
        value: &str,
    ) -> Result<String>;

    /// Remove a DNS record.
    ///
    /// Deletes a DNS record using its unique identifier.
    async fn remove_record(&self, record_id: &str) -> Result<()>;

    /// Get all records for a domain.
    async fn get_records(&self, domain: &str) -> Result<Vec<Record>>;

    /// Remove TXT DNS records by domain.
    ///
    /// Deletes all TXT DNS records matching the given domain.
    async fn remove_txt_records(&self, domain: &str) -> Result<()> {
        for record in self.get_records(domain).await? {
            if record.r#type != "TXT" {
                continue;
            }
            debug!(domain = %domain, id = %record.id, "removing txt record");
            self.remove_record(&record.id).await?;
        }
        Ok(())
    }
}

/// A DNS-01 client.
#[derive(Debug, Serialize, Deserialize)]
#[enum_dispatch(Dns01Api)]
#[serde(rename_all = "lowercase")]
pub enum Dns01Client {
    Cloudflare(CloudflareClient),
}

impl Dns01Client {
    pub async fn new_cloudflare(
        base_domain: String,
        api_token: String,
        api_url: Option<String>,
    ) -> Result<Self> {
        let client = CloudflareClient::new(base_domain, api_token, api_url).await?;
        Ok(Self::Cloudflare(client))
    }
}