wrangler 0.0.0-e4ca780f → 0.0.0-e4ef867c

package/src/kv/index.ts

@@ -1,594 +0,0 @@
-import { StringDecoder } from "node:string_decoder";
-import { readConfig } from "../config";
-import { confirm } from "../dialogs";
-import {
-	demandOneOfOption,
-	printWranglerBanner,
-	CommandLineArgsError,
-} from "../index";
-import { logger } from "../logger";
-import * as metrics from "../metrics";
-import { parseJSON, readFileSync, readFileSyncToBuffer } from "../parse";
-import { requireAuth } from "../user";
-import {
-	createKVNamespace,
-	deleteKVNamespace,
-	getKVNamespaceId,
-	isValidKVNamespaceBinding,
-	listKVNamespaces,
-	deleteKVBulkKeyValue,
-	deleteKVKeyValue,
-	getKVKeyValue,
-	isKVKeyValue,
-	listKVNamespaceKeys,
-	putKVBulkKeyValue,
-	putKVKeyValue,
-	unexpectedKVKeyValueProps,
-} from "./helpers";
-import type { CommonYargsArgv } from "../yargs-types";
-import type { KeyValue } from "./helpers";
-
-export function kvNamespace(kvYargs: CommonYargsArgv) {
-	return kvYargs
-		.command(
-			"create <namespace>",
-			"Create a new namespace",
-			(yargs) => {
-				return yargs
-					.positional("namespace", {
-						describe: "The name of the new namespace",
-						type: "string",
-						demandOption: true,
-					})
-					.option("preview", {
-						type: "boolean",
-						describe: "Interact with a preview namespace",
-					});
-			},
-			async (args) => {
-				await printWranglerBanner();
-
-				if (!isValidKVNamespaceBinding(args.namespace)) {
-					throw new CommandLineArgsError(
-						`The namespace binding name "${args.namespace}" is invalid. It can only have alphanumeric and _ characters, and cannot begin with a number.`
-					);
-				}
-
-				const config = readConfig(args.config, args);
-				if (!config.name) {
-					logger.warn(
-						"No configured name present, using `worker` as a prefix for the title"
-					);
-				}
-
-				const name = config.name || "worker";
-				const environment = args.env ? `-${args.env}` : "";
-				const preview = args.preview ? "_preview" : "";
-				const title = `${name}${environment}-${args.namespace}${preview}`;
-
-				const accountId = await requireAuth(config);
-
-				// TODO: generate a binding name stripping non alphanumeric chars
-
-				logger.log(`🌀 Creating namespace with title "${title}"`);
-				const namespaceId = await createKVNamespace(accountId, title);
-				await metrics.sendMetricsEvent("create kv namespace", {
-					sendMetrics: config.send_metrics,
-				});
-
-				logger.log("✨ Success!");
-				const envString = args.env ? ` under [env.${args.env}]` : "";
-				const previewString = args.preview ? "preview_" : "";
-				logger.log(
-					`Add the following to your configuration file in your kv_namespaces array${envString}:`
-				);
-				logger.log(
-					`{ binding = "${args.namespace}", ${previewString}id = "${namespaceId}" }`
-				);
-
-				// TODO: automatically write this block to the wrangler.toml config file??
-			}
-		)
-		.command(
-			"list",
-			"Outputs a list of all KV namespaces associated with your account id.",
-			(listArgs) => listArgs,
-			async (args) => {
-				const config = readConfig(args.config, args);
-
-				const accountId = await requireAuth(config);
-
-				// TODO: we should show bindings if they exist for given ids
-
-				logger.log(
-					JSON.stringify(await listKVNamespaces(accountId), null, "  ")
-				);
-				await metrics.sendMetricsEvent("list kv namespaces", {
-					sendMetrics: config.send_metrics,
-				});
-			}
-		)
-		.command(
-			"delete",
-			"Deletes a given namespace.",
-			(yargs) => {
-				return yargs
-					.option("binding", {
-						type: "string",
-						requiresArg: true,
-						describe: "The name of the namespace to delete",
-					})
-					.option("namespace-id", {
-						type: "string",
-						requiresArg: true,
-						describe: "The id of the namespace to delete",
-					})
-					.check(demandOneOfOption("binding", "namespace-id"))
-					.option("preview", {
-						type: "boolean",
-						describe: "Interact with a preview namespace",
-					});
-			},
-			async (args) => {
-				await printWranglerBanner();
-				const config = readConfig(args.config, args);
-
-				let id;
-				try {
-					id = getKVNamespaceId(args, config);
-				} catch (e) {
-					throw new CommandLineArgsError(
-						"Not able to delete namespace.\n" + ((e as Error).message ?? e)
-					);
-				}
-
-				const accountId = await requireAuth(config);
-
-				logger.log(`Deleting KV namespace ${id}.`);
-				await deleteKVNamespace(accountId, id);
-				logger.log(`Deleted KV namespace ${id}.`);
-				await metrics.sendMetricsEvent("delete kv namespace", {
-					sendMetrics: config.send_metrics,
-				});
-
-				// TODO: recommend they remove it from wrangler.toml
-
-				// test-mf wrangler kv:namespace delete --namespace-id 2a7d3d8b23fc4159b5afa489d6cfd388
-				// Are you sure you want to delete namespace 2a7d3d8b23fc4159b5afa489d6cfd388? [y/n]
-				// n
-				// 💁  Not deleting namespace 2a7d3d8b23fc4159b5afa489d6cfd388
-				// ➜  test-mf wrangler kv:namespace delete --namespace-id 2a7d3d8b23fc4159b5afa489d6cfd388
-				// Are you sure you want to delete namespace 2a7d3d8b23fc4159b5afa489d6cfd388? [y/n]
-				// y
-				// 🌀  Deleting namespace 2a7d3d8b23fc4159b5afa489d6cfd388
-				// ✨  Success
-				// ⚠️  Make sure to remove this "kv-namespace" entry from your configuration file!
-				// ➜  test-mf
-
-				// TODO: do it automatically
-
-				// TODO: delete the preview namespace as well?
-			}
-		);
-}
-
-export const kvKey = (kvYargs: CommonYargsArgv) => {
-	return kvYargs
-		.command(
-			"put <key> [value]",
-			"Writes a single key/value pair to the given namespace.",
-			(yargs) => {
-				return yargs
-					.positional("key", {
-						type: "string",
-						describe: "The key to write to",
-						demandOption: true,
-					})
-					.positional("value", {
-						type: "string",
-						describe: "The value to write",
-					})
-					.option("binding", {
-						type: "string",
-						requiresArg: true,
-						describe: "The binding of the namespace to write to",
-					})
-					.option("namespace-id", {
-						type: "string",
-						requiresArg: true,
-						describe: "The id of the namespace to write to",
-					})
-					.check(demandOneOfOption("binding", "namespace-id"))
-					.option("preview", {
-						type: "boolean",
-						describe: "Interact with a preview namespace",
-					})
-					.option("ttl", {
-						type: "number",
-						describe: "Time for which the entries should be visible",
-					})
-					.option("expiration", {
-						type: "number",
-						describe: "Time since the UNIX epoch after which the entry expires",
-					})
-					.option("metadata", {
-						type: "string",
-						describe: "Arbitrary JSON that is associated with a key",
-						coerce: (jsonStr: string): KeyValue["metadata"] => {
-							try {
-								return JSON.parse(jsonStr);
-							} catch (_) {}
-						},
-					})
-					.option("path", {
-						type: "string",
-						requiresArg: true,
-						describe: "Read value from the file at a given path",
-					})
-					.check(demandOneOfOption("value", "path"));
-			},
-			async ({ key, ttl, expiration, metadata, ...args }) => {
-				await printWranglerBanner();
-				const config = readConfig(args.config, args);
-				const namespaceId = getKVNamespaceId(args, config);
-				// One of `args.path` and `args.value` must be defined
-				const value = args.path
-					? readFileSyncToBuffer(args.path)
-					: // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
-					  args.value!;
-
-				const metadataLog = metadata
-					? ` with metadata "${JSON.stringify(metadata)}"`
-					: "";
-
-				if (args.path) {
-					logger.log(
-						`Writing the contents of ${args.path} to the key "${key}" on namespace ${namespaceId}${metadataLog}.`
-					);
-				} else {
-					logger.log(
-						`Writing the value "${value}" to key "${key}" on namespace ${namespaceId}${metadataLog}.`
-					);
-				}
-
-				const accountId = await requireAuth(config);
-
-				await putKVKeyValue(accountId, namespaceId, {
-					key,
-					value,
-					expiration,
-					expiration_ttl: ttl,
-					metadata: metadata as KeyValue["metadata"],
-				});
-				await metrics.sendMetricsEvent("write kv key-value", {
-					sendMetrics: config.send_metrics,
-				});
-			}
-		)
-		.command(
-			"list",
-			"Outputs a list of all keys in a given namespace.",
-			(yargs) => {
-				return yargs
-					.option("binding", {
-						type: "string",
-						requiresArg: true,
-						describe: "The name of the namespace to list",
-					})
-					.option("namespace-id", {
-						type: "string",
-						requiresArg: true,
-						describe: "The id of the namespace to list",
-					})
-					.check(demandOneOfOption("binding", "namespace-id"))
-					.option("preview", {
-						type: "boolean",
-						// In the case of listing keys we will default to non-preview mode
-						default: false,
-						describe: "Interact with a preview namespace",
-					})
-					.option("prefix", {
-						type: "string",
-						requiresArg: true,
-						describe: "A prefix to filter listed keys",
-					});
-			},
-			async ({ prefix, ...args }) => {
-				// TODO: support for limit+cursor (pagination)
-				const config = readConfig(args.config, args);
-				const namespaceId = getKVNamespaceId(args, config);
-
-				const accountId = await requireAuth(config);
-
-				const results = await listKVNamespaceKeys(
-					accountId,
-					namespaceId,
-					prefix
-				);
-				logger.log(JSON.stringify(results, undefined, 2));
-				await metrics.sendMetricsEvent("list kv keys", {
-					sendMetrics: config.send_metrics,
-				});
-			}
-		)
-		.command(
-			"get <key>",
-			"Reads a single value by key from the given namespace.",
-			(yargs) => {
-				return yargs
-					.positional("key", {
-						describe: "The key value to get.",
-						type: "string",
-						demandOption: true,
-					})
-					.option("binding", {
-						type: "string",
-						requiresArg: true,
-						describe: "The name of the namespace to get from",
-					})
-					.option("namespace-id", {
-						type: "string",
-						requiresArg: true,
-						describe: "The id of the namespace to get from",
-					})
-					.check(demandOneOfOption("binding", "namespace-id"))
-					.option("preview", {
-						type: "boolean",
-						describe: "Interact with a preview namespace",
-					})
-					.option("preview", {
-						type: "boolean",
-						// In the case of getting key values we will default to non-preview mode
-						default: false,
-						describe: "Interact with a preview namespace",
-					})
-					.option("text", {
-						type: "boolean",
-						default: false,
-						describe: "Decode the returned value as a utf8 string",
-					});
-			},
-			async ({ key, ...args }) => {
-				const config = readConfig(args.config, args);
-				const namespaceId = getKVNamespaceId(args, config);
-
-				const accountId = await requireAuth(config);
-				const bufferKVValue = Buffer.from(
-					await getKVKeyValue(accountId, namespaceId, key)
-				);
-
-				if (args.text) {
-					const decoder = new StringDecoder("utf8");
-					logger.log(decoder.write(bufferKVValue));
-				} else {
-					process.stdout.write(bufferKVValue);
-				}
-				await metrics.sendMetricsEvent("read kv value", {
-					sendMetrics: config.send_metrics,
-				});
-			}
-		)
-		.command(
-			"delete <key>",
-			"Removes a single key value pair from the given namespace.",
-			(yargs) => {
-				return yargs
-					.positional("key", {
-						describe: "The key value to delete",
-						type: "string",
-						demandOption: true,
-					})
-					.option("binding", {
-						type: "string",
-						requiresArg: true,
-						describe: "The name of the namespace to delete from",
-					})
-					.option("namespace-id", {
-						type: "string",
-						requiresArg: true,
-						describe: "The id of the namespace to delete from",
-					})
-					.check(demandOneOfOption("binding", "namespace-id"))
-					.option("preview", {
-						type: "boolean",
-						describe: "Interact with a preview namespace",
-					});
-			},
-			async ({ key, ...args }) => {
-				await printWranglerBanner();
-				const config = readConfig(args.config, args);
-				const namespaceId = getKVNamespaceId(args, config);
-
-				logger.log(`Deleting the key "${key}" on namespace ${namespaceId}.`);
-
-				const accountId = await requireAuth(config);
-
-				await deleteKVKeyValue(accountId, namespaceId, key);
-				await metrics.sendMetricsEvent("delete kv key-value", {
-					sendMetrics: config.send_metrics,
-				});
-			}
-		);
-};
-
-export const kvBulk = (kvYargs: CommonYargsArgv) => {
-	return kvYargs
-		.command(
-			"put <filename>",
-			"Upload multiple key-value pairs to a namespace",
-			(yargs) => {
-				return yargs
-					.positional("filename", {
-						describe: `The JSON file of key-value pairs to upload, in form [{"key":..., "value":...}"...]`,
-						type: "string",
-						demandOption: true,
-					})
-					.option("binding", {
-						type: "string",
-						requiresArg: true,
-						describe: "The name of the namespace to insert values into",
-					})
-					.option("namespace-id", {
-						type: "string",
-						requiresArg: true,
-						describe: "The id of the namespace to insert values into",
-					})
-					.check(demandOneOfOption("binding", "namespace-id"))
-					.option("preview", {
-						type: "boolean",
-						describe: "Interact with a preview namespace",
-					});
-			},
-			async ({ filename, ...args }) => {
-				await printWranglerBanner();
-				// The simplest implementation I could think of.
-				// This could be made more efficient with a streaming parser/uploader
-				// but we'll do that in the future if needed.
-
-				const config = readConfig(args.config, args);
-				const namespaceId = getKVNamespaceId(args, config);
-				const content = parseJSON(readFileSync(filename), filename);
-
-				if (!Array.isArray(content)) {
-					throw new Error(
-						`Unexpected JSON input from "${filename}".\n` +
-							`Expected an array of key-value objects but got type "${typeof content}".`
-					);
-				}
-
-				const errors: string[] = [];
-				const warnings: string[] = [];
-				for (let i = 0; i < content.length; i++) {
-					const keyValue = content[i];
-					if (!isKVKeyValue(keyValue)) {
-						errors.push(
-							`The item at index ${i} is ${JSON.stringify(keyValue)}`
-						);
-					} else {
-						const props = unexpectedKVKeyValueProps(keyValue);
-						if (props.length > 0) {
-							warnings.push(
-								`The item at index ${i} contains unexpected properties: ${JSON.stringify(
-									props
-								)}.`
-							);
-						}
-					}
-				}
-				if (warnings.length > 0) {
-					logger.warn(
-						`Unexpected key-value properties in "${filename}".\n` +
-							warnings.join("\n")
-					);
-				}
-				if (errors.length > 0) {
-					throw new Error(
-						`Unexpected JSON input from "${filename}".\n` +
-							`Each item in the array should be an object that matches:\n\n` +
-							`interface KeyValue {\n` +
-							`  key: string;\n` +
-							`  value: string;\n` +
-							`  expiration?: number;\n` +
-							`  expiration_ttl?: number;\n` +
-							`  metadata?: object;\n` +
-							`  base64?: boolean;\n` +
-							`}\n\n` +
-							errors.join("\n")
-					);
-				}
-
-				const accountId = await requireAuth(config);
-				await putKVBulkKeyValue(accountId, namespaceId, content);
-				await metrics.sendMetricsEvent("write kv key-values (bulk)", {
-					sendMetrics: config.send_metrics,
-				});
-
-				logger.log("Success!");
-			}
-		)
-		.command(
-			"delete <filename>",
-			"Delete multiple key-value pairs from a namespace",
-			(yargs) => {
-				return yargs
-					.positional("filename", {
-						describe: `The JSON file of keys to delete, in the form ["key1", "key2", ...]`,
-						type: "string",
-						demandOption: true,
-					})
-					.option("binding", {
-						type: "string",
-						requiresArg: true,
-						describe: "The name of the namespace to delete from",
-					})
-					.option("namespace-id", {
-						type: "string",
-						requiresArg: true,
-						describe: "The id of the namespace to delete from",
-					})
-					.check(demandOneOfOption("binding", "namespace-id"))
-					.option("preview", {
-						type: "boolean",
-						describe: "Interact with a preview namespace",
-					})
-					.option("force", {
-						type: "boolean",
-						alias: "f",
-						describe: "Do not ask for confirmation before deleting",
-					});
-			},
-			async ({ filename, ...args }) => {
-				await printWranglerBanner();
-				const config = readConfig(args.config, args);
-				const namespaceId = getKVNamespaceId(args, config);
-
-				if (!args.force) {
-					const result = await confirm(
-						`Are you sure you want to delete all the keys read from "${filename}" from kv-namespace with id "${namespaceId}"?`
-					);
-					if (!result) {
-						logger.log(`Not deleting keys read from "${filename}".`);
-						return;
-					}
-				}
-
-				const content = parseJSON(readFileSync(filename), filename) as string[];
-
-				if (!Array.isArray(content)) {
-					throw new Error(
-						`Unexpected JSON input from "${filename}".\n` +
-							`Expected an array of strings but got:\n${content}`
-					);
-				}
-
-				const errors: string[] = [];
-				for (let i = 0; i < content.length; i++) {
-					const key = content[i];
-					if (typeof key !== "string") {
-						errors.push(
-							`The item at index ${i} is type: "${typeof key}" - ${JSON.stringify(
-								key
-							)}`
-						);
-					}
-				}
-
-				if (errors.length > 0) {
-					throw new Error(
-						`Unexpected JSON input from "${filename}".\n` +
-							`Expected an array of strings.\n` +
-							errors.join("\n")
-					);
-				}
-
-				const accountId = await requireAuth(config);
-
-				await deleteKVBulkKeyValue(accountId, namespaceId, content);
-				await metrics.sendMetricsEvent("delete kv key-values (bulk)", {
-					sendMetrics: config.send_metrics,
-				});
-
-				logger.log("Success!");
-			}
-		);
-};

