feat: datasets create / key-value-stores create

Author: vladfrangu

src/commands/datasets/create.ts

@@ -0,0 +1,48 @@
+import { Args } from '@oclif/core';
+import chalk from 'chalk';
+
+import { ApifyCommand } from '../../lib/apify_command.js';
+import { tryToGetDataset } from '../../lib/commands/storages.js';
+import { error, success } from '../../lib/outputs.js';
+import { getLoggedClientOrThrow } from '../../lib/utils.js';
+
+export class DatasetsCreateCommand extends ApifyCommand<typeof DatasetsCreateCommand> {
+	static override description = 'Creates a new Dataset on your account';
+
+	static override args = {
+		datasetName: Args.string({
+			description: 'Optional name for the Dataset',
+			required: false,
+		}),
+	};
+
+	static override enableJsonFlag = true;
+
+	async run() {
+		const { datasetName } = this.args;
+
+		const client = await getLoggedClientOrThrow();
+
+		if (datasetName) {
+			const existing = await tryToGetDataset(client, datasetName);
+
+			if (existing) {
+				error({ message: 'Cannot create a Dataset with the same name!' });
+				return;
+			}
+		}
+
+		const newDataset = await client.datasets().getOrCreate(datasetName);
+
+		if (this.flags.json) {
+			return newDataset;
+		}
+
+		success({
+			message: `Dataset with ID ${chalk.yellow(newDataset.id)}${datasetName ? ` (called ${chalk.yellow(datasetName)})` : ''} was created.`,
+			stdout: true,
+		});
+
+		return undefined;
+	}
+}

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

@@ -0,0 +1,50 @@
+import { Args } from '@oclif/core';
+import chalk from 'chalk';
+
+import { ApifyCommand } from '../../lib/apify_command.js';
+import { tryToGetKeyValueStore } from '../../lib/commands/storages.js';
+import { error, success } from '../../lib/outputs.js';
+import { getLoggedClientOrThrow } from '../../lib/utils.js';
+
+export class KeyValueStoresCreateCommand extends ApifyCommand<typeof KeyValueStoresCreateCommand> {
+	static override description = 'Creates a new Key-value Store on your account';
+
+	static override hiddenAliases = ['kvs:create'];
+
+	static override args = {
+		keyValueStoreName: Args.string({
+			description: 'Optional name for the Key-value Store',
+			required: false,
+		}),
+	};
+
+	static override enableJsonFlag = true;
+
+	async run() {
+		const { keyValueStoreName } = this.args;
+
+		const client = await getLoggedClientOrThrow();
+
+		if (keyValueStoreName) {
+			const existing = await tryToGetKeyValueStore(client, keyValueStoreName);
+
+			if (existing) {
+				error({ message: 'Cannot create a Key-value Store with the same name!' });
+				return;
+			}
+		}
+
+		const newStore = await client.keyValueStores().getOrCreate(keyValueStoreName);
+
+		if (this.flags.json) {
+			return newStore;
+		}
+
+		success({
+			message: `Key-value Store with ID ${chalk.yellow(newStore.id)}${keyValueStoreName ? ` (called ${chalk.yellow(keyValueStoreName)})` : ''} was created.`,
+			stdout: true,
+		});
+
+		return undefined;
+	}
+}

src/lib/commands/storages.ts

@@ -0,0 +1,57 @@
+import type { ApifyClient, Dataset, DatasetClient, KeyValueStore, KeyValueStoreClient } from 'apify-client';
+
+import { getLocalUserInfo } from '../utils.js';
+
+type ReturnTypeForStorage<T extends 'dataset' | 'keyValueStore'> = T extends 'dataset'
+	? {
+			dataset: Dataset;
+			datasetClient: DatasetClient;
+		}
+	: { keyValueStore: KeyValueStore; keyValueStoreClient: KeyValueStoreClient };
+
+async function tryToGetStorage<T extends 'dataset' | 'keyValueStore'>(
+	client: ApifyClient,
+	id: string,
+	storageType: T,
+): Promise<ReturnTypeForStorage<T> | null> {
+	const byIdOrName = await client
+		.dataset(id)
+		.get()
+		.catch(() => undefined);
+
+	if (byIdOrName) {
+		return {
+			[storageType]: byIdOrName,
+			[`${storageType}Client`]: client[storageType](byIdOrName.id),
+		} as ReturnTypeForStorage<T>;
+	}
+
+	const info = await getLocalUserInfo();
+
+	const byName = await client[storageType](`${info.username!}/${id}`)
+		.get()
+		.catch(() => undefined);
+
+	if (byName) {
+		return {
+			[storageType]: byName,
+			[`${storageType}Client`]: client[storageType](byName.id),
+		} as ReturnTypeForStorage<T>;
+	}
+
+	return null;
+}
+
+export async function tryToGetDataset(
+	client: ApifyClient,
+	datasetId: string,
+): Promise<{ dataset: Dataset | undefined; datasetClient: DatasetClient } | null> {
+	return tryToGetStorage(client, datasetId, 'dataset');
+}
+
+export async function tryToGetKeyValueStore(
+	client: ApifyClient,
+	keyValueStoreId: string,
+): Promise<{ keyValueStore: KeyValueStore | undefined; keyValueStoreClient: KeyValueStoreClient } | null> {
+	return tryToGetStorage(client, keyValueStoreId, 'keyValueStore');
+}