blueprint-runner 0.2.0-alpha.2

//! Tangle Protocol Configuration
//!
//! Provides configuration for running blueprints on Tangle EVM contracts.

use alloy_primitives::Address;
use serde::{Deserialize, Serialize};
use std::error::Error;
use std::sync::{Arc, Mutex};

use crate::BlueprintConfig;
use crate::config::{BlueprintEnvironment, BlueprintSettings, Protocol, ProtocolSettingsT};
use crate::error::RunnerError;

/// Protocol settings for Tangle
///
/// This contains the EVM-specific configuration for connecting to Tangle contracts.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct TangleProtocolSettings {
    /// The blueprint ID registered in the Tangle contract
    pub blueprint_id: u64,
    /// The service ID for the Tangle blueprint instance
    ///
    /// Note: This will be `None` if running in Registration Mode.
    pub service_id: Option<u64>,
    /// The Tangle core contract address
    pub tangle_contract: Address,
    /// The MultiAssetDelegation (restaking) contract address
    pub restaking_contract: Address,
    /// The operator status registry contract used for heartbeats
    pub status_registry_contract: Address,
}

impl Default for TangleProtocolSettings {
    fn default() -> Self {
        Self {
            blueprint_id: 0,
            service_id: None,
            // Default to zero address - must be configured
            tangle_contract: Address::ZERO,
            restaking_contract: Address::ZERO,
            status_registry_contract: Address::ZERO,
        }
    }
}

impl ProtocolSettingsT for TangleProtocolSettings {
    fn load(_settings: BlueprintSettings) -> Result<Self, Box<dyn Error + Send + Sync>> {
        use crate::error::ConfigError;

        // Parse blueprint_id from environment
        let blueprint_id: u64 = std::env::var("BLUEPRINT_ID")
            .map_err(|_| ConfigError::MissingBlueprintId)?
            .parse()
            .map_err(|_| ConfigError::MissingBlueprintId)?;

        // Parse service_id from environment (optional)
        let service_id: Option<u64> = std::env::var("SERVICE_ID")
            .ok()
            .and_then(|s| s.parse().ok());

        // Parse contract addresses from environment
        let tangle_contract = std::env::var("TANGLE_CONTRACT")
            .ok()
            .and_then(|s| s.parse().ok())
            .unwrap_or(Address::ZERO);

        let restaking_contract = std::env::var("RESTAKING_CONTRACT")
            .ok()
            .and_then(|s| s.parse().ok())
            .unwrap_or(Address::ZERO);

        let status_registry_contract = std::env::var("STATUS_REGISTRY_CONTRACT")
            .ok()
            .and_then(|s| s.parse().ok())
            .unwrap_or(Address::ZERO);

        Ok(Self {
            blueprint_id,
            service_id,
            tangle_contract,
            restaking_contract,
            status_registry_contract,
        })
    }

    fn protocol_name(&self) -> &'static str {
        "tangle"
    }

    fn protocol(&self) -> Protocol {
        Protocol::Tangle
    }
}

/// Runtime configuration for Tangle blueprints
#[derive(Clone, Debug)]
pub struct TangleConfig {
    /// RPC endpoint for operator registration announcements
    pub rpc_address: String,
    /// Whether to exit after registration
    pub exit_after_register: bool,
    /// Custom registration inputs supplied by the blueprint
    registration_inputs: Arc<Mutex<Vec<u8>>>,
}

impl Default for TangleConfig {
    fn default() -> Self {
        Self::new("")
    }
}

impl TangleConfig {
    /// Create a new TangleConfig with the given RPC address
    #[must_use]
    pub fn new(rpc_address: impl Into<String>) -> Self {
        Self {
            rpc_address: rpc_address.into(),
            exit_after_register: true,
            registration_inputs: Arc::new(Mutex::new(Vec::new())),
        }
    }

    /// Set whether to exit after registration
    #[must_use]
    pub fn with_exit_after_register(mut self, should_exit: bool) -> Self {
        self.exit_after_register = should_exit;
        self
    }

    /// Provide custom registration inputs (TLV payload) for the blueprint.
    #[must_use]
    pub fn with_registration_inputs(self, inputs: impl Into<Vec<u8>>) -> Self {
        self.set_registration_inputs(inputs);
        self
    }

    /// Update the registration payload in-place.
    pub fn set_registration_inputs(&self, inputs: impl Into<Vec<u8>>) {
        let mut guard = self
            .registration_inputs
            .lock()
            .unwrap_or_else(|err| err.into_inner());
        *guard = inputs.into();
    }

    /// Accessor for the current registration inputs.
    #[must_use]
    pub fn registration_inputs(&self) -> Vec<u8> {
        match self.registration_inputs.lock() {
            Ok(guard) => guard.clone(),
            Err(err) => err.into_inner().clone(),
        }
    }
}

impl BlueprintConfig for TangleConfig {
    async fn register(&self, env: &BlueprintEnvironment) -> Result<(), RunnerError> {
        let inputs = self.registration_inputs();
        register_impl(&self.rpc_address, &inputs, env).await
    }

    async fn requires_registration(&self, env: &BlueprintEnvironment) -> Result<bool, RunnerError> {
        requires_registration_impl(env).await
    }

    fn should_exit_after_registration(&self) -> bool {
        self.exit_after_register
    }

