ai-sdk for solid (createMutable)

import {
  createMutable,
  modifyMutable,
  produce,
  reconcile,
  unwrap,
} from "solid-js/store"
import {
  AbstractChat,
  type ChatInit,
  type ChatState,
  type ChatStatus,
  type UIMessage,
} from "ai"

class SolidChatState<UI_MESSAGE extends UIMessage>
  implements ChatState<UI_MESSAGE>
{
  private readonly _data: {
    messages: UI_MESSAGE[]
    status: ChatStatus
    error: Error | undefined
  }

  constructor(messages: UI_MESSAGE[] = []) {
    this._data = createMutable({
      messages,
      status: "ready",
      error: undefined,
    })
  }

  get messages(): UI_MESSAGE[] {
    return this._data.messages
  }
  set messages(messages: UI_MESSAGE[]) {
    this._data.messages = messages
  }

  get status(): ChatStatus {
    return this._data.status
  }
  set status(status: ChatStatus) {
    this._data.status = status
  }

  get error(): Error | undefined {
    return this._data.error
  }
  set error(error: Error | undefined) {
    this._data.error = error
  }

  pushMessage = (m: UI_MESSAGE) => {
    this._data.messages.push(createMutable(m))
  }

  popMessage = () => {
    this._data.messages.pop()
  }

  replaceMessage = (i: number, m: UI_MESSAGE) => {
    modifyMutable(this._data.messages[i], reconcile(m, { merge: true }))
  }

  snapshot = <T>(value: T): T => unwrap(value)
}

export class Chat<
  UI_MESSAGE extends UIMessage = UIMessage,
> extends AbstractChat<UI_MESSAGE> {
  constructor(init: ChatInit<UI_MESSAGE>) {
    super({
      ...init,
      state: new SolidChatState(init.messages ?? []),
    })
  }
}