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

use blockstack_lib::burnchains::Txid;
use blockstack_lib::types::chainstate::BurnchainHeaderHash;

use crate::stacks_node;

mod sqlite_peg_queue;

pub use sqlite_peg_queue::SqlitePegQueue;

pub trait PegQueue {
    type Error: std::error::Error;

    fn sbtc_op(&self) -> Result<Option<SbtcOp>, Self::Error>;
    fn poll<N: stacks_node::StacksNode>(&self, stacks_node: &N) -> Result<(), Self::Error>;

    fn acknowledge(
        &self,
        txid: &Txid,
        burn_header_hash: &BurnchainHeaderHash,
    ) -> Result<(), Self::Error>;
}

#[derive(Debug, serde::Serialize, serde::Deserialize)]
pub enum SbtcOp {
    PegIn(stacks_node::PegInOp),
    PegOutRequest(stacks_node::PegOutRequestOp),
}

impl SbtcOp {
    pub fn as_peg_in(&self) -> Option<&stacks_node::PegInOp> {
        match self {
            Self::PegIn(op) => Some(op),
            _ => None,
        }
    }

    pub fn as_peg_out_request(&self) -> Option<&stacks_node::PegOutRequestOp> {
        match self {
            Self::PegOutRequest(op) => Some(op),
            _ => None,
        }
    }
}