Merge branch 'main' of https://github.com/IAmKio/eth-global-london-2024

username_registry/script/UsernameRegistry.s.sol

@@ -0,0 +1,17 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity ^0.8.13;
+
+import {Script, console2} from "forge-std/Script.sol";
+import {UsernameRegistry} from "../src/contracts/UsernameRegistry.sol";
+
+contract UsernameRegistryScript is Script {
+    function setUp() public {}
+
+    function run() public {
+        uint256 deployerPrivateKey = vm.envUint("DEPLOYER_PRIVATE_KEY");
+        vm.startBroadcast(deployerPrivateKey);
+        UsernameRegistry unreg = new UsernameRegistry(msg.sender);
+        console2.log("UsernameRegistry deployed at address", address(unreg));
+        vm.stopBroadcast();
+    }
+}

username_registry/src/contracts/UsernameRegistry.sol

@@ -0,0 +1,46 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity 0.8.20;
+
+import "solady/auth/Ownable.sol";
+import "../interfaces/IUsernameRegistry.sol";
+
+contract UsernameRegistry is Ownable, IUsernameRegistry {
+    mapping(address => string) public addressToUsername;
+    mapping(string => address) public usernameToAddress;
+
+    constructor(address _owner) {
+        _setOwner(_owner);
+    }
+
+    // External functions
+
+    function registerUserData(
+        address _scw,
+        string memory _un
+    ) external onlyOwner {
+        _setSCW(_scw, _un);
+        _setUsername(_scw, _un);
+        emit UserRegistered(_scw, _un);
+    }
+
+    function getUsernameForWallet(
+        address _scw
+    ) external view returns (string memory un) {
+        return addressToUsername[_scw];
+    }
+
+    function getWalletForUsername(
+        string memory _un
+    ) external view returns (address scw) {
+        return usernameToAddress[_un];
+    }
+
+    // Internal functions
+
+    function _setSCW(address _scw, string memory _un) internal {
+        usernameToAddress[_un] = _scw;
+    }
+    function _setUsername(address _scw, string memory _un) internal {
+        addressToUsername[_scw] = _un;
+    }
+}

username_registry/src/interfaces/IUsernameRegistry.sol

@@ -0,0 +1,16 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity 0.8.20;
+
+interface IUsernameRegistry {
+    event UserRegistered(address indexed scw, string un);
+
+    function registerUserData(address _scw, string memory _un) external;
+
+    function getUsernameForWallet(
+        address _scw
+    ) external view returns (string memory un);
+
+    function getWalletForUsername(
+        string memory _un
+    ) external view returns (address scw);
+}

username_registry/test/UsernameRegistry.t.sol

@@ -0,0 +1,68 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity 0.8.20;
+
+import "forge-std/Test.sol";
+import {UsernameRegistry} from "../src/contracts/UsernameRegistry.sol";
+
+contract UsernameRegistryTest is Test {
+    UsernameRegistry registry;
+    Account internal owner;
+    Account internal user;
+
+    event UserRegistered(address indexed scw, string un);
+
+    function setUp() public {
+        owner = makeAccount("owner");
+        user = makeAccount("user");
+        registry = new UsernameRegistry(owner.addr);
+    }
+
+    function test_registerUserData() public {
+        string memory username = "john";
+        vm.prank(owner.addr);
+
+        registry.registerUserData(user.addr, username);
+
+        assertEq(registry.getUsernameForWallet(user.addr), username);
+        assertEq(registry.getWalletForUsername(username), user.addr);
+    }
+
+    function test_revertOnlyOwnerCanRegister() public {
+        string memory username = "john";
+        vm.prank(user.addr);
+        vm.expectRevert();
+        registry.registerUserData(user.addr, username);
+    }
+
+    // Test getUsernameForWallet
+    function test_getUsernameForWallet() public {
+        string memory username = "john";
+        vm.prank(owner.addr);
+
+        registry.registerUserData(user.addr, username);
+
+        string memory result = registry.getUsernameForWallet(user.addr);
+        assertEq(result, username);
+    }
+
+    // Test getWalletForUsername
+    function test_getWalletForUsername() public {
+        string memory username = "john";
+        vm.prank(owner.addr);
+
+        registry.registerUserData(user.addr, username);
+
+        address result = registry.getWalletForUsername(username);
+        assertEq(result, user.addr);
+    }
+
+    // Test UserRegistered event
+    function test_emitUserRegisteredEvent() public {
+        string memory username = "john";
+        vm.expectEmit(true, true, false, true);
+        emit UserRegistered(user.addr, username);
+        vm.prank(owner.addr);
+
+        registry.registerUserData(user.addr, username);
+    }
+}