126 lines
4.9 KiB
TypeScript
126 lines
4.9 KiB
TypeScript
import { ethers, upgrades } from "hardhat";
|
||
import * as fs from "fs";
|
||
import * as path from "path";
|
||
|
||
/**
|
||
* 升级 YTAssetFactory 合约
|
||
*
|
||
* 升级步骤：
|
||
* 1. 部署新的 YTAssetFactory 实现合约
|
||
* 2. 使用 upgrades.upgradeProxy() 进行 UUPS 升级
|
||
* 3. 验证新功能
|
||
*
|
||
* 注意：
|
||
* - 这是 UUPS 代理升级，代理地址保持不变
|
||
* - 所有状态数据已保留
|
||
*/
|
||
async function main() {
|
||
const [deployer] = await ethers.getSigners();
|
||
console.log("\n==========================================");
|
||
console.log("🔄 升级 YTAssetFactory 系统");
|
||
console.log("==========================================");
|
||
console.log("升级账户:", deployer.address);
|
||
console.log("账户余额:", ethers.formatEther(await ethers.provider.getBalance(deployer.address)), "ETH\n");
|
||
|
||
// ========== 读取部署信息 ==========
|
||
const deploymentsPath = path.join(__dirname, "../../deployments-vault-system.json");
|
||
if (!fs.existsSync(deploymentsPath)) {
|
||
throw new Error("未找到部署信息文件 deployments-vault-system.json，请先运行部署脚本");
|
||
}
|
||
|
||
const deployments = JSON.parse(fs.readFileSync(deploymentsPath, "utf-8"));
|
||
|
||
if (!deployments.contracts?.YTAssetFactory?.proxy) {
|
||
throw new Error("未找到 YTAssetFactory 部署信息");
|
||
}
|
||
|
||
console.log("📋 当前部署的合约:");
|
||
console.log(" YTAssetFactory Proxy: ", deployments.contracts.YTAssetFactory.proxy);
|
||
console.log(" YTAssetFactory Implementation: ", deployments.contracts.YTAssetFactory.implementation);
|
||
console.log("");
|
||
|
||
// ========== Phase 1: 升级 YTAssetFactory ==========
|
||
console.log("🔄 Phase 1: 升级 YTAssetFactory 代理合约");
|
||
console.log(" 当前 YTAssetFactory Proxy:", deployments.contracts.YTAssetFactory.proxy);
|
||
console.log(" 当前 YTAssetFactory Implementation:", deployments.contracts.YTAssetFactory.implementation);
|
||
|
||
// 获取新的 YTAssetFactory 合约工厂
|
||
const YTAssetFactoryV2 = await ethers.getContractFactory("YTAssetFactory");
|
||
|
||
console.log("\n 正在验证新实现合约...");
|
||
const upgradedFactory = await upgrades.upgradeProxy(
|
||
deployments.contracts.YTAssetFactory.proxy,
|
||
YTAssetFactoryV2,
|
||
{
|
||
kind: "uups"
|
||
}
|
||
);
|
||
await upgradedFactory.waitForDeployment();
|
||
|
||
console.log(" ✅ YTAssetFactory 已升级！");
|
||
|
||
// 获取新的实现合约地址
|
||
const upgradedFactoryAddress = await upgradedFactory.getAddress();
|
||
const newFactoryImplAddress = await upgrades.erc1967.getImplementationAddress(upgradedFactoryAddress);
|
||
console.log(" 新 YTAssetFactory Implementation:", newFactoryImplAddress);
|
||
console.log("");
|
||
|
||
// ========== Phase 2: 验证升级结果 ==========
|
||
console.log("🔄 Phase 2: 验证升级结果");
|
||
|
||
console.log(" YTAssetFactory Proxy (不变):", upgradedFactoryAddress);
|
||
console.log(" Owner:", await upgradedFactory.owner());
|
||
console.log(" Vault Implementation:", await upgradedFactory.vaultImplementation());
|
||
console.log(" USDC Address:", await upgradedFactory.usdcAddress());
|
||
console.log("");
|
||
|
||
// ========== 保存更新的部署信息 ==========
|
||
// 保存旧的实现地址作为历史记录
|
||
if (!deployments.upgradeHistory) {
|
||
deployments.upgradeHistory = [];
|
||
}
|
||
|
||
deployments.upgradeHistory.push({
|
||
timestamp: new Date().toISOString(),
|
||
contract: "YTAssetFactory",
|
||
oldImplementation: deployments.contracts.YTAssetFactory.implementation,
|
||
newImplementation: newFactoryImplAddress,
|
||
upgrader: deployer.address
|
||
});
|
||
|
||
// 更新当前实现地址
|
||
deployments.contracts.YTAssetFactory.implementation = newFactoryImplAddress;
|
||
deployments.lastUpdate = new Date().toISOString();
|
||
|
||
fs.writeFileSync(deploymentsPath, JSON.stringify(deployments, null, 2));
|
||
console.log("💾 升级信息已保存到:", deploymentsPath);
|
||
|
||
// ========== 升级总结 ==========
|
||
console.log("\n🎉 升级总结:");
|
||
console.log("=====================================");
|
||
console.log("旧 YTAssetFactory Implementation:");
|
||
console.log(" ", deployments.upgradeHistory[deployments.upgradeHistory.length - 1].oldImplementation);
|
||
console.log("");
|
||
console.log("新 YTAssetFactory Implementation:");
|
||
console.log(" ", newFactoryImplAddress);
|
||
console.log("");
|
||
console.log("Factory Proxy (不变):");
|
||
console.log(" ", deployments.contracts.YTAssetFactory.proxy);
|
||
console.log("=====================================\n");
|
||
|
||
console.log("✅ 升级完成！");
|
||
console.log("");
|
||
console.log("📌 重要提示:");
|
||
console.log(" 1. YTAssetFactory 代理地址保持不变");
|
||
console.log(" 2. 所有状态数据已保留");
|
||
console.log(" 3. 建议运行验证脚本确认升级成功");
|
||
console.log(" 4. 主网升级前务必充分测试\n");
|
||
}
|
||
|
||
main()
|
||
.then(() => process.exit(0))
|
||
.catch((error) => {
|
||
console.error(error);
|
||
process.exit(1);
|
||
});
|