@agentuity/cli 0.1.15 → 0.1.17

package/src/cmd/cloud/queue/dlq.ts

@@ -0,0 +1,203 @@
+import { z } from 'zod';
+import { createCommand, createSubcommand } from '../../../types';
+import * as tui from '../../../tui';
+import { createQueueAPIClient, getQueueApiOptions } from './util';
+import { getCommand } from '../../../command-prefix';
+import {
+	listDeadLetterMessages,
+	replayDeadLetterMessage,
+	purgeDeadLetter,
+	type DeadLetterMessage,
+	MessageSchema,
+} from '@agentuity/server';
+import { ErrorCode } from '../../../errors';
+
+const DlqListResponseSchema = z.object({
+	messages: z.array(
+		z.object({
+			id: z.string(),
+			offset: z.number(),
+			failure_reason: z.string().nullable(),
+			delivery_attempts: z.number(),
+			moved_at: z.string(),
+		})
+	),
+	total: z.number().optional(),
+});
+
+const listDlqSubcommand = createSubcommand({
+	name: 'list',
+	aliases: ['ls'],
+	description: 'List messages in the dead letter queue',
+	tags: ['read-only', 'fast', 'requires-auth'],
+	requires: { auth: true },
+	examples: [
+		{
+			command: getCommand('cloud queue dlq list my-queue'),
+			description: 'List DLQ messages',
+		},
+	],
+	schema: {
+		args: z.object({
+			queue_name: z.string().min(1).describe('Queue name'),
+		}),
+		options: z.object({
+			limit: z.coerce.number().optional().describe('Maximum number of messages to return'),
+			offset: z.coerce.number().optional().describe('Offset for pagination'),
+		}),
+		response: DlqListResponseSchema,
+	},
+
+	async handler(ctx) {
+		const { args, opts, options } = ctx;
+		const client = await createQueueAPIClient(ctx);
+		const result = await listDeadLetterMessages(
+			client,
+			args.queue_name,
+			{
+				limit: opts.limit,
+				offset: opts.offset,
+			},
+			getQueueApiOptions(ctx)
+		);
+
+		if (!options.json) {
+			if (result.messages.length === 0) {
+				tui.info('No messages in dead letter queue');
+			} else {
+				const tableData = result.messages.map((m: DeadLetterMessage) => ({
+					ID: m.id.substring(0, 8) + '...',
+					Offset: m.offset,
+					Reason: m.failure_reason?.substring(0, 30) || 'Unknown',
+					Attempts: m.delivery_attempts,
+					'Moved At': new Date(m.moved_at).toLocaleString(),
+				}));
+				tui.table(tableData, ['ID', 'Offset', 'Reason', 'Attempts', 'Moved At']);
+			}
+		}
+
+		return {
+			messages: result.messages.map((m: DeadLetterMessage) => ({
+				id: m.id,
+				offset: m.offset,
+				failure_reason: m.failure_reason ?? null,
+				delivery_attempts: m.delivery_attempts,
+				moved_at: m.moved_at,
+			})),
+			total: result.total,
+		};
+	},
+});
+
+const ReplayResponseSchema = z.object({
+	success: z.boolean(),
+	message: MessageSchema,
+});
+
+const replayDlqSubcommand = createSubcommand({
+	name: 'replay',
+	description: 'Replay a message from the dead letter queue',
+	tags: ['mutating', 'updates-resource', 'requires-auth'],
+	requires: { auth: true },
+	examples: [
+		{
+			command: getCommand('cloud queue dlq replay my-queue msg-123'),
+			description: 'Replay a DLQ message',
+		},
+	],
+	schema: {
+		args: z.object({
+			queue_name: z.string().min(1).describe('Queue name'),
+			message_id: z.string().min(1).describe('Message ID'),
+		}),
+		response: ReplayResponseSchema,
+	},
+
+	async handler(ctx) {
+		const { args, options } = ctx;
+		const client = await createQueueAPIClient(ctx);
+		const message = await replayDeadLetterMessage(
+			client,
+			args.queue_name,
+			args.message_id,
+			getQueueApiOptions(ctx)
+		);
+
+		if (!options.json) {
+			tui.success(`Replayed message: ${message.id}`);
+			tui.info(`  New offset: ${message.offset}`);
+		}
+
+		return {
+			success: true,
+			message,
+		};
+	},
+});
+
+const PurgeResponseSchema = z.object({
+	success: z.boolean(),
+	queue_name: z.string(),
+});
+
+const purgeDlqSubcommand = createSubcommand({
+	name: 'purge',
+	description: 'Purge all messages from the dead letter queue',
+	tags: ['mutating', 'deletes-resource', 'requires-auth'],
+	requires: { auth: true },
+	examples: [
+		{
+			command: getCommand('cloud queue dlq purge my-queue --confirm'),
+			description: 'Purge all DLQ messages',
+		},
+	],
+	schema: {
+		args: z.object({
+			queue_name: z.string().min(1).describe('Queue name'),
+		}),
+		options: z.object({
+			confirm: z.boolean().default(false).describe('Skip confirmation prompt'),
+		}),
+		response: PurgeResponseSchema,
+	},
+
+	async handler(ctx) {
+		const { args, opts, options } = ctx;
+
+		if (!opts.confirm) {
+			tui.fatal('Use --confirm to confirm DLQ purge', ErrorCode.INVALID_ARGUMENT);
+		}
+
+		const client = await createQueueAPIClient(ctx);
+		await purgeDeadLetter(client, args.queue_name, getQueueApiOptions(ctx));
+
+		if (!options.json) {
+			tui.success(`Purged dead letter queue for: ${args.queue_name}`);
+		}
+
+		return {
+			success: true,
+			queue_name: args.queue_name,
+		};
+	},
+});
+
+export const dlqSubcommand = createCommand({
+	name: 'dlq',
+	description: 'Dead letter queue operations',
+	tags: ['requires-auth'],
+	requires: { auth: true },
+	examples: [
+		{
+			command: getCommand('cloud queue dlq list my-queue'),
+			description: 'List DLQ messages',
+		},
+		{
+			command: getCommand('cloud queue dlq replay my-queue msg-123'),
+			description: 'Replay a message',
+		},
+	],
+	subcommands: [listDlqSubcommand, replayDlqSubcommand, purgeDlqSubcommand],
+});
+
+export default dlqSubcommand;

