sanityCheck.h

#pragma once

#include <fstream>

#include "logger.h"

static bool isValidIpAddress(char* ipAddress)
{
    int a, b, c, d;
    int r = sscanf(ipAddress, "%d.%d.%d.%d", &a, &b, &c, &d);
    if (r == 4 && (0 <= a && a < 256)
        && (0 <= b && b < 256)
        && (0 <= c && c < 256)
        && (0 <= d && d < 256))
    {
        return true;
    }
    return false;
}

static void sanityCheckTxType(uint16_t type)
{
    if (type > 1)
    {
        LOG("unknown input type %u\n", type);
        exit(1);
    }
}

static void sanityCheckSeed(const char* privKey)
{
	if (privKey == NULL)
	{
		LOG("privKey is null\n");
		exit(1);
	}
    if (strlen(privKey) != 55)
    {
        LOG("Seed must be 55-char length\n");
        exit(1);
    }
    if (memcmp(privKey, DEFAULT_SEED, 55) == 0)
    {
        LOG("WARNING: You are using default seed\n");
    }
}

static void sanityCheckNode(char* ip, int port)
{
	if (port < 1024)
	{
		LOG("invalid port number\n");
		exit(1);
	}
	if (ip == NULL)
	{
		LOG("invalid ipv4 address\n");
		exit(1);
	}
	if (!isValidIpAddress(ip))
	{
		LOG("invalid ipv4 address: %s\n", ip);
		exit(1);		
	}
}

static void sanityCheckAmountTransferAsset(long long amount)
{
    if (amount <= 0)
    {
        LOG("Invalid amount\n");
        exit(1);
    }
}

static void sanityCheckIdentity(const char* identity)
{
	if (identity == NULL)
    {
		LOG("identity is null\n");
		exit(1);
	}
	for (int i = 0; i < 60; i++)
    {
		if (identity[i] < 'A' || identity[i] > 'Z')
        {
			LOG("invalid identity at position %d (%c)", i, identity[i]);
			exit(1);
		}
	}
    if (!checkSumIdentity(identity))
    {
        LOG("Identity checksum failed: %s\n", identity);
        exit(1);
    }
}

static void sanityCheckTxHash(char* txHash)
{
    if (txHash == NULL)
    {
        LOG("identity is null\n");
        exit(1);
    }
    for (int i = 0; i < 60; i++)
    {
        if (txHash[i] < 'a' || txHash[i] > 'z')
        {
            LOG("invalid identity at position %d (%c)", i, txHash[i]);
            exit(1);
        }
    }
}

static void sanityCheckTxAmount(int64_t amount)
{
	if (amount < 0)
    {
		LOG("invalid amount %lld\n", amount);
		exit(1);
	}
}

static void sanityCheckExtraDataSize(int data_size)
{
	if (data_size > 1024)
    {
		LOG("Invalid data size %d, data size must be smaller than 1024\n", data_size);
		exit(1);
	}
}

static void sanityCheckRawPacketSize(int data_size)
{
	if (data_size > 1024)
    {
		LOG("Invalid data size %d, data size must be smaller than 1024\n", data_size);
		exit(1);
	}
}

static void sanityFileExist (const std::string name)
{
    std::ifstream f(name.c_str());
    if (!f.good())
    {
        LOG("File %s does not exist\n", name.c_str());
        exit(1);
    }
}

static void sanityCheckSpecialCommand(int cmd)
{
    if (cmd == -1)
    {
        LOG("Invalid special command");
        exit(1);
    }
}

static void sanityCheckNumberOfUnit(long long unit)
{
    if (unit <= 0)
    {
        LOG("Invalid number of unit\n");
        exit(1);
    }
}

static void sanityCheckNumberOfDecimal(char unit)
{
    if (unit < 0)
    {
        LOG("Invalid number of decimal\n");
        exit(1);
    }
}

static void sanityCheckValidString(const char* str)
{
    if (str == nullptr)
    {
        LOG("Invalid string\n");
        exit(1);
    }
}

static void sanityCheckUnitofMeasurement(const char* str)
{
    if (str == nullptr)
    {
        LOG("Invalid string\n");
        exit(1);
    }
    if (strlen(str) != 7)
    {
        LOG("UnitofMeasurement length must be 7, they are powers of the corresponding SI base units:\n");
        LOG("AMPERE\n");
        LOG("CANDELA\n");
        LOG("KELVIN\n");
        LOG("KILOGRAM\n");
        LOG("METER\n");
        LOG("MOLE\n");
        LOG("SECOND\n");
        exit(1);
    }
}

static void sanityCheckMainAuxStatus(const char* str)
{
    if (str == nullptr)
    {
        LOG("Invalid MAIN/AUX string\n");
        exit(1);
    }
    int len = strlen(str);
    if (len == 4)
    {
        if (memcmp(str, "MAIN", 4) != 0)
        {
            LOG("Invalid MAIN/AUX string. Expected (MAIN/AUX), have %s\n", str);
            exit(1);
        }
    } 
    else if (len == 3)
    {
        if (memcmp(str, "AUX", 3) != 0)
        {
            LOG("Invalid MAIN/AUX string. Expected (MAIN/AUX), have %s\n", str);
            exit(1);
        }
    }
    else
    {
        LOG("Invalid MAIN/AUX string. Expected (MAIN/AUX), have %s\n", str);
        exit(1);
    }
}

static void checkValidEpoch(int epoch)
{
    if (epoch >= 0 &&  epoch < 1000)
    {
        return;
    }
    LOG("Invalid epoch number\n");
    exit(1);
}

static void checkValidSolutionThreshold(int thres)
{
    if (thres >= 0)
    {
        return;
    }
    LOG("Invalid solution threshold\n");
    exit(1);
}