import { Schema } from "@effect/schema"
import type { Types } from "effect"
import type { StaticType } from "../../../StaticType"
import type { HasFields } from "./util"


export interface MutableClass<
    Self,
    Fields extends Schema.Struct.Fields,
    I, R, C,
    Inherited,
    Proto,
>
extends Omit<
    StaticType<Schema.Class<Self, Fields, I, R, C, Inherited, Proto>>,
    "extend"
>
{
    new(
        ...args: ConstructorParameters<Schema.Class<Self, Fields, I, R, C, Inherited, Proto>>
    ): Omit<
        InstanceType<Schema.Class<Self, Fields, I, R, C, Inherited, Proto>>,
        keyof Fields
    > &
    Types.Mutable<Schema.Struct.Type<Fields>>

    extend<Extended>(identifier: string): <newFields extends Schema.Struct.Fields>(
        fields:        newFields | HasFields<newFields>,
        annotations?:  Schema.Annotations.Schema<Extended>,
    ) => MutableClass<
        Extended,
        Fields & newFields,
        I & Schema.Struct.Encoded<newFields>,
        R | Schema.Struct.Context<newFields>,
        C & Schema.Struct.Constructor<newFields>,
        Self,
        Proto
    >
}


interface MutableInput<
    Self,
    Fields extends Schema.Struct.Fields,
    I, R, C,
    Inherited,
    Proto
> extends StaticType<Schema.Class<Self, Fields, I, R, C, Inherited, Proto>> {
    new(
        props: any,
        options?: any,
      ): Schema.Struct.Type<Fields> & Omit<Inherited, keyof Fields> & Proto
}

export const Mutable = <
    Self,
    Fields extends Schema.Struct.Fields,
    I, R, C,
    Inherited,
    Proto
>(
    self: MutableInput<Self, Fields, I, R, C, Inherited, Proto>
) => self as MutableClass<Self, Fields, I, R, C, Inherited, Proto>