Check for balances spent to op_return in assert_runes

Author: lifofifoX

src/index.rs

@@ -499,6 +499,23 @@ impl Index {
     )
   }
 
+  pub fn get_output_script(&self, output: &OutPoint) -> Result<Option<ScriptBuf>> {
+    let rtx = self.database.begin_read()?;
+    let outpoint_to_utxo_entry = rtx.open_table(OUTPOINT_TO_UTXO_ENTRY)?;
+
+    let entry = outpoint_to_utxo_entry
+      .get(&output.store())?
+      .ok_or_else(|| anyhow!("UTXO entry not found for output: {:?}", output))?;
+
+    let parsed = entry.value().parse(self);
+
+    Ok(
+      parsed
+        .script_pubkey()
+        .map(|bytes| ScriptBuf::from_bytes(bytes.to_vec())),
+    )
+  }
+
   pub fn has_address_index(&self) -> bool {
     self.index_addresses
   }
@@ -744,6 +761,7 @@ impl Index {
                 .value()
                 .parse(self)
                 .script_pubkey()
+                .unwrap()
                 .to_vec(),
             )
           } else {
@@ -6621,7 +6639,7 @@ mod tests {
 
     let (event_sender, mut event_receiver) = tokio::sync::mpsc::channel(1024);
     let context = Context::builder()
-      .arg("--index-runes")
+      .args(["--index-runes", "--index-addresses"])
       .event_sender(event_sender)
       .build();
 

src/index/testing.rs

@@ -128,7 +128,22 @@ impl Context {
     let balances = balances.as_mut();
     balances.sort_by_key(|(outpoint, _)| *outpoint);
 
-    for (_, balances) in balances.iter_mut() {
+    let mut burned_balances = 0;
+
+    for (outpoint, balances) in balances.iter_mut() {
+      if self
+        .index
+        .get_output_script(outpoint)
+        .ok()
+        .flatten()
+        .map_or(false, |script| script.is_op_return())
+      {
+        for (_, balance) in balances.iter() {
+          println!("burned_balances: {}", burned_balances);
+          burned_balances += *balance;
+        }
+      }
+
       balances.sort_by_key(|(id, _)| *id);
     }
 
@@ -146,7 +161,7 @@ impl Context {
 
     for (id, entry) in runes {
       pretty_assert_eq!(
-        outstanding.get(id).copied().unwrap_or_default(),
+        outstanding.get(id).copied().unwrap_or_default() - burned_balances,
         entry.supply() - entry.burned
       );
     }

src/index/updater.rs

@@ -556,7 +556,7 @@ impl Updater<'_> {
               entry
             } else if let Some(entry) = outpoint_to_utxo_entry.remove(&outpoint)? {
               if self.index.index_addresses {
-                let script_pubkey = entry.value().parse(self.index).script_pubkey();
+                let script_pubkey = entry.value().parse(self.index).script_pubkey().unwrap();
                 if !script_pubkey_to_outpoint.remove(script_pubkey, outpoint)? {
                   panic!("script pubkey entry ({script_pubkey:?}, {outpoint:?}) not found");
                 }
@@ -845,7 +845,7 @@ impl Updater<'_> {
 
         let utxo_entry = utxo_entry.parse(self.index);
         if self.index.index_addresses {
-          let script_pubkey = utxo_entry.script_pubkey();
+          let script_pubkey = utxo_entry.script_pubkey().unwrap();
           script_pubkey_to_outpoint.insert(script_pubkey, &outpoint.store())?;
         }
 

src/index/utxo_entry.rs

@@ -153,8 +153,8 @@ impl<'a> ParsedUtxoEntry<'a> {
     ranges
   }
 
-  pub fn script_pubkey(&self) -> &'a [u8] {
-    self.script_pubkey.unwrap()
+  pub fn script_pubkey(&self) -> Option<&'a [u8]> {
+    self.script_pubkey
   }
 
   pub fn inscriptions(&self) -> &'a [u8] {
@@ -279,8 +279,8 @@ impl UtxoEntryBuf {
     }
 
     if index.index_addresses {
-      assert!(a_parsed.script_pubkey().is_empty());
-      assert!(b_parsed.script_pubkey().is_empty());
+      assert!(a_parsed.script_pubkey().unwrap().is_empty());
+      assert!(b_parsed.script_pubkey().unwrap().is_empty());
       merged.push_script_pubkey(&[], index);
     }