add proxies

Aut-Labs · Dec 2, 2023 · 374f330 · 374f330
1 parent cd3101f
commit 374f330
Show file tree

Hide file tree

Showing 2 changed files with 48 additions and 0 deletions.
diff --git a/contracts/AutIDProxy.sol b/contracts/AutIDProxy.sol
@@ -0,0 +1,26 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity 0.8.19;
+
+import "@openzeppelin/contracts/proxy/Proxy.sol";
+import "@openzeppelin/contracts/utils/StorageSlot.sol";
+import "@openzeppelin/contracts/access/Ownable.sol";
+
+contract LocalReputationProxy is Proxy, Ownable {
+    bytes32 private constant _IMPLEMENTATION_SLOT = bytes32(uint256(keccak256("id.aut.os.autid.proxy.implementation")) - 1);
+
+    constructor(address implementation_) Ownable() {
+        upgrade(implementation_);
+    }
+
+    function implementation() external view returns (address) {
+        return _implementation();
+    }
+
+    function upgrade(address to) public onlyOwner {
+        StorageSlot.getAddressSlot(_IMPLEMENTATION_SLOT).value = to;
+    }
+
+    function _implementation() internal view override returns (address) {
+        return StorageSlot.getAddressSlot(_IMPLEMENTATION_SLOT).value;
+    }
+}
diff --git a/contracts/LocalReputationProxy.sol b/contracts/LocalReputationProxy.sol
@@ -0,0 +1,22 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity 0.8.19;
+
+import "@openzeppelin/contracts/proxy/Proxy.sol";
+import "@openzeppelin/contracts/utils/StorageSlot.sol";
+import "@openzeppelin/contracts/access/Ownable.sol";
+
+contract LocalReputationProxy is Proxy, Ownable {
+    bytes32 private constant _IMPLEMENTATION_SLOT = bytes32(uint256(keccak256("id.aut.os.lr.proxy.implementation")) - 1);
+
+    constructor(address implementation) Ownable() {
+        upgrade(implementation);
+    }
+
+    function upgrade(address to) public onlyOwner {
+        StorageSlot.getAddressSlot(_IMPLEMENTATION_SLOT).value = to;
+    }
+
+    function _implementation() internal view override returns (address) {
+        return StorageSlot.getAddressSlot(_IMPLEMENTATION_SLOT).value;
+    }
+}