Now uses updateConfigFile to display workers/bindings instructions

Author: jonesphillip

packages/wrangler/src/pipelines/cli/create.ts

@@ -125,7 +125,7 @@ export const pipelinesCreateCommand = createCommand({
 
 			if (streamTable) {
 				const stream = await getStream(config, streamTable.id);
-				displayUsageExamples(stream, config);
+				await displayUsageExamples(stream, config, args);
 			} else {
 				logger.log(
 					`\nRun 'wrangler pipelines get ${pipeline.id}' to view full details.`

packages/wrangler/src/pipelines/cli/setup.ts

@@ -67,7 +67,7 @@ export const pipelinesSetupCommand = createCommand({
 			await setupSinkConfiguration(config, setupConfig);
 			const created = await reviewAndCreateStreamSink(config, setupConfig);
 			await setupSQLTransformationWithValidation(config, setupConfig, created);
-			await createPipelineIfNeeded(config, setupConfig, created);
+			await createPipelineIfNeeded(config, setupConfig, created, args);
 		} catch (error) {
 			if (error instanceof UserError) {
 				throw error;
@@ -778,7 +778,8 @@ async function reviewAndCreateStreamSink(
 async function createPipelineIfNeeded(
 	config: Config,
 	setupConfig: SetupConfig,
-	created: { stream?: Stream; sink?: Sink }
+	created: { stream?: Stream; sink?: Sink },
+	args: { env?: string }
 ): Promise<void> {
 	if (!setupConfig.pipelineConfig) {
 		throw new UserError("Pipeline configuration is missing");
@@ -792,7 +793,7 @@ async function createPipelineIfNeeded(
 		logger.log("\n✨ Setup complete!");
 
 		if (created.stream) {
-			displayUsageExamples(created.stream, config);
+			await displayUsageExamples(created.stream, config, args);
 		}
 	} catch (error) {
 		logger.error(

packages/wrangler/src/pipelines/cli/streams/utils.ts

@@ -1,4 +1,4 @@
-import { formatConfigSnippet } from "../../../config";
+import { updateConfigFile } from "../../../config";
 import { logger } from "../../../logger";
 import formatLabelledValues from "../../../utils/render-labelled-values";
 import type { Config } from "../../../config";
@@ -205,7 +205,11 @@ function generateExampleData(stream: Stream): string {
 	}
 }
 
-export function displayUsageExamples(stream: Stream, config: Config) {
+export async function displayUsageExamples(
+	stream: Stream,
+	config: Config,
+	args: { env?: string }
+) {
 	const bindingName = generateStreamBindingName(stream.name);
 	const exampleData = generateExampleData(stream);
 
@@ -214,17 +218,19 @@ export function displayUsageExamples(stream: Stream, config: Config) {
 	// Worker binding example (always shown since worker_binding is always enabled)
 	logger.log("\nWorker Integration:");
 
-	const configSnippet = {
-		pipelines: [
-			{
-				pipeline: stream.name,
-				binding: bindingName,
-			},
-		],
-	};
-
-	logger.log("Add to your configuration file:");
-	logger.log(formatConfigSnippet(configSnippet, config.configPath));
+	await updateConfigFile(
+		(customBindingName) => ({
+			pipelines: [
+				{
+					pipeline: stream.name,
+					binding: customBindingName ?? bindingName,
+				},
+			],
+		}),
+		config.configPath,
+		args.env,
+		false // Don't offer to update automatically
+	);
 
 	logger.log("\nIn your Worker:");
 	logger.log(`await env.${bindingName}.send([${exampleData}]);`);