214 lines
9.5 KiB
TypeScript
214 lines
9.5 KiB
TypeScript
import { ethers, upgrades } from "hardhat";
|
||
import * as fs from "fs";
|
||
import * as path from "path";
|
||
|
||
/**
|
||
* 升级 Lending 或 Configurator 合约
|
||
* 使用 upgrades.upgradeProxy() 进行 UUPS 升级
|
||
*/
|
||
async function main() {
|
||
const [deployer] = await ethers.getSigners();
|
||
console.log("\n==========================================");
|
||
console.log("🔄 升级 Lending 借贷池系统");
|
||
console.log("==========================================");
|
||
console.log("升级账户:", deployer.address);
|
||
console.log("账户余额:", ethers.formatEther(await ethers.provider.getBalance(deployer.address)), "ETH\n");
|
||
|
||
// ========== 读取部署信息 ==========
|
||
const deploymentsPath = path.join(__dirname, "../../deployments-lending.json");
|
||
if (!fs.existsSync(deploymentsPath)) {
|
||
throw new Error("未找到部署信息文件，请先运行部署脚本");
|
||
}
|
||
|
||
const network = await ethers.provider.getNetwork();
|
||
const chainId = network.chainId.toString();
|
||
const deployments = JSON.parse(fs.readFileSync(deploymentsPath, "utf-8"))[chainId];
|
||
|
||
if (!deployments) {
|
||
throw new Error(`未找到网络 ${chainId} 的部署信息`);
|
||
}
|
||
|
||
console.log("📋 当前部署的合约:");
|
||
console.log(" LendingPriceFeed Proxy:", deployments.lendingPriceFeed);
|
||
if (deployments.lendingPriceFeedImpl) {
|
||
console.log(" LendingPriceFeed Impl:", deployments.lendingPriceFeedImpl);
|
||
}
|
||
console.log(" Configurator Proxy:", deployments.configuratorProxy);
|
||
console.log(" Configurator Impl:", deployments.configuratorImpl);
|
||
console.log(" Lending Proxy:", deployments.lendingProxy);
|
||
console.log(" Lending Impl:", deployments.lendingImpl, "\n");
|
||
|
||
// ========== 选择要升级的合约 ==========
|
||
// 修改这里来选择升级哪个合约
|
||
// 1 = LendingPriceFeed, 2 = Configurator, 3 = Lending
|
||
const UPGRADE_CONTRACT = 1; // 修改这个数字来选择要升级的合约
|
||
|
||
if (UPGRADE_CONTRACT === 1) {
|
||
// ========== 升级 LendingPriceFeed ==========
|
||
console.log("🔄 Phase 1: 升级 LendingPriceFeed 合约");
|
||
|
||
if (!deployments.lendingPriceFeed) {
|
||
throw new Error("未找到 LendingPriceFeed Proxy 地址，请先运行部署脚本");
|
||
}
|
||
|
||
console.log(" 当前 LendingPriceFeed Proxy:", deployments.lendingPriceFeed);
|
||
if (deployments.lendingPriceFeedImpl) {
|
||
console.log(" 当前 LendingPriceFeed Implementation:", deployments.lendingPriceFeedImpl);
|
||
}
|
||
|
||
// 获取新的 LendingPriceFeed 合约工厂
|
||
const LendingPriceFeedV2 = await ethers.getContractFactory("LendingPriceFeed");
|
||
|
||
console.log("\n 正在验证新实现合约...");
|
||
const upgradedPriceFeed = await upgrades.upgradeProxy(
|
||
deployments.lendingPriceFeed,
|
||
LendingPriceFeedV2,
|
||
{
|
||
kind: "uups"
|
||
}
|
||
);
|
||
await upgradedPriceFeed.waitForDeployment();
|
||
|
||
console.log(" ✅ LendingPriceFeed 已升级！");
|
||
|
||
// 获取新的实现合约地址
|
||
const upgradedPriceFeedAddress = await upgradedPriceFeed.getAddress();
|
||
const newPriceFeedImplAddress = await upgrades.erc1967.getImplementationAddress(upgradedPriceFeedAddress);
|
||
console.log(" 新 LendingPriceFeed Implementation:", newPriceFeedImplAddress);
|
||
|
||
// 验证升级
|
||
console.log("\n 验证升级结果:");
|
||
console.log(" LendingPriceFeed Proxy (不变):", upgradedPriceFeedAddress);
|
||
console.log(" Owner:", await upgradedPriceFeed.owner());
|
||
console.log(" USDC Address:", await upgradedPriceFeed.usdcAddress());
|
||
|
||
// 保存新的实现地址
|
||
deployments.lendingPriceFeedImpl = newPriceFeedImplAddress;
|
||
deployments.lastUpgradeTime = new Date().toISOString();
|
||
|
||
const allDeployments = JSON.parse(fs.readFileSync(deploymentsPath, "utf-8"));
|
||
allDeployments[chainId] = deployments;
|
||
fs.writeFileSync(deploymentsPath, JSON.stringify(allDeployments, null, 2));
|
||
|
||
console.log("\n✅ LendingPriceFeed 升级完成！");
|
||
console.log("=====================================");
|
||
console.log("旧实现:", deployments.lendingPriceFeedImpl || "未记录");
|
||
console.log("新实现:", newPriceFeedImplAddress);
|
||
console.log("=====================================\n");
|
||
|
||
} else if (UPGRADE_CONTRACT === 3) {
|
||
// ========== 升级 Lending ==========
|
||
console.log("🔄 Phase 1: 升级 Lending 合约");
|
||
|
||
if (!deployments.lendingProxy) {
|
||
throw new Error("未找到 Lending Proxy 地址，请先运行配置脚本");
|
||
}
|
||
|
||
console.log(" 当前 Lending Proxy:", deployments.lendingProxy);
|
||
console.log(" 当前 Lending Implementation:", deployments.lendingImpl);
|
||
|
||
// 获取新的 Lending 合约工厂
|
||
// 注意：如果你有 LendingV2，请替换为 "LendingV2"
|
||
const LendingV2 = await ethers.getContractFactory("Lending");
|
||
|
||
console.log("\n 正在验证新实现合约...");
|
||
// upgrades.upgradeProxy 会自动验证存储布局兼容性
|
||
const upgradedLending = await upgrades.upgradeProxy(
|
||
deployments.lendingProxy,
|
||
LendingV2,
|
||
{
|
||
kind: "uups"
|
||
}
|
||
);
|
||
await upgradedLending.waitForDeployment();
|
||
|
||
console.log(" ✅ Lending 已升级！");
|
||
|
||
// 获取新的实现合约地址
|
||
const upgradedLendingAddress = await upgradedLending.getAddress();
|
||
const newLendingImplAddress = await upgrades.erc1967.getImplementationAddress(upgradedLendingAddress);
|
||
console.log(" 新 Lending Implementation:", newLendingImplAddress);
|
||
|
||
// 验证升级
|
||
console.log("\n 验证升级结果:");
|
||
console.log(" Lending Proxy (不变):", upgradedLendingAddress);
|
||
console.log(" Owner:", await upgradedLending.owner());
|
||
console.log(" Base Token:", await upgradedLending.baseToken());
|
||
|
||
// 保存新的实现地址
|
||
deployments.lendingImpl = newLendingImplAddress;
|
||
deployments.lendingUpgradeTimestamp = new Date().toISOString();
|
||
|
||
} else if (UPGRADE_CONTRACT === 2) {
|
||
// ========== 升级 Configurator ==========
|
||
console.log("🔄 Phase 1: 升级 Configurator 合约");
|
||
|
||
console.log(" 当前 Configurator Proxy:", deployments.configuratorProxy);
|
||
console.log(" 当前 Configurator Implementation:", deployments.configuratorImpl);
|
||
|
||
// 获取新的 Configurator 合约工厂
|
||
// 注意：如果你有 ConfiguratorV2，请替换为 "ConfiguratorV2"
|
||
const ConfiguratorV2 = await ethers.getContractFactory("Configurator");
|
||
|
||
console.log("\n 正在验证新实现合约...");
|
||
const upgradedConfigurator = await upgrades.upgradeProxy(
|
||
deployments.configuratorProxy,
|
||
ConfiguratorV2,
|
||
{
|
||
kind: "uups"
|
||
}
|
||
);
|
||
await upgradedConfigurator.waitForDeployment();
|
||
|
||
console.log(" ✅ Configurator 已升级！");
|
||
|
||
// 获取新的实现合约地址
|
||
const upgradedConfiguratorAddress = await upgradedConfigurator.getAddress();
|
||
const newConfiguratorImplAddress = await upgrades.erc1967.getImplementationAddress(upgradedConfiguratorAddress);
|
||
console.log(" 新 Configurator Implementation:", newConfiguratorImplAddress);
|
||
|
||
// 验证升级
|
||
console.log("\n 验证升级结果:");
|
||
console.log(" Configurator Proxy (不变):", upgradedConfiguratorAddress);
|
||
console.log(" Owner:", await upgradedConfigurator.owner());
|
||
|
||
// 保存新的实现地址
|
||
deployments.configuratorImpl = newConfiguratorImplAddress;
|
||
deployments.configuratorUpgradeTimestamp = new Date().toISOString();
|
||
|
||
const allDeployments2 = JSON.parse(fs.readFileSync(deploymentsPath, "utf-8"));
|
||
allDeployments2[chainId] = deployments;
|
||
fs.writeFileSync(deploymentsPath, JSON.stringify(allDeployments2, null, 2));
|
||
|
||
console.log("\n✅ Configurator 升级完成！");
|
||
console.log("=====================================");
|
||
console.log("旧实现:", deployments.configuratorImpl);
|
||
console.log("新实现:", newConfiguratorImplAddress);
|
||
console.log("=====================================\n");
|
||
} else {
|
||
throw new Error(`无效的升级选项: ${UPGRADE_CONTRACT}。请设置 UPGRADE_CONTRACT 为 1 (LendingPriceFeed), 2 (Configurator), 或 3 (Lending)`);
|
||
}
|
||
|
||
// ========== 保存部署信息（最终）==========
|
||
const allDeployments = JSON.parse(fs.readFileSync(deploymentsPath, "utf-8"));
|
||
allDeployments[chainId] = deployments;
|
||
fs.writeFileSync(deploymentsPath, JSON.stringify(allDeployments, null, 2));
|
||
|
||
console.log("\n💾 升级信息已保存到:", deploymentsPath);
|
||
|
||
console.log("\n✅ 升级流程全部完成！");
|
||
console.log("⚠️ 重要提示:");
|
||
console.log(" 1. 代理地址保持不变，用户无需更改合约地址");
|
||
console.log(" 2. 所有状态数据已保留");
|
||
console.log(" 3. 建议在测试网充分测试后再升级主网");
|
||
console.log(" 4. 当前升级的合约:", UPGRADE_CONTRACT === 1 ? "LendingPriceFeed" : (UPGRADE_CONTRACT === 2 ? "Configurator" : "Lending"), "\n");
|
||
}
|
||
|
||
main()
|
||
.then(() => process.exit(0))
|
||
.catch((error) => {
|
||
console.error(error);
|
||
process.exit(1);
|
||
});
|
||
|