UNPKG

trigger.dev

Version:

A Command-Line Interface for Trigger.dev projects

197 lines • 9.16 kB
import { Option as CommandOption } from "commander"; import { z } from "zod"; import { CommonCommandOptions, commonOptions, wrapCommandAction } from "../cli/common.js"; import { watchConfig } from "../config.js"; import { startDevSession } from "../dev/devSession.js"; import { createLockFile } from "../dev/lock.js"; import { chalkError } from "../utilities/cliOutput.js"; import { resolveLocalEnvVars } from "../utilities/localEnvVars.js"; import { printDevBanner, printStandloneInitialBanner } from "../utilities/initialBanner.js"; import { logger } from "../utilities/logger.js"; import { runtimeChecks } from "../utilities/runtimeCheck.js"; import { getProjectClient } from "../utilities/session.js"; import { login } from "./login.js"; import { updateTriggerPackages } from "./update.js"; import { readConfigHasSeenMCPInstallPrompt, writeConfigHasSeenMCPInstallPrompt, } from "../utilities/configFiles.js"; import { confirm, isCancel, log } from "@clack/prompts"; import { installMcpServer } from "./install-mcp.js"; import { tryCatch } from "@trigger.dev/core/utils"; import { VERSION } from "@trigger.dev/core"; import { initiateRulesInstallWizard } from "./install-rules.js"; const DevCommandOptions = CommonCommandOptions.extend({ debugOtel: z.boolean().default(false), config: z.string().optional(), projectRef: z.string().optional(), skipUpdateCheck: z.boolean().default(false), envFile: z.string().optional(), keepTmpFiles: z.boolean().default(false), maxConcurrentRuns: z.coerce.number().optional(), mcp: z.boolean().default(false), mcpPort: z.coerce.number().optional().default(3333), analyze: z.boolean().default(false), disableWarnings: z.boolean().default(false), skipMCPInstall: z.boolean().default(false), skipRulesInstall: z.boolean().default(false), rulesInstallManifestPath: z.string().optional(), rulesInstallBranch: z.string().optional(), }); export function configureDevCommand(program) { return commonOptions(program .command("dev") .description("Run your Trigger.dev tasks locally") .option("-c, --config <config file>", "The name of the config file") .option("-p, --project-ref <project ref>", "The project ref. Required if there is no config file.") .option("--env-file <env file>", "Path to the .env file to use for the dev session. Defaults to .env in the project directory.") .option("--max-concurrent-runs <max concurrent runs>", "The maximum number of concurrent runs to allow in the dev session") .option("--debug-otel", "Enable OpenTelemetry debugging") .option("--skip-update-check", "Skip checking for @trigger.dev package updates") .option("--keep-tmp-files", "Keep temporary files after the dev session ends, helpful for debugging") .option("--mcp", "Start the MCP server") .option("--mcp-port", "The port to run the MCP server on", "3333") .addOption(new CommandOption("--analyze", "Analyze the build output and import timings").hideHelp()) .addOption(new CommandOption("--skip-mcp-install", "Skip the Trigger.dev MCP server install wizard").hideHelp()) .addOption(new CommandOption("--skip-rules-install", "Skip the Trigger.dev Agent rules install wizard").hideHelp()) .addOption(new CommandOption("--rules-install-manifest-path <path>", "The path to the rules install manifest").hideHelp()) .addOption(new CommandOption("--rules-install-branch <branch>", "The branch to install the rules from").hideHelp()) .addOption(new CommandOption("--disable-warnings", "Suppress warnings output").hideHelp())).action(async (options) => { wrapCommandAction("dev", DevCommandOptions, options, async (opts) => { await devCommand(opts); }); }); } export async function devCommand(options) { runtimeChecks(); // Only show these install prompts if the user is in a terminal (not in a Coding Agent) if (process.stdout.isTTY) { const skipMCPInstall = typeof options.skipMCPInstall === "boolean" && options.skipMCPInstall; if (!skipMCPInstall) { const hasSeenMCPInstallPrompt = readConfigHasSeenMCPInstallPrompt(); if (!hasSeenMCPInstallPrompt) { const installChoice = await confirm({ message: "Would you like to install the Trigger.dev MCP server?", initialValue: true, }); writeConfigHasSeenMCPInstallPrompt(true); const skipInstall = isCancel(installChoice) || !installChoice; if (!skipInstall) { log.step("Welcome to the Trigger.dev MCP server install wizard 🧙"); const [installError] = await tryCatch(installMcpServer({ yolo: false, tag: VERSION, logLevel: options.logLevel, })); if (installError) { log.error(`Failed to install MCP server: ${installError.message}`); } } } } const skipRulesInstall = typeof options.skipRulesInstall === "boolean" && options.skipRulesInstall; if (!skipRulesInstall) { await tryCatch(initiateRulesInstallWizard({ manifestPath: options.rulesInstallManifestPath, branch: options.rulesInstallBranch, })); } } const authorization = await login({ embedded: true, silent: true, defaultApiUrl: options.apiUrl, profile: options.profile, }); if (!authorization.ok) { if (authorization.error === "fetch failed") { logger.log(`${chalkError("X Error:")} Connecting to the server failed. Please check your internet connection or contact eric@trigger.dev for help.`); } else { logger.log(`${chalkError("X Error:")} You must login first. Use the \`login\` CLI command.\n\n${authorization.error}`); } process.exitCode = 1; return; } let watcher; try { const devInstance = await startDev({ ...options, cwd: process.cwd(), login: authorization }); watcher = devInstance.watcher; await devInstance.waitUntilExit(); } finally { await watcher?.stop(); } } async function startDev(options) { logger.debug("Starting dev CLI", { options }); let watcher; try { if (options.logLevel) { logger.loggerLevel = options.logLevel; } await printStandloneInitialBanner(true, options.profile); let displayedUpdateMessage = false; if (!options.skipUpdateCheck) { displayedUpdateMessage = await updateTriggerPackages(options.cwd, { ...options }, true, true); } const removeLockFile = await createLockFile(options.cwd); let devInstance; printDevBanner(displayedUpdateMessage); const envVars = resolveLocalEnvVars(options.envFile); if (envVars.TRIGGER_PROJECT_REF) { logger.debug("Using project ref from env", { ref: envVars.TRIGGER_PROJECT_REF }); } watcher = await watchConfig({ cwd: options.cwd, async onUpdate(config) { logger.debug("Updated config, rerendering", { config }); if (devInstance) { devInstance.stop(); } devInstance = await bootDevSession(config); }, overrides: { project: options.projectRef ?? envVars.TRIGGER_PROJECT_REF, }, configFile: options.config, }); logger.debug("Initial config", watcher.config); // eslint-disable-next-line no-inner-declarations async function bootDevSession(configParam) { const projectClient = await getProjectClient({ accessToken: options.login.auth.accessToken, apiUrl: options.login.auth.apiUrl, projectRef: configParam.project, env: "dev", profile: options.profile, }); if (!projectClient) { process.exit(1); } return startDevSession({ name: projectClient.name, rawArgs: options, rawConfig: configParam, client: projectClient.client, initialMode: "local", dashboardUrl: options.login.dashboardUrl, showInteractiveDevSession: true, keepTmpFiles: options.keepTmpFiles, }); } devInstance = await bootDevSession(watcher.config); const waitUntilExit = async () => { }; return { watcher, stop: async () => { devInstance?.stop(); await watcher?.stop(); removeLockFile(); }, waitUntilExit, }; } catch (error) { await watcher?.stop(); throw error; } } //# sourceMappingURL=dev.js.map