/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright 2025 Oxide Computer Company
 * Copyright 2025 Bill Sommerfeld
 */

/*
 * Tests for modern wcsftime (XPG5 and later) and wcsftime_l.   The libc tests
 * depend on locale/ar, locale/de, locale/en, and locale/ja. We limit
 * ourselves to these locales, plus C.UTF-8.
 */

#include <err.h>
#include <stdlib.h>
#include <xlocale.h>
#include <locale.h>
#include <sys/sysmacros.h>
#include <sys/debug.h>
#include <stdbool.h>
#include <string.h>


/*
 * We're just testing that the desired locale reaches the underlying
 * formatter so we only look at one attribute: the full month name.
 */
static const struct test_locale {
        const char *name;
        const char *monthname;
} locales[] = {
        { "C.UTF-8", "December"},
        {"ja_JP.UTF-8", "12月"},
        {"de_DE.UTF-8", "Dezember"},
        {"en_US.UTF-8", "December"},
};

struct tm sample_tm = { .tm_mon = 11 };

#define WCSSIZE 100
#define CSIZE 200

/*
 * Test that the correct decimal point is recognized.
 */
bool
test_locale(const char *name, const char *monthname)
{
        bool result_ok = true;
        wchar_t wfmt[WCSSIZE];
        wchar_t wcs[WCSSIZE];
        char cstring[CSIZE];

        size_t len;

        locale_t loc = newlocale(LC_ALL_MASK, name, NULL);

        if (loc == NULL) {
                err(EXIT_FAILURE, "INTERNAL TEST FAILURE: failed to "
                    "construct locale %s", name);
        }

        if (mbstowcs_l(wfmt, "%B", WCSSIZE, loc) == (size_t)-1) {
                err(EXIT_FAILURE, "INTERNAL TEST FAILURE: failed to "
                    "construct wide char format string");
        }

        len = wcsftime_l(wcs, WCSSIZE, wfmt, &sample_tm, loc);
        if (len == (size_t)-1) {
                warn("wcsftime_l returned -1");
                result_ok = false;
        }

        if (wcstombs_l(cstring, wcs, CSIZE, loc) == (size_t)-1) {
                err(EXIT_FAILURE, "INTERNAL TEST FAILURE: failed to "
                    "convert wide char string back to multibyte string");
        }

        if (strcmp(cstring, monthname)) {
                warn("Wrong monthname for locale %s month %d: "
                    "got %s expected %s", name, sample_tm.tm_mon+1,
                    cstring, monthname);
                result_ok = false;
        }

        (void) memset(wcs, 0, sizeof (wcs));
        (void) memset(cstring, 0, sizeof (cstring));

        if (uselocale(loc) == NULL) {
                err(EXIT_FAILURE, "INTERNAL TEST FAILURE: failed to "
                    "set locale %s", name);
        }

        len = wcsftime(wcs, WCSSIZE, wfmt, &sample_tm);
        if (len == (size_t)-1) {
                warn("wcsftime_l returned -1");
                result_ok = false;
        }

        if (wcstombs(cstring, wcs, CSIZE) == (size_t)-1) {
                err(EXIT_FAILURE, "INTERNAL TEST FAILURE: failed to "
                    "convert wide char string back to multibyte string");
        }

        if (strcmp(cstring, monthname)) {
                warn("Wrong monthname for locale %s month %d: "
                    "got %s expected %s", name, sample_tm.tm_mon+1,
                    cstring, monthname);
                result_ok = false;
        }

        return (result_ok);
}


int
main(void)
{
        int ret = EXIT_SUCCESS;

        for (size_t i = 0; i < ARRAY_SIZE(locales); i++) {
                if (!test_locale(locales[i].name, locales[i].monthname))
                        ret = EXIT_FAILURE;
        }

        if (ret == EXIT_SUCCESS) {
                (void) printf("All tests completed successfully\n");
        }

        return (ret);
}