Index ActionState over the types of state used

mbg · Jul 7, 2026 · Jun 16, 2026 · Jun 16, 2026 · Jun 23, 2026 · Jun 23, 2026
commit 708e10653366421552971c6d675904adccdcc2f4
diff --git a/src/action-common.ts b/src/action-common.ts
@@ -2,13 +2,36 @@ import { Env } from "./environment";
 import { FeatureEnablement } from "./feature-flags";
 import { Logger } from "./logging";
 
-export interface ActionState {
-  /** The logger that is in use. */
-  logger: Logger;
+/** Describes different state features that an Action may have. */
+export interface FeatureState {
+  Logger: {
+    /** The logger that is in use. */
+    logger: Logger;
+  };
+  Env: {
+    /** Information about environment variables. */
+    env: Env;
+  };
+  FeatureFlags: {
+    /** Information about enabled feature flags. */
+    features: FeatureEnablement;
+  };
+}
 
-  /** Information about environment variables. */
-  env: Env;
+/** Identifies a type of state an Action may have. */
+export type StateFeature = keyof FeatureState;
 
-  /** Information about enabled feature flags. */
-  features: FeatureEnablement;
-}
+/** Constructs the union of all state types identifies by `Fs`. */
+export type FieldsOf<Fs extends readonly StateFeature[]> = Fs extends [
+  infer Head extends StateFeature,
+]
+  ? FeatureState[Head]
+  : Fs extends [
+        infer Head extends StateFeature,
+        ...infer Tail extends StateFeature[],
+      ]
+    ? FeatureState[Head] & FieldsOf<Tail>
+    : never;
+
+/** Describes the state of an Action that has access to the state corresponding to `Fs`. */
+export type ActionState<Fs extends readonly StateFeature[]> = FieldsOf<Fs>;
@@ -602,7 +602,7 @@ async function downloadCacheWithTime(
 }
 
 async function loadUserConfig(
-  actionState: ActionState,
+  actionState: ActionState<["Logger", "Env", "FeatureFlags"]>,
   configFile: string,
   workspacePath: string,
   apiDetails: api.GitHubApiCombinedDetails,
@@ -1160,7 +1160,7 @@ export async function initConfig(
     logger.debug("No configuration file was provided");
   } else {
     logger.debug(`Using configuration file: ${inputs.configFile}`);
-    const actionState: ActionState = { logger, features, env: getEnv() };
+    const actionState = { logger, features, env: getEnv() };
     userConfig = await loadUserConfig(
       actionState,
       inputs.configFile,

@@ -51,7 +51,7 @@ export function getConfigFileInput(
  * @returns The `UserConfig`, if it could be fetched and parsed successfully.
  */
 export async function getRemoteConfig(
-  actionState: ActionState,
+  actionState: ActionState<["Logger", "Env", "FeatureFlags"]>,
   configFile: string,
   apiDetails: api.GitHubApiCombinedDetails,
 ): Promise<UserConfig> {

@@ -80,7 +80,7 @@ function parseOldRemoteFileAddress(
  * @throws `ConfigurationError` if the format of `configFile` is not valid.
  */
 export async function parseRemoteFileAddress(
-  actionState: ActionState,
+  actionState: ActionState<["FeatureFlags", "Env"]>,
   configFile: string,
 ): Promise<RemoteFileAddress> {
   // Try to parse the input using the old format. If successful, return the

@@ -10,7 +10,7 @@ import test, {
 import nock from "nock";
 import * as sinon from "sinon";
 
-import { ActionState } from "./action-common";
+import { ActionState, StateFeature } from "./action-common";
 import { ActionsEnv, ActionsEnvVars, getActionVersion } from "./actions-util";
 import { AnalysisKind } from "./analyses";
 import * as apiClient from "./api-client";
@@ -189,18 +189,25 @@ export function getTestActionsEnv(): ActionsEnv {
   };
 }
 
+/** For testing purposes, we make all available state features accessible in `TestEnv`. */
+type AllState = ["Logger", "Env", "FeatureFlags"];
+
 /**
  * Wraps a function that accepts an `ActionEnv` for testing in different environments.
  */
-export class TestEnv<Args extends readonly any[], R> {
-  private readonly fn: (state: ActionState, ...args: Args) => R;
+export class TestEnv<
+  Args extends readonly any[],
+  R,
+  Fs extends ReadonlyArray<AllState[number]>,
+> {
+  private readonly fn: (state: ActionState<Fs>, ...args: Args) => R;
   private args?: Args;
-  private state: ActionState;
+  private state: ActionState<AllState>;
 
   constructor(
-    fn: (state: ActionState, ...args: Args) => R,
+    fn: (state: ActionState<Fs>, ...args: Args) => R,
     args?: Args,
-    initialState?: ActionState,
+    initialState?: ActionState<AllState>,
   ) {
     this.fn = fn;
     this.args = args;
@@ -211,11 +218,11 @@ export class TestEnv<Args extends readonly any[], R> {
     };
   }
 
-  private clone(): TestEnv<Args, R> {
+  private clone(): TestEnv<Args, R, Fs> {
     return new TestEnv(this.fn, this.args, { ...this.state });
   }
 
-  public getState(): ActionState {
+  public getState(): ActionState<AllState> {
     return this.state;
   }
 
@@ -229,13 +236,13 @@ export class TestEnv<Args extends readonly any[], R> {
     return result;
   }
 
-  public withFeatures(enabled: Feature[]): TestEnv<Args, R> {
+  public withFeatures(enabled: Feature[]): TestEnv<Args, R, Fs> {
     const result = this.clone();
     result.state.features = createFeatures(enabled);
     return result;
   }
 
-  public withEnv(env: Env): TestEnv<Args, R> {
+  public withEnv(env: Env): TestEnv<Args, R, Fs> {
     const result = this.clone();
     result.state.env = env;
     return result;
@@ -245,7 +252,7 @@ export class TestEnv<Args extends readonly any[], R> {
     if (!this.args) {
       throw new Error("Trying to call function in TestEnv without arguments.");
     }
-    return this.fn(this.state, ...this.args);
+    return this.fn(this.state as ActionState<Fs>, ...this.args);
   }
 
   public passes<T>(
@@ -259,9 +266,11 @@ export class TestEnv<Args extends readonly any[], R> {
 }
 
 /** Utility function to construct a `TestEnv`. */
-export function callee<Args extends readonly any[], R>(
-  fn: (state: ActionState, ...args: Args) => R,
-): TestEnv<Args, R> {
+export function callee<
+  Args extends readonly any[],
+  R,
+  Fs extends readonly StateFeature[],
+>(fn: (state: ActionState<Fs>, ...args: Args) => R): TestEnv<Args, R, Fs> {
   return new TestEnv(fn);
 }