package/src/cmd/cloud/queue/get.ts

@@ -0,0 +1,158 @@
+import { z } from 'zod';
+import { createCommand } from '../../../types';
+import * as tui from '../../../tui';
+import { createQueueAPIClient, getQueueApiOptions } from './util';
+import { getCommand } from '../../../command-prefix';
+import { getQueue, getMessage, QueueSchema, MessageSchema, type Message } from '@agentuity/server';
+
+const GetResponseSchema = z.union([
+	z.object({ type: z.literal('queue'), queue: QueueSchema }),
+	z.object({ type: z.literal('message'), message: MessageSchema }),
+]);
+
+function displayMessage(message: Message): void {
+	const details: Record<string, unknown> = {
+		ID: message.id,
+		'Queue ID': message.queue_id,
+		Offset: message.offset,
+		State: message.state,
+		'Delivery Attempts': message.delivery_attempts,
+	};
+
+	if (message.partition_key) {
+		details['Partition Key'] = message.partition_key;
+	}
+	if (message.idempotency_key) {
+		details['Idempotency Key'] = message.idempotency_key;
+	}
+
+	details.Published = message.published_at ? new Date(message.published_at).toLocaleString() : '-';
+	details.Created = message.created_at ? new Date(message.created_at).toLocaleString() : '-';
+
+	if (message.expires_at) {
+		details.Expires = new Date(message.expires_at).toLocaleString();
+	}
+	if (message.delivered_at) {
+		details.Delivered = new Date(message.delivered_at).toLocaleString();
+	}
+	if (message.acknowledged_at) {
+		details.Acknowledged = new Date(message.acknowledged_at).toLocaleString();
+	}
+
+	tui.table([details], undefined, { layout: 'vertical' });
+
+	tui.newline();
+	tui.info('Payload');
+	tui.json(message.payload);
+
+	if (message.metadata && Object.keys(message.metadata).length > 0) {
+		tui.newline();
+		tui.info('Metadata');
+		tui.json(message.metadata);
+	}
+}
+
+export const getSubcommand = createCommand({
+	name: 'get',
+	description: 'Get queue or message details',
+	tags: ['read-only', 'fast', 'requires-auth'],
+	requires: { auth: true },
+	examples: [
+		{ command: getCommand('cloud queue get my-queue'), description: 'Get queue details' },
+		{
+			command: getCommand('cloud queue get my-queue msg_abc123'),
+			description: 'Get message details',
+		},
+	],
+	schema: {
+		args: z.object({
+			queue_name: z.string().min(1).describe('Queue name'),
+			message_id: z.string().optional().describe('Message ID (msg_...) to get message details'),
+		}),
+		response: GetResponseSchema,
+	},
+	idempotent: true,
+
+	async handler(ctx) {
+		const { args, options } = ctx;
+		const client = await createQueueAPIClient(ctx);
+		const apiOptions = getQueueApiOptions(ctx);
+
+		if (args.message_id) {
+			const message = await getMessage(client, args.queue_name, args.message_id, apiOptions);
+
+			if (!options.json) {
+				displayMessage(message);
+			}
+
+			return { type: 'message' as const, message };
+		}
+
+		const queue = await getQueue(client, args.queue_name, apiOptions);
+
+		if (!options.json) {
+			const details: Record<string, unknown> = {
+				Name: queue.name,
+				ID: queue.id,
+				Type: queue.queue_type,
+			};
+
+			if (queue.description) {
+				details.Description = queue.description;
+			}
+
+			if (queue.paused_at) {
+				details.Status = 'Paused';
+				details['Paused At'] = new Date(queue.paused_at).toLocaleString();
+			} else {
+				details.Status = 'Active';
+			}
+
+			details.Created = new Date(queue.created_at).toLocaleString();
+			details.Updated = new Date(queue.updated_at).toLocaleString();
+
+			tui.table([details], undefined, { layout: 'vertical' });
+
+			if (
+				queue.message_count !== undefined ||
+				queue.dlq_count !== undefined ||
+				queue.next_offset !== undefined
+			) {
+				tui.newline();
+				tui.info('Stats');
+				tui.table(
+					[
+						{
+							Messages: queue.message_count ?? 0,
+							'DLQ Count': queue.dlq_count ?? 0,
+							'Next Offset': queue.next_offset ?? 0,
+						},
+					],
+					undefined,
+					{ layout: 'vertical' }
+				);
+			}
+
+			if (queue.default_visibility_timeout_seconds !== undefined) {
+				tui.newline();
+				tui.info('Settings');
+				const settings: Record<string, unknown> = {};
+				if (queue.default_ttl_seconds != null) {
+					settings['Default TTL'] = `${queue.default_ttl_seconds}s`;
+				}
+				settings['Visibility Timeout'] = `${queue.default_visibility_timeout_seconds}s`;
+				settings['Max Retries'] = queue.default_max_retries;
+				settings['Retry Backoff'] = `${queue.default_retry_backoff_ms}ms`;
+				settings['Max Backoff'] = `${queue.default_retry_max_backoff_ms}ms`;
+				settings['Backoff Multiplier'] = `${queue.default_retry_multiplier}x`;
+				settings['Max In-Flight'] = queue.max_in_flight_per_client;
+
+				tui.table([settings], undefined, { layout: 'vertical' });
+			}
+		}
+
+		return { type: 'queue' as const, queue };
+	},
+});
+
+export default getSubcommand;