    fn update_registration_inputs(&self, inputs: Vec<u8>) -> Result<(), RunnerError> {
        self.set_registration_inputs(inputs);
        Ok(())
    }
}

/// Check if operator registration is required
async fn requires_registration_impl(env: &BlueprintEnvironment) -> Result<bool, RunnerError> {
    use super::error::TangleError;
    use blueprint_client_tangle::{TangleClient, TangleClientConfig, TangleSettings};

    let settings = env.protocol_settings.tangle()?;

    // Create the client config from environment
    let client_config = TangleClientConfig {
        http_rpc_endpoint: env.http_rpc_endpoint.clone(),
        ws_rpc_endpoint: env.ws_rpc_endpoint.clone(),
        settings: TangleSettings {
            blueprint_id: settings.blueprint_id,
            service_id: settings.service_id,
            tangle_contract: settings.tangle_contract,
            restaking_contract: settings.restaking_contract,
            status_registry_contract: settings.status_registry_contract,
        },
        keystore_uri: env.keystore_uri.clone(),
        data_dir: env.data_dir.clone(),
        test_mode: env.test_mode,
        dry_run: env.dry_run,
    };

    // Create the EVM client
    let client = TangleClient::new(client_config)
        .await
        .map_err(|e| TangleError::Contract(e.to_string()))?;

    // Check if operator is registered for the blueprint
    let is_registered = client
        .is_operator_registered(settings.blueprint_id, client.account())
        .await
        .map_err(|e| TangleError::Contract(e.to_string()))?;

    Ok(!is_registered)
}

/// Register the operator on the Tangle contract
async fn register_impl(
    rpc_address: &str,
    registration_inputs: &[u8],
    env: &BlueprintEnvironment,
) -> Result<(), RunnerError> {
    use super::error::TangleError;
    use alloy_primitives::{B256, Bytes};
    use alloy_provider::ProviderBuilder;
    use alloy_signer_local::PrivateKeySigner;
    use blueprint_client_tangle::contracts::ITangle;
    use blueprint_crypto::k256::K256Ecdsa;
    use blueprint_keystore::backends::Backend;
    use blueprint_keystore::backends::eigenlayer::EigenlayerBackend;

    let settings = env.protocol_settings.tangle()?;

    blueprint_core::info!(
        "Starting Tangle registration: blueprint_id={}, rpc_address={}, tangle_contract={:?}",
        settings.blueprint_id,
        rpc_address,
        settings.tangle_contract
    );

    // 1. Get ECDSA key from keystore
    let ecdsa_public = env
        .keystore()
        .first_local::<K256Ecdsa>()
        .map_err(|e| TangleError::Keystore(e.to_string()))?;

    let ecdsa_secret = env
        .keystore()
        .expose_ecdsa_secret(&ecdsa_public)
        .map_err(|e| TangleError::Keystore(e.to_string()))?
        .ok_or_else(|| TangleError::Keystore("No ECDSA secret found in keystore".into()))?;

    // 2. Create wallet/signer from secret key
    let secret_bytes = ecdsa_secret.0.to_bytes();
    let secret_b256 = B256::from_slice(&secret_bytes);
    let wallet = PrivateKeySigner::from_bytes(&secret_b256)
        .map_err(|e| TangleError::Keystore(format!("Failed to create signer: {e}")))?;

    let operator_address = wallet.address();
    blueprint_core::info!("Operator address: {}", operator_address);

    // 3. Create provider with signer
    let provider = ProviderBuilder::new()
        .wallet(wallet)
        .connect(env.http_rpc_endpoint.as_str())
        .await
        .map_err(|e| TangleError::Contract(format!("Failed to connect to RPC: {e}")))?;

    // 4. Create contract instance with signed provider
    let tangle_contract = ITangle::new(settings.tangle_contract, &provider);

    // 5. Prepare operator preferences
    let ecdsa_point = ecdsa_public.0.to_encoded_point(false);
    let ecdsa_bytes = Bytes::copy_from_slice(ecdsa_point.as_bytes());

    blueprint_core::info!(
        "Sending registerOperator transaction for blueprint_id={}",
        settings.blueprint_id,
    );

    let pending_tx = if registration_inputs.is_empty() {
        tangle_contract
            .registerOperator_1(settings.blueprint_id, ecdsa_bytes, rpc_address.to_string())
            .send()
            .await
    } else {
        let inputs = Bytes::copy_from_slice(registration_inputs);
        tangle_contract
            .registerOperator_0(
                settings.blueprint_id,
                ecdsa_bytes,
                rpc_address.to_string(),
                inputs,
            )
            .send()
            .await
    }
    .map_err(|e| TangleError::Transaction(format!("Failed to send transaction: {e}")))?;

    blueprint_core::info!(
        "Transaction sent, waiting for confirmation: {:?}",
        pending_tx.tx_hash()
    );

    // 6. Wait for transaction confirmation
    let receipt = pending_tx
        .get_receipt()
        .await
        .map_err(|e| TangleError::Transaction(format!("Failed to get receipt: {e}")))?;

    if !receipt.status() {
        return Err(TangleError::Transaction("Transaction reverted".into()).into());
    }

    blueprint_core::info!(
        "Registration successful! tx_hash={:?}, block={:?}",
        receipt.transaction_hash,
        receipt.block_number
    );

    Ok(())
}