在输入一个接受单个枚举作为参数并返回一个Map器函数的工厂函数时寻找一些帮助。

// enumeration of possible partners 
enum Partner {
    Google = 'google',
    Microsoft = 'microsoft'
}

// lets say it's a domain entity, we'll map partner DTOs to it
type Entity = {
    id: string
}

// DTO and mapper function for Partner.Google
type GoogleDto = {
    google_id: string
}

function mapFromGoogle(dto: GoogleDto): Entity {
    return {
        id: dto.google_id
    }
}

// DTO and mapper function for Partner.Microsoft
type MicrosoftDto = {
    ms_id: number
}

function mapFromMicrosoft(dto: MicrosoftDto): Entity {
    return {
        id: String(dto.ms_id)
    }
}

// And here I am trying to use a conditional type
// to check which enum value is passed as an argument
// and to provide a correct return type

function toEntity<T extends Partner>(partner: T): T extends Partner.Google ? typeof mapFromGoogle : typeof mapFromMicrosoft {
    switch(partner) {
        case Partner.Google:
            return mapFromGoogle // Type '(dto: GoogleDto) => Entity' is not assignable to type 'T extends Partner.Google ? (dto: GoogleDto) => Entity : (dto: MicrosoftDto) => Entity'
        case Partner.Microsoft:
            return mapFromMicrosoft // Type '(dto: MicrosoftDto) => Entity' is not assignable to type 'T extends Partner.Google ? (dto: GoogleDto) => Entity : (dto: MicrosoftDto) => Entity'
        default:
        throw new Error('Unsupported partner')
    }
}

const e1 = toEntity(Partner.Google)({ google_id: 'id' }) 
const e2 = toEntity(Partner.Google)({ google_id: 'id', ms_id: 4 }) // 'ms_id' does not exist in type 'GoogleDto'
const e3 = toEntity(Partner.Microsoft)({ ms_id: 10 })
const e4 = toEntity(Partner.Microsoft)({ ms_id: 10, google_id: 'asd' }) // 'google_id' does not exist in type 'MicrosoftDto'
const e5 = toEntity(Partner.Google)({}) // Property 'google_id' is missing in type '{}' but required in type 'GoogleDto'
const e6 = toEntity(Partner.Microsoft)({}) // Property 'ms_id' is missing in type '{}' but required in type 'MicrosoftDto'

生成的函数按预期工作，它正确地依赖于提供的伙伴，如果提供的DTO ID无效，则会出错。
但是，当我尝试从case块中的toEntity函数返回特定的Map器函数时，TS显示错误。
问有没有人能给我指出正确的方向来解决这个案子。
我尝试过将枚举转换为联合，但也不起作用。还尝试过删除依赖于TS来推断返回的Map器函数类型的返回类型，但在这种情况下，当调用toEntity时，类型检查不起作用。