package/src/cmd/cloud/queue/index.ts

@@ -0,0 +1,66 @@
+import { createCommand } from '../../../types';
+import { listSubcommand } from './list';
+import { createSubcommand } from './create';
+import { getSubcommand } from './get';
+import { deleteSubcommand } from './delete';
+import { publishSubcommand } from './publish';
+import { messagesSubcommand } from './messages';
+import { receiveSubcommand } from './receive';
+import { ackSubcommand } from './ack';
+import { nackSubcommand } from './nack';
+import { dlqSubcommand } from './dlq';
+import { destinationsSubcommand } from './destinations';
+import { sourcesSubcommand } from './sources';
+import { pauseSubcommand } from './pause';
+import { resumeSubcommand } from './resume';
+import { statsSubcommand } from './stats';
+import { getCommand } from '../../../command-prefix';
+
+export const command = createCommand({
+	name: 'queue',
+	aliases: ['queues'],
+	description: 'Manage message queues',
+	tags: ['requires-auth'],
+	examples: [
+		{
+			command: getCommand('cloud queue list'),
+			description: 'List all queues',
+		},
+		{
+			command: getCommand('cloud queue create worker --name my-tasks'),
+			description: 'Create a worker queue',
+		},
+		{
+			command: getCommand('cloud queue publish my-queue \'{"task":"process"}\''),
+			description: 'Publish a message',
+		},
+		{
+			command: getCommand('cloud queue receive my-queue'),
+			description: 'Receive a message from a worker queue',
+		},
+		{
+			command: getCommand('cloud queue stats'),
+			description: 'View analytics for all queues',
+		},
+	],
+	subcommands: [
+		listSubcommand,
+		createSubcommand,
+		getSubcommand,
+		deleteSubcommand,
+		publishSubcommand,
+		messagesSubcommand,
+		receiveSubcommand,
+		ackSubcommand,
+		nackSubcommand,
+		dlqSubcommand,
+		destinationsSubcommand,
+		sourcesSubcommand,
+		pauseSubcommand,
+		resumeSubcommand,
+		statsSubcommand,
+	],
+	requires: { auth: true },
+});
+
+export default command;

