TypeScript：泛型检查枚举类型

Question

有一个进程枚举。它们有不同的步骤，也用枚举表示，如下所示。

enum Process {
  Simple = "simple",
  Advanced = "advanced"
}

enum SimpleStep {
  A = "A",
  B = "B"
}

enum AdvancedStep {
  A = "A",
  B = "B",
  C = "C"
}

通过以下语句，我创建了一系列步骤。

const SIMPLE_STEPS = Object.keys(SimpleStep).map(
  (k: string) => SimpleStep[k]
);

const ADVANCED_STEPS = Object.keys(AdvancedStep).map(
  k => AdvancedStep[k]
);

const ALL_STEPS = {
  [Process.Simple]: SIMPLE_STEPS,
  [Process.Advanced]: ADVANCED_STEPS
};

我写了下面的函数来获取步数。

// ???: Check if S is a step of Process
const getStepNumber = <P extends Process, S>(process: P, step: S) => {
  return ALL_STEPS[process].indexOf(step) + 1;
};

// returns 2, which is correct
console.log('step number of B', getStepNumber(Process.Advanced, AdvancedStep.B)); 

// returns 0. Is it possible to prevent at compile-time?
console.log('step number of C', getStepNumber(Process.Simple, AdvancedStep.C));

正如您在代码示例中看到的那样，是否可以使用泛型来防止在编译时以错误的步骤调用函数？

这里是 playground，如果你想尝试整个例子：TS Playground

Answer 1

一个选项是引入一个条件类型，它允许您根据提供给函数的 Process 推断需要的步骤枚举（即 SimpleStep 或 AdvancedStep）。这可以按如下方式完成：

type StepFromProcess<P extends Process> =
    P extends Process.Simple ? SimpleStep : AdvancedStep

然后您可以更改函数以使用该类型：

const getStepNumber = <P extends Process>(process: P, step: StepFromProcess<P>) => ...

编译器现在将阻止您进行此（无效）调用：

console.log('step number of C', getStepNumber(Process.Simple, AdvancedStep.C));

TypeScript：泛型检查枚举类型

TypeScript: Generics check for enum type

generics

typescript

typescript-generics

typescript-typings