diff --git a/packages/world/ts/node/render-solidity/renderSystemLibrary.ts b/packages/world/ts/node/render-solidity/renderSystemLibrary.ts index cfd468e947..dfa8301a46 100644 --- 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 index 6759b52831..eaa2a0b4c5 100644 --- a/test/system-libraries/src/namespaces/a/codegen/libraries/ASystemLib.sol +++ b/test/system-libraries/src/namespaces/a/codegen/libraries/ASystemLib.sol @@ -109,8 +109,12 @@ library ASystemLib { return CallWrapper(self.toResourceId(), from); } - function asRoot(ASystemType self) internal pure returns (RootCallWrapper memory) { - return RootCallWrapper(self.toResourceId(), address(0)); + function callAsRoot(ASystemType self) internal view returns (RootCallWrapper memory) { + return RootCallWrapper(self.toResourceId(), msg.sender); + } + + function callAsRootFrom(ASystemType self, address from) internal pure returns (RootCallWrapper memory) { + return RootCallWrapper(self.toResourceId(), from); } function toResourceId(ASystemType self) internal pure returns (ResourceId) { diff --git a/test/system-libraries/src/namespaces/b/codegen/libraries/BSystemLib.sol b/test/system-libraries/src/namespaces/b/codegen/libraries/BSystemLib.sol index 246b4fbfd2..d51e2177a9 100644 --- a/test/system-libraries/src/namespaces/b/codegen/libraries/BSystemLib.sol +++ b/test/system-libraries/src/namespaces/b/codegen/libraries/BSystemLib.sol @@ -72,8 +72,12 @@ library BSystemLib { return CallWrapper(self.toResourceId(), from); } - function asRoot(BSystemType self) internal pure returns (RootCallWrapper memory) { - return RootCallWrapper(self.toResourceId(), address(0)); + function callAsRoot(BSystemType self) internal view returns (RootCallWrapper memory) { + return RootCallWrapper(self.toResourceId(), msg.sender); + } + + function callAsRootFrom(BSystemType self, address from) internal pure returns (RootCallWrapper memory) { + return RootCallWrapper(self.toResourceId(), from); } function toResourceId(BSystemType self) internal pure returns (ResourceId) { diff --git a/test/system-libraries/src/namespaces/root/RootSystem.sol b/test/system-libraries/src/namespaces/root/RootSystem.sol index 45149421ef..d2c28c0d23 100644 --- 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 index 441452f4f7..f0be5f46a8 100644 --- a/test/system-libraries/src/namespaces/root/codegen/libraries/RootSystemLib.sol +++ b/test/system-libraries/src/namespaces/root/codegen/libraries/RootSystemLib.sol @@ -55,8 +55,12 @@ library RootSystemLib { return CallWrapper(self.toResourceId(), from); } - function asRoot(RootSystemType self) internal pure returns (RootCallWrapper memory) { - return RootCallWrapper(self.toResourceId(), address(0)); + function callAsRoot(RootSystemType self) internal view returns (RootCallWrapper memory) { + return RootCallWrapper(self.toResourceId(), msg.sender); + } + + function callAsRootFrom(RootSystemType self, address from) internal pure returns (RootCallWrapper memory) { + return RootCallWrapper(self.toResourceId(), from); } function toResourceId(RootSystemType self) internal pure returns (ResourceId) {