import { BackendIdMismatchError, ServerAheadError, UnknownError } from '@livestore/common'
import { Rpc, RpcGroup, Schema } from '@livestore/utils/effect'

import * as SyncMessage from './sync-message-types.ts'

const commonPayloadFields = {
  /**
   * While the storeId is already implied by the Durable Object, we still need the explicit storeId
   * since a DO doesn't know its own id.name value. 🫠
   * https://community.cloudflare.com/t/how-can-i-get-the-name-of-a-durable-object-from-itself/505961/8
   */
  storeId: Schema.String,
  /** Needed for various reasons (e.g. auth) */
  payload: Schema.optional(Schema.JsonValue),
}

export class SyncDoRpc extends RpcGroup.make(
  Rpc.make('SyncDoRpc.Pull', {
    payload: {
      /** Omitting the cursor will start from the beginning */
      cursor: SyncMessage.PullRequest.fields.cursor,
      // TODO rename
      /** Whether to keep the pull stream alive and wait for more events */
      rpcContext: Schema.optional(
        Schema.Struct({
          callerContext: Schema.Struct({
            bindingName: Schema.String,
            durableObjectId: Schema.String,
          }),
        }),
      ),
      ...commonPayloadFields,
    },
    success: Schema.Struct({
      rpcRequestId: Schema.String,
      ...SyncMessage.PullResponse.fields,
    }),
    error: Schema.Union(UnknownError, BackendIdMismatchError),
    stream: true,
  }),
  Rpc.make('SyncDoRpc.Push', {
    payload: {
      ...SyncMessage.PushRequest.fields,
      ...commonPayloadFields,
    },
    success: SyncMessage.PushAck,
    error: Schema.Union(UnknownError, ServerAheadError, BackendIdMismatchError),
  }),
  Rpc.make('SyncDoRpc.Ping', {
    payload: {
      ...commonPayloadFields,
    },
    success: Schema.Void,
  }),
) {}