javascript - 在 Typescript 中，如何使用字符串类型联合来约束来自另一个类型或接口(interface)的键

Question

如果我有:

type Op = 'AND' | 'OR';

我知道我可以写:

type Exp<T> = {
   [_ in Op]: T[];
};

这样下面所有的行都是有效的

let a: Exp<number> = { OR: [4] };
let b: Exp<number> = { AND: [5] };
let c: Exp<number> = { AND: [1], OR: [2, 3] };

我的问题是，我可以这样写吗，a 和 b 有效，但 c 无效？即只允许有一个键，且键的类型为Op。

即使我像这样迭代所有 Op 类型:

type Exp<T> = { AND: T[] } | { OR: T[] };

它仍然不会执行我希望它执行的操作。我猜这是由于声明合并造成的？

typescript 很新。寻找一种优雅的方式来表达这一点。

Answer 1

有几种不同的方法可以实现您想要的，哪种方法最好取决于您的实际用例，我无法从示例中得出。

这些是实现您想要的目标的一些方法。

方式 1:传递 key 。

type Exp<T, K extends Op> = Record<K, T[]>

let a: Exp<number, "OR"> = { OR: [4] }; //pass
let b: Exp<number, "AND"> = { AND: [5] }; //pass
let c: Exp<number, "And"> = { AND: [1], OR: [2, 3] }; //fail
let c: Exp<number, Op> = { AND: [1], OR: [2, 3] }; // pass

方法 2:使 OP 的每个 key 都成为异或，在这种情况下，您不再需要传递 key 。

export type XOR<T, U> = (T | U) extends object ? (Without<T, U> & U) | (Without<U, T> & T) : T | U;
export type Without<T, U> = { [P in Exclude<keyof T, keyof U>]?: never };

type Op = 'AND' | 'OR';

type Exp<T> = XOR<Pick<Record<Op, T[]>, "AND">, Pick<Record<Op, T[]>, "OR">>

let a: Exp<number> = { OR: [4] }; //pass
let b: Exp<number> = { AND: [5] }; //pass
let c: Exp<number> = { AND: [1], OR: [2, 3] }; //fail

方式 3:请注意，这是自动生成的版本，它并不漂亮，我这样做只是为了证明它是可能的。如果没有重载，这个解决方案不能达到无限深度，这是因为联合不能被迭代，但是这显示它到深度九。我发布这个只是因为它引起了人们的兴趣。

// add an element to the end of a tuple
type Push<L extends any[], T> =
  ((r: any, ...x: L) => void) extends ((...x: infer L2) => void) ?
  { [K in keyof L2]-?: K extends keyof L ? L[K] : T } : never

// convert a union to an intersection: X | Y | Z ==> X & Y & Z
type UnionToIntersection<U> =
  (U extends any ? (k: U) => void : never) extends ((k: infer I) => void) ? I : never

// convert a union to an overloaded function X | Y ==> ((x: X)=>void) & ((y:Y)=>void)     
type UnionToOvlds<U> = UnionToIntersection<U extends any ? (f: U) => void : never>;

// convert a union to a tuple X | Y => [X, Y]
// a union of too many elements will become an array instead
type UnionToTuple<U> = UTT0<U> extends infer T ? T extends any[] ?
  Exclude<U, T[number]> extends never ? T : U[] : never : never

// each type function below pulls the last element off the union and 
// pushes it onto the list it builds
type UTT0<U> = UnionToOvlds<U> extends ((a: infer A) => void) ? Push<UTT1<Exclude<U, A>>, A> : []
type UTT1<U> = UnionToOvlds<U> extends ((a: infer A) => void) ? Push<UTT2<Exclude<U, A>>, A> : []
type UTT2<U> = UnionToOvlds<U> extends ((a: infer A) => void) ? Push<UTT3<Exclude<U, A>>, A> : []
type UTT3<U> = UnionToOvlds<U> extends ((a: infer A) => void) ? Push<UTT4<Exclude<U, A>>, A> : []
type UTT4<U> = UnionToOvlds<U> extends ((a: infer A) => void) ? Push<UTT5<Exclude<U, A>>, A> : []
type UTT5<U> = UnionToOvlds<U> extends ((a: infer A) => void) ? Push<UTT6<Exclude<U, A>>, A> : []
type UTT6<U> = UnionToOvlds<U> extends ((a: infer A) => void) ? Push<UTT7<Exclude<U, A>>, A> : []
type UTT7<U> = UnionToOvlds<U> extends ((a: infer A) => void) ? Push<UTT8<Exclude<U, A>>, A> : []
type UTT8<U> = UnionToOvlds<U> extends ((a: infer A) => void) ? Push<UTT9<Exclude<U, A>>, A> : []
type UTT9<U> = UnionToOvlds<U> extends ((a: infer A) => void) ? Push<UTTX<Exclude<U, A>>, A> : []
type UTTX<U> = []; // bail out

export type XOR<T, U> = (T | U) extends object ? (Without<T, U> & U) | (Without<U, T> & T) : T | U;
export type Without<T, U> = { [P in Exclude<keyof T, keyof U>]?: never };

type DeepXOR<Obj, T extends any[]> = {
    [K in keyof T]: T[K] extends keyof Obj
        ? XOR<Pick<Obj, T[K]>, ((..._: T) => any) extends ((_: any, ..._1: infer TAIL) => any)
            ? DeepXOR<Obj, TAIL>[0] extends never
                ? never
                : DeepXOR<Obj, TAIL>[0]
            : never
            >
        : never;
};


interface Person {
    name: string;
    age: number;
    dob: Date;
}


type PersonXOR = DeepXOR<Person, UnionToTuple<keyof Person>>[0]

const test: PersonXOR = {name: "shanon"}; // pass
const test1: PersonXOR = {age: 24}; // pass
const test2: PersonXOR = {dob: new Date()}; // pass
const test3: PersonXOR = {name: "shanon", dob: new Date()} // fail

编辑方式 4:

@dragomirtitian提醒我你实际上可以通过连接每个对象/ key 对(单数对)并将其与其他对的一部分相交并将它们的类型值设置为从不来创建严格的联合。可能是最好最干净的方式

type Op = 'AND' | 'OR' | "NOT";


type Exp<T> = Op extends infer D ?
    D extends any ? {
   [_ in D]: T[];
}: never : never;

type UnionKeys<T> = T extends any ? keyof T : never;
type StrictUnionHelper<T, TAll> = T extends any ? T & Partial<Record<Exclude<UnionKeys<TAll>, keyof T>, never>> : never;
type StrictUnion<T> = StrictUnionHelper<T, T>

let a: StrictUnion<Exp<number>> = { OR: [4] }; // no error
let b: StrictUnion<Exp<number>> = { AND: [5] }; // no error
let c: StrictUnion<Exp<number>> = { AND: [1], OR: [2, 3] }; // error