import type { DummyState } from './DummyState'

import { BaseRecord } from '@aries-framework/core'
import { v4 as uuid } from 'uuid'

export interface DummyStorageProps {
  id?: string
  createdAt?: Date
  connectionId: string
  threadId: string
  state: DummyState
}

export class DummyRecord extends BaseRecord implements DummyStorageProps {
  public connectionId!: string
  public threadId!: string
  public state!: DummyState

  public static readonly type = 'DummyRecord'
  public readonly type = DummyRecord.type

  public constructor(props: DummyStorageProps) {
    super()
    if (props) {
      this.id = props.id ?? uuid()
      this.createdAt = props.createdAt ?? new Date()
      this.state = props.state
      this.connectionId = props.connectionId
      this.threadId = props.threadId
    }
  }

  public getTags() {
    return {
      ...this._tags,
      threadId: this.threadId,
      connectionId: this.connectionId,
      state: this.state,
    }
  }

  public assertState(expectedStates: DummyState | DummyState[]) {
    if (!Array.isArray(expectedStates)) {
      expectedStates = [expectedStates]
    }

    if (!expectedStates.includes(this.state)) {
      throw new Error(`Dummy record is in invalid state ${this.state}. Valid states are: ${expectedStates.join(', ')}.`)
    }
  }
}