import { createServer } from 'node:http';
import { WebSocketServer } from 'ws';
import { runtime } from '@livon/runtime';
import { schemaModule } from '@livon/schema';
import { nodeWsTransport } from '@livon/node-ws-transport';
import { serverSchema } from './schema.js';

const httpServer = createServer();
const wsServer = new WebSocketServer({ server: httpServer, path: '/ws' });

runtime(
  nodeWsTransport({ server: wsServer }),
  schemaModule(serverSchema, { explain: true }),
);

httpServer.listen(3002, '127.0.0.1');

import { runtime } from '@livon/runtime';
import { clientWsTransport } from '@livon/client-ws-transport';
import { api } from './generated/api.js';

runtime(clientWsTransport({ url: 'ws://127.0.0.1:3002/ws' }), api);

api({
  onMessage: (payload) => {
    console.log(payload.text);
  },
});

await api.sendMessage({
  author: 'Alice',
  text: 'Hello from LIVON',
});

import {
  and,
  api,
  object,
  operation,
  string,
  subscription,
} from '@livon/schema';

const MessageInput = object({
  name: 'MessageInput',
  shape: {
    author: string(),
    text: string(),
  },
});

const WithId = object({ name: 'WithId', shape: { id: string() } });

const MessageWithId = and({ left: MessageInput, right: WithId });

const sendMessage = operation({
  input: MessageInput,
  output: MessageWithId,
  exec: async (input) => ({ ...input, id: 'msg-1' }),
  publish: {
    onMessage: (output) => output,
  },
});

export const ApiSchema = api({
  operations: {sendMessage},
  subscriptions: {onMessage: subscription({payload: MessageWithId})},
});

export const serverSchema = ApiSchema;