TraitExpression work

src/Trait.ts

@@ -69,24 +69,6 @@ export namespace Trait {
         StaticMembers<Trait.OwnImplClass<T>>
     )
 
-    export type OwnClass<T> = (
-        AbstractClass<Trait.OwnInstance<T>, any[]> &
-        Extend<[
-            Trait.OwnStaticAbstract<T>,
-            StaticMembers<Trait.OwnImplClass<T>>,
-        ]>
-    )
-    export interface OwnClassFn extends Fn {
-        return: Trait.OwnClass<this["arg0"]>
-    }
-
-    export type OwnInstance<T> = (
-        Extend<[Trait.OwnAbstract<T>, Trait.OwnImplInstance<T>]>
-    )
-    export interface OwnInstanceFn extends Fn {
-        return: Trait.OwnInstance<this["arg0"]>
-    }
-
     export type Supertraits<T> = (
         TraitExpression.Traits<Trait.OwnSuperExpression<T>>
     )

src/TraitExpression.ts

@@ -1,6 +1,6 @@
 import { Fn, Pipe, Tuples } from "hotscript"
 import { AbstractClass, Class, Simplify } from "type-fest"
-import { Trait, Traits } from "./Trait"
+import { Trait, TraitInstance, Traits } from "./Trait"
 import { TraitBuilder } from "./TraitBuilder"
 import { ExtendFn, ExtendPlain, SimplifyFn, StaticMembersFn } from "./util"
 
@@ -49,13 +49,14 @@ export class TraitExpression<
     ) {
         return new TraitBuilder<
             This,
+
             Simplify<
                 ExtendPlain<Traits.MapAbstract<T>>
             >,
             Simplify<
                 ExtendPlain<Traits.MapStaticAbstract<T>>
             >,
-            (
+
             AbstractClass<
                 Simplify<
                     ExtendPlain<Traits.MapImplInstance<T>>
@@ -64,7 +65,6 @@ export class TraitExpression<
             Simplify<
                 ExtendPlain<Traits.MapImplStaticMembers<T>>
             >
-            )
         >(
             this,
             {} as any,
@@ -100,26 +100,34 @@ export namespace TraitExpression {
 }
 
 
-export type Implements<Exp extends TraitExpression<any, any>> = (
-    Pipe<Exp, [
-        TraitExpression.TraitsFn,
-        Tuples.Map<Trait.OwnAbstractFn>,
-        ExtendFn,
-        SimplifyFn,
-    ]>
+export type Implements<
+    Exp extends TraitExpression<any, Trait<any, any, any, any>[]>
+> = (
+    Simplify<
+        ExtendPlain<
+            Traits.MapAbstract<
+                TraitExpression.Traits<Exp>
+            >
+        >
+    >
 )
 
-export type ImplementsStatic<Exp extends TraitExpression<any, any>> = (
-    Pipe<Exp, [
-        TraitExpression.TraitsFn,
-        Tuples.Map<Trait.OwnStaticAbstractFn>,
-        ExtendFn,
-        SimplifyFn,
-    ]>
+export type ImplementsStatic<
+    Exp extends TraitExpression<any, Trait<any, any, any, any>[]>
+> = (
+    Simplify<
+        ExtendPlain<
+            Traits.MapStaticAbstract<
+                TraitExpression.Traits<Exp>
+            >
+        >
+    >
 )
 
 
-export type TraitExpressionClass<Exp extends TraitExpression<any, any>> = (
+export type TraitExpressionClass<
+    Exp extends TraitExpression<any, Trait<any, any, any, any>[]>
+> = (
     AbstractClass<
         TraitExpressionInstance<Exp>,
         ConstructorParameters<TraitExpression.Superclass<Exp>>
@@ -127,7 +135,9 @@ export type TraitExpressionClass<Exp extends TraitExpression<any, any>> = (
     TraitExpressionStaticMembers<Exp>
 )
 
-export type TraitExpressionConcreteClass<Exp extends TraitExpression<any, any>> = (
+export type TraitExpressionConcreteClass<
+    Exp extends TraitExpression<any, Trait<any, any, any, any>[]>
+> = (
     Class<
         TraitExpressionInstance<Exp>,
         ConstructorParameters<TraitExpression.Superclass<Exp>>
@@ -135,18 +145,35 @@ export type TraitExpressionConcreteClass<Exp extends TraitExpression<any, any>>
     TraitExpressionStaticMembers<Exp>
 )
 
-export type TraitExpressionInstance<Exp extends TraitExpression<any, any>> = (
-    Pipe<TraitExpression.Traits<Exp>, [
-        Tuples.Map<Trait.OwnInstanceFn>,                   // Map all the traits to their instance representation
-        Tuples.Prepend<                                    // Add the instance of the superclass at the top of the list
-            InstanceType<TraitExpression.Superclass<Exp>>
+export type TraitExpressionInstance<
+    Exp extends TraitExpression<any, Trait<any, any, any, any>[]>
+> = (
+    // Pipe<TraitExpression.Traits<Exp>, [
+    //     Tuples.Map<Trait.OwnInstanceFn>,                   // Map all the traits to their instance representation
+    //     Tuples.Prepend<                                    // Add the instance of the superclass at the top of the list
+    //         InstanceType<TraitExpression.Superclass<Exp>>
+    //     >,
+    //     ExtendFn,                                          // Reduce to a single object that extends all the objects in the list
+    //     SimplifyFn,                                        // Make readable for IDEs
+    // ]>
+    Simplify<
+        ExtendPlain<[
+            InstanceType<TraitExpression.Superclass<Exp>>,
+            ...MapTraitsInstance<
+                TraitExpression.Traits<Exp>
             >,
-        ExtendFn,                                          // Reduce to a single object that extends all the objects in the list
-        SimplifyFn,                                        // Make readable for IDEs
         ]>
+    >
 )
+type MapTraitsInstance<T extends Trait<any, any, any, any>[]> = {
+    [K in keyof T]: K extends keyof []
+        ? T[K]
+        : TraitInstance<T[K]>
+}
 
-export type TraitExpressionStaticMembers<Exp extends TraitExpression<any, any>> = (
+export type TraitExpressionStaticMembers<
+    Exp extends TraitExpression<any, Trait<any, any, any, any>[]>
+> = (
     Pipe<TraitExpression.Traits<Exp>, [
         Tuples.Map<Trait.OwnClassFn>,                     // Map all the traits to their class representation
         Tuples.Prepend<TraitExpression.Superclass<Exp>>,  // Add the superclass at the top of the list