/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the MIT license as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 */

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

#ifdef WINDOWS
#include <time.h>
#include <WinSock2.h>
#include <sysinfoapi.h>
#else
#include <unistd.h>
#include <termios.h>
#include <errno.h>
#endif

#include <toolsdef.h>

int64_t atomic_add_fetch_64(int64_t volatile* ptr, int64_t val) {
#ifdef WINDOWS
    return InterlockedExchangeAdd64((int64_t volatile*)(ptr), (int64_t)(val)) + (int64_t)(val);
#elif defined(_TD_NINGSI_60)
    return __sync_add_and_fetch((ptr), (val));
#else
    return __atomic_add_fetch((ptr), (val), __ATOMIC_SEQ_CST);
#endif
}

FORCE_INLINE int32_t toolsGetNumberOfCores() {
#ifdef WINDOWS
    SYSTEM_INFO info;
    GetSystemInfo(&info);
    return (int32_t)info.dwNumberOfProcessors;
#else
    return (int32_t)sysconf(_SC_NPROCESSORS_ONLN);
#endif
}

void errorWrongValue(char *program, char *wrong_arg, char *wrong_value) {
    fprintf(stderr, "%s %s: %s is an invalid value\n",
            program, wrong_arg, wrong_value);
    fprintf(stderr, "Try `%s --help' or `%s --usage' for more "
            "information.\n", program, program);
}

void errorPrintReqArg(char *program, char *wrong_arg) {
    fprintf(stderr,
            "%s: option requires an argument -- '%s'\n",
            program, wrong_arg);
    fprintf(stderr,
            "Try `%s --help' or `%s --usage' for more "
            "information.\n", program, program);
}

void errorPrintReqArg2(char *program, char *wrong_arg) {
    fprintf(stderr,
            "%s: option requires a number argument '-%s'\n",
            program, wrong_arg);
    fprintf(stderr,
            "Try `%s --help' or `%s --usage' for more information.\n",
            program, program);
}

void errorPrintReqArg3(char *program, char *wrong_arg) {
    fprintf(stderr,
            "%s: option '%s' requires an argument\n",
            program, wrong_arg);
    fprintf(stderr,
            "Try `taosdump --help' or `taosdump --usage' for more "
            "information.\n");
}

int setConsoleEcho(bool on) {
#if defined(WINDOWS)
    HANDLE hStdin = GetStdHandle(STD_INPUT_HANDLE);
    DWORD  mode = 0;
    GetConsoleMode(hStdin, &mode);
    if (on) {
        mode |= ENABLE_ECHO_INPUT;
    } else {
        mode &= ~ENABLE_ECHO_INPUT;
    }
    SetConsoleMode(hStdin, mode);

#else
#define ECHOFLAGS (ECHO | ECHOE | ECHOK | ECHONL)
    int err;
    struct termios term;

    if (tcgetattr(STDIN_FILENO, &term) == -1) {
        perror("Cannot get the attribution of the terminal");
        return -1;
    }

    if (on)
        term.c_lflag |= ECHOFLAGS;
    else
        term.c_lflag &= ~ECHOFLAGS;

    err = tcsetattr(STDIN_FILENO, TCSAFLUSH, &term);
    if (err == -1 || err == EINTR) {
        perror("Cannot set the attribution of the terminal");
        return -1;
    }

#endif
    return 0;
}