main.c

// Copyright (c) 2024-2026 Steffen Illhardt,
// Licensed under the MIT license ( https://opensource.org/license/mit/ ).

#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <wchar.h>
#include "wtswidth.h"

static inline void help(void)
{
  static const char msg[] =
    "* 'wtswidth' - Windows Terminal string width, version 1.3. *\n"
    "Count the number of columns required to represent strings in the Windows Terminal.\n"
    "Write the widths as list of numbers of character cells the passed strings occupy.\n"
    "In case the wrong syntax is used, this message is displayed and 1 is returned.\n\n"
    "Usage:\n"
    "wtswidth a s1 [s2 [... sN]]\n"
    "  a       Number of columns that a character with ambiguous width occupies.\n"
    "           This can be either 1 for \"Narrow\" or 2 for \"Wide\". The default\n"
    "           setting in Windows Terminal is \"Narrow\" (1).\n"
    "  s1..sN  Strings to be measured.\n\n";

  fwrite(msg, 1, sizeof(msg) - 1, stderr);
}

#if defined(_WIN32) || defined(__CYGWIN__)
// On Windows, the command line is always available as wide string anyway. So we just use that.
#  include <windows.h>
int main(void)
{
  int argc = 0;
  wchar_t **const argv = CommandLineToArgvW(GetCommandLineW(), &argc);
  if (argc < 3)
  {
    LocalFree(argv);
    help();
    return 1;
  }

  if (wcstol(argv[1], NULL, 10) == 2)
    wtssetambwidth(wide);

  for (int i = 2; i < argc; ++i)
    printf_s("%d\n", wts16width((const char16_t *)argv[i], (int)wcslen(argv[i])));

  LocalFree(argv);
  return 0;
}
#else
// On *nix we assume to get UTF-8 as this is virtually standard.
int main(int argc, char *argv[])
{
  if (argc < 3)
  {
    help();
    return 1;
  }

  if (strtol(argv[1], NULL, 10) == 2)
    wtssetambwidth(wide);

  for (int i = 2; i < argc; ++i)
    printf("%d\n", wts8width(argv[i], (int)strlen(argv[i])));

  return 0;
}
#endif