eatthepie/src/commands/completeDraw.js

Command line app for interacting with Eat The Pie, the world lottery on World Chain.

import inquirer from "inquirer";
import chalk from "chalk";
import { loadConfig } from "../utils/config.js";
import { createPublicClient, createWalletClient } from "../utils/ethereum.js";
import { completeDraw } from "../services/gameService.js";

/**
 * Error messages that require special handling
 */
const ERROR_MESSAGES = {
  DRAW_NOT_INITIATED: "Draw not initiated for this game",
  ALREADY_SET: "Random has already been set",
};

/**
 * Error response messages
 */
const ERROR_RESPONSES = {
  DRAW_NOT_INITIATED: "🎲 Draw has not yet been initiated for this game.",
  ALREADY_SET: "✅ Draw has already been completed for this game.",
};

/**
 * Success messages
 */
const SUCCESS_MESSAGES = {
  DRAW_COMPLETED: "🎲 Draw completed successfully! ✨",
};

/**
 * Validation messages
 */
const VALIDATION = {
  GAME_NUMBER: "⚠️ Please enter a valid game number",
};

/**
 * Prompt messages
 */
const PROMPT_MESSAGES = {
  GAME_NUMBER: "Enter the game number to complete the draw for:",
};

/**
 * Handles the process of completing the draw for a specific game.
 */
async function completeDrawHandler() {
  try {
    console.log(chalk.cyan("\n🎲 Starting draw completion process..."));

    // Initialize clients and configuration
    const config = await loadConfig();
    const publicClient = createPublicClient(config);
    const walletClient = createWalletClient(config);

    // Get game number from user
    const gameNumber = await promptForGameNumber();

    // Process draw completion
    await processCompleteDraw(
      walletClient,
      publicClient,
      config.contractAddress,
      gameNumber
    );
  } catch (error) {
    handleCompleteDrawError(error);
  }
}

/**
 * Prompts the user for a game number
 * @returns {Promise<number>} The selected game number
 */
async function promptForGameNumber() {
  const { gameNumber } = await inquirer.prompt([
    {
      type: "number",
      name: "gameNumber",
      message: "🎮 " + PROMPT_MESSAGES.GAME_NUMBER,
      validate: (input) => input > 0 || VALIDATION.GAME_NUMBER,
    },
  ]);
  return gameNumber;
}

/**
 * Processes the draw completion transaction and displays results
 * @param {WalletClient} walletClient - The wallet client instance
 * @param {PublicClient} publicClient - The public client instance
 * @param {string} contractAddress - The lottery contract address
 * @param {number} gameNumber - The game number to complete the draw for
 */
async function processCompleteDraw(
  walletClient,
  publicClient,
  contractAddress,
  gameNumber
) {
  console.log(chalk.yellow("\n🎯 Completing draw..."));

  const txHash = await completeDraw(
    walletClient,
    publicClient,
    contractAddress,
    gameNumber
  );

  displaySuccessMessages(txHash);

  // Wait for transaction confirmation
  await waitForTransactionConfirmation(publicClient, txHash);
}

/**
 * Waits for a transaction to be confirmed and displays the confirmation
 * @param {PublicClient} publicClient - The public client instance
 * @param {string} txHash - The transaction hash to wait for
 */
async function waitForTransactionConfirmation(publicClient, txHash) {
  console.log(chalk.yellow("\n⏳ Waiting for transaction to be confirmed..."));

  const receipt = await publicClient.waitForTransactionReceipt({
    hash: txHash,
    confirmations: 1,
  });

  console.log(chalk.cyan("📦 Block Number:"), receipt.blockNumber);
  console.log(chalk.green("\n✅ Transaction confirmed successfully!"));
  console.log(chalk.cyan("🎲 Draw has been completed!"));
}

/**
 * Displays success messages after completing the draw
 * @param {string} txHash - The transaction hash
 */
function displaySuccessMessages(txHash) {
  console.log(chalk.yellow("\n📝 Transaction Hash:"), txHash);
  console.log(chalk.green(SUCCESS_MESSAGES.DRAW_COMPLETED));
}

/**
 * Handles errors that occur during the draw completion process
 * @param {Error} error - The error to handle
 */
function handleCompleteDrawError(error) {
  // Map of error messages to their corresponding responses
  const errorHandlers = {
    [ERROR_MESSAGES.DRAW_NOT_INITIATED]: () =>
      console.log(chalk.yellow(ERROR_RESPONSES.DRAW_NOT_INITIATED)),
    [ERROR_MESSAGES.ALREADY_SET]: () =>
      console.log(chalk.yellow(ERROR_RESPONSES.ALREADY_SET)),
  };

  // Check if error matches any known error types
  for (const [errorMessage, handler] of Object.entries(errorHandlers)) {
    if (error.message.includes(errorMessage)) {
      handler();
      return;
    }
  }

  console.error(chalk.red("\n❌ Error:"), error.shortMessage || error.message);
  console.error(
    chalk.red(
      "\n⚠️ Make sure your settings are correct.\n🔧 Run 'config' to view them and 'setup' to reset them."
    )
  );
  process.exit(1);
}

export default {
  command: "complete-draw",
  describe: "🎲 Complete the draw",
  handler: completeDrawHandler,
};