Ocean indexing scaffold

Author: Jouzo

lib/Cargo.lock

@@ -18,6 +18,8 @@ lto = true
 vsdb = { git = "https://github.com/defich/vsdb.git" }
 vsdbsled = { git = "https://github.com/defich/vsdbsled.git" }
 ethereum = { git = "https://github.com/defich/ethereum.git" }
+bitcoin = { git = "https://github.com/Jouzo/rust-bitcoin.git" }
+bitcoin-io = { git = "https://github.com/Jouzo/rust-bitcoin.git" }
 
 [workspace.dependencies]
 
@@ -115,3 +117,7 @@ lru = "0.12"
 #### Precompile dependencies
 sp-io = "24.0"
 substrate-bn = "0.6"
+
+#### Ocean dependencies
+dftx-rs = { git = "https://github.com/Jouzo/dftx-rs.git" }
+bitcoin = "0.31"

lib/Cargo.toml

@@ -12,3 +12,6 @@ keccak-hash.workspace = true
 log.workspace = true
 serde.workspace = true
 thiserror.workspace = true
+hex.workspace = true
+dftx-rs.workspace = true
+bitcoin.workspace = true

lib/ain-ocean/Cargo.toml

@@ -0,0 +1,14 @@
+use dftx_rs::{vault::PlaceAuctionBid, Transaction};
+
+use super::BlockContext;
+use crate::indexer::{Index, Result};
+
+impl Index for PlaceAuctionBid {
+    fn index(&self, context: &BlockContext, tx: Transaction) -> Result<()> {
+        todo!()
+    }
+
+    fn invalidate(&self) {
+        todo!()
+    }
+}

lib/ain-ocean/src/indexer/auction.rs

@@ -0,0 +1,83 @@
+use std::io::Write;
+
+use dftx_rs::{masternode::*, Transaction};
+use log::debug;
+
+use super::BlockContext;
+use crate::{
+    indexer::{Index, Result},
+    model::masternode::{Masternode, MasternodeBlock},
+};
+
+impl Index for CreateMasternode {
+    fn index(&self, context: &BlockContext, tx: Transaction) -> Result<()> {
+        debug!("[CreateMasternode] Indexing...");
+
+        let masternode = Masternode {
+            id: tx.txid().to_string(),
+            owner_address: tx.output[1].script_pubkey.to_hex_string(),
+            operator_address: tx.output[1].script_pubkey.to_hex_string(),
+            creation_height: context.height,
+            resign_height: -1,
+            resign_tx: None,
+            minted_blocks: 0,
+            timelock: self.timelock.0.unwrap_or_default(),
+            block: MasternodeBlock {
+                hash: context.hash.to_string(),
+                height: context.height,
+                time: context.time,
+                median_time: context.median_time,
+            },
+            collateral: tx.output[1].value.to_string(),
+            sort: None,
+            history: None,
+        };
+
+        let mut file = std::fs::OpenOptions::new()
+            .write(true)
+            .create(true)
+            .append(true)
+            .open("/tmp/masternode")
+            .expect("Unable to open file");
+
+        writeln!(file, "{:?}", masternode).expect("Unable to write to file");
+
+        Ok(())
+    }
+
+    fn invalidate(&self) {
+        todo!()
+    }
+}
+
+impl Index for UpdateMasternode {
+    fn index(&self, context: &BlockContext, tx: Transaction) -> Result<()> {
+        debug!("[UpdateMasternode] Indexing...");
+        // TODO
+        // Get mn
+        // Update fields
+        Ok(())
+    }
+
+    fn invalidate(&self) {
+        // TODO
+        // Get mn
+        // Restore from history
+    }
+}
+
+impl Index for ResignMasternode {
+    fn index(&self, context: &BlockContext, tx: Transaction) -> Result<()> {
+        debug!("[ResignMasternode] Indexing...");
+        // TODO
+        // Get mn
+        // Set resign tx and resign height
+        Ok(())
+    }
+
+    fn invalidate(&self) {
+        // TODO
+        // Get mn
+        // Set resign height to -1
+    }
+}

lib/ain-ocean/src/indexer/masternode.rs

