feat(ampd): add Stacks handlers with custom ITS logic & verify logic for contract deployments

ampd/Cargo.toml

@@ -12,6 +12,7 @@ axum = "0.7.5"
 base64 = "0.21.2"
 bcs = { workspace = true }
 clap = { version = "4.2.7", features = ["derive", "cargo"] }
+clarity = { git = "https://github.com/stacks-network/stacks-core", tag = "2.5.0.0.7", default-features = false, features = ["slog_json"] }
 config = "0.13.2"
 cosmrs = { version = "0.14.0", features = ["cosmwasm", "grpc"] }
 cosmwasm-std = { workspace = true, features = ["stargate"] }

ampd/src/config.rs

@@ -132,6 +132,19 @@ mod tests {
             type = 'StellarVerifierSetVerifier'
             cosmwasm_contract = '{}'
             http_url = 'http://localhost:8000'
+
+            [[handlers]]
+            type = 'StacksMsgVerifier'
+            cosmwasm_contract = '{}'
+            http_url = 'http://localhost:8000'
+            its_address = 'its_address'
+            reference_native_interchain_token_address = 'interchain_token_address'
+            reference_token_manager_address = 'token_manager_address'
+
+            [[handlers]]
+            type = 'StacksVerifierSetVerifier'
+            cosmwasm_contract = '{}'
+            http_url = 'http://localhost:8000'
             ",
             TMAddress::random(PREFIX),
             TMAddress::random(PREFIX),
@@ -143,10 +156,12 @@ mod tests {
             TMAddress::random(PREFIX),
             TMAddress::random(PREFIX),
             TMAddress::random(PREFIX),
+            TMAddress::random(PREFIX),
+            TMAddress::random(PREFIX),
         );
 
         let cfg: Config = toml::from_str(config_str.as_str()).unwrap();
-        assert_eq!(cfg.handlers.len(), 10);
+        assert_eq!(cfg.handlers.len(), 12);
     }
 
     #[test]
@@ -350,6 +365,22 @@ mod tests {
                     ),
                     http_url: Url::from_str("http://127.0.0.1").unwrap(),
                 },
+                HandlerConfig::StacksMsgVerifier {
+                    cosmwasm_contract: TMAddress::from(
+                        AccountId::new("axelar", &[0u8; 32]).unwrap(),
+                    ),
+                    http_url: Url::from_str("http://127.0.0.1").unwrap(),
+                    its_address: "its_address".to_string(),
+                    reference_native_interchain_token_address: "interchain_token_address"
+                        .to_string(),
+                    reference_token_manager_address: "token_manager_address".to_string(),
+                },
+                HandlerConfig::StacksVerifierSetVerifier {
+                    cosmwasm_contract: TMAddress::from(
+                        AccountId::new("axelar", &[0u8; 32]).unwrap(),
+                    ),
+                    http_url: Url::from_str("http://127.0.0.1").unwrap(),
+                },
             ],
             ..Config::default()
         }

ampd/src/handlers/config.rs

@@ -63,6 +63,17 @@ pub enum Config {
         cosmwasm_contract: TMAddress,
         http_url: Url,
     },
+    StacksMsgVerifier {
+        cosmwasm_contract: TMAddress,
+        http_url: Url,
+        its_address: String,
+        reference_native_interchain_token_address: String,
+        reference_token_manager_address: String,
+    },
+    StacksVerifierSetVerifier {
+        cosmwasm_contract: TMAddress,
+        http_url: Url,
+    },
 }
 
 fn validate_evm_verifier_set_verifier_configs<'de, D>(configs: &[Config]) -> Result<(), D::Error>
@@ -159,6 +170,16 @@ where
         Config::StellarVerifierSetVerifier,
         "Stellar verifier set verifier"
     )?;
+    ensure_unique_config!(
+        &configs,
+        Config::StacksMsgVerifier,
+        "Stacks message verifier"
+    )?;
+    ensure_unique_config!(
+        &configs,
+        Config::StacksVerifierSetVerifier,
+        "Stacks verifier set verifier"
+    )?;
 
     Ok(configs)
 }
@@ -305,5 +326,45 @@ mod tests {
                 Err(e) if e.to_string().contains("only one Stellar verifier set verifier config is allowed")
             )
         );
+
+        let configs = vec![
+            Config::StacksMsgVerifier {
+                cosmwasm_contract: TMAddress::random(PREFIX),
+                http_url: "http://localhost:8080/".parse().unwrap(),
+                its_address: "its_address".to_string(),
+                reference_native_interchain_token_address: "interchain_token_address".to_string(),
+                reference_token_manager_address: "token_manager_address".to_string(),
+            },
+            Config::StacksMsgVerifier {
+                cosmwasm_contract: TMAddress::random(PREFIX),
+                http_url: "http://localhost:8080/".parse().unwrap(),
+                its_address: "its_address".to_string(),
+                reference_native_interchain_token_address: "interchain_token_address".to_string(),
+                reference_token_manager_address: "token_manager_address".to_string(),
+            },
+        ];
+
+        assert!(
+            matches!(deserialize_handler_configs(to_value(configs).unwrap()),
+                Err(e) if e.to_string().contains("only one Stacks message verifier config is allowed")
+            )
+        );
+
+        let configs = vec![
+            Config::StacksVerifierSetVerifier {
+                cosmwasm_contract: TMAddress::random(PREFIX),
+                http_url: "http://localhost:8080/".parse().unwrap(),
+            },
+            Config::StacksVerifierSetVerifier {
+                cosmwasm_contract: TMAddress::random(PREFIX),
+                http_url: "http://localhost:8080/".parse().unwrap(),
+            },
+        ];
+
+        assert!(
+            matches!(deserialize_handler_configs(to_value(configs).unwrap()),
+                Err(e) if e.to_string().contains("only one Stacks verifier set verifier config is allowed")
+            )
+        );
     }
 }

ampd/src/handlers/mod.rs

@@ -5,6 +5,8 @@ pub mod evm_verify_verifier_set;
 pub mod multisig;
 pub mod mvx_verify_msg;
 pub mod mvx_verify_verifier_set;
+pub mod stacks_verify_msg;
+pub mod stacks_verify_verifier_set;
 pub(crate) mod stellar_verify_msg;
 pub(crate) mod stellar_verify_verifier_set;
 pub mod sui_verify_msg;