refactor telemetry command

Author: emily-shen

packages/wrangler/src/index.ts

@@ -74,6 +74,13 @@ import {
 } from "./kv";
 import { logBuildFailure, logger, LOGGER_LEVELS } from "./logger";
 import { getMetricsDispatcher } from "./metrics";
+import {
+	metricsAlias,
+	telemetryDisableCommand,
+	telemetryEnableCommand,
+	telemetryNamespace,
+	telemetryStatusCommand,
+} from "./metrics/commands";
 import { mTlsCertificateCommands } from "./mtls-certificate/cli";
 import { writeOutput } from "./output";
 import { pages } from "./pages";
@@ -932,6 +939,30 @@ export function createCLIParser(argv: string[]) {
 	]);
 	registry.registerNamespace("whoami");
 
+	registry.define([
+		{
+			command: "wrangler telemetry",
+			definition: telemetryNamespace,
+		},
+		{
+			command: "wrangler metrics",
+			definition: metricsAlias,
+		},
+		{
+			command: "wrangler telemetry disable",
+			definition: telemetryDisableCommand,
+		},
+		{
+			command: "wrangler telemetry enable",
+			definition: telemetryEnableCommand,
+		},
+		{
+			command: "wrangler telemetry status",
+			definition: telemetryStatusCommand,
+		},
+	]);
+	registry.registerNamespace("telemetry");
+
 	/******************************************************/
 	/*               DEPRECATED COMMANDS                  */
 	/******************************************************/

packages/wrangler/src/metrics/commands.ts

@@ -1,11 +1,14 @@
 import chalk from "chalk";
-import { defineAlias, defineCommand, defineNamespace } from "../core";
+import {
+	createAlias,
+	createCommand,
+	createNamespace,
+} from "../core/create-command";
 import { getWranglerSendMetricsFromEnv } from "../environment-variables/misc-variables";
 import { logger } from "../logger";
 import { readMetricsConfig, updateMetricsPermission } from "./metrics-config";
 
-defineNamespace({
-	command: "wrangler telemetry",
+export const telemetryNamespace = createNamespace({
 	metadata: {
 		description: "📈 Configure whether Wrangler collects telemetry",
 		owner: "Workers: Authoring and Testing",
@@ -14,13 +17,11 @@ defineNamespace({
 	},
 });
 
-defineAlias({
-	command: "wrangler metrics",
+export const metricsAlias = createAlias({
 	aliasOf: "wrangler telemetry",
 });
 
-defineCommand({
-	command: "wrangler telemetry disable",
+export const telemetryDisableCommand = createCommand({
 	metadata: {
 		description: "Disable Wrangler telemetry collection",
 		owner: "Workers: Authoring and Testing",
@@ -35,8 +36,7 @@ defineCommand({
 	},
 });
 
-defineCommand({
-	command: "wrangler telemetry enable",
+export const telemetryEnableCommand = createCommand({
 	metadata: {
 		description: "Enable Wrangler telemetry collection",
 		owner: "Workers: Authoring and Testing",
@@ -51,8 +51,7 @@ defineCommand({
 	},
 });
 
-defineCommand({
-	command: "wrangler telemetry status",
+export const telemetryStatusCommand = createCommand({
 	metadata: {
 		description: "Check whether Wrangler telemetry collection is enabled",
 		owner: "Workers: Authoring and Testing",