feat(world): deployment salt by msg.sender

.changeset/selfish-pears-marry.md

@@ -0,0 +1,5 @@
+---
+"@latticexyz/world": minor
+---
+
+Added `msg.sender` to the `WorldFactory` deployment salt, so the World address is derived from the creator of the world.

packages/world/gas-report.json

@@ -57,13 +57,13 @@
     "file": "test/Factories.t.sol",
     "test": "testCreate2Factory",
     "name": "deploy contract via Create2",
-    "gasUsed": 4763472
+    "gasUsed": 4797613
   },
   {
     "file": "test/Factories.t.sol",
-    "test": "testWorldFactory",
+    "test": "testWorldFactoryGas",
     "name": "deploy world via WorldFactory",
-    "gasUsed": 13045360
+    "gasUsed": 13045646
   },
   {
     "file": "test/World.t.sol",

packages/world/src/IWorldFactory.sol

@@ -14,10 +14,11 @@ interface IWorldFactory {
   event WorldDeployed(address indexed newContract);
 
   /**
-   * @notice Returns the total count of deployed World contracts.
-   * @return The total number of World contracts deployed by this factory.
+   * @notice Returns the total count of deployed World contracts per account.
+   * @param account The account.
+   * @return The total number of World contracts deployed by this factory per account.
    */
-  function worldCount() external view returns (uint256);
+  function accountCount(address account) external view returns (uint256);
 
   /**
    * @notice Deploys a new World contract.

packages/world/src/WorldFactory.sol

@@ -17,8 +17,8 @@ contract WorldFactory is IWorldFactory {
   /// @notice Address of the core module to be set in the World instances.
   IModule public immutable coreModule;
 
-  /// @notice Counter to keep track of the number of World instances deployed.
-  uint256 public worldCount;
+  /// @notice Counters to keep track of the number of World instances deployed per address.
+  mapping(address => uint256) public accountCount;
 
   /// @param _coreModule The address of the core module.
   constructor(IModule _coreModule) {
@@ -33,7 +33,8 @@ contract WorldFactory is IWorldFactory {
   function deployWorld() public returns (address worldAddress) {
     // Deploy a new World and increase the WorldCount
     bytes memory bytecode = type(World).creationCode;
-    worldAddress = Create2.deploy(bytecode, worldCount++);
+    uint256 salt = uint256(keccak256(abi.encode(msg.sender, accountCount[msg.sender]++)));
+    worldAddress = Create2.deploy(bytecode, salt);
     IBaseWorld world = IBaseWorld(worldAddress);
 
     // Initialize the World and transfer ownership to the caller

packages/world/test/Factories.t.sol

@@ -51,17 +51,23 @@ contract FactoriesTest is Test, GasReporter {
 
     // Confirm worldFactory was deployed correctly
     IWorldFactory worldFactory = IWorldFactory(calculatedAddress);
-    assertEq(uint256(worldFactory.worldCount()), uint256(0));
+    assertEq(uint256(worldFactory.accountCount(address(0))), uint256(0));
   }
 
-  function testWorldFactory() public {
+  function testWorldFactory(address account) public {
+    vm.startPrank(account);
+
     // Deploy WorldFactory with current CoreModule
     CoreModule coreModule = createCoreModule();
     address worldFactoryAddress = address(new WorldFactory(coreModule));
     IWorldFactory worldFactory = IWorldFactory(worldFactoryAddress);
 
     // Address we expect for World
-    address calculatedAddress = calculateAddress(worldFactoryAddress, bytes32(0), type(World).creationCode);
+    address calculatedAddress = calculateAddress(
+      worldFactoryAddress,
+      keccak256(abi.encode(account, 0)),
+      type(World).creationCode
+    );
 
     // Check for HelloWorld event from World
     vm.expectEmit(true, true, true, true);
@@ -77,13 +83,47 @@ contract FactoriesTest is Test, GasReporter {
     // Set the store address manually
     StoreSwitch.setStoreAddress(calculatedAddress);
 
+    // Confirm accountCount (which is salt) has incremented
+    assertEq(uint256(worldFactory.accountCount(account)), uint256(1));
+
     // Confirm correct Core is installed
     assertTrue(InstalledModules.get(address(coreModule), keccak256(new bytes(0))));
 
-    // Confirm worldCount (which is salt) has incremented
-    assertEq(uint256(worldFactory.worldCount()), uint256(1));
+    // Confirm the msg.sender is owner of the root namespace of the new world
+    assertEq(NamespaceOwner.get(ROOT_NAMESPACE_ID), account);
+
+    // Deploy another world
+
+    // Address we expect for World
+    calculatedAddress = calculateAddress(
+      worldFactoryAddress,
+      keccak256(abi.encode(account, 1)),
+      type(World).creationCode
+    );
+
+    // Check for HelloWorld event from World
+    vm.expectEmit(true, true, true, true);
+    emit HelloWorld(WORLD_VERSION);
+
+    // Check for WorldDeployed event from Factory
+    vm.expectEmit(true, false, false, false);
+    emit WorldDeployed(calculatedAddress);
+    worldFactory.deployWorld();
+
+    // Confirm accountCount (which is salt) has incremented
+    assertEq(uint256(worldFactory.accountCount(account)), uint256(2));
+
+    // Set the store address manually
+    StoreSwitch.setStoreAddress(calculatedAddress);
+
+    // Confirm correct Core is installed
+    assertTrue(InstalledModules.get(address(coreModule), keccak256(new bytes(0))));
 
     // Confirm the msg.sender is owner of the root namespace of the new world
-    assertEq(NamespaceOwner.get(ROOT_NAMESPACE_ID), address(this));
+    assertEq(NamespaceOwner.get(ROOT_NAMESPACE_ID), account);
+  }
+
+  function testWorldFactoryGas() public {
+    testWorldFactory(address(this));
   }
 }