feat: add retry for contract name input (#318)

Author: i582

CHANGELOG.md

@@ -5,6 +5,12 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## Unreleased
+
+### Changed
+
+- If the input contract name is invalid, blueprint does not exit, but asks for the name again
+
 ## [0.40.0] - 2025-08-18
 
 ### Added

src/cli/create.ts

@@ -4,12 +4,13 @@ import { lstat, mkdir, open, readdir, readFile } from 'fs/promises';
 
 import arg from 'arg';
 import { Project } from '@tact-lang/compiler';
+import chalk from 'chalk';
 
 import { getConfig } from '../config/utils';
 import { getRootTactConfig, TactConfig, updateRootTactConfig } from '../config/tact.config';
 import { Args, extractFirstArg, Runner } from './Runner';
 import { executeTemplate, TEMPLATES_DIR } from '../template';
-import { assertValidContractName, selectOption, toSnakeCase } from '../utils';
+import { validateContractName, selectOption, toSnakeCase } from '../utils';
 import { UIProvider } from '../ui/UIProvider';
 import { buildOne } from '../build';
 import { helpArgs, helpMessages, templateTypes } from './constants';
@@ -85,8 +86,15 @@ export const create: Runner = async (_args: Args, ui: UIProvider) => {
         return;
     }
 
-    const name = extractFirstArg(localArgs) ?? (await ui.input('Contract name (PascalCase)'));
-    assertValidContractName(name);
+    const argName = extractFirstArg(localArgs);
+    if (argName !== undefined) {
+        const error = validateContractName(argName);
+        if (error) {
+            throw new Error(error);
+        }
+    }
+
+    const name = argName ?? (await requestContractName('Contract name (PascalCase)', ui));
 
     const which = (
         await selectOption(templateTypes, {
@@ -120,3 +128,16 @@ export const create: Runner = async (_args: Args, ui: UIProvider) => {
         await createFiles(path.join(TEMPLATES_DIR, lang, template), process.cwd(), replaces);
     }
 };
+
+export async function requestContractName(message: string, ui: UIProvider): Promise<string> {
+    while (true) {
+        const name = await ui.input(message);
+        const error = validateContractName(name);
+        if (error !== undefined) {
+            ui.write(chalk.redBright(`Error: `) + error);
+            // ask user again
+            continue;
+        }
+        return name;
+    }
+}

src/cli/rename.ts

@@ -5,10 +5,11 @@ import arg from 'arg';
 
 import { COMPILABLES_DIR, CONTRACTS_DIR, SCRIPTS_DIR, TACT_ROOT_CONFIG, TESTS_DIR, WRAPPERS_DIR } from '../paths';
 import { Args, extractFirstArg, extractSecondArg, Runner, RunnerContext } from './Runner';
-import { assertValidContractName, findContracts, toLowerCase, toSnakeCase, extractFile } from '../utils';
+import { validateContractName, findContracts, toLowerCase, toSnakeCase, extractFile } from '../utils';
 import { UIProvider } from '../ui/UIProvider';
 import { helpArgs, helpMessages } from './constants';
 import { selectContract } from './build';
+import { requestContractName } from './create';
 
 export function renameExactIfRequired(
     str: string,
@@ -85,8 +86,16 @@ export const rename: Runner = async (_args: Args, ui: UIProvider, _context: Runn
 
     const oldName = await selectContract(ui, extractFirstArg(localArgs));
 
-    const newName = extractSecondArg(localArgs) ?? (await ui.input('New contract name (PascalCase)'));
-    assertValidContractName(newName);
+    const argName = extractSecondArg(localArgs);
+    if (argName !== undefined) {
+        const error = validateContractName(argName);
+        if (error) {
+            throw new Error(error);
+        }
+    }
+
+    const newName = argName ?? (await requestContractName('New contract name (PascalCase)', ui));
+
     const contracts = await findContracts();
     if (contracts.includes(newName)) {
         ui.write(`Contract with name ${newName} already exists.`);

src/utils/string.utils.ts

@@ -20,14 +20,16 @@ export function toPascalCase(str: string): string {
         .replace(/\s+/g, ''); // Removes all spaces
 }
 
-export function assertValidContractName(name: string) {
-    if (name.length === 0) throw new Error(`Contract name cannot be empty.`);
+export function validateContractName(name: string): string | undefined {
+    if (name.length === 0) return `Contract name cannot be empty.`;
 
     if (name.toLowerCase() === 'contract') {
-        throw new Error(`Contract name 'contract' is reserved. Please choose a different name.`);
+        return `Contract name 'contract' is reserved. Please choose a different name.`;
     }
 
     if (!isPascalCase(name)) {
-        throw new Error(`Contract name '${name}' is not in PascalCase. Please try ${toPascalCase(name)}.`);
+        return `Contract name '${name}' is not in PascalCase. Please try ${toPascalCase(name)}.`;
     }
+
+    return undefined;
 }