spiracha 1.4.0 → 2.0.0

package/src/lib/kiro-exporter-types.ts

@@ -0,0 +1,100 @@
+import os from 'node:os';
+import path from 'node:path';
+import type { ExportFormat, JsonValue } from './shared';
+
+export const getDefaultKiroDataDir = (_env: NodeJS.ProcessEnv = process.env, homeDir = os.homedir()): string => {
+    return path.join(homeDir, 'Library', 'Application Support', 'Kiro', 'User', 'globalStorage', 'kiro.kiroagent');
+};
+
+export const DEFAULT_KIRO_DATA_DIR = getDefaultKiroDataDir();
+
+export const resolveKiroDataDir = (): string => {
+    const configured =
+        process.env.SPIRACHA_KIRO_DATA_DIR?.trim() ||
+        process.env.SPIRACHA_KIRO_AGENT_DIR?.trim() ||
+        process.env.SPIRACHA_KIRO_DIR?.trim();
+    return configured ? configured : DEFAULT_KIRO_DATA_DIR;
+};
+
+export const resolveKiroWorkspaceSessionsDir = (): string => {
+    const configured = process.env.SPIRACHA_KIRO_WORKSPACE_SESSIONS_DIR?.trim();
+    return configured ? configured : path.join(resolveKiroDataDir(), 'workspace-sessions');
+};
+
+export type KiroWorkspaceGroup = {
+    assistantMessageCount: number;
+    directoryName: string;
+    imageCount: number;
+    key: string;
+    label: string;
+    lastActiveAtIso: string | null;
+    lastActiveAtMs: number | null;
+    messageCount: number;
+    promptLogCount: number;
+    sessionCount: number;
+    uri: string;
+    userMessageCount: number;
+    worktree: string;
+};
+
+export type KiroSessionSummary = {
+    assistantMessageCount: number;
+    autonomyMode: string | null;
+    createdAtIso: string | null;
+    createdAtMs: number | null;
+    defaultModelTitle: string | null;
+    filePath: string;
+    imageCount: number;
+    lastActiveAtIso: string | null;
+    lastActiveAtMs: number | null;
+    messageCount: number;
+    promptLogCount: number;
+    renderablePartCount: number;
+    selectedModel: string | null;
+    selectedProfileId: string | null;
+    sessionId: string;
+    sessionType: string | null;
+    title: string;
+    userMessageCount: number;
+    workspaceDirectory: string | null;
+    workspaceKey: string;
+    workspaceLabel: string;
+    workspacePath: string | null;
+    worktree: string;
+};
+
+export type KiroPartType = 'image' | 'text' | 'unknown';
+
+export type KiroTranscriptPart = {
+    imageUrl?: string | null;
+    raw: Record<string, JsonValue>;
+    text?: string;
+    type: KiroPartType;
+};
+
+export type KiroTranscriptEntryType = 'message' | 'tool_call';
+
+export type KiroTranscriptEntry = {
+    entryType: KiroTranscriptEntryType;
+    entryId: string;
+    executionId: string | null;
+    parts: KiroTranscriptPart[];
+    promptLogCount: number;
+    raw: Record<string, JsonValue>;
+    role: string;
+    timestamp: string | null;
+};
+
+export type KiroSessionTranscript = {
+    entries: KiroTranscriptEntry[];
+    rawSession: Record<string, JsonValue>;
+    renderablePartCount: number;
+    session: KiroSessionSummary;
+};
+
+export type KiroExportOptions = {
+    includeCommentary: boolean;
+    includeMetadata: boolean;
+    includeTools: boolean;
+    outputFormat: ExportFormat;
+};

package/src/lib/kiro-transcript-phase.ts

@@ -0,0 +1,41 @@
+import type { KiroTranscriptEntry } from './kiro-exporter-types';
+
+export type KiroMessagePhase = 'commentary' | 'final_answer' | null;
+
+export const getFinalKiroAssistantMessageEntryIds = (entries: KiroTranscriptEntry[]): Set<string> => {
+    const finalEntryIds = new Set<string>();
+    let latestAssistantMessageEntryId: string | null = null;
+
+    const flushAssistantRun = () => {
+        if (latestAssistantMessageEntryId) {
+            finalEntryIds.add(latestAssistantMessageEntryId);
+            latestAssistantMessageEntryId = null;
+        }
+    };
+
+    for (const entry of entries) {
+        if (entry.entryType === 'tool_call') {
+            continue;
+        }
+
+        if (entry.role === 'user') {
+            flushAssistantRun();
+            continue;
+        }
+
+        if (entry.role === 'assistant') {
+            latestAssistantMessageEntryId = entry.entryId;
+        }
+    }
+
+    flushAssistantRun();
+    return finalEntryIds;
+};
+
+export const getKiroMessagePhase = (entry: KiroTranscriptEntry, finalAssistantMessageEntryIds: Set<string>) => {
+    if (entry.role !== 'assistant') {
+        return null;
+    }
+
+    return finalAssistantMessageEntryIds.has(entry.entryId) ? 'final_answer' : 'commentary';
+};

