Add callAsRoot and callAsRootFrom

latticexyz · Nov 22, 2024 · d3b592b · d3b592b
1 parent 18987ea
commit d3b592b
Show file tree

Hide file tree

Showing 5 changed files with 25 additions and 9 deletions.
diff --git a/packages/world/ts/node/render-solidity/renderSystemLibrary.ts b/packages/world/ts/node/render-solidity/renderSystemLibrary.ts
@@ -91,8 +91,12 @@ export function renderSystemLibrary(options: RenderSystemLibraryOptions) {
         return CallWrapper(self.toResourceId(), from);
       }
 
-      function asRoot(${userTypeName} self) internal pure returns (RootCallWrapper memory) {
-        return RootCallWrapper(self.toResourceId(), address(0));
+      function callAsRoot(${userTypeName} self) internal view returns (RootCallWrapper memory) {
+        return RootCallWrapper(self.toResourceId(), msg.sender);
+      }
+
+      function callAsRootFrom(${userTypeName} self, address from) internal pure returns (RootCallWrapper memory) {
+        return RootCallWrapper(self.toResourceId(), from);
       }
 
       function toResourceId(${userTypeName} self) internal pure returns (ResourceId) {

diff --git a/test/system-libraries/src/namespaces/a/codegen/libraries/ASystemLib.sol b/test/system-libraries/src/namespaces/a/codegen/libraries/ASystemLib.sol
diff --git a/test/system-libraries/src/namespaces/b/codegen/libraries/BSystemLib.sol b/test/system-libraries/src/namespaces/b/codegen/libraries/BSystemLib.sol
diff --git a/test/system-libraries/src/namespaces/root/RootSystem.sol b/test/system-libraries/src/namespaces/root/RootSystem.sol
@@ -6,6 +6,6 @@ import { aSystem } from "../a/codegen/libraries/ASystemLib.sol";
 
 contract RootSystem is System {
   function setValueInA(uint256 value) external {
-    aSystem.asRoot().setValue(value);
+    aSystem.callAsRoot().setValue(value);
   }
 }
diff --git a/test/system-libraries/src/namespaces/root/codegen/libraries/RootSystemLib.sol b/test/system-libraries/src/namespaces/root/codegen/libraries/RootSystemLib.sol