package/src/cmd/cloud/queue/list.ts

@@ -0,0 +1,81 @@
+import { z } from 'zod';
+import { createCommand } from '../../../types';
+import * as tui from '../../../tui';
+import { createQueueAPIClient, getQueueApiOptions } from './util';
+import { getCommand } from '../../../command-prefix';
+import { listQueues, type Queue } from '@agentuity/server';
+
+const QueueListResponseSchema = z.object({
+	queues: z.array(
+		z.object({
+			name: z.string(),
+			queue_type: z.string(),
+			message_count: z.number(),
+			dlq_count: z.number(),
+			created_at: z.string(),
+		})
+	),
+	total: z.number().optional(),
+});
+
+export const listSubcommand = createCommand({
+	name: 'list',
+	aliases: ['ls'],
+	description: 'List all queues',
+	tags: ['read-only', 'fast', 'requires-auth'],
+	requires: { auth: true },
+	examples: [
+		{ command: getCommand('cloud queue list'), description: 'List all queues' },
+		{ command: getCommand('cloud queue ls'), description: 'List all queues (alias)' },
+	],
+	schema: {
+		args: z.object({}),
+		options: z.object({
+			limit: z.coerce.number().optional().describe('Maximum number of queues to return'),
+			offset: z.coerce.number().optional().describe('Offset for pagination'),
+		}),
+		response: QueueListResponseSchema,
+	},
+	idempotent: true,
+
+	async handler(ctx) {
+		const { options, opts } = ctx;
+		const client = await createQueueAPIClient(ctx);
+		const result = await listQueues(
+			client,
+			{
+				limit: opts.limit,
+				offset: opts.offset,
+			},
+			getQueueApiOptions(ctx)
+		);
+
+		if (!options.json) {
+			if (result.queues.length === 0) {
+				tui.info('No queues found');
+			} else {
+				const tableData = result.queues.map((q: Queue) => ({
+					Name: q.name,
+					Type: q.queue_type,
+					Messages: q.message_count ?? 0,
+					DLQ: q.dlq_count ?? 0,
+					Created: new Date(q.created_at).toLocaleString(),
+				}));
+				tui.table(tableData, ['Name', 'Type', 'Messages', 'DLQ', 'Created']);
+			}
+		}
+
+		return {
+			queues: result.queues.map((q: Queue) => ({
+				name: q.name,
+				queue_type: q.queue_type,
+				message_count: q.message_count ?? 0,
+				dlq_count: q.dlq_count ?? 0,
+				created_at: q.created_at,
+			})),
+			total: result.total,
+		};
+	},
+});
+
+export default listSubcommand;

package/src/cmd/cloud/queue/messages.ts

