如何为函数参数生成TypeScript嵌套键类型与动态返回类型？

完整代码实现 #

// 生成所有嵌套键路径（点分隔形式）
type NestedKey<S> = S extends object
  ? {
      [K in keyof S & string]: K | `${K}.${NestedKey<S[K]>}`;
    }[keyof S & string]
  : never;

// 根据键路径获取对应嵌套类型
type PathValue<S, K extends NestedKey<S>> = K extends `${infer Key}.${infer Rest}`
  ? Key extends keyof S
    ? PathValue<S[Key], Rest & NestedKey<S[Key]>>
    : never
  : K extends keyof S
  ? S[K]
  : never;

// 定义返回的函数类型（重载形式）
type ReturnFn<S> = {
  (): S;
  <K extends NestedKey<S>>(target: K): PathValue<S, K>;
};

// 实现test函数
function test<S>(data: S): ReturnFn<S> {
  return ((target?: string) => {
    let result: any = data;
    target?.split(".").forEach(k => result = result[k]);
    return result;
  }) as ReturnFn<S>;
}

// 测试示例
const data = {
  suggestion: {a: {nested: true}}
};

const result = test(data);
const fullData = result(); // 类型为 { suggestion: { a: { nested: boolean; }; }; }
const suggestion = result("suggestion"); // 类型为 { a: { nested: boolean; }; }
const suggestionA = result("suggestion.a"); // 类型为 { nested: boolean; }
const nestedValue = result("suggestion.a.nested"); // 类型为 boolean

关键类型解析 #

NestedKey