All files / src/action/checks manager.ts

55.17% Statements 16/29
21.42% Branches 9/42
80% Functions 8/10
51.85% Lines 14/27

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139                                                                                    26x 23x                                                                                                     23x 26x         23x   26x   26x         26x 26x                                       23x 1x   22x 7x   15x    
/**
 * Check Manager
 *
 * Manages GitHub Check runs for Warden triggers.
 * Wraps the core github-checks module with action-specific logic.
 */
 
import type { SkillReport, UsageStats, AuxiliaryUsageMap } from '../../types/index.js';
import {
  aggregateSeverityCounts,
  determineConclusion,
} from '../../output/github-checks.js';
import { mergeAuxiliaryUsage } from '../../sdk/usage.js';
import type { TriggerResult } from '../triggers/executor.js';
 
// Re-export types and functions that are used directly
export {
  createCoreCheck,
  updateCoreCheck,
  createSkillCheck,
  updateSkillCheck,
  failSkillCheck,
  aggregateSeverityCounts,
  determineConclusion,
} from '../../output/github-checks.js';
 
export type {
  CheckOptions,
  UpdateSkillCheckOptions,
  CreateCheckResult,
  CoreCheckSummaryData,
  CheckConclusion,
} from '../../output/github-checks.js';
 
// -----------------------------------------------------------------------------
// Aggregate Functions
// -----------------------------------------------------------------------------
 
/**
 * Aggregate usage stats from multiple reports.
 */
export function aggregateUsage(reports: SkillReport[]): UsageStats | undefined {
  const reportsWithUsage = reports.filter((r) => r.usage);
  Eif (reportsWithUsage.length === 0) return undefined;
 
  const seed: UsageStats = {
    inputTokens: 0,
    outputTokens: 0,
    cacheReadInputTokens: 0,
    cacheCreationInputTokens: 0,
    cacheCreation5mInputTokens: 0,
    cacheCreation1hInputTokens: 0,
    webSearchRequests: 0,
    costUSD: 0,
  };
 
  return reportsWithUsage.reduce((acc, r) => {
    acc.inputTokens += r.usage?.inputTokens ?? 0;
    acc.outputTokens += r.usage?.outputTokens ?? 0;
    acc.cacheReadInputTokens = (acc.cacheReadInputTokens ?? 0) + (r.usage?.cacheReadInputTokens ?? 0);
    acc.cacheCreationInputTokens = (acc.cacheCreationInputTokens ?? 0) + (r.usage?.cacheCreationInputTokens ?? 0);
    acc.cacheCreation5mInputTokens = (acc.cacheCreation5mInputTokens ?? 0) + (r.usage?.cacheCreation5mInputTokens ?? 0);
    acc.cacheCreation1hInputTokens = (acc.cacheCreation1hInputTokens ?? 0) + (r.usage?.cacheCreation1hInputTokens ?? 0);
    acc.webSearchRequests = (acc.webSearchRequests ?? 0) + (r.usage?.webSearchRequests ?? 0);
    acc.costUSD += r.usage?.costUSD ?? 0;
    return acc;
  }, seed);
}
 
/**
 * Build core check summary data from trigger results.
 */
export function buildCoreSummaryData(
  results: TriggerResult[],
  reports: SkillReport[]
): {
  totalSkills: number;
  totalFindings: number;
  findingsBySeverity: Record<string, number>;
  totalDurationMs?: number;
  totalUsage?: UsageStats;
  totalAuxiliaryUsage?: AuxiliaryUsageMap;
  findings: SkillReport['findings'];
  skillResults: {
    name: string;
    findingCount: number;
    conclusion: 'success' | 'failure' | 'neutral' | 'cancelled';
    durationMs?: number;
    usage?: UsageStats;
    auxiliaryUsage?: AuxiliaryUsageMap;
  }[];
} {
  // Aggregate auxiliary usage across all reports
  let totalAuxiliaryUsage: AuxiliaryUsageMap | undefined;
  for (const r of reports) {
    Iif (r.auxiliaryUsage) {
      totalAuxiliaryUsage = mergeAuxiliaryUsage(totalAuxiliaryUsage, r.auxiliaryUsage);
    }
  }
 
  return {
    totalSkills: results.length,
    totalFindings: reports.reduce((sum, r) => sum + r.findings.length, 0),
    findingsBySeverity: aggregateSeverityCounts(reports),
    totalDurationMs: reports.some((r) => r.durationMs !== undefined)
      ? reports.reduce((sum, r) => sum + (r.durationMs ?? 0), 0)
      : undefined,
    totalUsage: aggregateUsage(reports),
    totalAuxiliaryUsage,
    findings: reports.flatMap((r) => r.findings),
    skillResults: results.map((r) => ({
      name: r.triggerName,
      findingCount: r.report?.findings.length ?? 0,
      conclusion: r.report
        ? determineConclusion(r.report.findings, r.failOn, r.failCheck)
        : ('failure' as const),
      durationMs: r.report?.durationMs,
      usage: r.report?.usage,
      auxiliaryUsage: r.report?.auxiliaryUsage,
    })),
  };
}
 
/**
 * Determine overall core check conclusion.
 */
export function determineCoreConclusion(
  shouldFailAction: boolean,
  totalFindings: number
): 'success' | 'failure' | 'neutral' {
  if (shouldFailAction) {
    return 'failure';
  }
  if (totalFindings > 0) {
    return 'neutral';
  }
  return 'success';
}