@@ -0,0 +1,160 @@
+import { z } from 'zod';
+import { createCommand } from '../../../types';
+import * as tui from '../../../tui';
+import { createQueueAPIClient, getQueueApiOptions } from './util';
+import { getCommand } from '../../../command-prefix';
+import { listMessages, getMessage, MessageSchema, type Message } from '@agentuity/server';
+
+const MessagesListResponseSchema = z.object({
+	messages: z.array(
+		z.object({
+			id: z.string(),
+			offset: z.number(),
+			state: z.string().optional(),
+			size: z.number().optional(),
+			created_at: z.string().optional(),
+		})
+	),
+	total: z.number().optional(),
+});
+
+const MessagesResponseSchema = z.union([
+	z.object({ type: z.literal('list'), data: MessagesListResponseSchema }),
+	z.object({ type: z.literal('message'), message: MessageSchema }),
+]);
+
+function displayMessage(message: Message): void {
+	const details: Record<string, unknown> = {
+		ID: message.id,
+		'Queue ID': message.queue_id,
+		Offset: message.offset,
+		State: message.state,
+		'Delivery Attempts': message.delivery_attempts,
+	};
+
+	if (message.partition_key) {
+		details['Partition Key'] = message.partition_key;
+	}
+	if (message.idempotency_key) {
+		details['Idempotency Key'] = message.idempotency_key;
+	}
+
+	details.Published = message.published_at ? new Date(message.published_at).toLocaleString() : '-';
+	details.Created = message.created_at ? new Date(message.created_at).toLocaleString() : '-';
+
+	if (message.expires_at) {
+		details.Expires = new Date(message.expires_at).toLocaleString();
+	}
+	if (message.delivered_at) {
+		details.Delivered = new Date(message.delivered_at).toLocaleString();
+	}
+	if (message.acknowledged_at) {
+		details.Acknowledged = new Date(message.acknowledged_at).toLocaleString();
+	}
+
+	tui.table([details], undefined, { layout: 'vertical' });
+
+	tui.newline();
+	tui.info('Payload');
+	tui.json(message.payload);
+
+	if (message.metadata && Object.keys(message.metadata).length > 0) {
+		tui.newline();
+		tui.info('Metadata');
+		tui.json(message.metadata);
+	}
+}
+
+export const messagesSubcommand = createCommand({
+	name: 'messages',
+	aliases: ['msgs'],
+	description: 'List messages in a queue or get a specific message',
+	tags: ['read-only', 'fast', 'requires-auth'],
+	requires: { auth: true },
+	examples: [
+		{
+			command: getCommand('cloud queue messages my-queue'),
+			description: 'List messages in a queue',
+		},
+		{
+			command: getCommand('cloud queue messages my-queue --limit 10'),
+			description: 'List first 10 messages',
+		},
+		{
+			command: getCommand('cloud queue messages my-queue msg_abc123'),
+			description: 'Get a specific message by ID',
+		},
+	],
+	schema: {
+		args: z.object({
+			queue_name: z.string().min(1).describe('Queue name'),
+			message_id: z
+				.string()
+				.optional()
+				.describe('Message ID (optional, to get a specific message)'),
+		}),
+		options: z.object({
+			limit: z.number().optional().describe('Maximum number of messages to return'),
+			offset: z.number().optional().describe('Offset for pagination'),
+		}),
+		response: MessagesResponseSchema,
+	},
+	idempotent: true,
+
+	async handler(ctx) {
+		const { args, opts, options } = ctx;
+		const client = await createQueueAPIClient(ctx);
+		const apiOptions = getQueueApiOptions(ctx);
+
+		if (args.message_id) {
+			const message = await getMessage(client, args.queue_name, args.message_id, apiOptions);
+
+			if (!options.json) {
+				displayMessage(message);
+			}
+
+			return { type: 'message' as const, message };
+		}
+
+		const result = await listMessages(
+			client,
+			args.queue_name,
+			{
+				limit: opts.limit,
+				offset: opts.offset,
+			},
+			apiOptions
+		);
+
+		if (!options.json) {
+			if (result.messages.length === 0) {
+				tui.info('No messages found');
+			} else {
+				const tableData = result.messages.map((m: Message) => ({
+					ID: m.id,
+					Offset: m.offset,
+					State: m.state,
+					Size: m.size,
+					Created: m.created_at ? new Date(m.created_at).toLocaleString() : 'N/A',
+				}));
+				tui.table(tableData, ['ID', 'Offset', 'State', 'Size', 'Created']);
+			}
+		}
+
+		return {
+			type: 'list' as const,
+			data: {
+				messages: result.messages.map((m: Message) => ({
+					id: m.id,
+					offset: m.offset,
+					state: m.state,
+					size: m.size,
+					created_at: m.created_at,
+				})),
+				total: result.total,
+			},
+		};
+	},
+});
+
+export default messagesSubcommand;