breez_sdk_core/persist/
reverseswap.rs

1use super::{db::SqliteStorage, error::PersistResult};
2use crate::{FullReverseSwapInfo, ReverseSwapInfoCached, ReverseSwapStatus};
3use rusqlite::{named_params, OptionalExtension, Row, TransactionBehavior};
4
5impl SqliteStorage {
6    pub(crate) fn insert_reverse_swap(&self, rsi: &FullReverseSwapInfo) -> PersistResult<()> {
7        let mut con = self.get_connection()?;
8        let tx = con.transaction_with_behavior(TransactionBehavior::Immediate)?;
9
10        tx.execute(
11            "INSERT INTO sync.reverse_swaps (id, created_at_block_height, preimage, private_key, claim_pubkey, timeout_block_height, invoice, onchain_amount_sat, sat_per_vbyte, receive_amount_sat, redeem_script)\
12            VALUES (:id, :created_at_block_height, :preimage, :private_key, :claim_pubkey, :timeout_block_height, :invoice, :onchain_amount_sat, :sat_per_vbyte, :receive_amount_sat, :redeem_script)",
13            named_params! {
14                ":id": rsi.id,
15                ":created_at_block_height": rsi.created_at_block_height,
16                ":preimage": rsi.preimage,
17                ":private_key": rsi.private_key,
18                ":claim_pubkey": rsi.claim_pubkey,
19                ":timeout_block_height": rsi.timeout_block_height,
20                ":invoice": rsi.invoice,
21                ":onchain_amount_sat": rsi.onchain_amount_sat,
22                ":sat_per_vbyte": rsi.sat_per_vbyte,
23                ":receive_amount_sat": rsi.receive_amount_sat,
24                ":redeem_script": rsi.redeem_script
25            },
26        )?;
27
28        tx.execute(
29            "INSERT INTO reverse_swaps_info (id, status)\
30            VALUES (:id, :status)",
31            named_params! {
32                ":id": rsi.id,
33                ":status": serde_json::to_value(rsi.cache.status)?
34            },
35        )?;
36
37        tx.commit()?;
38        Ok(())
39    }
40
41    pub(crate) fn update_reverse_swap_status(
42        &self,
43        id: &str,
44        status: &ReverseSwapStatus,
45    ) -> PersistResult<()> {
46        debug!("Persisting new status for reverse swap {id} to be {status:?}");
47
48        self.get_connection()?.execute(
49            "INSERT OR REPLACE INTO reverse_swaps_info(id, status) VALUES(:id, :status)",
50            named_params! {
51             ":status": serde_json::to_value(status)?,
52             ":id": id,
53            },
54        )?;
55
56        Ok(())
57    }
58
59    pub(crate) fn update_reverse_swap_lockup_txid(
60        &self,
61        id: &str,
62        lockup_txid: Option<String>,
63    ) -> PersistResult<()> {
64        debug!("Updating lockup_txid for reverse swap {id} to be - lockup_txid: {lockup_txid:?}");
65
66        self.get_connection()?.execute(
67            "UPDATE reverse_swaps_info SET lockup_txid = :lockup_txid WHERE id = :id",
68            named_params! {
69             ":id": id,
70             ":lockup_txid": lockup_txid,
71            },
72        )?;
73
74        Ok(())
75    }
76
77    pub(crate) fn update_reverse_swap_claim_txid(
78        &self,
79        id: &str,
80        claim_txid: Option<String>,
81    ) -> PersistResult<()> {
82        debug!("Updating claim_txid for reverse swap {id} to be - claim_txid: {claim_txid:?}");
83
84        self.get_connection()?.execute(
85            "UPDATE reverse_swaps_info SET claim_txid = :claim_txid WHERE id = :id",
86            named_params! {
87             ":id": id,
88             ":claim_txid": claim_txid,
89            },
90        )?;
91
92        Ok(())
93    }
94
95    pub(crate) fn get_reverse_swap(&self, id: &str) -> PersistResult<Option<FullReverseSwapInfo>> {
96        Ok(self
97            .get_connection()?
98            .query_row(
99                &self.select_reverse_swap_query("reverse_swaps.id = ?1", ""),
100                [id],
101                |row| self.sql_row_to_reverse_swap(row, ""),
102            )
103            .optional()?)
104    }
105
106    pub(crate) fn list_reverse_swaps(&self) -> PersistResult<Vec<FullReverseSwapInfo>> {
107        let con = self.get_connection()?;
108        let mut stmt = con.prepare(&self.select_reverse_swap_query("true", ""))?;
109
110        let vec: Vec<FullReverseSwapInfo> = stmt
111            .query_map([], |row| self.sql_row_to_reverse_swap(row, ""))?
112            .map(|i| i.unwrap())
113            .collect();
114
115        Ok(vec)
116    }
117
118    pub(crate) fn select_reverse_swap_fields(&self, prefix: &str) -> String {
119        format!(
120            "        
121        {prefix}id,
122        {prefix}created_at_block_height,
123        {prefix}preimage,
124        {prefix}private_key,
125        {prefix}timeout_block_height,
126        {prefix}claim_pubkey,
127        {prefix}invoice,
128        {prefix}onchain_amount_sat,
129        {prefix}sat_per_vbyte,
130        {prefix}receive_amount_sat,
131        {prefix}redeem_script,
132        {prefix}status,
133        {prefix}lockup_txid,
134        {prefix}claim_txid           
135        "
136        )
137    }
138
139    pub(crate) fn sql_row_to_reverse_swap(
140        &self,
141        row: &Row,
142        prefix: &str,
143    ) -> PersistResult<FullReverseSwapInfo, rusqlite::Error> {
144        Ok(FullReverseSwapInfo {
145            id: row.get(format!("{prefix}id").as_str())?,
146            created_at_block_height: row
147                .get(format!("{prefix}created_at_block_height").as_str())?,
148            preimage: row.get(format!("{prefix}preimage").as_str())?,
149            private_key: row.get(format!("{prefix}private_key").as_str())?,
150            timeout_block_height: row.get(format!("{prefix}timeout_block_height").as_str())?,
151            claim_pubkey: row.get(format!("{prefix}claim_pubkey").as_str())?,
152            invoice: row.get(format!("{prefix}invoice").as_str())?,
153            onchain_amount_sat: row.get(format!("{prefix}onchain_amount_sat").as_str())?,
154            sat_per_vbyte: row.get(format!("{prefix}sat_per_vbyte").as_str())?,
155            receive_amount_sat: row.get(format!("{prefix}receive_amount_sat").as_str())?,
156            redeem_script: row.get(format!("{prefix}redeem_script").as_str())?,
157            cache: ReverseSwapInfoCached {
158                // The status is stored in the main DB, which is empty when the node is restored.
159                // We therefore default to the Initial state. This will be updated at the end of sync().
160                status: serde_json::from_value(row.get(format!("{prefix}status").as_str())?)
161                    .unwrap_or(ReverseSwapStatus::Initial),
162                lockup_txid: row.get(format!("{prefix}lockup_txid").as_str())?,
163                claim_txid: row.get(format!("{prefix}claim_txid").as_str())?,
164            },
165        })
166    }
167
168    pub(crate) fn select_reverse_swap_query(&self, where_clause: &str, prefix: &str) -> String {
169        let fields = format!(
170            "        
171            reverse_swaps.id as {prefix}id,
172            created_at_block_height as {prefix}created_at_block_height,
173            preimage as {prefix}preimage,
174            private_key as {prefix}private_key,
175            timeout_block_height as {prefix}timeout_block_height,
176            claim_pubkey as {prefix}claim_pubkey,
177            invoice as {prefix}invoice,
178            onchain_amount_sat as {prefix}onchain_amount_sat,
179            sat_per_vbyte as {prefix}sat_per_vbyte,
180            receive_amount_sat as {prefix}receive_amount_sat,
181            redeem_script as {prefix}redeem_script,
182            status as {prefix}status,
183            lockup_txid as {prefix}lockup_txid,
184            claim_txid as {prefix}claim_txid         
185            "
186        );
187
188        format!(
189            "
190            SELECT
191             {fields}
192            FROM sync.reverse_swaps
193             LEFT JOIN reverse_swaps_info ON reverse_swaps.id = reverse_swaps_info.id
194            WHERE {}
195            ",
196            where_clause
197        )
198    }
199}
breez_sdk_core/persist/reverseswap.rs

breez_sdk_core/persist/
reverseswap.rs
