@0xobelisk/sui-cli 0.5.30 → 0.5.31

package/src/utils/checkBalance.ts

@@ -0,0 +1,55 @@
+import { Dubhe, NetworkType } from '@0xobelisk/sui-client';
+import chalk from 'chalk';
+import dotenv from 'dotenv';
+import { validatePrivateKey } from './utils';
+import { DubheCliError } from './errors';
+dotenv.config();
+
+export async function checkBalanceHandler(
+	network: 'mainnet' | 'testnet' | 'devnet' | 'loclnet',
+	amount: number = 2
+) {
+	try {
+		console.log(
+			chalk.blue(
+				`Note: You need at least 2 SUI for transaction fees and staking, and registering a Dapp will reserve 1 SUI for Dubhe Dapp Staking, which can be retrieved upon unregistering.`
+			)
+		);
+
+		const privateKey = process.env.PRIVATE_KEY;
+		if (!privateKey) {
+			throw new DubheCliError(
+				`Missing PRIVATE_KEY environment variable.
+  Run 'echo "PRIVATE_KEY=YOUR_PRIVATE_KEY" > .env'
+  in your contracts directory to use the default sui private key.`
+			);
+		}
+		const privateKeyFormat = validatePrivateKey(privateKey);
+		if (privateKeyFormat === false) {
+			throw new DubheCliError(`Please check your privateKey.`);
+		}
+
+		const dubhe = new Dubhe({
+			secretKey: process.env.PRIVATE_KEY,
+			networkType: network as NetworkType,
+		});
+
+		const balance = await dubhe.getBalance();
+		const balanceInSUI = Number(balance.totalBalance) / 1_000_000_000;
+
+		if (balanceInSUI < amount) {
+			// console.log(chalk.yellow(`Account balance is less than 2 SUI.`));
+			throw new DubheCliError(
+				`Account balance is less than ${amount} SUI. Please get more SUI.`
+			);
+		}
+
+		console.log(
+			chalk.green(
+				`Current account balance: ${balanceInSUI.toFixed(4)} SUI`
+			)
+		);
+	} catch (error) {
+		throw new DubheCliError('Failed to check balance: ' + error);
+	}
+}

package/src/utils/errors.ts

@@ -1,46 +1,46 @@
-import chalk from "chalk";
-import { ZodError } from "zod";
-import { fromZodError, ValidationError } from "zod-validation-error";
+import chalk from 'chalk';
+import { ZodError } from 'zod';
+import { fromZodError, ValidationError } from 'zod-validation-error';
 
 export class NotInsideProjectError extends Error {
-  name = "NotInsideProjectError";
-  message = "You are not inside a Dubhe project";
+	name = 'NotInsideProjectError';
+	message = 'You are not inside a Dubhe project';
 }
 
 export class DubheCliError extends Error {
-  name = "DubheCliError";
+	name = 'DubheCliError';
 }
 
 export class UpgradeError extends Error {
-  name = "UpgradeError";
+	name = 'UpgradeError';
 }
 
 export class FsIibError extends Error {
-  name = "FsIibError";
+	name = 'FsIibError';
 }
 
 export function logError(error: unknown) {
-  if (error instanceof ValidationError) {
-    console.log(chalk.redBright(error.message));
-  } else if (error instanceof ZodError) {
-    // TODO currently this error shouldn't happen, use `fromZodErrorCustom`
-    const validationError = fromZodError(error, {
-      prefixSeparator: "\n- ",
-      issueSeparator: "\n- ",
-    });
-    console.log(chalk.redBright(validationError.message));
-  } else if (error instanceof NotInsideProjectError) {
-    console.log(chalk.red(error.message));
-    console.log("");
-    // TODO add docs to the website and update the link to the specific page
-    console.log(
-      chalk.blue(
-        `To learn more about Dubhe's configuration, please go to https://github.com/0xobelisk`
-      )
-    );
-  } else if (error instanceof DubheCliError) {
-    console.log(chalk.red(error));
-  } else {
-    console.log(error);
-  }
+	if (error instanceof ValidationError) {
+		console.log(chalk.redBright(error.message));
+	} else if (error instanceof ZodError) {
+		// TODO currently this error shouldn't happen, use `fromZodErrorCustom`
+		const validationError = fromZodError(error, {
+			prefixSeparator: '\n- ',
+			issueSeparator: '\n- ',
+		});
+		console.log(chalk.redBright(validationError.message));
+	} else if (error instanceof NotInsideProjectError) {
+		console.log(chalk.red(error.message));
+		console.log('');
+		// TODO add docs to the website and update the link to the specific page
+		console.log(
+			chalk.blue(
+				`To learn more about Dubhe's configuration, please go to https://github.com/0xobelisk`
+			)
+		);
+	} else if (error instanceof DubheCliError) {
+		console.log(chalk.red(error));
+	} else {
+		console.log(error);
+	}
 }

