import { SortDirectionSchema } from '@agentuity/client';
import { z } from 'zod';
import { APIClient, APIResponseSchema } from '@agentuity/api';
import { SessionResponseError } from './util.ts';

const SessionSchema = z.object({
	id: z.string().describe('the session id'),
	created_at: z.string().describe('the creation timestamp'),
	updated_at: z.string().describe('the last update timestamp'),
	deleted: z.boolean().describe('whether the session is deleted'),
	deleted_at: z.string().nullable().describe('the deletion timestamp'),
	deleted_by: z.string().nullable().describe('who deleted the session'),
	start_time: z.string().describe('the session start time'),
	end_time: z.string().nullable().describe('the session end time'),
	duration: z.number().nullable().describe('the session duration in nanoseconds'),
	org_id: z.string().describe('the organization id'),
	project_id: z.string().describe('the project id'),
	deployment_id: z.string().describe('the deployment id'),
	agent_ids: z.array(z.string()).describe('the agent ids involved'),
	trigger: z.string().describe('how the session was triggered'),
	env: z.string().describe('the environment'),
	devmode: z.boolean().describe('whether dev mode was enabled'),
	pending: z.boolean().describe('whether the session is pending'),
	success: z.boolean().describe('whether the session succeeded'),
	error: z.string().nullable().describe('the error message if failed'),
	metadata: z
		.record(z.string(), z.unknown())
		.nullable()
		.optional()
		.describe('unencrypted key-value metadata'),
	cpu_time: z.number().nullable().describe('the CPU time in nanoseconds'),
	llm_cost: z.number().nullable().describe('the LLM cost'),
	llm_prompt_token_count: z.number().nullable().describe('the LLM prompt token count'),
	llm_completion_token_count: z.number().nullable().describe('the LLM completion token count'),
	total_cost: z.number().nullable().describe('the total cost'),
	method: z.string().nullable().describe('the HTTP method'),
	url: z.string().nullable().describe('the request URL'),
	route_id: z.string().nullable().describe('the route id'),
	thread_id: z.string().nullable().describe('the thread id'),
	timeline: z.unknown().nullable().optional().describe('the session timeline tree'),
	user_data: z.string().nullable().optional().describe('the user data as JSON'),
});

export { SessionSchema };

export const SessionListResponse = z.array(SessionSchema);

export const SessionListResponseSchema = APIResponseSchema(SessionListResponse);

export type SessionListResponse = z.infer<typeof SessionListResponseSchema>;
export type SessionList = z.infer<typeof SessionListResponse>;
export type Session = z.infer<typeof SessionSchema>;

export type SessionSortField = 'created' | 'updated' | 'duration' | 'startTime';

export const SessionListOptionsSchema = z.object({
	count: z.number().optional().describe('Deprecated alias for limit'),
	limit: z.number().optional().describe('Maximum number of sessions to return'),
	offset: z.number().optional().describe('Pagination offset'),
	sort: z
		.enum(['created', 'updated', 'duration', 'startTime'])
		.optional()
		.describe('Field to sort sessions by'),
	direction: SortDirectionSchema.optional().describe('Sort direction for session listing'),
	orgId: z.string().optional().describe('Optional organization id filter'),
	projectId: z.string().optional().describe('Optional project id filter'),
	deploymentId: z.string().optional().describe('Optional deployment id filter'),
	trigger: z.string().optional().describe('Optional trigger filter'),
	env: z.string().optional().describe('Optional environment filter'),
	devmode: z.boolean().optional().describe('Optional devmode filter'),
	success: z.boolean().optional().describe('Optional success status filter'),
	threadId: z.string().optional().describe('Optional thread id filter'),
	agentIdentifier: z.string().optional().describe('Optional agent identifier filter'),
	startAfter: z.string().optional().describe('Filter sessions starting after this timestamp'),
	startBefore: z.string().optional().describe('Filter sessions starting before this timestamp'),
	metadata: z
		.record(z.string(), z.unknown())
		.optional()
		.describe('Optional metadata match filter serialized as JSON'),
});

export type SessionListOptions = z.infer<typeof SessionListOptionsSchema>;

/**
 * List sessions
 *
 * @param client
 * @param options filtering and pagination options
 * @returns
 */
export async function sessionList(
	client: APIClient,
	options: SessionListOptions = {}
): Promise<SessionList> {
	const {
		count,
		limit,
		offset,
		sort,
		direction,
		orgId,
		projectId,
		deploymentId,
		trigger,
		env,
		devmode,
		success,
		threadId,
		agentIdentifier,
		startAfter,
		startBefore,
		metadata,
	} = options;
	const resolvedLimit = limit ?? count ?? 10;
	const params = new URLSearchParams({ limit: resolvedLimit.toString() });
	if (orgId) params.set('orgId', orgId);
	if (projectId) params.set('projectId', projectId);
	if (deploymentId) params.set('deploymentId', deploymentId);
	if (trigger) params.set('trigger', trigger);
	if (env) params.set('env', env);
	if (offset !== undefined) params.set('offset', offset.toString());
	if (sort) params.set('sort', sort);
	if (direction) params.set('direction', direction);
	if (devmode !== undefined) params.set('devmode', devmode.toString());
	if (success !== undefined) params.set('success', success.toString());
	if (threadId) params.set('threadId', threadId);
	if (agentIdentifier) params.set('agentIdentifier', agentIdentifier);
	if (startAfter) params.set('startAfter', startAfter);
	if (startBefore) params.set('startBefore', startBefore);
	if (metadata) params.set('metadata', JSON.stringify(metadata));

	const resp = await client.request<SessionListResponse>(
		'GET',
		`/session?${params.toString()}`,
		SessionListResponseSchema
	);

	if (resp.success) {
		return resp.data;
	}

	throw new SessionResponseError({ message: resp.message });
}