blueprint-manager 0.4.0-alpha.2

Tangle Blueprint manager and Runner
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

/// Common types used across the protocol abstraction layer
use blueprint_client_tangle::TangleEvent;
use blueprint_runner::config::{Protocol, ProtocolSettings};
use serde::{Deserialize, Serialize};

/// The type of protocol
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Serialize, Deserialize)]
pub enum ProtocolType {
    /// Tangle Network
    Tangle,
    /// EigenLayer AVS
    Eigenlayer,
}

impl From<Protocol> for ProtocolType {
    fn from(protocol: Protocol) -> Self {
        match protocol {
            Protocol::Tangle => ProtocolType::Tangle,
            Protocol::Eigenlayer => ProtocolType::Eigenlayer,
            _ => unreachable!("Protocol not supported"),
        }
    }
}

impl From<ProtocolType> for Protocol {
    fn from(protocol_type: ProtocolType) -> Self {
        match protocol_type {
            ProtocolType::Tangle => Protocol::Tangle,
            ProtocolType::Eigenlayer => Protocol::Eigenlayer,
        }
    }
}

impl From<&ProtocolSettings> for ProtocolType {
    fn from(settings: &ProtocolSettings) -> Self {
        match settings {
            ProtocolSettings::Eigenlayer(_) => ProtocolType::Eigenlayer,
            _ => ProtocolType::Tangle,
        }
    }
}

/// A protocol event containing block/event information
///
/// This is a unified type that can hold events from any protocol.
/// Protocol-specific handlers extract the data they need.
#[derive(Debug, Clone)]
pub enum ProtocolEvent {
    /// A Tangle network event
    Tangle(TangleProtocolEvent),
    /// An EigenLayer event (new tasks, responses, etc.)
    Eigenlayer(EigenlayerProtocolEvent),
}

/// Tangle-specific event data
#[derive(Debug, Clone)]
pub struct TangleProtocolEvent {
    /// Block number
    pub block_number: u64,
    /// Block hash
    pub block_hash: alloy_primitives::B256,
    /// Timestamp
    pub timestamp: u64,
    /// Logs in the block
    pub logs: Vec<alloy_rpc_types::Log>,
    /// Full event details
    pub inner: TangleEvent,
}

/// EigenLayer-specific event data
#[derive(Debug, Clone)]
pub struct EigenlayerProtocolEvent {
    /// Block number
    pub block_number: u64,
    /// Block hash
    pub block_hash: Vec<u8>,
    /// EVM logs for this block
    pub logs: Vec<alloy_rpc_types::Log>,
}

impl ProtocolEvent {
    /// Extract Tangle event data if this is a Tangle event
    #[must_use]
    pub fn as_tangle(&self) -> Option<&TangleProtocolEvent> {
        match self {
            ProtocolEvent::Tangle(evt) => Some(evt),
            _ => None,
        }
    }

    /// Extract EigenLayer event data if this is an EigenLayer event
    #[must_use]
    pub fn as_eigenlayer(&self) -> Option<&EigenlayerProtocolEvent> {
        match self {
            ProtocolEvent::Eigenlayer(evt) => Some(evt),
            _ => None,
        }
    }

    /// Get the block number for any protocol event
    #[must_use]
    pub fn block_number(&self) -> u64 {
        match self {
            ProtocolEvent::Tangle(evt) => evt.block_number,
            ProtocolEvent::Eigenlayer(evt) => evt.block_number,
        }
    }
}