@@ -0,0 +1,70 @@
+mod auction;
+mod masternode;
+mod oracle;
+mod pool;
+
+use dftx_rs::Transaction;
+
+pub(crate) type Result<T> = std::result::Result<T, Box<dyn std::error::Error>>;
+
+pub(crate) trait Index {
+    fn index(&self, context: &BlockContext, tx: Transaction) -> Result<()>;
+    fn invalidate(&self);
+}
+
+use bitcoin::BlockHash;
+use dftx_rs::{deserialize, Block, DfTx};
+
+pub(crate) struct BlockContext {
+    height: u32,
+    hash: BlockHash,
+    time: u64,
+    median_time: u64,
+}
+
+pub fn index_block(block: String, block_height: u32) -> Result<()> {
+    let hex = hex::decode(block)?;
+    let block = deserialize::<Block>(&hex)?;
+
+    let context = BlockContext {
+        height: block_height,
+        hash: block.block_hash(),
+        time: 0,        // TODO
+        median_time: 0, // TODO
+    };
+
+    for tx in block.txdata {
+        let bytes = tx.output[0].script_pubkey.as_bytes();
+        if bytes.len() > 2 && bytes[0] == 0x6a && bytes[1] <= 0x4e {
+            let offset = 1 + match bytes[1] {
+                0x4c => 2,
+                0x4d => 3,
+                0x4e => 4,
+                _ => 1,
+            };
+
+            let raw_tx = &bytes[offset..];
+
+            let dftx = deserialize::<DfTx>(raw_tx)?;
+
+            match dftx {
+                DfTx::CreateMasternode(data) => data.index(&context, tx)?,
+                DfTx::UpdateMasternode(data) => data.index(&context, tx)?,
+                DfTx::ResignMasternode(data) => data.index(&context, tx)?,
+                DfTx::AppointOracle(data) => data.index(&context, tx)?,
+                DfTx::RemoveOracle(data) => data.index(&context, tx)?,
+                DfTx::UpdateOracle(data) => data.index(&context, tx)?,
+                DfTx::SetOracleData(data) => data.index(&context, tx)?,
+                DfTx::PoolSwap(data) => data.index(&context, tx)?,
+                DfTx::CompositeSwap(data) => data.index(&context, tx)?,
+                DfTx::PlaceAuctionBid(data) => data.index(&context, tx)?,
+                _ => (),
+            }
+        }
+    }
+    Ok(())
+}
+
+pub fn invalidate_block() {
+    todo!()
+}

lib/ain-ocean/src/indexer/mod.rs

@@ -0,0 +1,44 @@
+use dftx_rs::{oracles::*, Transaction};
+
+use super::BlockContext;
+use crate::indexer::{Index, Result};
+
+impl Index for AppointOracle {
+    fn index(&self, context: &BlockContext, tx: Transaction) -> Result<()> {
+        todo!()
+    }
+
+    fn invalidate(&self) {
+        todo!()
+    }
+}
+
+impl Index for RemoveOracle {
+    fn index(&self, context: &BlockContext, tx: Transaction) -> Result<()> {
+        todo!()
+    }
+
+    fn invalidate(&self) {
+        todo!()
+    }
+}
+
+impl Index for UpdateOracle {
+    fn index(&self, context: &BlockContext, tx: Transaction) -> Result<()> {
+        todo!()
+    }
+
+    fn invalidate(&self) {
+        todo!()
+    }
+}
+
+impl Index for SetOracleData {
+    fn index(&self, context: &BlockContext, tx: Transaction) -> Result<()> {
+        todo!()
+    }
+
+    fn invalidate(&self) {
+        todo!()
+    }
+}

lib/ain-ocean/src/indexer/oracle.rs

@@ -0,0 +1,24 @@
+use dftx_rs::{pool::*, Transaction};
+
+use super::BlockContext;
+use crate::indexer::{Index, Result};
+
+impl Index for PoolSwap {
+    fn index(&self, context: &BlockContext, tx: Transaction) -> Result<()> {
+        todo!()
+    }
+
+    fn invalidate(&self) {
+        todo!()
+    }
+}
+
+impl Index for CompositeSwap {
+    fn index(&self, context: &BlockContext, tx: Transaction) -> Result<()> {
+        todo!()
+    }
+
+    fn invalidate(&self) {
+        todo!()
+    }
+}

lib/ain-ocean/src/indexer/pool.rs

@@ -1,6 +1,8 @@
 mod api;
 pub mod api_paged_response;
 pub mod error;
+mod indexer;
 mod model;
 
 pub use api::ocean_router;
+pub use indexer::{index_block, invalidate_block};

lib/ain-ocean/src/lib.rs

@@ -1,7 +1,4 @@
-use serde::{Serialize, Deserialize};
-
-#[derive(Serialize, Deserialize, Debug, Default)]
-#[serde(rename_all = "camelCase")]
+#[derive(Debug, Default)]
 pub struct Block {
     pub id: String,
     pub hash: String,
@@ -22,6 +19,3 @@ pub struct Block {
     pub size_stripped: i32,
     pub weight: i32,
 }
-
-
-