package/src/utils/generateAccount.ts

@@ -0,0 +1,100 @@
+import { Dubhe } from '@0xobelisk/sui-client';
+import * as fs from 'fs';
+import chalk from 'chalk';
+
+export async function generateAccountHandler(
+	force: boolean = false,
+	outputTsPath: string
+) {
+	const path = process.cwd();
+	let privateKey: string;
+
+	if (force) {
+		const dubhe = new Dubhe();
+		const keypair = dubhe.getKeypair();
+		privateKey = keypair.getSecretKey();
+
+		fs.writeFileSync(`${path}/.env`, `PRIVATE_KEY=${privateKey}`);
+		console.log(chalk.green(`File created at: ${path}/.env`));
+
+		if (outputTsPath) {
+			const dir = outputTsPath.substring(
+				0,
+				outputTsPath.lastIndexOf('/')
+			);
+			if (!fs.existsSync(dir)) {
+				fs.mkdirSync(dir, { recursive: true });
+			}
+			fs.writeFileSync(
+				outputTsPath,
+				`export const PRIVATEKEY = '${privateKey}';
+export const ACCOUNT = '${keypair.toSuiAddress()}';
+`
+			);
+			console.log(chalk.green(`File created at: ${outputTsPath}`));
+		}
+
+		console.log(
+			chalk.blue(`Force generate new Account: ${keypair.toSuiAddress()}`)
+		);
+		return;
+	}
+
+	// Check if .env file exists and has content
+	try {
+		const envContent = fs.readFileSync(`${path}/.env`, 'utf8');
+		const match = envContent.match(/PRIVATE_KEY=(.+)/);
+		if (match && match[1]) {
+			privateKey = match[1];
+			const dubhe = new Dubhe({ secretKey: privateKey });
+			const keypair = dubhe.getKeypair();
+
+			if (outputTsPath) {
+				const dir = outputTsPath.substring(
+					0,
+					outputTsPath.lastIndexOf('/')
+				);
+				if (!fs.existsSync(dir)) {
+					fs.mkdirSync(dir, { recursive: true });
+				}
+				fs.writeFileSync(
+					outputTsPath,
+					`export const PRIVATEKEY = '${privateKey}';
+export const ACCOUNT = '${keypair.toSuiAddress()}';
+`
+				);
+				console.log(chalk.green(`File created at: ${outputTsPath}`));
+			}
+
+			console.log(
+				chalk.blue(`Using existing Account: ${keypair.toSuiAddress()}`)
+			);
+			return;
+		}
+	} catch (error) {
+		// .env file doesn't exist or failed to read, continue to generate new account
+	}
+
+	// If no existing private key, generate new account
+	const dubhe = new Dubhe();
+	const keypair = dubhe.getKeypair();
+	privateKey = keypair.getSecretKey();
+	fs.writeFileSync(`${path}/.env`, `PRIVATE_KEY=${privateKey}`);
+	console.log(chalk.green(`File created at: ${path}/.env`));
+
+	if (outputTsPath) {
+		const dir = outputTsPath.substring(0, outputTsPath.lastIndexOf('/'));
+		if (!fs.existsSync(dir)) {
+			fs.mkdirSync(dir, { recursive: true });
+		}
+		fs.writeFileSync(
+			outputTsPath,
+			`export const PRIVATEKEY = '${privateKey}';
+export const ACCOUNT = '${keypair.toSuiAddress()}';
+`
+		);
+		console.log(chalk.green(`File created at: ${outputTsPath}`));
+	}
+
+	console.log(chalk.blue(`Generate new Account: ${keypair.toSuiAddress()}`));
+}

package/src/utils/publishHandler.ts

