/*
 * Copyright (C) 2007-2021 Apple Inc. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1.  Redistributions of source code must retain the above copyright
 *     notice, this list of conditions and the following disclaimer.
 * 2.  Redistributions in binary form must reproduce the above copyright
 *     notice, this list of conditions and the following disclaimer in the
 *     documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' AND ANY
 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#pragma once

#include <array>
#include <concepts>
#include <type_traits>
#include <wtf/Assertions.h>
#include <wtf/text/Latin1Character.h>

// The behavior of many of the functions in the <ctype.h> header is dependent
// on the current locale. But in the WebKit project, all uses of those functions
// are in code processing something that's not locale-specific. These equivalents
// for some of the <ctype.h> functions are named more explicitly, not dependent
// on the C library locale, and we should also optimize them as needed.

// All functions return false or leave the character unchanged if passed a character
// that is outside the range 0-7F. So they can be used on Unicode strings or
// characters if the intent is to do processing only if the character is ASCII.

namespace WTF {

template<typename T>
concept Character = std::convertible_to<std::remove_cv_t<T>, char>;

template<Character CharacterType> constexpr bool isASCII(CharacterType);
template<Character CharacterType> constexpr bool isASCIIAlpha(CharacterType);
template<Character CharacterType> constexpr bool isASCIIAlphanumeric(CharacterType);
template<Character CharacterType> constexpr bool isASCIIBinaryDigit(CharacterType);
template<Character CharacterType> constexpr bool isASCIIDigit(CharacterType);
template<Character CharacterType> constexpr bool isASCIIDigitOrPunctuation(CharacterType);
template<Character CharacterType> constexpr bool isASCIIHexDigit(CharacterType);
template<Character CharacterType> constexpr bool isASCIILower(CharacterType);
template<Character CharacterType> constexpr bool isASCIIOctalDigit(CharacterType);
template<Character CharacterType> constexpr bool isASCIIPrintable(CharacterType);
template<Character CharacterType> constexpr bool isASCIIGraphic(CharacterType);
template<Character CharacterType> constexpr bool isTabOrSpace(CharacterType);
template<Character CharacterType> constexpr bool isASCIIWhitespace(CharacterType);
template<Character CharacterType> constexpr bool isUnicodeCompatibleASCIIWhitespace(CharacterType);
template<Character CharacterType> constexpr bool isASCIIUpper(CharacterType);

// Inverse of isASCIIWhitespace for predicates
template<Character CharacterType> constexpr bool isNotASCIIWhitespace(CharacterType);

template<Character CharacterType> CharacterType toASCIILower(CharacterType);
template<Character CharacterType> CharacterType toASCIIUpper(CharacterType);

template<Character CharacterType> uint8_t toASCIIHexValue(CharacterType);
template<Character CharacterType> uint8_t toASCIIHexValue(CharacterType firstCharacter, CharacterType secondCharacter);

constexpr char lowerNibbleToASCIIHexDigit(uint8_t);
constexpr char upperNibbleToASCIIHexDigit(uint8_t);
constexpr char lowerNibbleToLowercaseASCIIHexDigit(uint8_t);
constexpr char upperNibbleToLowercaseASCIIHexDigit(uint8_t);

template<Character CharacterType> constexpr bool isASCIIAlphaCaselessEqual(CharacterType, char expectedASCIILowercaseLetter);

// The toASCIILowerUnchecked function can be used for comparing any input character
// to a lowercase English character. The isASCIIAlphaCaselessEqual function should
// be used for regular comparison of ASCII alpha characters, but switch statements
// in the CSS tokenizer, for example, instead make direct use toASCIILowerUnchecked.
template<Character CharacterType> constexpr CharacterType toASCIILowerUnchecked(CharacterType);

extern WTF_EXPORT_PRIVATE const std::array<uint8_t, 256> asciiCaseFoldTable;

template<Character CharacterType> constexpr bool isASCII(CharacterType character)
{
    return !(character & ~0x7F);
}

template<Character CharacterType> constexpr bool isASCIILower(CharacterType character)
{
    return character >= 'a' && character <= 'z';
}

template<Character CharacterType> constexpr CharacterType toASCIILowerUnchecked(CharacterType character)
{
    // This function can be used for comparing any input character
    // to a lowercase English character. The isASCIIAlphaCaselessEqual
    // below should be used for regular comparison of ASCII alpha
    // characters, but switch statements in CSS tokenizer instead make
    // direct use of this function.
    return character | 0x20;
}

template<Character CharacterType> constexpr bool isASCIIAlpha(CharacterType character)
{
    return isASCIILower(toASCIILowerUnchecked(character));
}

template<Character CharacterType> constexpr bool isASCIIDigit(CharacterType character)
{
    return character >= '0' && character <= '9';
}

template<Character CharacterType> constexpr bool isASCIIAlphanumeric(CharacterType character)
{
    return isASCIIDigit(character) || isASCIIAlpha(character);
}

template<Character CharacterType> constexpr bool isASCIIHexDigit(CharacterType character)
{
    return isASCIIDigit(character) || (toASCIILowerUnchecked(character) >= 'a' && toASCIILowerUnchecked(character) <= 'f');
}

template<Character CharacterType> constexpr bool isASCIIBinaryDigit(CharacterType character)
{
    return character == '0' || character == '1';
}

template<Character CharacterType> constexpr bool isASCIIOctalDigit(CharacterType character)
{
    return character >= '0' && character <= '7';
}

template<Character CharacterType> constexpr bool isASCIIPrintable(CharacterType character)
{
    return character >= ' ' && character <= '~';
}

template<Character CharacterType> constexpr bool isASCIIGraphic(CharacterType character)
{
    return character >= '!' && character <= '~';
}

template<Character CharacterType> constexpr bool isTabOrSpace(CharacterType character)
{
    return character == ' ' || character == '\t';
}

// Infra's "ASCII whitespace" <https://infra.spec.whatwg.org/#ascii-whitespace>
template<Character CharacterType> constexpr bool isASCIIWhitespace(CharacterType character)
{
    return character == ' ' || character == '\n' || character == '\t' || character == '\r' || character == '\f';
}

template<Character CharacterType> constexpr bool isASCIIWhitespaceWithoutFF(CharacterType character)
{
    // This is different from isASCIIWhitespace: JSON/HTTP/XML do not accept \f as a whitespace.
    // ECMA-404 specifies the following:
    // > Whitespace is any sequence of one or more of the following code points:
    // > character tabulation (U+0009), line feed (U+000A), carriage return (U+000D), and space (U+0020).
    //
    // This matches HTTP whitespace:
    // https://fetch.spec.whatwg.org/#http-whitespace-byte
    //
    // And XML whitespace:
    // https://www.w3.org/TR/2008/REC-xml-20081126/#NT-S
    return character == ' ' || character == '\n' || character == '\t' || character == '\r';
}

template<Character CharacterType> constexpr bool isUnicodeCompatibleASCIIWhitespace(CharacterType character)
{
    return isASCIIWhitespace(character) || character == '\v';
}

template<Character CharacterType> constexpr bool isASCIIUpper(CharacterType character)
{
    return character >= 'A' && character <= 'Z';
}

template<Character CharacterType> constexpr bool isNotASCIIWhitespace(CharacterType character)
{
    return !isASCIIWhitespace(character);
}

template<Character CharacterType> inline CharacterType toASCIILower(CharacterType character)
{
    return character | (isASCIIUpper(character) << 5);
}

template<> inline char toASCIILower(char character)
{
    return static_cast<char>(asciiCaseFoldTable[static_cast<uint8_t>(character)]);
}

template<> inline Latin1Character toASCIILower(Latin1Character character)
{
    return asciiCaseFoldTable[character];
}

template<Character CharacterType> inline CharacterType toASCIIUpper(CharacterType character)
{
    return character & ~(isASCIILower(character) << 5);
}

template<Character CharacterType> inline uint8_t toASCIIHexValue(CharacterType character)
{
    ASSERT(isASCIIHexDigit(character));
    return character < 'A' ? character - '0' : (character - 'A' + 10) & 0xF;
}

template<Character CharacterType> inline uint8_t toASCIIHexValue(CharacterType firstCharacter, CharacterType secondCharacter)
{
    return toASCIIHexValue(firstCharacter) << 4 | toASCIIHexValue(secondCharacter);
}

constexpr char lowerNibbleToASCIIHexDigit(uint8_t value)
{
    uint8_t nibble = value & 0xF;
    return nibble + (nibble < 10 ? '0' : 'A' - 10);
}

constexpr char upperNibbleToASCIIHexDigit(uint8_t value)
{
    uint8_t nibble = value >> 4;
    return nibble + (nibble < 10 ? '0' : 'A' - 10);
}

constexpr char lowerNibbleToLowercaseASCIIHexDigit(uint8_t value)
{
    uint8_t nibble = value & 0xF;
    return nibble + (nibble < 10 ? '0' : 'a' - 10);
}

constexpr char upperNibbleToLowercaseASCIIHexDigit(uint8_t value)
{
    uint8_t nibble = value >> 4;
    return nibble + (nibble < 10 ? '0' : 'a' - 10);
}

template<Character CharacterType> constexpr bool isASCIIAlphaCaselessEqual(CharacterType inputCharacter, char expectedASCIILowercaseLetter)
{
    // Name of this argument says this must be a lowercase letter, but it can actually be:
    //   - a lowercase letter
    //   - a numeric digit
    //   - a space
    //   - punctuation in the range 0x21-0x3F, including "-", "/", and "+"
    // It cannot be:
    //   - an uppercase letter
    //   - a non-ASCII character
    //   - other punctuation, such as underscore and backslash
    //   - a control character such as "\n"
    // FIXME: Would be nice to make both the function name and expectedASCIILowercaseLetter argument name clearer.
    ASSERT(toASCIILowerUnchecked(expectedASCIILowercaseLetter) == expectedASCIILowercaseLetter);
    if (toASCIILowerUnchecked(inputCharacter) == static_cast<CharacterType>(expectedASCIILowercaseLetter)) [[likely]]
        return true;
    return false;
}

template<Character CharacterType> constexpr bool isASCIIDigitOrPunctuation(CharacterType character)
{
    return (character >= '!' && character <= '@') || (character >= '[' && character <= '`') || (character >= '{' && character <= '~');
}

}

using WTF::isASCII;
using WTF::isASCIIAlpha;
using WTF::isASCIIAlphaCaselessEqual;
using WTF::isASCIIAlphanumeric;
using WTF::isASCIIBinaryDigit;
using WTF::isASCIIDigit;
using WTF::isASCIIDigitOrPunctuation;
using WTF::isASCIIHexDigit;
using WTF::isASCIILower;
using WTF::isASCIIOctalDigit;
using WTF::isASCIIPrintable;
using WTF::isASCIIUpper;
using WTF::isASCIIWhitespace;
using WTF::isASCIIWhitespaceWithoutFF;
using WTF::isNotASCIIWhitespace;
using WTF::isTabOrSpace;
using WTF::isUnicodeCompatibleASCIIWhitespace;
using WTF::lowerNibbleToASCIIHexDigit;
using WTF::lowerNibbleToLowercaseASCIIHexDigit;
using WTF::toASCIIHexValue;
using WTF::toASCIILower;
using WTF::toASCIILowerUnchecked;
using WTF::toASCIIUpper;
using WTF::upperNibbleToASCIIHexDigit;
using WTF::upperNibbleToLowercaseASCIIHexDigit;