import type { Infer } from './schema'
import { expectTypeOf } from 'vitest'
import { boolean, func, number, object, rpc, string } from './schema'

// see https://github.com/colinhacks/zod?tab=readme-ov-file#functions

declare module './schema' {
  interface TypeMeta {
    rpcDesc?: string
  }
}

describe('rpc.spec', () => {
  it('should do something', async () => {
    const rpcFn = object({ rpc: rpc() })
    type RpcFn = Infer<typeof rpcFn>
    expectTypeOf<RpcFn>().toMatchObjectType<{
      rpc: () => void
    }>()

    const rpcFn2 = object({ rpc: rpc(object({
      a: number(),
    }), string()) })
    type RpcFn2 = Infer<typeof rpcFn2>
    expectTypeOf<RpcFn2>().toMatchObjectType<{
      rpc: (info: { a: number }) => string | Promise<string>
    }>()

    const payloadSchema = object({
      id: string(),
      method: string(),
      params: object({
        a: string(),
        b: boolean(),
      }),
    })

    type PayloadSchema = Infer<typeof payloadSchema>

    expectTypeOf<PayloadSchema>().toMatchObjectType<{
      id: string
      method: string
      params: {
        a: string
        b: boolean
      }
    }>()

    const rpcSchema = object({
      echo: func(
        [string(), boolean()],
        string().optional(),
      ).meta({
        rpcDesc: 'Just echo the string',
      }),
      rpc: rpc(payloadSchema),
      noFunc: string(),
    })

    type RpcRaw = Infer<typeof rpcSchema>

    expectTypeOf<RpcRaw>().toMatchObjectType<{
      echo: (args_0: string, args_1: boolean) => string | undefined
      rpc: (info: {
        id: string
        method: string
        params: {
          a: string
          b: boolean
        }
      }) => void | Promise<void>
      noFunc: string
    }>()

    const api: RpcRaw = {
      echo: (a: string, b: boolean) => {
        return a + b
      },
      noFunc: 'noFunc',
      rpc(info) {
        const _ = info.id
        // return undefined
      },
    }

    expect(rpcSchema).toMatchInlineSnapshot(`
      Type {
        "_args": undefined,
        "_check": [Function],
        "_default": undefined,
        "_enumValues": undefined,
        "_info": undefined,
        "_meta": undefined,
        "_object": Object {
          "echo": Type {
            "_args": Array [
              Type {
                "_args": undefined,
                "_check": [Function],
                "_default": undefined,
                "_enumValues": undefined,
                "_info": undefined,
                "_meta": undefined,
                "_object": undefined,
                "_optional": undefined,
                "_ret": undefined,
                "_type": undefined,
                "_union": undefined,
                "type": "string",
              },
              Type {
                "_args": undefined,
                "_check": [Function],
                "_default": undefined,
                "_enumValues": undefined,
                "_info": undefined,
                "_meta": undefined,
                "_object": undefined,
                "_optional": undefined,
                "_ret": undefined,
                "_type": undefined,
                "_union": undefined,
                "type": "boolean",
              },
            ],
            "_check": [Function],
            "_default": undefined,
            "_enumValues": undefined,
            "_info": undefined,
            "_meta": Object {
              "rpcDesc": "Just echo the string",
            },
            "_object": undefined,
            "_optional": undefined,
            "_ret": Type {
              "_args": undefined,
              "_check": [Function],
              "_default": undefined,
              "_enumValues": undefined,
              "_info": undefined,
              "_meta": undefined,
              "_object": undefined,
              "_optional": true,
              "_ret": undefined,
              "_type": undefined,
              "_union": undefined,
              "type": "string",
            },
            "_type": undefined,
            "_union": undefined,
            "type": "function",
          },
          "noFunc": Type {
            "_args": undefined,
            "_check": [Function],
            "_default": undefined,
            "_enumValues": undefined,
            "_info": undefined,
            "_meta": undefined,
            "_object": undefined,
            "_optional": undefined,
            "_ret": undefined,
            "_type": undefined,
            "_union": undefined,
            "type": "string",
          },
          "rpc": Type {
            "_args": undefined,
            "_check": [Function],
            "_default": undefined,
            "_enumValues": undefined,
            "_info": Type {
              "_args": undefined,
              "_check": [Function],
              "_default": undefined,
              "_enumValues": undefined,
              "_info": undefined,
              "_meta": undefined,
              "_object": Object {
                "id": Type {
                  "_args": undefined,
                  "_check": [Function],
                  "_default": undefined,
                  "_enumValues": undefined,
                  "_info": undefined,
                  "_meta": undefined,
                  "_object": undefined,
                  "_optional": undefined,
                  "_ret": undefined,
                  "_type": undefined,
                  "_union": undefined,
                  "type": "string",
                },
                "method": Type {
                  "_args": undefined,
                  "_check": [Function],
                  "_default": undefined,
                  "_enumValues": undefined,
                  "_info": undefined,
                  "_meta": undefined,
                  "_object": undefined,
                  "_optional": undefined,
                  "_ret": undefined,
                  "_type": undefined,
                  "_union": undefined,
                  "type": "string",
                },
                "params": Type {
                  "_args": undefined,
                  "_check": [Function],
                  "_default": undefined,
                  "_enumValues": undefined,
                  "_info": undefined,
                  "_meta": undefined,
                  "_object": Object {
                    "a": Type {
                      "_args": undefined,
                      "_check": [Function],
                      "_default": undefined,
                      "_enumValues": undefined,
                      "_info": undefined,
                      "_meta": undefined,
                      "_object": undefined,
                      "_optional": undefined,
                      "_ret": undefined,
                      "_type": undefined,
                      "_union": undefined,
                      "type": "string",
                    },
                    "b": Type {
                      "_args": undefined,
                      "_check": [Function],
                      "_default": undefined,
                      "_enumValues": undefined,
                      "_info": undefined,
                      "_meta": undefined,
                      "_object": undefined,
                      "_optional": undefined,
                      "_ret": undefined,
                      "_type": undefined,
                      "_union": undefined,
                      "type": "boolean",
                    },
                  },
                  "_optional": undefined,
                  "_ret": undefined,
                  "_type": undefined,
                  "_union": undefined,
                  "type": "object",
                },
              },
              "_optional": undefined,
              "_ret": undefined,
              "_type": undefined,
              "_union": undefined,
              "type": "object",
            },
            "_meta": undefined,
            "_object": undefined,
            "_optional": undefined,
            "_ret": Type {
              "_args": undefined,
              "_check": [Function],
              "_default": undefined,
              "_enumValues": undefined,
              "_info": undefined,
              "_meta": undefined,
              "_object": undefined,
              "_optional": true,
              "_ret": undefined,
              "_type": undefined,
              "_union": undefined,
              "type": "none",
            },
            "_type": undefined,
            "_union": undefined,
            "type": "rpc",
          },
        },
        "_optional": undefined,
        "_ret": undefined,
        "_type": undefined,
        "_union": undefined,
        "type": "object",
      }
    `)
  })
})