standupPrompt.ts

/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the MIT License. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

import type { RefRow, SessionRow } from '../../../platform/chronicle/common/sessionStore';

/** A session row annotated with its source. */
export interface AnnotatedSession extends SessionRow {
	/** Where this session came from: 'vscode', 'cli', or 'cloud'. */
	source: 'vscode' | 'cli' | 'cloud';
}

/** A ref row annotated with its source. */
export interface AnnotatedRef extends RefRow {
	source: 'vscode' | 'cli' | 'cloud';
}

/** Sessions query — SQLite dialect, last 24 hours */
export const SESSIONS_QUERY_SQLITE = `SELECT *
	FROM sessions
	WHERE updated_at >= datetime('now', '-1 day')
	ORDER BY updated_at DESC`;

/** Build refs query for a list of session IDs */
export function buildRefsQuery(sessionIds: string[]): string {
	const ids = sessionIds.map(s => `'${s.replace(/'/g, '\'\'')}'`).join(',');
	return `SELECT session_id, ref_type, ref_value FROM session_refs WHERE session_id IN (${ids})`;
}

/** Build files query for a list of session IDs */
export function buildFilesQuery(sessionIds: string[]): string {
	const ids = sessionIds.map(s => `'${s.replace(/'/g, '\'\'')}'`).join(',');
	return `SELECT session_id, file_path, tool_name FROM session_files WHERE session_id IN (${ids})`;
}

/** Build turns query for a list of session IDs (user messages + assistant response summaries, truncated) */
export function buildTurnsQuery(sessionIds: string[]): string {
	const ids = sessionIds.map(s => `'${s.replace(/'/g, '\'\'')}'`).join(',');
	return `SELECT session_id, turn_index, substr(user_message, 1, 120) as user_message, substr(assistant_response, 1, 200) as assistant_response FROM turns WHERE session_id IN (${ids}) AND (user_message IS NOT NULL OR assistant_response IS NOT NULL) ORDER BY session_id, turn_index`;
}

/** A file row from the session_files table. */
export interface SessionFileInfo {
	session_id: string;
	file_path: string;
	tool_name?: string;
}

/** A turn summary from the turns table. */
export interface SessionTurnInfo {
	session_id: string;
	turn_index: number;
	user_message?: string;
	assistant_response?: string;
}

/**
 * Build a standup prompt from pre-fetched session and ref data.
 */
export function buildStandupPrompt(
	sessions: AnnotatedSession[],
	refs: AnnotatedRef[],
	turns: SessionTurnInfo[],
	files: SessionFileInfo[],
	extra?: string,
): string {
	if (sessions.length === 0) {
		return 'The user ran /standup but no sessions were found. Let them know there\'s no recent activity to report.';
	}

	const sessionLines = sessions.map(s => {
		const branch = s.branch ?? 'unknown';
		const repo = s.repository ?? 'unknown';
		const agent = s.agent_name ?? s.source;

		// Include turn summaries for this session (first few user messages + assistant responses)
		const sessionTurns = turns.filter(t => t.session_id === s.id).slice(0, 5);

		// Use first turn's user_message as summary when sessions.summary is empty
		const firstTurnMessage = sessionTurns[0]?.user_message;
		const summary = s.summary || firstTurnMessage || 'No summary';

		const turnLines = sessionTurns
			.filter(t => t.user_message || t.assistant_response)
			.map(t => {
				const parts: string[] = [];
				if (t.user_message) { parts.push(`User: ${t.user_message}`); }
				if (t.assistant_response) { parts.push(`Assistant: ${t.assistant_response}`); }
				return `    - ${parts.join(' → ')}`;
			});

		// Include files touched in this session (capped to avoid noise)
		const sessionFiles = files.filter(f => f.session_id === s.id);
		const uniqueFiles = [...new Set(sessionFiles.map(f => f.file_path))];
		const shownFiles = uniqueFiles.slice(0, 5);
		const fileLines = shownFiles.length > 0
			? [`    - Files (${uniqueFiles.length} total): ${shownFiles.join(', ')}${uniqueFiles.length > 5 ? `, +${uniqueFiles.length - 5} more` : ''}`]
			: [];

		return [
			`- ${s.id} | ${repo} (${branch}) | ${agent} | ${summary} | updated ${s.updated_at}`,
			...turnLines,
			...fileLines,
		].join('\n');
	});

	const refLines = refs.map(r => `- ${r.session_id} | ${r.ref_type}: ${r.ref_value}`);

	let prompt = `The user ran /chronicle standup. Generate a concise standup update from the pre-fetched data below.

## Pre-fetched Session Data (last 24 hours)

### Sessions (${sessions.length})
${sessionLines.join('\n')}

### References (PRs, Issues, Commits)
${refLines.length > 0 ? refLines.join('\n') : 'No references found.'}

## Instructions

1. Analyze the turn data (user messages and assistant responses) to understand the actual work done in each session.
2. Use file paths to identify which components, modules, or areas of the codebase were affected.
3. For any PR/issue references, mention them with links.
4. If a session has no turns or summary, note it briefly but don't skip it entirely.

## Output Format

Format the update grouped by work stream (branch/feature). Use this structure:

Standup for <date>:

**✅ Done**

**Feature name** (\`branch-name\` branch, \`repo-name\`)
  - Summary of what was accomplished (1-2 sentences grounded in the user messages and assistant responses)
  - Key files: list 2-3 most important files changed
  - Tools used: mention key tools if visible (e.g., apply_patch, run_in_terminal, search)
  - PR: [#123](link) — merged/closed (if applicable)

**🚧 In Progress**

**Feature name** (\`branch-name\` branch, \`repo-name\`)
  - Summary of current work (1-2 sentences based on turn content)
  - Key files: list 2-3 most important files being worked on
  - PR: [#789](link) — draft/open (if applicable)

Formatting rules:
- Use the turn data (user messages AND assistant responses) to understand WHAT was done, not just that something happened
- Use file paths to identify which components/areas were affected
- Group related sessions on the same branch into one entry
- Link PRs and issues using markdown link syntax
- Classify as Done if the session has no recent activity or the work appears complete, In Progress otherwise
- If a session has no branch or repo, still include it under an "Other" section`;

	if (extra) {
		prompt += `\n\nAdditional context: ${extra}`;
	}

	return prompt;
}