@@ -12,7 +12,11 @@ import {
 	updateVersionInFile,
 	saveContractData,
 	validatePrivateKey,
-	schema, getSchemaHub, updateDubheDependency, switchEnv, delay,
+	schema,
+	getSchemaHub,
+	updateDubheDependency,
+	switchEnv,
+	delay,
 } from './utils';
 import { DubheConfig } from '@0xobelisk/sui-common';
 import * as fs from 'fs';
@@ -22,9 +26,12 @@ async function getDappsObjectId(
 	network: 'mainnet' | 'testnet' | 'devnet' | 'localnet'
 ) {
 	switch (network) {
-		case "localnet": {
+		case 'localnet': {
 			const path = process.cwd();
-			return await getSchemaHub(`${path}/contracts/dubhe-framework`, network)
+			return await getSchemaHub(
+				`${path}/contracts/dubhe-framework`,
+				network
+			);
 		}
 		case 'testnet':
 			return '0x8dbf8d28ac027ba214c9e0951b09f6842843be6cb87242b7d9a326a2677cd47a';
@@ -33,12 +40,18 @@ async function getDappsObjectId(
 	}
 }
 
-function removeEnvContent(filePath: string, networkType: 'mainnet' | 'testnet' | 'devnet' | 'localnet'): void {
+function removeEnvContent(
+	filePath: string,
+	networkType: 'mainnet' | 'testnet' | 'devnet' | 'localnet'
+): void {
 	if (!fs.existsSync(filePath)) {
 		return;
 	}
 	const content = fs.readFileSync(filePath, 'utf-8');
-	const regex = new RegExp(`\\[env\\.${networkType}\\][\\s\\S]*?(?=\\[|$)`, 'g');
+	const regex = new RegExp(
+		`\\[env\\.${networkType}\\][\\s\\S]*?(?=\\[|$)`,
+		'g'
+	);
 	const updatedContent = content.replace(regex, '');
 	fs.writeFileSync(filePath, updatedContent, 'utf-8');
 }
@@ -56,12 +69,20 @@ const chainIds: { [key: string]: string } = {
 	mainnet: '35834a8a',
 };
 
-function updateEnvFile(filePath: string, networkType: 'mainnet' | 'testnet' | 'devnet' | 'localnet', operation: 'publish' | 'upgrade', chainId: string, publishedId: string): void {
+function updateEnvFile(
+	filePath: string,
+	networkType: 'mainnet' | 'testnet' | 'devnet' | 'localnet',
+	operation: 'publish' | 'upgrade',
+	chainId: string,
+	publishedId: string
+): void {
 	const envFilePath = path.resolve(filePath);
 	const envContent = fs.readFileSync(envFilePath, 'utf-8');
 	const envLines = envContent.split('\n');
 
-	const networkSectionIndex = envLines.findIndex(line => line.trim() === `[env.${networkType}]`);
+	const networkSectionIndex = envLines.findIndex(
+		line => line.trim() === `[env.${networkType}]`
+	);
 	const config: EnvConfig = {
 		chainId: chainId,
 		originalPublishedId: '',
@@ -76,14 +97,18 @@ function updateEnvFile(filePath: string, networkType: 'mainnet' | 'testnet' | 'd
 			config.latestPublishedId = publishedId;
 			config.publishedVersion = 1;
 		} else {
-			throw new Error(`Network type [env.${networkType}] not found in the file and cannot upgrade.`);
+			throw new Error(
+				`Network type [env.${networkType}] not found in the file and cannot upgrade.`
+			);
 		}
 	} else {
 		for (let i = networkSectionIndex + 1; i < envLines.length; i++) {
 			const line = envLines[i].trim();
 			if (line.startsWith('[')) break; // End of the current network section
 
-			const [key, value] = line.split('=').map(part => part.trim().replace(/"/g, ''));
+			const [key, value] = line
+				.split('=')
+				.map(part => part.trim().replace(/"/g, ''));
 			switch (key) {
 				case 'original-published-id':
 					config.originalPublishedId = value;
@@ -115,9 +140,11 @@ latest-published-id = "${config.latestPublishedId}"
 published-version = "${config.publishedVersion}"
 `;
 
-	const newEnvContent = networkSectionIndex === -1
-		? envContent + updatedSection
-		: envLines.slice(0, networkSectionIndex).join('\n') + updatedSection;
+	const newEnvContent =
+		networkSectionIndex === -1
+			? envContent + updatedSection
+			: envLines.slice(0, networkSectionIndex).join('\n') +
+			  updatedSection;
 
 	fs.writeFileSync(envFilePath, newEnvContent, 'utf-8');
 }
@@ -137,7 +164,7 @@ function getLastSegment(input: string): string {
 	return segments.length > 0 ? segments[segments.length - 1] : '';
 }
 
-function buildContract(projectPath: string):  string[][] {
+function buildContract(projectPath: string): string[][] {
 	let modules: any, dependencies: any;
 	try {
 		const buildResult = JSON.parse(
@@ -168,7 +195,7 @@ async function publishContract(
 	projectPath: string
 ) {
 	const dappsObjectId = await getDappsObjectId(network);
-	console.log("dappsObjectId", dappsObjectId);
+	console.log('dappsObjectId', dappsObjectId);
 	const chainId = await client.getChainIdentifier();
 	removeEnvContent(`${projectPath}/Move.lock`, network);
 	console.log('\n🚀 Starting Contract Publication...');
@@ -227,7 +254,7 @@ async function publishContract(
 		}
 		if (
 			object.type === 'created' &&
-			object.objectType.includes("schema_hub")
+			object.objectType.includes('schema_hub')
 		) {
 			console.log(`  ├─ Schema Hub: ${object.objectId}`);
 			schemaHubId = object.objectId;
@@ -236,14 +263,20 @@ async function publishContract(
 
 	console.log(`  └─ Transaction: ${result.digest}`);
 
-	updateEnvFile(`${projectPath}/Move.lock`, network, 'publish', chainId, packageId);
+	updateEnvFile(
+		`${projectPath}/Move.lock`,
+		network,
+		'publish',
+		chainId,
+		packageId
+	);
 
 	console.log('\n⚡ Executing Deploy Hook...');
 	await delay(5000);
 
 	const deployHookTx = new Transaction();
 	deployHookTx.setGasBudget(2000000000);
-	const [txCoin] = deployHookTx.splitCoins(deployHookTx.gas, ["1000000000"]);
+	const [txCoin] = deployHookTx.splitCoins(deployHookTx.gas, ['1000000000']);
 	deployHookTx.moveCall({
 		target: `${packageId}::deploy_hook::run`,
 		arguments: [
@@ -251,7 +284,7 @@ async function publishContract(
 			deployHookTx.object(dappsObjectId),
 			deployHookTx.object(upgradeCapId),
 			deployHookTx.object('0x6'),
-			txCoin
+			txCoin,
 		],
 	});
 
@@ -276,14 +309,18 @@ async function publishContract(
 		deployHookResult.objectChanges?.map(object => {
 			if (
 				object.type === 'created' &&
-				object.objectType.includes('_schema') && !object.objectType.includes("dynamic_field")
+				object.objectType.includes('_schema') &&
+				!object.objectType.includes('dynamic_field')
 			) {
 				console.log(`  ├─ ${object.objectType}`);
 				console.log(`     └─ ID: ${object.objectId}`);
 
 				let structure: Record<string, string> = {};
 				for (let schemaKey in dubheConfig.schemas) {
-					if (capitalizeAndRemoveUnderscores(schemaKey) === getLastSegment(object.objectType)) {
+					if (
+						capitalizeAndRemoveUnderscores(schemaKey) ===
+						getLastSegment(object.objectType)
+					) {
 						structure = dubheConfig.schemas[schemaKey].structure;
 					}
 				}
@@ -303,7 +340,7 @@ async function publishContract(
 			upgradeCapId,
 			schemaHubId,
 			version,
-			schemas,
+			schemas
 		);
 		console.log('\n✅ Contract Publication Complete\n');
 	} else {
@@ -319,7 +356,7 @@ async function publishContract(
 export async function publishDubheFramework(
 	client: SuiClient,
 	dubhe: Dubhe,
-	network: 'mainnet' | 'testnet' | 'devnet' | 'localnet',
+	network: 'mainnet' | 'testnet' | 'devnet' | 'localnet'
 ) {
 	const path = process.cwd();
 	const projectPath = `${path}/contracts/dubhe-framework`;
@@ -335,7 +372,6 @@ export async function publishDubheFramework(
 	const keypair = dubhe.getKeypair();
 	console.log(`  └─ Account: ${keypair.toSuiAddress()}`);
 
-
 	console.log('\n📦 Building Contract...');
 	const [modules, dependencies] = buildContract(projectPath);
 
@@ -381,10 +417,7 @@ export async function publishDubheFramework(
 			console.log(`  ├─ Upgrade Cap: ${object.objectId}`);
 			upgradeCapId = object.objectId;
 		}
-		if (
-			object.type === 'created' &&
-			object.objectType.includes("dapps")
-		) {
+		if (object.type === 'created' && object.objectType.includes('dapps')) {
 			console.log(`  ├─ Dapps: ${object.objectId}`);
 			schemaHubId = object.objectId;
 		}
@@ -392,23 +425,29 @@ export async function publishDubheFramework(
 
 	console.log(`  └─ Transaction: ${result.digest}`);
 
-	updateEnvFile(`${projectPath}/Move.lock`, network, 'publish', chainId, packageId);
+	updateEnvFile(
+		`${projectPath}/Move.lock`,
+		network,
+		'publish',
+		chainId,
+		packageId
+	);
 
 	saveContractData(
-		"dubhe-framework",
+		'dubhe-framework',
 		network,
 		packageId,
 		upgradeCapId,
 		schemaHubId,
 		version,
-		schemas,
+		schemas
 	);
 }
 
 export async function publishHandler(
 	dubheConfig: DubheConfig,
 	network: 'mainnet' | 'testnet' | 'devnet' | 'localnet',
-	contractName?: string,
+	contractName?: string
 ) {
 	await switchEnv(network);
 
@@ -428,7 +467,7 @@ in your contracts directory to use the default sui private key.`
 	const dubhe = new Dubhe({ secretKey: privateKeyFormat });
 	const client = new SuiClient({ url: getFullnodeUrl(network) });
 
-	if (contractName == "dubhe-framework") {
+	if (contractName == 'dubhe-framework') {
 		await publishDubheFramework(client, dubhe, network);
 	} else {
 		const path = process.cwd();

package/src/utils/storeConfig.ts

@@ -0,0 +1,79 @@
+import * as fsAsync from 'fs/promises';
+import { mkdirSync, writeFileSync } from 'fs';
+import { exit } from 'process';
+import { dirname } from 'path';
+import { DeploymentJsonType, schema } from './utils';
+import { DubheConfig } from '@0xobelisk/sui-common';
+
+async function getDeploymentJson(
+	projectPath: string,
+	network: string
+): Promise<DeploymentJsonType> {
+	try {
+		const data = await fsAsync.readFile(
+			`${projectPath}/.history/sui_${network}/latest.json`,
+			'utf8'
+		);
+		return JSON.parse(data) as DeploymentJsonType;
+	} catch (error) {
+		throw new Error(
+			`read .history/sui_${network}/latest.json failed. ${error}`
+		);
+	}
+}
+
+function storeConfig(
+	network: string,
+	packageId: string,
+	schemas: schema[],
+	outputPath: string
+) {
+	let code = `type NetworkType = 'testnet' | 'mainnet' | 'devnet' | 'localnet';
+
+export const NETWORK: NetworkType = '${network}';
+
+export const PACKAGE_ID = '${packageId}'
+
+${schemas
+	.map(
+		schema =>
+			`export const ${schema.name.split('::')[2]}_Object_Id = '${
+				schema.objectId
+			}'`
+	)
+	.join('\n')}
+`;
+	// if (outputPath) {
+	writeOutput(code, outputPath, 'storeConfig');
+	// writeOutput(code, `${path}/src/chain/config.ts`, 'storeConfig');
+	// }
+}
+
+async function writeOutput(
+	output: string,
+	fullOutputPath: string,
+	logPrefix?: string
+): Promise<void> {
+	mkdirSync(dirname(fullOutputPath), { recursive: true });
+
+	writeFileSync(fullOutputPath, output);
+	if (logPrefix !== undefined) {
+		console.log(`${logPrefix}: ${fullOutputPath}`);
+	}
+}
+
+export async function storeConfigHandler(
+	dubheConfig: DubheConfig,
+	network: 'mainnet' | 'testnet' | 'devnet' | 'localnet',
+	outputPath: string
+) {
+	const path = process.cwd();
+	const contractPath = `${path}/contracts/${dubheConfig.name}`;
+	const deployment = await getDeploymentJson(contractPath, network);
+	storeConfig(
+		deployment.network,
+		deployment.packageId,
+		deployment.schemas,
+		outputPath
+	);
+}