// This file is part of the Luau programming language and is licensed under MIT License; see LICENSE.txt for details
#include "Luau/Module.h"

#include "Luau/Clone.h"
#include "Luau/Common.h"
#include "Luau/ConstraintGraphBuilder.h"
#include "Luau/Normalize.h"
#include "Luau/RecursionCounter.h"
#include "Luau/Scope.h"
#include "Luau/TypeInfer.h"
#include "Luau/TypePack.h"
#include "Luau/TypeVar.h"
#include "Luau/VisitTypeVar.h"

#include <algorithm>

LUAU_FASTFLAG(LuauLowerBoundsCalculation);
LUAU_FASTFLAG(LuauNormalizeFlagIsConservative);
LUAU_FASTFLAG(DebugLuauDeferredConstraintResolution);
LUAU_FASTFLAGVARIABLE(LuauForceExportSurfacesToBeNormal, false);

namespace Luau
{

static bool contains(Position pos, Comment comment)
{
    if (comment.location.contains(pos))
        return true;
    else if (comment.type == Lexeme::BrokenComment &&
             comment.location.begin <= pos) // Broken comments are broken specifically because they don't have an end
        return true;
    else if (comment.type == Lexeme::Comment && comment.location.end == pos)
        return true;
    else
        return false;
}

bool isWithinComment(const SourceModule& sourceModule, Position pos)
{
    auto iter = std::lower_bound(sourceModule.commentLocations.begin(), sourceModule.commentLocations.end(),
        Comment{Lexeme::Comment, Location{pos, pos}}, [](const Comment& a, const Comment& b) {
            return a.location.end < b.location.end;
        });

    if (iter == sourceModule.commentLocations.end())
        return false;

    if (contains(pos, *iter))
        return true;

    // Due to the nature of std::lower_bound, it is possible that iter points at a comment that ends
    // at pos.  We'll try the next comment, if it exists.
    ++iter;
    if (iter == sourceModule.commentLocations.end())
        return false;

    return contains(pos, *iter);
}

struct ForceNormal : TypeVarOnceVisitor
{
    const TypeArena* typeArena = nullptr;

    ForceNormal(const TypeArena* typeArena)
        : typeArena(typeArena)
    {
    }

    bool visit(TypeId ty) override
    {
        if (ty->owningArena != typeArena)
            return false;

        asMutable(ty)->normal = true;
        return true;
    }

    bool visit(TypeId ty, const FreeTypeVar& ftv) override
    {
        visit(ty);
        return true;
    }

    bool visit(TypePackId tp, const FreeTypePack& ftp) override
    {
        return true;
    }
};

Module::~Module()
{
    unfreeze(interfaceTypes);
    unfreeze(internalTypes);
}

void Module::clonePublicInterface(InternalErrorReporter& ice)
{
    LUAU_ASSERT(interfaceTypes.typeVars.empty());
    LUAU_ASSERT(interfaceTypes.typePacks.empty());

    CloneState cloneState;

    ScopePtr moduleScope = FFlag::DebugLuauDeferredConstraintResolution ? nullptr : getModuleScope();
    Scope2* moduleScope2 = FFlag::DebugLuauDeferredConstraintResolution ? getModuleScope2() : nullptr;

    TypePackId returnType = FFlag::DebugLuauDeferredConstraintResolution ? moduleScope2->returnType : moduleScope->returnType;
    std::optional<TypePackId> varargPack = FFlag::DebugLuauDeferredConstraintResolution ? std::nullopt : moduleScope->varargPack;
    std::unordered_map<Name, TypeFun>* exportedTypeBindings =
        FFlag::DebugLuauDeferredConstraintResolution ? nullptr : &moduleScope->exportedTypeBindings;

    returnType = clone(returnType, interfaceTypes, cloneState);

    if (moduleScope)
    {
        moduleScope->returnType = returnType;
        if (varargPack)
        {
            varargPack = clone(*varargPack, interfaceTypes, cloneState);
            moduleScope->varargPack = varargPack;
        }
    }
    else
    {
        LUAU_ASSERT(moduleScope2);
        moduleScope2->returnType = returnType; // TODO varargPack
    }

    ForceNormal forceNormal{&interfaceTypes};

    if (FFlag::LuauLowerBoundsCalculation)
    {
        normalize(returnType, interfaceTypes, ice);
        if (FFlag::LuauForceExportSurfacesToBeNormal)
            forceNormal.traverse(returnType);
        if (varargPack)
        {
            normalize(*varargPack, interfaceTypes, ice);
            if (FFlag::LuauForceExportSurfacesToBeNormal)
                forceNormal.traverse(*varargPack);
        }
    }

    if (exportedTypeBindings)
    {
        for (auto& [name, tf] : *exportedTypeBindings)
        {
            tf = clone(tf, interfaceTypes, cloneState);
            if (FFlag::LuauLowerBoundsCalculation)
            {
                normalize(tf.type, interfaceTypes, ice);

                if (FFlag::LuauNormalizeFlagIsConservative)
                {
                    // We're about to freeze the memory.  We know that the flag is conservative by design.  Cyclic tables
                    // won't be marked normal.  If the types aren't normal by now, they never will be.
                    forceNormal.traverse(tf.type);
                    for (GenericTypeDefinition param : tf.typeParams)
                    {
                        forceNormal.traverse(param.ty);

                        if (param.defaultValue)
                        {
                            normalize(*param.defaultValue, interfaceTypes, ice);
                            forceNormal.traverse(*param.defaultValue);
                        }
                    }
                }
            }
        }
    }

    for (TypeId ty : returnType)
    {
        if (get<GenericTypeVar>(follow(ty)))
        {
            auto t = asMutable(ty);
            t->ty = AnyTypeVar{};
            t->normal = true;
        }
    }

    for (auto& [name, ty] : declaredGlobals)
    {
        ty = clone(ty, interfaceTypes, cloneState);
        if (FFlag::LuauLowerBoundsCalculation)
        {
            normalize(ty, interfaceTypes, ice);

            if (FFlag::LuauForceExportSurfacesToBeNormal)
                forceNormal.traverse(ty);
        }
    }

    freeze(internalTypes);
    freeze(interfaceTypes);
}

ScopePtr Module::getModuleScope() const
{
    LUAU_ASSERT(!scopes.empty());
    return scopes.front().second;
}

Scope2* Module::getModuleScope2() const
{
    LUAU_ASSERT(!scope2s.empty());
    return scope2s.front().second.get();
}

} // namespace Luau