import type { Activity, JulesClient } from '@google/jules-sdk';
import type {
BashOutputsResult,
BashOutput,
BashOutputsSummary,
} from './types.js';
export async function getBashOutputs(
client: JulesClient,
sessionId: string,
activityIds?: string[],
): Promise<BashOutputsResult> {
if (!sessionId) {
throw new Error('sessionId is required');
}
const session = client.session(sessionId);
await session.activities.hydrate();
if (activityIds) {
const activities = await client.select({
from: 'activities',
where: { id: { in: activityIds } },
order: 'asc',
});
return aggregateOutput(activities as Activity[], sessionId);
}
const activities = await session.activities.select({
order: 'asc',
});
return aggregateOutput(activities, sessionId);
}
function aggregateOutput(activities: Activity[], sessionId: string) {
const outputs: BashOutput[] = [];
const summary: BashOutputsSummary = {
totalCommands: 0,
succeeded: 0,
failed: 0,
};
for (const activity of activities) {
collectActivityOutput(activity, outputs, summary);
}
return {
sessionId,
outputs,
summary,
};
}
function collectActivityOutput(activity: Activity, outputs: BashOutput[], summary: BashOutputsSummary) {
for (const artifact of activity.artifacts) {
if (artifact.type === 'bashOutput') {
outputs.push({
command: artifact.command,
stdout: artifact.stdout,
stderr: artifact.stderr,
exitCode: artifact.exitCode,
activityId: activity.id,
});
summary.totalCommands++;
if (artifact.exitCode === 0 || artifact.exitCode === null) {
summary.succeeded++;
} else {
summary.failed++;
}
}
}
}
