RIOT/tests/test_struct_tm_utility/main.c

/**
 * @ingroup  tests
 * @{
 * @file
 * @brief    Test the `struct tm` helpers in "tm.h" of the module "timex".
 * @author   René Kijewski <rene.kijewski@fu-berlin.de>
 * @}
 */

#include <stdio.h>

#include "shell.h"
#include "posix_io.h"
#include "board_uart0.h"
#include "tm.h"

#define SHELL_BUFSIZE (UART0_BUFSIZE)

static const char MON_NAMES[12][3] = {
    "JAN", "FEB", "MAR", "APR",
    "MAY", "JUN", "JUL", "AUG",
    "SEP", "OCT", "NOV", "DEC",
};
static const char DAY_NAMES[7][3] = {
    "SUN", "MON", "TUE", "WED", "THU", "FRI", "SAT"
};
static const char BOOL_NAMES[2][3] = { "NO", "YES" };

static void cmd_days_in(int argc, char **argv)
{
    int mon;
    if ((argc != 2) || (sscanf(argv[1], "%d", &mon) != 1) || (mon < 1) || (mon > 12)) {
        printf("Usage: %s <Month[1..12]>\n", argv[0]);
    }
    else {
        printf("There are %d days in %.3s in common years.\n",
               TM_MON_DAYS[mon - 1], MON_NAMES[mon - 1]);
    }
}

static void cmd_leap_year(int argc, char **argv)
{
    int year;
    if ((argc != 2) || (sscanf(argv[1], "%d", &year) != 1)) {
        printf("Usage: %s <Year>\n", argv[0]);
    }
    else {
        printf("Was %d a leap year? %.3s.\n",
               year, BOOL_NAMES[tm_is_leap_year(year)]);
    }
}

static void cmd_doomsday(int argc, char **argv)
{
    int year;
    if ((argc != 2) || (sscanf(argv[1], "%d", &year) != 1)) {
        printf("Usage: %s <Year>\n", argv[0]);
    }
    else {
        printf("What weekday was MAR 0 of %d? %.3s.\n",
               year, DAY_NAMES[tm_doomsday(year) % 7]);
    }
}

static void cmd_day(int argc, char **argv)
{
    int year, mon, day;
    if ((argc != 4) || (sscanf(argv[1], "%d", &year) != 1)
                    || (sscanf(argv[2], "%d", &mon) != 1)
                    || (sscanf(argv[3], "%d", &day) != 1)) {
        printf("Usage: %s <Year> <Month[1..12]> <Day[1..31]>\n", argv[0]);
    }
    else {
        if (!tm_is_valid_date(year, mon - 1, day)) {
            puts("The supplied date is invalid, but no error should occur.");
        }

        int wday, yday;
        tm_get_wyday(year, mon - 1, day, &wday, &yday);
        printf("What weekday was %04d-%02d-%02d? The %d(th) day of the year was a %.3s.\n",
               year, mon, day, yday + 1, DAY_NAMES[wday]);
    }
}

static const shell_command_t shell_commands[] = {
    { "days_in", "Tells you the number of days in a month.", cmd_days_in },
    { "leap_year", "Tells you if a supplied year is a leap year.", cmd_leap_year },
    { "doomsday", "Tells you the wday Doomsday of the supplied year.", cmd_doomsday },
    { "day", "Tells you the day of the supplied date.", cmd_day },
    { NULL, NULL, NULL }
};

static int shell_readc(void)
{
    char c;
    int result = posix_read(uart0_handler_pid, &c, 1);
    if (result != 1) {
        return -1;
    }
    return (unsigned char) c;
}

static void shell_putchar(int c)
{
    putchar(c);
}

int main(void)
{
    board_uart0_init();
    posix_open(uart0_handler_pid, 0);

    shell_t shell;
    shell_init(&shell, shell_commands, SHELL_BUFSIZE, shell_readc, shell_putchar);

    puts("`struct tm` utility shell.");
    shell_run(&shell);

    return 0;
}