feat: add module registration helper

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rhinestone/modulekit",
-  "version": "0.4.12",
+  "version": "0.4.13",
   "description": "A development kit for building and testing smart account modules.",
   "license": "GPL-3.0",
   "author": {

src/deployment/RegistryDeployer.sol

@@ -81,6 +81,28 @@
         return registry.calcModuleAddress(salt, initCode);
     }
 
+    // <---- REGISTRATION ---->
+
+    function registerModule(
+        address module,
+        bytes memory metadata,
+        bytes memory resolverContext
+    )
+        public
+    {
+        ResolverUID _resolverUID = findResolver();
+        registry.registerModule({
+            resolverUID: _resolverUID,
+            moduleAddress: module,
+            metadata: metadata,
+            resolverContext: resolverContext
+        });
+    }
+
+    function findModule(address moduleAddress) public view returns (ModuleRecord memory) {
+        return registry.findModule(moduleAddress);
+    }
+
     // <---- ATTESTATIONS ---->
 
     function mockAttestToModule(
@@ -90,7 +112,7 @@
     )
         public
     {
         require(isContract(mockAttester), "MockAttester is not deployed on this network");
 
         SchemaUID _schemaUID = findSchema();
         AttestationRequest memory request = AttestationRequest({
@@ -108,7 +130,7 @@
             )
         );
 
         require(success, "Mock attestation failed");
     }
 
     function isModuleAttestedMock(address module) public view returns (bool) {
@@ -153,10 +175,6 @@
         });
     }
 
-    function findModule(address moduleAddress) public view returns (ModuleRecord memory) {
-        return registry.findModule(moduleAddress);
-    }
-
     function setRegistry(address _registry) public {
         registry = IRegistry(_registry);
     }
@@ -172,7 +190,7 @@
     // <---- OTHER ---->
     function isContract(address _addr) internal returns (bool isContract) {
         uint32 size;
         assembly {
             size := extcodesize(_addr)
         }
         return (size > 0);

test/RegistryDeployer.t.sol

@@ -71,6 +71,23 @@ contract RegistryDeployerTest is RegistryDeployer, BaseTest {
         assertEq(moduleRecord.metadata, metadata);
     }
 
+    function testRegisterModule() public onMainnet {
+        address module = address(new MockValidator());
+
+        ModuleRecord memory moduleRecord = findModule(module);
+        assertEq(ResolverUID.unwrap(moduleRecord.resolverUID), bytes32(0));
+        assertEq(moduleRecord.metadata, "");
+
+        bytes memory metadata = hex"41414141414141";
+        bytes memory resolverContext = "";
+
+        registerModule({ module: module, metadata: metadata, resolverContext: resolverContext });
+
+        moduleRecord = findModule(module);
+        assertEq(ResolverUID.unwrap(moduleRecord.resolverUID), ResolverUID.unwrap(resolverUID));
+        assertEq(moduleRecord.metadata, metadata);
+    }
+
     function testMockAttestModule() public onTestnet {
         deployModule();