#define WIN32_LEAN_AND_MEAN
#include <windows.h>

#include <stdio.h>
#include <stdlib.h>
#include <wchar.h>

#include "mcp/server.h"
#include "storage/store.h"

static void
usage(FILE *stream)
{
  fwprintf(stream, L"Usage: tiny-convmem-mcp --data-dir <path>\n");
}

int
wmain(int argc, wchar_t **argv)
{
  wchar_t *absolute = NULL;
  DWORD required;
  tiny_store_t *store = NULL;
  tiny_store_status_t status;
  int result;
  if (argc == 2 && wcscmp(argv[1], L"--help") == 0)
    {
      usage(stdout);
      return 0;
    }
  if (argc != 3 || wcscmp(argv[1], L"--data-dir") != 0 || argv[2][0] == L'\0')
    {
      usage(stderr);
      return 2;
    }
  required = GetFullPathNameW(argv[2], 0, NULL, NULL);
  if (required == 0 || (absolute = malloc((size_t)required * sizeof(wchar_t))) == NULL
      || GetFullPathNameW(argv[2], required, absolute, NULL) == 0)
    {
      fprintf(stderr, "unable to resolve --data-dir\n");
      free(absolute);
      return 2;
    }
  status = tiny_store_open(&store, absolute);
  free(absolute);
  if (status != TINY_STORE_OK)
    {
      fprintf(stderr, "unable to open conversation store: %s\n", tiny_store_status_name(status));
      return status == TINY_STORE_BUSY ? 3 : 1;
    }
  result = tiny_convmem_server_run(stdin, stdout, store);
  tiny_store_close(store);
  if (result != 0)
    {
      fprintf(stderr, "tiny-convmem-mcp stopped with error code %d\n", result);
    }
  return result;
}