feat: datasets rm / key-value-stores rm

Author: vladfrangu

src/commands/datasets/rm.ts

@@ -0,0 +1,60 @@
+import { Args } from '@oclif/core';
+import type { ApifyApiError } from 'apify-client';
+import chalk from 'chalk';
+
+import { ApifyCommand } from '../../lib/apify_command.js';
+import { confirmAction } from '../../lib/commands/confirm.js';
+import { tryToGetDataset } from '../../lib/commands/storages.js';
+import { error, info, success } from '../../lib/outputs.js';
+import { getLoggedClientOrThrow } from '../../lib/utils.js';
+
+export class DatasetsRmCommand extends ApifyCommand<typeof DatasetsRmCommand> {
+	static override description = 'Deletes a Dataset';
+
+	static override args = {
+		datasetNameOrId: Args.string({
+			description: 'The Dataset ID or name to delete',
+			required: true,
+		}),
+	};
+
+	async run() {
+		const { datasetNameOrId } = this.args;
+
+		const client = await getLoggedClientOrThrow();
+
+		const existingDataset = await tryToGetDataset(client, datasetNameOrId);
+
+		if (!existingDataset) {
+			error({
+				message: `Dataset with ID or name "${datasetNameOrId}" not found.`,
+			});
+
+			return;
+		}
+
+		const confirmed = await confirmAction({ type: 'Dataset' });
+
+		if (!confirmed) {
+			info({ message: 'Dataset deletion has been aborted.' });
+			return;
+		}
+
+		const { id, name } = existingDataset.dataset;
+
+		try {
+			await existingDataset.datasetClient.delete();
+
+			success({
+				message: `Dataset with ID ${chalk.yellow(id)}${name ? ` (called ${chalk.yellow(name)})` : ''} has been deleted.`,
+				stdout: true,
+			});
+		} catch (err) {
+			const casted = err as ApifyApiError;
+
+			error({
+				message: `Failed to delete Dataset with ID ${chalk.yellow(id)}\n  ${casted.message || casted}`,
+			});
+		}
+	}
+}

src/commands/key-value-stores/rm.ts

@@ -0,0 +1,62 @@
+import { Args } from '@oclif/core';
+import type { ApifyApiError } from 'apify-client';
+import chalk from 'chalk';
+
+import { ApifyCommand } from '../../lib/apify_command.js';
+import { confirmAction } from '../../lib/commands/confirm.js';
+import { tryToGetKeyValueStore } from '../../lib/commands/storages.js';
+import { error, info, success } from '../../lib/outputs.js';
+import { getLoggedClientOrThrow } from '../../lib/utils.js';
+
+export class KeyValueStoresRmCommand extends ApifyCommand<typeof KeyValueStoresRmCommand> {
+	static override description = 'Deletes a Key-value Store';
+
+	static override hiddenAliases = ['kvs:rm'];
+
+	static override args = {
+		keyValueStoreNameOrId: Args.string({
+			description: 'The Key-value Store ID or name to delete',
+			required: true,
+		}),
+	};
+
+	async run() {
+		const { keyValueStoreNameOrId } = this.args;
+
+		const client = await getLoggedClientOrThrow();
+
+		const existingKvs = await tryToGetKeyValueStore(client, keyValueStoreNameOrId);
+
+		if (!existingKvs) {
+			error({
+				message: `Key-value Store with ID or name "${keyValueStoreNameOrId}" not found.`,
+			});
+
+			return;
+		}
+
+		const confirmed = await confirmAction({ type: 'Key-value Store' });
+
+		if (!confirmed) {
+			info({ message: 'Key-value Store deletion has been aborted.' });
+			return;
+		}
+
+		const { id, name } = existingKvs.keyValueStore;
+
+		try {
+			await existingKvs.keyValueStoreClient.delete();
+
+			success({
+				message: `Key-value Store with ID ${chalk.yellow(id)}${name ? ` (called ${chalk.yellow(name)})` : ''} has been deleted.`,
+				stdout: true,
+			});
+		} catch (err) {
+			const casted = err as ApifyApiError;
+
+			error({
+				message: `Failed to delete Key-value Store with ID ${chalk.yellow(id)}\n  ${casted.message || casted}`,
+			});
+		}
+	}
+}

src/lib/commands/confirm.ts

@@ -10,8 +10,8 @@ const yesNoConfirmation = ({ type }: { type: string }): DistinctQuestion => ({
 const inputValidation = ({
 	type,
 	expectedValue,
-	failureMessage,
-}: { type: string; expectedValue: string; failureMessage: string }): DistinctQuestion => ({
+	failureMessage = 'That is not the correct input!',
+}: { type: string; expectedValue: string; failureMessage?: string }): DistinctQuestion => ({
 	name: 'confirmed',
 	type: 'input',
 	message: `Are you sure you want to delete this ${type}? If so, please type in "${expectedValue}":`,
@@ -28,7 +28,7 @@ export async function confirmAction({
 	expectedValue,
 	type,
 	failureMessage,
-}: { expectedValue?: string; type: string; failureMessage: string }): Promise<boolean> {
+}: { type: string; expectedValue?: string; failureMessage?: string }): Promise<boolean> {
 	const result = await inquirer.prompt<{ confirmed: boolean | string }>(
 		expectedValue ? inputValidation({ type, expectedValue, failureMessage }) : yesNoConfirmation({ type }),
 	);

src/lib/commands/storages.ts

@@ -45,13 +45,13 @@ async function tryToGetStorage<T extends 'dataset' | 'keyValueStore'>(
 export async function tryToGetDataset(
 	client: ApifyClient,
 	datasetId: string,
-): Promise<{ dataset: Dataset | undefined; datasetClient: DatasetClient } | null> {
+): Promise<{ dataset: Dataset; datasetClient: DatasetClient } | null> {
 	return tryToGetStorage(client, datasetId, 'dataset');
 }
 
 export async function tryToGetKeyValueStore(
 	client: ApifyClient,
 	keyValueStoreId: string,
-): Promise<{ keyValueStore: KeyValueStore | undefined; keyValueStoreClient: KeyValueStoreClient } | null> {
+): Promise<{ keyValueStore: KeyValueStore; keyValueStoreClient: KeyValueStoreClient } | null> {
 	return tryToGetStorage(client, keyValueStoreId, 'keyValueStore');
 }