如何告诉TypeScript object[foo]的类型是所有可能的object.prop类型的子集

type SubsetType<T, K extends keyof T> = T[K];

type PropKeys<T> = keyof T;

type SubsetTypeOfFoo<T, K extends keyof T> = SubsetType<T, K>[];

function getSubsetTypeOfFoo<T, K extends keyof T>(obj: T, prop: K): SubsetTypeOfFoo<T, K> {
  return obj[prop];
}

interface Person {
  name: string;
  age: number;
  address: string;
}

const person: Person = {
  name: "John",
  age: 30,
  address: "123 Street",
};

const nameOrAge: SubsetTypeOfFoo<Person, "name" | "age"> = getSubsetTypeOfFoo(person, "name");
// nameOrAge 的类型为 string[]，即 person.name 的类型的子集

const nameOrAddress: SubsetTypeOfFoo<Person, "name" | "address"> = getSubsetTypeOfFoo(person, "address");
// nameOrAddress 的类型为 string[]，即 person.address 的类型的子集