如何在TypeScript中兼顾常量对象类型校验与精确类型推断？

我有一个静态对象定义，希望确保它符合指定类型，同时还能利用该常量定义进行类型推断。示例代码如下：

const TREE_BRANCH_TYPES = ['Type1', 'Type2'] as const;

type TreeBranchType = typeof TREE_BRANCH_TYPES[number];
type TreeRoot = Record<TreeBranchType, TreeBranch>;
type TreeBranch = { leaves: readonly TreeLeaf[] };
type TreeLeaf = { name: string };

const tree: TreeRoot = {
    Type1: {
        leaves: [
            { name: 'Type1-Leaf1' },
            { name: 'Type1-Leaf2' },
        ]
    },
    Type2: {
        leaves: [
            { name: 'Type2-Leaf1' },
            { name: 'Type2-Leaf2' },
        ]
    }
} as const;

type TreeLeafName = typeof tree[TreeBranchType]['leaves'][number]['name'];

function findLeaf(type: TreeBranchType, name: TreeLeafName): TreeLeaf | undefined {
    return tree[type].leaves.find(leaf => leaf.name === name);
}

上述代码中，编译器会要求tree符合TreeRoot类型定义，但TreeLeafName会被推断为string类型；若移除tree的类型标注，TreeLeafName则会被推断为"Type1-Leaf1" | "Type1-Leaf2" | "Type2-Leaf1" | "Type2-Leaf2"这样的具体字符串字面量联合类型。请问是否有办法兼顾两者：既保证tree常量定义的类型安全，又能实现TreeLeafName的精确类型推断？

使用泛型辅助函数可以完美解决这个问题：它既能约束对象符合目标类型，又能保留对象的字面量类型信息，让类型推断可以精确到具体的字符串联合类型。

具体实现代码如下：

const TREE_BRANCH_TYPES = ['Type1', 'Type2'] as const;

type TreeBranchType = typeof TREE_BRANCH_TYPES[number];
type TreeBranch = { leaves: readonly TreeLeaf[] };
type TreeLeaf = { name: string };
type TreeRoot = Record<TreeBranchType, TreeBranch>;

// 泛型辅助函数：约束输入必须符合TreeRoot，同时返回原始字面量类型
const asTreeRoot = <T extends TreeRoot>(obj: T) => obj;

const tree = asTreeRoot({
    Type1: {
        leaves: [
            { name: 'Type1-Leaf1' },
            { name: 'Type1-Leaf2' },
        ]
    },
    Type2: {
        leaves: [
            { name: 'Type2-Leaf1' },
            { name: 'Type2-Leaf2' },
        ]
    }
} as const);

// 现在TreeLeafName会被精确推断为具体的字符串字面量联合类型
type TreeLeafName = typeof tree[TreeBranchType]['leaves'][number]['name'];
// 类型为："Type1-Leaf1" | "Type1-Leaf2" | "Type2-Leaf1" | "Type2-Leaf2"

function findLeaf(type: TreeBranchType, name: TreeLeafName): TreeLeaf | undefined {
    return tree[type].leaves.find(leaf => leaf.name === name);
}

原理说明 #

• 泛型函数asTreeRoot接收一个泛型参数T，要求T必须是TreeRoot的子类型，这就保证了传入的对象符合我们定义的类型规范。
• 函数返回值直接是T，也就是传入对象的原始字面量类型（因为我们用了as const断言），这样TypeScript就能保留所有精确的类型信息，让TreeLeafName被推断为具体的字符串联合类型。

内容的提问来源于stack exchange，提问作者RileyE