为类的方法声明一个附加成员

Question

我用打字稿装饰器装饰了一个方法，向该方法添加了一个成员：

export class MyClass {
    @AddMyMember()
    public myMethod(param1:string) {
        // ...
    }
}

所以我应该可以这样使用它：

const myClassInstance = new MyClass();
myClassInstance.myMethod.myMember(); // error of course here, as myMember has not been declared yet...

但我找不到有效声明 myMember 的方法：

export declare interface MyClass {
// Something great here maybe?....
}

Answer 1

合并问题

没有办法声明不同类型的类的成员，声明合并只能添加成员而不改变现有成员的类型。

我们可以做的就是使用交集类型。如果一个交集类型的两个成员具有相同的名称和不同的类型，则该交集上的成员将是两个原始类型的交集。

class _MyClass {
    public constructor (t: string) {

    }
    public myMethod(param1:string) {
    }

    static staticF(){}
}

type ReplaceInstanceType<T extends new(...a: any[])=> any, TNew> = {
    new (... a: (T extends new (...p: infer P) => unknown ? P: [])) : TNew
} & Pick<T, keyof T>

export type MyClass = _MyClass & {
    myMethod: {
        myMember(): void
    }
}


export const MyClass: ReplaceInstanceType<typeof _MyClass, MyClass> = _MyClass as any;

let a = new MyClass("") // can new up
a.myMethod("") // can call myMethod
a.myMethod.myMember(); // can call  myMember
MyClass.staticF

装饰器问题

问题的第二部分是装饰器不能改变它所应用的成员的类型（这是设计使然）。您可以改为使用 HOC 创建类，并传入成员名称列表（使用 keyof 进行编译检查）以添加额外的成员：

type ReplaceInstanceType<T extends new(...a: any[])=> any, TNew> = {
    new (... a: (T extends new (...p: infer P) => unknown ? P: [])) : TNew
} & Pick<T, keyof T>

function addMemebers<T extends new(...a: any[])=> any, K extends keyof InstanceType<T>>(members: K[], cls: T) : ReplaceInstanceType<T, InstanceType<T> & Record<K, {myMember(): void}>>{
    return null as any; // replace with implementation
}
export const MyClass = addMemebers(["myMethod"], class {
    public constructor (t: string) {

    }
    public myMethod(param1:string) {
    }

    static staticF(){}
})


let a = new MyClass("") // can new up
a.myMethod("") // can call myMethod
a.myMethod.myMember(); // can call  myMember
MyClass.staticF