fix(scripts): support overloaded initialize in deploy and app:deploy task

- deployContract() accepts optional initializer signature param,
  passed to hre.upgrades.deployProxy for overload disambiguation
- estimateDeploymentCost() accepts optional initializer param,
  passed to encodeFunctionData
- app:deploy task adds --requireTcbUpToDate flag and passes 6 args
  with explicit initializer signature

Author: Leechael

kms/auth-eth/hardhat.config.ts

@@ -244,6 +244,7 @@ task("kms:get-app-implementation", "Get current DstackApp implementation address
 
 task("app:deploy", "Deploy DstackApp with a UUPS proxy")
   .addFlag("allowAnyDevice", "Allow any device to boot this app")
+  .addFlag("requireTcbUpToDate", "Require TCB status to be UpToDate")
   .addOptionalParam("device", "Initial device ID", "", types.string)
   .addOptionalParam("hash", "Initial compose hash", "", types.string)
   .setAction(async (taskArgs, hre) => {
@@ -267,14 +268,14 @@ task("app:deploy", "Deploy DstackApp with a UUPS proxy")
       console.log("Initial compose hash:", composeHash === "0x0000000000000000000000000000000000000000000000000000000000000000" ? "none" : composeHash);
     }
 
-    // Use standard deployment - all cases use the same 6-parameter initializer
     const appContract = await deployContract(hre, "DstackApp", [
-      deployerAddress, 
-      false, 
-      taskArgs.allowAnyDevice,
+      deployerAddress,
+      false,                          // _disableUpgrades
+      taskArgs.requireTcbUpToDate,    // _requireTcbUpToDate
+      taskArgs.allowAnyDevice,        // _allowAnyDevice
       deviceId,
       composeHash
-    ]);
+    ], false, "initialize(address,bool,bool,bool,bytes32,bytes32)");
 
     if (!appContract) {
       return;

kms/auth-eth/lib/deployment-helpers.ts

@@ -62,14 +62,15 @@ export async function accountBalance(ethers: any, address: string) {
 export async function estimateDeploymentCost(
     hre: HardhatRuntimeEnvironment,
     contractName: string,
-    initializerArgs: any[] = []
+    initializerArgs: any[] = [],
+    initializer?: string
 ) {
     console.log("Estimating deployment costs...");
     const factory = await hre.ethers.getContractFactory(contractName);
 
     // Get the data for initialize function
     const initData = factory.interface.encodeFunctionData(
-        "initialize",
+        initializer || "initialize",
         initializerArgs
     );
 

kms/auth-eth/scripts/deploy.ts

@@ -6,7 +6,7 @@ import { HardhatRuntimeEnvironment } from "hardhat/types";
 import * as helpers from "../lib/deployment-helpers";
 
 // This function should be called directly by Hardhat tasks
-export async function deployContract(hre: HardhatRuntimeEnvironment, contractName: string, initializerArgs: any[] = [], quiet: boolean = false) {
+export async function deployContract(hre: HardhatRuntimeEnvironment, contractName: string, initializerArgs: any[] = [], quiet: boolean = false, initializer?: string) {
   try {
     function log(...msgs: any[]) {
       if (!quiet) {
@@ -29,7 +29,8 @@ export async function deployContract(hre: HardhatRuntimeEnvironment, contractNam
       await helpers.estimateDeploymentCost(
         hre,
         contractName,
-        initializerArgs
+        initializerArgs,
+        initializer
       );
 
       // Prompt for confirmation
@@ -43,7 +44,7 @@ export async function deployContract(hre: HardhatRuntimeEnvironment, contractNam
     log("Deploying proxy...");
     const contract = await hre.upgrades.deployProxy(contractFactory,
       initializerArgs,
-      { kind: 'uups' }
+      { kind: 'uups', ...(initializer ? { initializer } : {}) }
     );
     log("Waiting for deployment...");
     await contract.waitForDeployment();