-
Notifications
You must be signed in to change notification settings - Fork 8
/
Copy pathUUPSProxy.sol
51 lines (42 loc) · 2.49 KB
/
UUPSProxy.sol
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.21;
import {OwnableUpgradeable} from "@openzeppelin/contracts-upgradeable/access/OwnableUpgradeable.sol";
import {ERC1967Proxy} from "@openzeppelin/contracts/proxy/ERC1967/ERC1967Proxy.sol";
import {UUPSUpgradeable} from "@openzeppelin/contracts/proxy/utils/UUPSUpgradeable.sol";
/// Контракт логики
contract Logic is UUPSUpgradeable, OwnableUpgradeable {
uint256 private _value;
/**
* @dev Так как в обновляемых контрактах нет конструктора, то нам необходимо использовать функцию initialize()
* Дополнительно необходимо инициализировать контракт OwnableUpgradeable, так как он тоже обновляемый
* Адрес _initialOwner станет владельцем контракта Logic и сможет обновлять имплементацию для прокси
*/
function initialize(address _initialOwner) external initializer {
__Ownable_init(_initialOwner);
}
function store(uint256 _newValue) public {
_value = _newValue;
}
function retrieve() public view returns (uint256) {
return _value;
}
/**
* @notice Проверка возможности обновлять контракт
* @dev Согласно абстрактному контракту UUPSUpgradeable, нам обязательно необходимо переопределить данную функцию
* При помощи собственной реализации функции мы будем определять возможность обновления контракта логики для контракта прокси
* В рамках этого примера обновлять контракт логики может только владелец контракта логики
*/
function _authorizeUpgrade(address newImplementation) internal override onlyOwner {}
}
/// Контракт прокси
contract LogicProxy is ERC1967Proxy {
constructor(
address _logic,
bytes memory _data
) ERC1967Proxy(_logic, _data) {}
/// @notice Возвращает адрес установленного контракта логики для прокси
function getImplementation() public view returns (address) {
return _implementation();
}
receive() external payable {}
}