feat: select included credit policy

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Author: asizikov

src/pipeline/includedCreditsPolicy.test.ts

@@ -0,0 +1,71 @@
+import { describe, expect, it } from 'vitest'
+
+import type { ReportFormatMetadata } from './reportAdapters'
+import {
+  NATIVE_AI_CREDITS_STANDARD_INCLUDED_CREDITS_POLICY,
+  NATIVE_AI_CREDITS_SUMMER_PROMO_INCLUDED_CREDITS_POLICY,
+  TRANSITION_PERIOD_INCLUDED_CREDITS_POLICY,
+  resolveIncludedCreditsPolicy,
+} from './includedCreditsPolicy'
+
+const TRANSITION_PERIOD_REPORT_METADATA = {
+  format: 'transition-period-billing-preview',
+  label: 'Transition Period Billing Preview report',
+  supported: true,
+} satisfies ReportFormatMetadata
+
+const NATIVE_AI_CREDITS_REPORT_METADATA = {
+  format: 'native-ai-credits',
+  label: 'Native AI Credits report',
+  supported: false,
+} satisfies ReportFormatMetadata
+
+describe('resolveIncludedCreditsPolicy', () => {
+  it('returns the transition-period billing preview policy for transition reports', () => {
+    expect(resolveIncludedCreditsPolicy(TRANSITION_PERIOD_REPORT_METADATA)).toBe(
+      TRANSITION_PERIOD_INCLUDED_CREDITS_POLICY,
+    )
+    expect(resolveIncludedCreditsPolicy('transition-period-billing-preview', {
+      startDate: '2026-09-01',
+      endDate: '2026-09-30',
+    })).toBe(TRANSITION_PERIOD_INCLUDED_CREDITS_POLICY)
+  })
+
+  it('returns the native AI Credits summer promo policy for native report periods before September 2026', () => {
+    expect(resolveIncludedCreditsPolicy(NATIVE_AI_CREDITS_REPORT_METADATA, {
+      startDate: '2026-06-01',
+      endDate: '2026-06-30',
+    })).toBe(NATIVE_AI_CREDITS_SUMMER_PROMO_INCLUDED_CREDITS_POLICY)
+  })
+
+  it('uses period start to keep native report periods starting on 2026-08-31 in the summer promo policy', () => {
+    expect(resolveIncludedCreditsPolicy(NATIVE_AI_CREDITS_REPORT_METADATA, {
+      startDate: '2026-08-31',
+      endDate: '2026-09-30',
+    })).toBe(NATIVE_AI_CREDITS_SUMMER_PROMO_INCLUDED_CREDITS_POLICY)
+  })
+
+  it('returns the native AI Credits standard policy for native report periods starting on 2026-09-01 onward', () => {
+    expect(resolveIncludedCreditsPolicy(NATIVE_AI_CREDITS_REPORT_METADATA, {
+      startDate: '2026-09-01',
+      endDate: '2026-09-30',
+    })).toBe(NATIVE_AI_CREDITS_STANDARD_INCLUDED_CREDITS_POLICY)
+    expect(resolveIncludedCreditsPolicy(NATIVE_AI_CREDITS_REPORT_METADATA, {
+      startDate: '2026-10-01',
+      endDate: '2026-10-31',
+    })).toBe(NATIVE_AI_CREDITS_STANDARD_INCLUDED_CREDITS_POLICY)
+  })
+
+  it('defaults native AI Credits reports without a valid period start to the standard policy', () => {
+    expect(resolveIncludedCreditsPolicy(NATIVE_AI_CREDITS_REPORT_METADATA)).toBe(
+      NATIVE_AI_CREDITS_STANDARD_INCLUDED_CREDITS_POLICY,
+    )
+    expect(resolveIncludedCreditsPolicy(NATIVE_AI_CREDITS_REPORT_METADATA, {
+      endDate: '2026-08-31',
+    })).toBe(NATIVE_AI_CREDITS_STANDARD_INCLUDED_CREDITS_POLICY)
+    expect(resolveIncludedCreditsPolicy(NATIVE_AI_CREDITS_REPORT_METADATA, {
+      startDate: '8/31/26',
+      endDate: '2026-08-31',
+    })).toBe(NATIVE_AI_CREDITS_STANDARD_INCLUDED_CREDITS_POLICY)
+  })
+})

