@0xobelisk/sui-cli 1.2.0-pre.1 → 1.2.0-pre.100

package/src/utils/publishHandler.ts

@@ -3,19 +3,46 @@ import { execSync } from 'child_process';
 import chalk from 'chalk';
 import {
   saveContractData,
-  updateDubheDependency,
+  updateMoveTomlAddress,
   switchEnv,
   delay,
-  getDubheSchemaId,
-  initializeDubhe
+  getDubheDappHub,
+  initializeDubhe,
+  saveMetadata,
+  getOriginalDubhePackageId,
+  updatePublishedToml,
+  updateEphemeralPubFile,
+  getEphemeralPubFilePath
 } from './utils';
 import { DubheConfig } from '@0xobelisk/sui-common';
 import * as fs from 'fs';
 import * as path from 'path';
 
-const MAX_RETRIES = 60; // 60s timeout
-const RETRY_INTERVAL = 1000; // 1s retry interval
-const SPINNER = ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏'];
+/**
+ * Temporarily add localnet to Move.toml [environments] section before building.
+ * Sui CLI 1.40+ requires the active environment to be declared in Move.toml even
+ * when --build-env is specified. This patches the file and returns the original
+ * content so the caller can restore it in a finally block.
+ * Returns null if no changes were needed.
+ */
+function patchMoveTomlWithLocalnetEnv(moveTomlPath: string, chainId: string): string | null {
+  if (!fs.existsSync(moveTomlPath)) return null;
+  const content = fs.readFileSync(moveTomlPath, 'utf-8');
+
+  if (content.includes('localnet')) {
+    return null;
+  }
+
+  let updatedContent: string;
+  if (content.includes('[environments]')) {
+    updatedContent = content.replace('[environments]', `[environments]\nlocalnet = "${chainId}"`);
+  } else {
+    updatedContent = content.trimEnd() + `\n\n[environments]\nlocalnet = "${chainId}"\n`;
+  }
+
+  fs.writeFileSync(moveTomlPath, updatedContent, 'utf-8');
+  return content;
+}
 
 async function removeEnvContent(
   filePath: string,
@@ -127,21 +154,53 @@ published-version = "${config.publishedVersion}"
 // 	return segments.length > 0 ? segments[segments.length - 1] : '';
 // }
 
-function buildContract(projectPath: string): string[][] {
+/**
+ * Build a Move package and return [modules, dependencies] as base64 arrays.
+ *
+ * For localnet (ephemeral) networks:
+ *   - Uses --build-env testnet so dependency addresses are resolved via testnet
+ *     Published.toml (no need to add 'localnet' to Move.toml [environments]).
+ *   - Optionally reads a Pub.localnet.toml pubfile for already-published local deps.
+ *   - This matches the Sui docs approach:
+ *     https://docs.sui.io/guides/developer/packages/move-package-management
+ *
+ * For persistent networks (testnet/mainnet/devnet): uses -e <network> as before.
+ */
+function buildContract(
+  projectPath: string,
+  network?: 'mainnet' | 'testnet' | 'devnet' | 'localnet',
+  pubfilePath?: string
+): string[][] {
   let modules: any, dependencies: any;
   try {
+    let buildEnvFlag: string;
+    if (network === 'localnet') {
+      // Ephemeral approach: resolve deps via testnet configuration.
+      // --pubfile-path supplies already-published local dep addresses (e.g. dubhe).
+      buildEnvFlag = ' --build-env testnet';
+      if (pubfilePath) {
+        buildEnvFlag += ` --pubfile-path ${pubfilePath}`;
+      }
+    } else {
+      buildEnvFlag = network ? ` -e ${network}` : '';
+    }
+
+    // --no-tree-shaking avoids on-chain RPC calls during build.
     const buildResult = JSON.parse(
-      execSync(`sui move build --dump-bytecode-as-base64 --path ${projectPath}`, {
-        encoding: 'utf-8',
-        stdio: 'pipe'
-      })
+      execSync(
+        `sui move build --dump-bytecode-as-base64 --no-tree-shaking${buildEnvFlag} --path ${projectPath}`,
+        {
+          encoding: 'utf-8',
+          stdio: 'pipe'
+        }
+      )
     );
     modules = buildResult.modules;
     dependencies = buildResult.dependencies;
   } catch (error: any) {
     console.error(chalk.red('  └─ Build failed'));
-    console.error(error.stdout);
-    process.exit(1);
+    console.error(error.stdout || error.stderr);
+    throw new Error(`Build failed: ${error.stdout || error.stderr || error.message}`);
   }
   return [modules, dependencies];
 }
@@ -154,132 +213,28 @@ interface ObjectChange {
 }
 
 async function waitForNode(dubhe: Dubhe): Promise<string> {
-  let retryCount = 0;
-  let spinnerIndex = 0;
-  const startTime = Date.now();
-  let isInterrupted = false;
-  let chainId = '';
-  let hasShownBalanceWarning = false;
-
-  const handleInterrupt = () => {
-    isInterrupted = true;
-    process.stdout.write('\r' + ' '.repeat(50) + '\r');
-    console.log('\n  └─ Operation cancelled by user');
-    process.exit(0);
-  };
-  process.on('SIGINT', handleInterrupt);
-
-  try {
-    // 第一阶段：等待获取 chainId
-    while (retryCount < MAX_RETRIES && !isInterrupted && !chainId) {
-      try {
-        chainId = await dubhe.suiInteractor.currentClient.getChainIdentifier();
-      } catch (error) {
-        // 忽略错误，继续重试
-      }
-
-      if (isInterrupted) break;
-
-      if (!chainId) {
-        retryCount++;
-        if (retryCount === MAX_RETRIES) {
-          console.log(chalk.red(`  └─ Failed to connect to node after ${MAX_RETRIES} attempts`));
-          console.log(chalk.red('  └─ Please check if the Sui node is running.'));
-          process.exit(1);
-        }
-
-        const elapsedTime = Math.floor((Date.now() - startTime) / 1000);
-        const spinner = SPINNER[spinnerIndex % SPINNER.length];
-        spinnerIndex++;
-
-        process.stdout.write(`\r  ├─ ${spinner} Waiting for node... (${elapsedTime}s)`);
-        await new Promise((resolve) => setTimeout(resolve, RETRY_INTERVAL));
-      }
-    }
-
-    // 显示 chainId
-    process.stdout.write('\r' + ' '.repeat(50) + '\r');
-    console.log(`  ├─ ChainId: ${chainId}`);
-
-    // 第二阶段：检查部署账户余额
-    retryCount = 0;
-    while (retryCount < MAX_RETRIES && !isInterrupted) {
-      try {
-        const address = dubhe.getAddress();
-        const coins = await dubhe.suiInteractor.currentClient.getCoins({
-          owner: address,
-          coinType: '0x2::sui::SUI'
-        });
-
-        if (coins.data.length > 0) {
-          const balance = coins.data.reduce((sum, coin) => sum + Number(coin.balance), 0);
-          if (balance > 0) {
-            process.stdout.write('\r' + ' '.repeat(50) + '\r');
-            console.log(`  ├─ Deployer balance: ${balance / 10 ** 9} SUI`);
-            return chainId;
-          } else if (!hasShownBalanceWarning) {
-            process.stdout.write('\r' + ' '.repeat(50) + '\r');
-            console.log(
-              chalk.yellow(
-                `  ├─ Deployer balance: 0 SUI, please ensure your account has sufficient SUI balance`
-              )
-            );
-            hasShownBalanceWarning = true;
-          }
-        } else if (!hasShownBalanceWarning) {
-          process.stdout.write('\r' + ' '.repeat(50) + '\r');
-          console.log(
-            chalk.yellow(
-              `  ├─ No SUI coins found in deployer account, please ensure your account has sufficient SUI balance`
-            )
-          );
-          hasShownBalanceWarning = true;
-        }
-
-        retryCount++;
-        if (retryCount === MAX_RETRIES) {
-          console.log(
-            chalk.red(`  └─ Deployer account has no SUI balance after ${MAX_RETRIES} attempts`)
-          );
-          console.log(chalk.red('  └─ Please ensure your account has sufficient SUI balance.'));
-          process.exit(1);
-        }
-
-        const elapsedTime = Math.floor((Date.now() - startTime) / 1000);
-        const spinner = SPINNER[spinnerIndex % SPINNER.length];
-        spinnerIndex++;
-
-        process.stdout.write(`\r  ├─ ${spinner} Checking deployer balance... (${elapsedTime}s)`);
-        await new Promise((resolve) => setTimeout(resolve, RETRY_INTERVAL));
-      } catch (error) {
-        if (isInterrupted) break;
-
-        retryCount++;
-        if (retryCount === MAX_RETRIES) {
-          console.log(
-            chalk.red(`  └─ Failed to check deployer balance after ${MAX_RETRIES} attempts`)
-          );
-          console.log(chalk.red('  └─ Please check your account and network connection.'));
-          process.exit(1);
-        }
-
-        const elapsedTime = Math.floor((Date.now() - startTime) / 1000);
-        const spinner = SPINNER[spinnerIndex % SPINNER.length];
-        spinnerIndex++;
+  const chainId = await dubhe.suiInteractor.currentClient.getChainIdentifier();
+  console.log(`  ├─ ChainId: ${chainId}`);
+  const address = dubhe.getAddress();
+  const coins = await dubhe.suiInteractor.currentClient.getCoins({
+    owner: address,
+    coinType: '0x2::sui::SUI'
+  });
 
-        process.stdout.write(`\r  ├─ ${spinner} Checking deployer balance... (${elapsedTime}s)`);
-        await new Promise((resolve) => setTimeout(resolve, RETRY_INTERVAL));
-      }
+  if (coins.data.length > 0) {
+    const balance = coins.data.reduce((sum, coin) => sum + Number(coin.balance), 0);
+    if (balance > 0) {
+      console.log(`  ├─ Deployer balance: ${balance / 10 ** 9} SUI`);
+      return chainId;
+    } else {
+      console.log(
+        chalk.yellow(
+          `  ├─ Deployer balance: 0 SUI, please ensure your account has sufficient SUI balance`
+        )
+      );
     }
-  } finally {
-    process.removeListener('SIGINT', handleInterrupt);
   }
-
-  if (isInterrupted) {
-    process.exit(0);
-  }
-
-  throw new Error('Failed to connect to node');
+  return chainId;
 }
 
 async function publishContract(
@@ -301,7 +256,61 @@ async function publishContract(
   console.log(`  └─ Account: ${dubhe.getAddress()}`);
 
   console.log('\n📦 Building Contract...');
-  const [modules, dependencies] = buildContract(projectPath);
+  // For localnet: pass the ephemeral pubfile so the build system can resolve
+  // the dubhe dependency that was just published in publishDubheFramework().
+  const pubfilePath =
+    network === 'localnet' ? getEphemeralPubFilePath(process.cwd(), network) : undefined;
+
+  // For localnet: temporarily remove Published.toml to prevent the testnet address
+  // (if it exists) from being picked up by --build-env testnet, which would cause
+  // PublishErrorNonZeroAddress. The pubfile supplies dubhe's localnet address instead.
+  const contractPublishedTomlPath = `${projectPath}/Published.toml`;
+  let savedContractPublishedToml: string | null = null;
+  if (network === 'localnet' && fs.existsSync(contractPublishedTomlPath)) {
+    savedContractPublishedToml = fs.readFileSync(contractPublishedTomlPath, 'utf-8');
+    fs.unlinkSync(contractPublishedTomlPath);
+  }
+
+  // For localnet: also temporarily remove dubhe's Published.toml when building the
+  // contract package. After publishDubheFramework restores dubhe/Published.toml, its
+  // testnet entry's original-id may be "0x0", which collides with the contract package's
+  // own address (also 0x0 before publish), triggering a spurious cyclic-dependency error.
+  const dubhePublishedTomlPath = path.join(path.dirname(projectPath), 'dubhe', 'Published.toml');
+  let savedDubhePublishedToml: string | null = null;
+  if (network === 'localnet' && fs.existsSync(dubhePublishedTomlPath)) {
+    savedDubhePublishedToml = fs.readFileSync(dubhePublishedTomlPath, 'utf-8');
+    fs.unlinkSync(dubhePublishedTomlPath);
+  }
+
+  // Sui CLI 1.40+ checks that the active environment is declared in Move.toml
+  // even when --build-env is specified. Temporarily inject localnet into [environments]
+  // for both the contract and its dubhe dependency.
+  const contractMoveTomlPath = `${projectPath}/Move.toml`;
+  const dubheMoveTomlPath = path.join(path.dirname(projectPath), 'dubhe', 'Move.toml');
+  let savedContractMoveToml: string | null = null;
+  let savedDubheMoveToml: string | null = null;
+  if (network === 'localnet') {
+    savedContractMoveToml = patchMoveTomlWithLocalnetEnv(contractMoveTomlPath, chainId);
+    savedDubheMoveToml = patchMoveTomlWithLocalnetEnv(dubheMoveTomlPath, chainId);
+  }
+
+  let modules: any, dependencies: any;
+  try {
+    [modules, dependencies] = buildContract(projectPath, network, pubfilePath);
+  } finally {
+    if (savedContractPublishedToml !== null) {
+      fs.writeFileSync(contractPublishedTomlPath, savedContractPublishedToml, 'utf-8');
+    }
+    if (savedDubhePublishedToml !== null) {
+      fs.writeFileSync(dubhePublishedTomlPath, savedDubhePublishedToml, 'utf-8');
+    }
+    if (savedContractMoveToml !== null) {
+      fs.writeFileSync(contractMoveTomlPath, savedContractMoveToml, 'utf-8');
+    }
+    if (savedDubheMoveToml !== null) {
+      fs.writeFileSync(dubheMoveTomlPath, savedDubheMoveToml, 'utf-8');
+    }
+  }
 
   console.log('\n🔄 Publishing Contract...');
   const tx = new Transaction();
@@ -311,63 +320,30 @@ async function publishContract(
   const [upgradeCap] = tx.publish({ modules, dependencies });
   tx.transferObjects([upgradeCap], dubhe.getAddress());
 
-  let result: any = null;
-  let retryCount = 0;
-  let spinnerIndex = 0;
-  const startTime = Date.now();
-  let isInterrupted = false;
-
-  const handleInterrupt = () => {
-    isInterrupted = true;
-    process.stdout.write('\r' + ' '.repeat(50) + '\r');
-    console.log('\n  └─ Operation cancelled by user');
-    process.exit(0);
-  };
-  process.on('SIGINT', handleInterrupt);
-
+  let result;
   try {
-    while (retryCount < MAX_RETRIES && !result && !isInterrupted) {
-      try {
-        result = await dubhe.signAndSendTxn({ tx });
-      } catch (error) {
-        if (isInterrupted) break;
-
-        retryCount++;
-        if (retryCount === MAX_RETRIES) {
-          console.log(chalk.red(`  └─ Publication failed after ${MAX_RETRIES} attempts`));
-          console.log(chalk.red('  └─ Please check your network connection and try again later.'));
-          process.exit(1);
-        }
-
-        const elapsedTime = Math.floor((Date.now() - startTime) / 1000);
-        const spinner = SPINNER[spinnerIndex % SPINNER.length];
-        spinnerIndex++;
-
-        process.stdout.write(`\r  ├─ ${spinner} Retrying... (${elapsedTime}s)`);
-        await new Promise((resolve) => setTimeout(resolve, RETRY_INTERVAL));
-      }
-    }
-  } finally {
-    process.removeListener('SIGINT', handleInterrupt);
-  }
-
-  if (isInterrupted) {
-    process.exit(0);
+    result = await dubhe.signAndSendTxn({ tx });
+  } catch (error: any) {
+    console.error(chalk.red('  └─ Publication failed'));
+    console.error(error.message);
+    throw new Error(`Contract publication failed: ${error.message}`);
   }
 
-  process.stdout.write('\r' + ' '.repeat(50) + '\r');
-
   if (!result || result.effects?.status.status === 'failure') {
-    console.log(chalk.red('  └─ Publication failed'));
-    process.exit(1);
+    throw new Error('Contract publication transaction failed');
   }
 
   console.log('  ├─ Processing publication results...');
   let version = 1;
   let packageId = '';
-  let schemaId = '';
-  let schemas = dubheConfig.schemas;
+  let dappHub = '';
+  let components = dubheConfig.components;
+  let resources = dubheConfig.resources;
+  let enums = dubheConfig.enums;
   let upgradeCapId = '';
+  let startCheckpoint = '';
+
+  let printObjects: any[] = [];
 
   result.objectChanges!.map((object: ObjectChange) => {
     if (object.type === 'published') {
@@ -382,24 +358,35 @@ async function publishContract(
       console.log(`  ├─ Upgrade Cap: ${object.objectId}`);
       upgradeCapId = object.objectId || '';
     }
+    if (
+      object.type === 'created' &&
+      object.objectType &&
+      object.objectType.includes('dapp_service::DappHub')
+    ) {
+      dappHub = object.objectId || '';
+    }
+    if (object.type === 'created') {
+      printObjects.push(object);
+    }
   });
 
   console.log(`  └─ Transaction: ${result.digest}`);
 
   updateEnvFile(`${projectPath}/Move.lock`, network, 'publish', chainId, packageId);
+  updatePublishedToml(projectPath, network, chainId, packageId, packageId, 1);
 
   console.log('\n⚡ Executing Deploy Hook...');
   await delay(5000);
 
+  startCheckpoint = await dubhe.suiInteractor.currentClient.getLatestCheckpointSequenceNumber();
+
   const deployHookTx = new Transaction();
   let args = [];
-  if (dubheConfig.name !== 'dubhe') {
-    let dubheSchemaId = await getDubheSchemaId(network);
-    args.push(deployHookTx.object(dubheSchemaId));
-  }
+  let dubheDappHub = dubheConfig.name === 'dubhe' ? dappHub : await getDubheDappHub(network);
+  args.push(deployHookTx.object(dubheDappHub));
   args.push(deployHookTx.object('0x6'));
   deployHookTx.moveCall({
-    target: `${packageId}::${dubheConfig.name}_genesis::run`,
+    target: `${packageId}::genesis::run`,
     arguments: args
   });
 
@@ -409,49 +396,46 @@ async function publishContract(
   } catch (error: any) {
     console.error(chalk.red('  └─ Deploy hook execution failed'));
     console.error(error.message);
-    process.exit(1);
+    throw new Error(`genesis::run failed: ${error.message}`);
   }
 
   if (deployHookResult.effects?.status.status === 'success') {
     console.log('  ├─ Hook execution successful');
     console.log(`  ├─ Transaction: ${deployHookResult.digest}`);
 
-    console.log('\n📋 Created Schemas:');
-    deployHookResult.objectChanges?.map((object: ObjectChange) => {
-      if (
-        object.type === 'created' &&
-        object.objectType &&
-        object.objectType.includes('schema::Schema')
-      ) {
-        schemaId = object.objectId || '';
-      }
-      if (
-        object.type === 'created' &&
-        object.objectType &&
-        object.objectType.includes('schema') &&
-        !object.objectType.includes('dynamic_field')
-      ) {
-        console.log(`  ├─ Type: ${object.objectType}`);
-        console.log(`  └─ ID: ${object.objectId}`);
-      }
+    console.log('\n📋 Created Objects:');
+    printObjects.map((object: ObjectChange) => {
+      console.log(`  ├─ ID: ${object.objectId}`);
+      console.log(`  └─ Type: ${object.objectType}`);
     });
 
-    saveContractData(
+    await saveContractData(
       dubheConfig.name,
       network,
+      startCheckpoint,
       packageId,
-      schemaId,
+      dubheDappHub,
       upgradeCapId,
       version,
-      schemas
+      components,
+      resources,
+      enums
     );
+
+    await saveMetadata(dubheConfig.name, network, packageId);
+
+    // Insert package id to dubhe config
+    let config = JSON.parse(fs.readFileSync(`${process.cwd()}/dubhe.config.json`, 'utf-8'));
+    config.original_package_id = packageId;
+    config.dubhe_object_id = dubheDappHub;
+    config.original_dubhe_package_id = await getOriginalDubhePackageId(network);
+    config.start_checkpoint = startCheckpoint;
+
+    fs.writeFileSync(`${process.cwd()}/dubhe.config.json`, JSON.stringify(config, null, 2));
+
     console.log('\n✅ Contract Publication Complete\n');
   } else {
-    console.log(chalk.yellow('  └─ Deploy hook execution failed'));
-    console.log(chalk.yellow('     Please republish or manually call deploy_hook::run'));
-    console.log(chalk.yellow('     Please check the transaction digest:'));
-    console.log(chalk.yellow(`     ${deployHookResult.digest}`));
-    process.exit(1);
+    throw new Error(`genesis::run transaction failed. Digest: ${deployHookResult.digest}`);
   }
 }
 
@@ -460,15 +444,13 @@ async function checkDubheFramework(projectPath: string): Promise<boolean> {
     console.log(chalk.yellow('\nℹ️ Dubhe Framework Files Not Found'));
     console.log(chalk.yellow('  ├─ Expected Path:'), projectPath);
     console.log(chalk.yellow('  ├─ To set up Dubhe Framework:'));
-    console.log(chalk.yellow('  │  1. Create directory: mkdir -p contracts/dubhe-framework'));
+    console.log(chalk.yellow('  │  1. Create directory: mkdir -p contracts/dubhe'));
     console.log(
       chalk.yellow(
-        '  │  2. Clone repository: git clone https://github.com/0xobelisk/dubhe-framework contracts/dubhe-framework'
+        '  │  2. Clone repository: git clone https://github.com/0xobelisk/dubhe contracts/dubhe'
       )
     );
-    console.log(
-      chalk.yellow('  │  3. Or download from: https://github.com/0xobelisk/dubhe-framework')
-    );
+    console.log(chalk.yellow('  │  3. Or download from: https://github.com/0xobelisk/dubhe'));
     console.log(chalk.yellow('  └─ After setup, restart the local node'));
     return false;
   }
@@ -479,8 +461,8 @@ export async function publishDubheFramework(
   dubhe: Dubhe,
   network: 'mainnet' | 'testnet' | 'devnet' | 'localnet'
 ) {
-  const path = process.cwd();
-  const projectPath = `${path}/contracts/dubhe-framework`;
+  const cwd = process.cwd();
+  const projectPath = `${cwd}/src/dubhe`;
 
   if (!(await checkDubheFramework(projectPath))) {
     return;
@@ -492,66 +474,66 @@ export async function publishDubheFramework(
   const chainId = await waitForNode(dubhe);
 
   await removeEnvContent(`${projectPath}/Move.lock`, network);
-  const [modules, dependencies] = buildContract(projectPath);
-  const tx = new Transaction();
-  const [upgradeCap] = tx.publish({ modules, dependencies });
-  tx.transferObjects([upgradeCap], dubhe.getAddress());
+  await updateMoveTomlAddress(projectPath, '0x0');
+
+  const startCheckpoint =
+    await dubhe.suiInteractor.currentClient.getLatestCheckpointSequenceNumber();
+
+  // For localnet: --build-env testnet is used to resolve git dependencies, but the
+  // Move CLI will also read Published.toml and use any existing testnet address for
+  // dubhe — causing PublishErrorNonZeroAddress if a testnet entry already exists.
+  // Fix: temporarily remove Published.toml before the build, then restore it.
+  // This ensures the dubhe package compiles with address 0x0 (from Move.toml).
+  const publishedTomlPath = `${projectPath}/Published.toml`;
+  let savedPublishedTomlContent: string | null = null;
+  if (network === 'localnet' && fs.existsSync(publishedTomlPath)) {
+    savedPublishedTomlContent = fs.readFileSync(publishedTomlPath, 'utf-8');
+    fs.unlinkSync(publishedTomlPath);
+  }
 
-  let result: any = null;
-  let retryCount = 0;
-  let spinnerIndex = 0;
-  const startTime = Date.now();
-  let isInterrupted = false;
-
-  const handleInterrupt = () => {
-    isInterrupted = true;
-    process.stdout.write('\r' + ' '.repeat(50) + '\r');
-    console.log('\n  └─ Operation cancelled by user');
-    process.exit(0);
-  };
-  process.on('SIGINT', handleInterrupt);
+  // Sui CLI 1.40+ checks that the active environment is declared in Move.toml
+  // even when --build-env is specified. Temporarily inject localnet into [environments].
+  const moveTomlPath = `${projectPath}/Move.toml`;
+  let savedMoveTomlContent: string | null = null;
+  if (network === 'localnet') {
+    savedMoveTomlContent = patchMoveTomlWithLocalnetEnv(moveTomlPath, chainId);
+  }
 
+  let modules: any, dependencies: any;
   try {
-    while (retryCount < MAX_RETRIES && !result && !isInterrupted) {
-      try {
-        result = await dubhe.signAndSendTxn({ tx });
-      } catch (error) {
-        if (isInterrupted) break;
-
-        retryCount++;
-        if (retryCount === MAX_RETRIES) {
-          console.log(chalk.red(`  └─ Publication failed after ${MAX_RETRIES} attempts`));
-          console.log(chalk.red('  └─ Please check your network connection and try again later.'));
-          process.exit(1);
-        }
-
-        const elapsedTime = Math.floor((Date.now() - startTime) / 1000);
-        const spinner = SPINNER[spinnerIndex % SPINNER.length];
-        spinnerIndex++;
-
-        process.stdout.write(`\r  ├─ ${spinner} Retrying... (${elapsedTime}s)`);
-        await new Promise((resolve) => setTimeout(resolve, RETRY_INTERVAL));
-      }
-    }
+    // For localnet: use --build-env testnet (no pubfile needed — dubhe has no local deps).
+    // For testnet/mainnet: use -e <network> as usual.
+    [modules, dependencies] = buildContract(projectPath, network);
   } finally {
-    process.removeListener('SIGINT', handleInterrupt);
+    // Always restore Published.toml and Move.toml (successful build or error)
+    if (savedPublishedTomlContent !== null) {
+      fs.writeFileSync(publishedTomlPath, savedPublishedTomlContent, 'utf-8');
+    }
+    if (savedMoveTomlContent !== null) {
+      fs.writeFileSync(moveTomlPath, savedMoveTomlContent, 'utf-8');
+    }
   }
 
-  if (isInterrupted) {
-    process.exit(0);
-  }
+  const tx = new Transaction();
+  const [upgradeCap] = tx.publish({ modules, dependencies });
+  tx.transferObjects([upgradeCap], dubhe.getAddress());
 
-  process.stdout.write('\r' + ' '.repeat(50) + '\r');
+  let result;
+  try {
+    result = await dubhe.signAndSendTxn({ tx });
+  } catch (error: any) {
+    console.error(chalk.red('  └─ Publication failed'));
+    console.error(error.message);
+    throw new Error(`Dubhe framework publication failed: ${error.message}`);
+  }
 
   if (!result || result.effects?.status.status === 'failure') {
-    console.log(chalk.red('  └─ Publication failed'));
-    process.exit(1);
+    throw new Error('Dubhe framework publication transaction failed');
   }
 
   let version = 1;
   let packageId = '';
-  let schemaId = '';
-  let schemas: Record<string, string> = {};
+  let dappHub = '';
   let upgradeCapId = '';
 
   result.objectChanges!.map((object: ObjectChange) => {
@@ -565,14 +547,20 @@ export async function publishDubheFramework(
     ) {
       upgradeCapId = object.objectId || '';
     }
+    if (
+      object.type === 'created' &&
+      object.objectType &&
+      object.objectType.includes('dapp_service::DappHub')
+    ) {
+      dappHub = object.objectId || '';
+    }
   });
 
   await delay(3000);
-
   const deployHookTx = new Transaction();
   deployHookTx.moveCall({
-    target: `${packageId}::dubhe_genesis::run`,
-    arguments: [deployHookTx.object('0x6')]
+    target: `${packageId}::genesis::run`,
+    arguments: [deployHookTx.object(dappHub), deployHookTx.object('0x6')]
   });
 
   let deployHookResult;
@@ -581,30 +569,47 @@ export async function publishDubheFramework(
   } catch (error: any) {
     console.error(chalk.red('  └─ Deploy hook execution failed'));
     console.error(error.message);
-    process.exit(1);
+    throw new Error(`Dubhe genesis::run failed: ${error.message}`);
   }
 
-  if (deployHookResult.effects?.status.status === 'success') {
-    deployHookResult.objectChanges?.map((object: ObjectChange) => {
-      if (
-        object.type === 'created' &&
-        object.objectType &&
-        object.objectType.includes('dubhe_schema::Schema')
-      ) {
-        schemaId = object.objectId || '';
-      }
-    });
+  if (deployHookResult.effects?.status.status !== 'success') {
+    throw new Error('Deploy hook execution failed');
   }
 
-  saveContractData('dubhe-framework', network, packageId, schemaId, upgradeCapId, version, schemas);
+  await updateMoveTomlAddress(projectPath, packageId);
+  await saveContractData(
+    'dubhe',
+    network,
+    startCheckpoint,
+    packageId,
+    dappHub,
+    upgradeCapId,
+    version,
+    {},
+    {},
+    {}
+  );
 
   updateEnvFile(`${projectPath}/Move.lock`, network, 'publish', chainId, packageId);
-  await delay(1000);
+  updatePublishedToml(projectPath, network, chainId, packageId, packageId, 1);
+
+  // For localnet: write dubhe's published address to Pub.localnet.toml so that
+  // the counter package build (next step) can resolve the dubhe dependency.
+  if (network === 'localnet') {
+    const pubfilePath = getEphemeralPubFilePath(cwd, network);
+    updateEphemeralPubFile(pubfilePath, chainId, 'testnet', {
+      source: projectPath,
+      publishedAt: packageId,
+      originalId: packageId,
+      upgradeCap: upgradeCapId
+    });
+  }
 }
 
 export async function publishHandler(
   dubheConfig: DubheConfig,
   network: 'mainnet' | 'testnet' | 'devnet' | 'localnet',
+  _force: boolean,
   gasBudget?: number
 ) {
   await switchEnv(network);
@@ -613,12 +618,12 @@ export async function publishHandler(
     network
   });
 
-  if (network === 'localnet') {
+  const path = process.cwd();
+  const projectPath = `${path}/src/${dubheConfig.name}`;
+
+  if (network === 'localnet' && dubheConfig.name !== 'dubhe') {
     await publishDubheFramework(dubhe, network);
   }
 
-  const path = process.cwd();
-  const projectPath = `${path}/contracts/${dubheConfig.name}`;
-  await updateDubheDependency(`${projectPath}/Move.toml`, network);
   await publishContract(dubhe, dubheConfig, network, projectPath, gasBudget);
 }