package/src/lib/kiro-transcript.ts

@@ -0,0 +1,115 @@
+import type {
+    KiroExportOptions,
+    KiroSessionSummary,
+    KiroSessionTranscript,
+    KiroTranscriptEntry,
+    KiroTranscriptPart,
+} from './kiro-exporter-types';
+import { getFinalKiroAssistantMessageEntryIds, getKiroMessagePhase } from './kiro-transcript-phase';
+import {
+    cleanExtractedText,
+    cleanInlineTitle,
+    type MetadataEntry,
+    renderDocumentTitle,
+    renderMetadataBlock,
+    renderSection,
+} from './shared';
+
+const getSessionTitle = (session: KiroSessionSummary): string => {
+    return cleanInlineTitle(session.title || session.sessionId);
+};
+
+const buildMetadataEntries = (session: KiroSessionSummary): MetadataEntry[] => [
+    { key: 'exported_from', value: 'kiro_workspace_sessions' },
+    { key: 'session_id', value: session.sessionId },
+    { key: 'title', value: session.title },
+    { key: 'source_session_path', value: session.filePath },
+    { key: 'workspace_key', value: session.workspaceKey },
+    { key: 'workspace_directory', value: session.workspaceDirectory },
+    { key: 'workspace_path', value: session.workspacePath },
+    { key: 'selected_model', value: session.selectedModel },
+    { key: 'default_model_title', value: session.defaultModelTitle },
+    { key: 'selected_profile_id', value: session.selectedProfileId },
+    { key: 'autonomy_mode', value: session.autonomyMode },
+    { key: 'session_type', value: session.sessionType },
+    { key: 'created_at_iso', value: session.createdAtIso },
+    { key: 'last_active_at_iso', value: session.lastActiveAtIso },
+    { key: 'message_count', value: session.messageCount },
+    { key: 'image_count', value: session.imageCount },
+    { key: 'prompt_log_count', value: session.promptLogCount },
+];
+
+const roleTitle = (role: string): string => {
+    if (role === 'assistant') {
+        return 'Assistant';
+    }
+
+    if (role === 'user') {
+        return 'User';
+    }
+
+    if (role === 'system') {
+        return 'System';
+    }
+
+    return role ? cleanInlineTitle(role) : 'Message';
+};
+
+const renderTextPart = (part: KiroTranscriptPart, title: string, options: KiroExportOptions): string => {
+    const text = cleanExtractedText(part.text ?? '').trim();
+    return text ? renderSection(title, text, options.outputFormat) : '';
+};
+
+const renderImagePart = (part: KiroTranscriptPart, options: KiroExportOptions): string => {
+    if (!options.includeCommentary) {
+        return '';
+    }
+
+    const text = cleanExtractedText(part.text ?? 'Image attachment').trim();
+    return renderSection('Attachment', text, options.outputFormat);
+};
+
+const renderPart = (
+    entry: KiroTranscriptEntry,
+    part: KiroTranscriptPart,
+    options: KiroExportOptions,
+    finalAssistantMessageEntryIds: Set<string>,
+): string => {
+    if (entry.entryType === 'tool_call') {
+        return options.includeTools && part.type === 'text' ? renderTextPart(part, 'Tool call', options) : '';
+    }
+
+    if (getKiroMessagePhase(entry, finalAssistantMessageEntryIds) === 'commentary' && !options.includeCommentary) {
+        return '';
+    }
+
+    switch (part.type) {
+        case 'text':
+            return renderTextPart(part, roleTitle(entry.role), options);
+        case 'image':
+            return renderImagePart(part, options);
+        case 'unknown':
+            return '';
+    }
+};
+
+export const renderKiroTranscript = (transcript: KiroSessionTranscript, options: KiroExportOptions): string | null => {
+    const finalAssistantMessageEntryIds = getFinalKiroAssistantMessageEntryIds(transcript.entries);
+    const sections = transcript.entries.flatMap((entry) =>
+        entry.parts.map((part) => renderPart(entry, part, options, finalAssistantMessageEntryIds)).filter(Boolean),
+    );
+    if (sections.length === 0) {
+        return null;
+    }
+
+    const parts = [
+        renderDocumentTitle(getSessionTitle(transcript.session), options.outputFormat),
+        '',
+        options.includeMetadata
+            ? renderMetadataBlock(buildMetadataEntries(transcript.session), options.outputFormat)
+            : '',
+        ...sections,
+    ].filter(Boolean);
+
+    return `${parts.join('\n').trimEnd()}\n`;
+};