ENS naming for contracts

Cargo.toml

@@ -22,6 +22,7 @@ members = [
     "crates/forge/",
     "crates/script-sequence/",
     "crates/macros/",
+    "crates/enscribe/",
     "crates/test-utils/",
     "crates/lint/",
 ]
@@ -197,6 +198,8 @@ foundry-test-utils = { path = "crates/test-utils" }
 foundry-wallets = { path = "crates/wallets" }
 foundry-linking = { path = "crates/linking" }
 
+enscribe = { path = "crates/enscribe" }
+
 # solc & compilation utilities
 foundry-block-explorers = { version = "0.17.0", default-features = false }
 foundry-compilers = { version = "0.16.1", default-features = false }

crates/enscribe/Cargo.toml

@@ -0,0 +1,37 @@
+[package]
+name = "enscribe"
+description = "setting ens names for smart contracts"
+
+version.workspace = true
+edition.workspace = true
+rust-version.workspace = true
+authors.workspace = true
+license.workspace = true
+homepage.workspace = true
+repository.workspace = true
+
+[lints]
+workspace = true
+
+[dependencies]
+eyre.workspace = true
+
+foundry-cli.workspace = true
+foundry-common.workspace = true
+foundry-config.workspace = true
+
+alloy-ens = { workspace = true, features = ["provider"] }
+alloy-contract.workspace = true
+alloy-provider = { workspace = true, features = ["reqwest", "ws", "ipc"] }
+alloy-primitives.workspace = true
+alloy-sol-types.workspace = true
+reqwest.workspace = true
+serde_json.workspace = true
+serde.workspace = true
+uuid = { version = "1.16.0", features = ["v4"] }
+
+[dev-dependencies]
+wiremock = "0.6"
+tokio.workspace = true
+reqwest.workspace = true
+serial_test = "3.2.0"

crates/enscribe/src/abi.rs

@@ -0,0 +1,67 @@
+use alloy_sol_types::sol;
+sol! {
+    /// ENS Registry contract.
+    #[sol(rpc)]
+    contract EnsRegistry {
+        /// Sets subnode record
+        function setSubnodeRecord(
+                bytes32 node,
+                bytes32 label,
+                address owner,
+                address resolver,
+                uint64 ttl
+            ) external;
+
+        /// checks if an ens name record already exists
+        function recordExists(bytes32 node) external returns (bool);
+        /// returns the owner of this node
+        function owner(bytes32 node) external returns (address);
+    }
+
+    /// ENS Name Wrapper contract
+    #[sol(rpc)]
+    contract NameWrapper {
+        function isWrapped(bytes32 node) external returns (bool);
+        function setSubnodeRecord(
+                bytes32 node,
+                string label,
+                address owner,
+                address resolver,
+                uint64 ttl,
+                uint32 fuses,
+                uint64 expiry,
+            ) external;
+    }
+
+    /// ENS Public Resolver contract
+    #[sol(rpc)]
+    contract PublicResolver {
+        function setAddr(bytes32 node, address addr) external;
+        function addr(bytes32 node) external returns (address);
+        function setName(bytes32 node,  string newName) external;
+    }
+
+    /// ENS Reverse Registrar contract
+    #[sol(rpc)]
+    contract ReverseRegistrar {
+        function setName(string memory name) external returns (bytes32);
+        function setNameForAddr(address addr, address owner, address resolver, string name) external;
+    }
+
+    /// Enscribe contract
+    #[sol(rpc)]
+    contract Enscribe {
+        function setName(
+            address contractAddress,
+            string label,
+            string parentName,
+            bytes32 parentNode
+        ) external returns (bool success);
+    }
+
+    /// Ownable contract
+    #[sol(rpc)]
+    contract Ownable {
+        function owner() external returns (address);
+    }
+}

crates/enscribe/src/lib.rs

@@ -0,0 +1,8 @@
+//! enscribe does ENS name setting for contracts.
+
+#[allow(clippy::too_many_arguments)]
+mod abi;
+pub(crate) mod logger;
+pub mod name;
+
+pub use name::set_primary_name;

crates/enscribe/src/logger.rs

@@ -0,0 +1,59 @@
+use eyre::Result;
+use serde_json::json;
+use uuid::Uuid;
+
+static METRICS_API_URL: &str = "https://app.enscribe.xyz/api/v1/config";
+
+pub(crate) struct MetricLogger {
+    corelation_id: String,
+    client: reqwest::Client,
+    deployer_address: String,
+    network: u64,
+    op_type: String,
+    contract_type: String,
+    contract_addr: String,
+    ens_name: String,
+}
+
+impl MetricLogger {
+    pub(crate) fn new(
+        deployer_address: String,
+        network: u64,
+        op_type: String,
+        contract_type: String,
+        contract_addr: String,
+        ens_name: String,
+    ) -> Self {
+        let client = reqwest::Client::new();
+        let corelation_id = Uuid::new_v4().to_string();
+
+        Self {
+            corelation_id,
+            client,
+            deployer_address,
+            network,
+            op_type,
+            contract_type,
+            contract_addr,
+            ens_name,
+        }
+    }
+
+    pub(crate) async fn log(&self, step: &str, txn_hash: &str) -> Result<()> {
+        self.client.post(METRICS_API_URL).json(&json!({
+            "co_id": self.corelation_id,
+            "step": step.to_owned(),
+            "txn_hash": txn_hash,
+            "contract_type": self.contract_type,
+            "contract_address": self.contract_addr,
+            "ens_name": self.ens_name,
+            "deployer_address": self.deployer_address,
+            "network": self.network,
+            "timestamp": std::time::SystemTime::now().duration_since(std::time::UNIX_EPOCH)?.as_secs(),
+            "source": "forge",
+            "op_type": self.op_type,
+        })).send().await?;
+
+        Ok(())
+    }
+}