src/pipeline/includedCreditsPolicy.ts

@@ -1,3 +1,5 @@
+import type { ReportFormat, ReportFormatMetadata } from './reportAdapters'
+
 export type QuotaUnit = 'pru' | 'aic'
 export type OrganizationIncludedCreditTier = 'business' | 'enterprise'
 
@@ -27,8 +29,15 @@ export type IncludedCreditsPolicy = {
   readonly organizationPlans: OrganizationIncludedCreditPlans
 }
 
+export type ReportPeriod = {
+  readonly startDate?: string | null
+  readonly endDate?: string | null
+}
+
 const COPILOT_BUSINESS_LABEL = 'Copilot Business'
 const COPILOT_ENTERPRISE_LABEL = 'Copilot Enterprise'
+const NATIVE_AI_CREDITS_STANDARD_POLICY_START_DATE = '2026-09-01'
+const ISO_DATE_PATTERN = /^(\d{4})-(\d{2})-(\d{2})$/
 
 export const TRANSITION_PERIOD_INCLUDED_CREDITS_POLICY = {
   id: 'transition-period-billing-preview',
@@ -101,3 +110,46 @@ export const NATIVE_AI_CREDITS_STANDARD_INCLUDED_CREDITS_POLICY = {
     },
   },
 } as const satisfies IncludedCreditsPolicy
+
+function getReportFormat(reportMetadataOrFormat: ReportFormat | ReportFormatMetadata): ReportFormat {
+  return typeof reportMetadataOrFormat === 'string'
+    ? reportMetadataOrFormat
+    : reportMetadataOrFormat.format
+}
+
+function isValidIsoDate(value: string): boolean {
+  const match = ISO_DATE_PATTERN.exec(value)
+  if (!match) return false
+
+  const year = Number(match[1])
+  const month = Number(match[2])
+  const day = Number(match[3])
+  const date = new Date(Date.UTC(year, month - 1, day))
+
+  return (
+    date.getUTCFullYear() === year
+    && date.getUTCMonth() === month - 1
+    && date.getUTCDate() === day
+  )
+}
+
+function isBeforeIsoDate(value: string | null | undefined, boundary: string): boolean {
+  if (!value || !isValidIsoDate(value)) return false
+
+  return value < boundary
+}
+
+export function resolveIncludedCreditsPolicy(
+  reportMetadataOrFormat: ReportFormat | ReportFormatMetadata,
+  reportPeriod: ReportPeriod = {},
+): IncludedCreditsPolicy {
+  if (getReportFormat(reportMetadataOrFormat) === 'transition-period-billing-preview') {
+    return TRANSITION_PERIOD_INCLUDED_CREDITS_POLICY
+  }
+
+  if (isBeforeIsoDate(reportPeriod.startDate, NATIVE_AI_CREDITS_STANDARD_POLICY_START_DATE)) {
+    return NATIVE_AI_CREDITS_SUMMER_PROMO_INCLUDED_CREDITS_POLICY
+  }
+
+  return NATIVE_AI_CREDITS_STANDARD_INCLUDED_CREDITS_POLICY
+}

src/pipeline/runPipeline.ts

@@ -1,5 +1,6 @@
 import type { Aggregator } from './aggregators/base'
 import { createAicIncludedCreditsAllocator, type AicIncludedCreditsOverrides } from './aicIncludedCredits'
+import { resolveIncludedCreditsPolicy } from './includedCreditsPolicy'
 import {
   InvalidReportError,
   parseTokenUsageHeader,
@@ -92,6 +93,7 @@ export async function runPipeline(
   const { includedCreditsOverrides = {}, progressResolution = 500, onProgress } = options ?? {}
   const reportAdapter = await validateFileFormat(file)
   const reportMetadata = reportAdapter.metadata
+  const includedCreditsPolicy = resolveIncludedCreditsPolicy(reportMetadata)
   let lastProgressStage: PipelineProgress['stage'] | null = null
   let lastProgressPercent = -1
   let lastProgressTimestamp = 0
@@ -138,6 +140,7 @@ export async function runPipeline(
   }
 
   const aicIncludedCreditAllocator = await createAicIncludedCreditsAllocator(file, includedCreditsOverrides, {
+    includedCreditsPolicy,
     onProgress: (streamProgress) => {
       emitProgress('analyzing', 0, streamProgress)
     },