//
// Copyright 2017 The ANGLE Project Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
//

#include "compiler/translator/HashNames.h"

#include "compiler/translator/ImmutableString.h"
#include "compiler/translator/ImmutableStringBuilder.h"
#include "compiler/translator/IntermNode.h"
#include "compiler/translator/Symbol.h"

namespace sh
{

namespace
{
constexpr const ImmutableString kHashedNamePrefix("webgl_");

ImmutableString HashName(const ImmutableString &name, ShHashFunction64 hashFunction)
{
    ASSERT(!name.empty());
    ASSERT(hashFunction);
    khronos_uint64_t number = (*hashFunction)(name.data(), name.length());

    // Build the hashed name in place.
    static const unsigned int kHexStrMaxLength = sizeof(number) * 2;
    static const size_t kHashedNameMaxLength   = kHashedNamePrefix.length() + kHexStrMaxLength;

    ImmutableStringBuilder hashedName(kHashedNameMaxLength);
    hashedName << kHashedNamePrefix;

    hashedName.appendHex(number);

    return hashedName;
}

void AddToNameMapIfNotMapped(const ImmutableString &name,
                             const ImmutableString &hashedName,
                             NameMap *nameMap)
{
    if (nameMap)
    {
        NameMap::const_iterator it = nameMap->find(name.data());
        if (it != nameMap->end())
        {
            return;
        }
        nameMap->insert(name.data(), hashedName.data());
    }
}

}  // anonymous namespace

ImmutableString HashName(const ImmutableString &name,
                         char prefix,
                         ShHashFunction64 hashFunction,
                         NameMap *nameMap)
{
    if (hashFunction == nullptr)
    {
        size_t kPrefixLength = 2;
        if (!prefix || name.length() + kPrefixLength > kESSLMaxIdentifierLength)
        {
            // If the identifier length is already close to the limit, we can't prefix it. This is
            // not a problem since there are no builtins or ANGLE's internal variables that would
            // have as long names and could conflict.
            return name;
        }
        ImmutableString res = BuildConcatenatedImmutableString('_', prefix, name);
        AddToNameMapIfNotMapped(name, res, nameMap);
        return res;
    }

    // Has a hash function. If this name has already been mapped, reuse the
    // existing entry so the same source identifier always maps to the same
    // output identifier.
    if (nameMap)
    {
        NameMap::const_iterator it = nameMap->find(name.data());
        if (it != nameMap->end())
        {
            return ImmutableString(it->second);
        }
    }

    const ImmutableString baseHashedName = HashName(name, hashFunction);
    if (!nameMap)
    {
        return baseHashedName;
    }

    // Ensure the chosen output identifier is not already assigned to a
    // different source identifier. If it is, append a numeric suffix until a
    // unique name is found.
    ImmutableString hashedName = baseHashedName;
    unsigned int suffix        = 0u;
    while (nameMap->containsHashedName(hashedName.data()))
    {
        ++suffix;
        hashedName = BuildConcatenatedImmutableString(baseHashedName, '_', suffix);
    }
    nameMap->insert(name.data(), hashedName.data());
    return hashedName;
}

ImmutableString HashName(const TSymbol *symbol,
                         char prefix,
                         ShHashFunction64 hashFunction,
                         NameMap *nameMap)
{
    if (symbol->symbolType() == SymbolType::Empty)
    {
        return kEmptyImmutableString;
    }
    if (symbol->symbolType() == SymbolType::AngleInternal ||
        symbol->symbolType() == SymbolType::BuiltIn)
    {
        return symbol->name();
    }
    return HashName(symbol->name(), prefix, hashFunction, nameMap);
}

}  // namespace sh