package/src/logger.ts

@@ -1,119 +0,0 @@
-import { format } from "node:util";
-import chalk from "chalk";
-import CLITable from "cli-table3";
-import { formatMessagesSync } from "esbuild";
-import { getEnvironmentVariableFactory } from "./environment-variables/factory";
-import type { BuildFailure } from "esbuild";
-export const LOGGER_LEVELS = {
-	none: -1,
-	error: 0,
-	warn: 1,
-	info: 2,
-	log: 3,
-	debug: 4,
-} as const;
-
-export type LoggerLevel = keyof typeof LOGGER_LEVELS;
-
-/** A map from LOGGER_LEVEL to the error `kind` needed by `formatMessagesSync()`. */
-const LOGGER_LEVEL_FORMAT_TYPE_MAP = {
-	error: "error",
-	warn: "warning",
-	info: undefined,
-	log: undefined,
-	debug: undefined,
-} as const;
-
-const getLogLevelFromEnv = getEnvironmentVariableFactory({
-	variableName: "WRANGLER_LOG",
-});
-
-function getLoggerLevel(): LoggerLevel {
-	const fromEnv = getLogLevelFromEnv()?.toLowerCase();
-	if (fromEnv !== undefined) {
-		if (fromEnv in LOGGER_LEVELS) return fromEnv as LoggerLevel;
-		const expected = Object.keys(LOGGER_LEVELS)
-			.map((level) => `"${level}"`)
-			.join(" | ");
-		console.warn(
-			`Unrecognised WRANGLER_LOG value ${JSON.stringify(
-				fromEnv
-			)}, expected ${expected}, defaulting to "log"...`
-		);
-	}
-	return "log";
-}
-
-export type TableRow<Keys extends string> = Record<Keys, string>;
-
-export class Logger {
-	constructor() {}
-
-	loggerLevel = getLoggerLevel();
-	columns = process.stdout.columns;
-
-	debug = (...args: unknown[]) => this.doLog("debug", args);
-	info = (...args: unknown[]) => this.doLog("info", args);
-	log = (...args: unknown[]) => this.doLog("log", args);
-	warn = (...args: unknown[]) => this.doLog("warn", args);
-	error = (...args: unknown[]) => this.doLog("error", args);
-	table<Keys extends string>(data: TableRow<Keys>[]) {
-		const keys: Keys[] =
-			data.length === 0 ? [] : (Object.keys(data[0]) as Keys[]);
-		const t = new CLITable({
-			head: keys.map((k) => chalk.bold.blue(k)),
-		});
-		t.push(...data.map((row) => keys.map((k) => row[k])));
-		return this.doLog("log", [t.toString()]);
-	}
-
-	private doLog(messageLevel: Exclude<LoggerLevel, "none">, args: unknown[]) {
-		if (LOGGER_LEVELS[this.loggerLevel] >= LOGGER_LEVELS[messageLevel]) {
-			console[messageLevel](this.formatMessage(messageLevel, format(...args)));
-		}
-	}
-
-	private formatMessage(
-		level: Exclude<LoggerLevel, "none">,
-		message: string
-	): string {
-		const kind = LOGGER_LEVEL_FORMAT_TYPE_MAP[level];
-		if (kind) {
-			// Format the message using the esbuild formatter.
-			// The first line of the message is the main `text`,
-			// subsequent lines are put into the `notes`.
-			const [firstLine, ...otherLines] = message.split("\n");
-			const notes =
-				otherLines.length > 0
-					? otherLines.map((text) => ({ text }))
-					: undefined;
-			return formatMessagesSync([{ text: firstLine, notes }], {
-				color: true,
-				kind,
-				terminalWidth: this.columns,
-			})[0];
-		} else {
-			return message;
-		}
-	}
-}
-
-/**
- * A drop-in replacement for `console` for outputting logging messages.
- *
- * Errors and Warnings will get additional formatting to highlight them to the user.
- * You can also set a `logger.loggerLevel` value to one of "debug", "log", "warn" or "error",
- * to filter out logging messages.
- */
-export const logger = new Logger();
-
-/**
- * Logs all errors/warnings associated with an esbuild BuildFailure in the same
- * style esbuild would.
- */
-export function logBuildFailure(failure: BuildFailure) {
-	let logs = formatMessagesSync(failure.errors, { kind: "error", color: true });
-	for (const log of logs) console.error(log);
-	logs = formatMessagesSync(failure.warnings, { kind: "warning", color: true });
-	for (const log of logs) console.warn(log);
-}