traitify-ts/src/TraitExpression.ts

import { AbstractClass, Class, Simplify } from "type-fest"
import { Trait, TraitInstance, TraitStaticMembers, TraitTuple } from "./Trait"
import { TraitBuilder } from "./TraitBuilder"
import { ExtendPlain, StaticMembers } from "./util"


export class TraitExpression<
    Superclass   extends AbstractClass<object>,
    const Traits extends Trait<any, any, any, any>[],
> {
    constructor(
        readonly superclass:  Superclass,
        readonly traits:      Traits,
    ) {}

    get extends(): (
        AbstractClass<
            Simplify<
                ExtendPlain<[
                    InstanceType<Superclass>,
                    ...TraitTuple.MapImplInstance<Traits>,
                ]>
            >,

            ConstructorParameters<Superclass>
        > &

        Simplify<
            ExtendPlain<[
                StaticMembers<Superclass>,
                ...TraitTuple.MapImplStaticMembers<Traits>,
            ]>
        >
    ) {
        return this.traits.reduce(
            (previous, trait) => trait.apply(previous),
            this.superclass,
        ) as any
    }

    implementsStatic(target: ImplementsStatic<typeof this>, context: any) {}

    subtrait<
        This extends TraitExpression<typeof TraitExpression.NullSuperclass, any>
    >(
        this: This
    ) {
        return new TraitBuilder<
            This,

            Simplify<
                ExtendPlain<TraitTuple.MapAbstract<Traits>>
            >,
            Simplify<
                ExtendPlain<TraitTuple.MapStaticAbstract<Traits>>
            >,

            AbstractClass<
                Simplify<
                    ExtendPlain<TraitTuple.MapImplInstance<Traits>>
                >
            > &
            Simplify<
                ExtendPlain<TraitTuple.MapImplStaticMembers<Traits>>
            >
        >(
            this,
            {} as any,
            {} as any,
            Super => class extends Super {} as any,
        )
    }
}

export namespace TraitExpression {
    export class NullSuperclass {
        static readonly _tag = "@thilawyn/traitify-ts/TraitExpression.NullSuperclass"
        constructor(...args: any[]) {}
    }

    export type Superclass<T> = (
        T extends TraitExpression<infer Superclass, any>
            ? Superclass
            : never
    )

    export type Traits<T> = (
        T extends TraitExpression<any, infer Traits>
            ? Traits
            : never
    )
}


export type Implements<
    Exp extends TraitExpression<any, Trait<any, any, any, any>[]>
> = (
    Simplify<
        ExtendPlain<
            TraitTuple.MapAbstract<
                TraitExpression.Traits<Exp>
            >
        >
    >
)

export type ImplementsStatic<
    Exp extends TraitExpression<any, Trait<any, any, any, any>[]>
> = (
    Simplify<
        ExtendPlain<
            TraitTuple.MapStaticAbstract<
                TraitExpression.Traits<Exp>
            >
        >
    >
)


export type TraitExpressionClass<
    Exp extends TraitExpression<any, Trait<any, any, any, any>[]>
> = (
    AbstractClass<
        TraitExpressionInstance<Exp>,
        ConstructorParameters<TraitExpression.Superclass<Exp>>
    > &
    TraitExpressionStaticMembers<Exp>
)

export type TraitExpressionConcreteClass<
    Exp extends TraitExpression<any, Trait<any, any, any, any>[]>
> = (
    Class<
        TraitExpressionInstance<Exp>,
        ConstructorParameters<TraitExpression.Superclass<Exp>>
    > &
    TraitExpressionStaticMembers<Exp>
)

export type TraitExpressionInstance<
    Exp extends TraitExpression<any, Trait<any, any, any, any>[]>
> = (
    Simplify<
        ExtendPlain<[
            InstanceType<TraitExpression.Superclass<Exp>>,
            ...MapTraitsInstance<TraitExpression.Traits<Exp>>,
        ]>
    >
)
type MapTraitsInstance<T> = {
    [K in keyof T]: K extends keyof []
        ? T[K]
        : T[K] extends Trait<any, any, any, any>
            ? TraitInstance<T[K]>
            : never
}

export type TraitExpressionStaticMembers<
    Exp extends TraitExpression<any, Trait<any, any, any, any>[]>
> = (
    Simplify<
        ExtendPlain<[
            StaticMembers<TraitExpression.Superclass<Exp>>,
            ...MapTraitsStaticMembers<TraitExpression.Traits<Exp>>,
        ]>
    >
)
type MapTraitsStaticMembers<T> = {
    [K in keyof T]: K extends keyof []
        ? T[K]
        : T[K] extends Trait<any, any, any, any>
            ? TraitStaticMembers<T[K]>
            : never
}