2022-06-03 13:32:20 -07:00
|
|
|
// This file is part of the Luau programming language and is licensed under MIT License; see LICENSE.txt for details
|
|
|
|
|
2022-09-01 16:00:14 -07:00
|
|
|
#include "Luau/Anyification.h"
|
2022-08-04 14:27:28 -07:00
|
|
|
#include "Luau/ApplyTypeFunction.h"
|
2022-11-18 10:45:14 -08:00
|
|
|
#include "Luau/Clone.h"
|
2023-03-24 10:34:14 -07:00
|
|
|
#include "Luau/Common.h"
|
2022-06-03 13:32:20 -07:00
|
|
|
#include "Luau/ConstraintSolver.h"
|
2022-10-21 10:33:43 -07:00
|
|
|
#include "Luau/DcrLogger.h"
|
2022-06-03 13:32:20 -07:00
|
|
|
#include "Luau/Instantiation.h"
|
2022-06-23 18:44:07 -07:00
|
|
|
#include "Luau/Location.h"
|
2022-10-21 10:33:43 -07:00
|
|
|
#include "Luau/Metamethods.h"
|
2022-09-01 16:00:14 -07:00
|
|
|
#include "Luau/ModuleResolver.h"
|
2022-06-03 13:32:20 -07:00
|
|
|
#include "Luau/Quantify.h"
|
2023-05-19 11:59:59 -07:00
|
|
|
#include "Luau/Simplify.h"
|
2022-06-03 13:32:20 -07:00
|
|
|
#include "Luau/ToString.h"
|
2023-01-03 19:33:19 +02:00
|
|
|
#include "Luau/Type.h"
|
2023-05-19 11:59:59 -07:00
|
|
|
#include "Luau/TypeFamily.h"
|
|
|
|
#include "Luau/TypeUtils.h"
|
2022-06-03 13:32:20 -07:00
|
|
|
#include "Luau/Unifier.h"
|
2023-01-03 19:33:19 +02:00
|
|
|
#include "Luau/VisitType.h"
|
2022-06-03 13:32:20 -07:00
|
|
|
|
|
|
|
LUAU_FASTFLAGVARIABLE(DebugLuauLogSolver, false);
|
|
|
|
|
|
|
|
namespace Luau
|
|
|
|
{
|
|
|
|
|
2023-02-24 10:24:22 -08:00
|
|
|
size_t HashBlockedConstraintId::operator()(const BlockedConstraintId& bci) const
|
|
|
|
{
|
|
|
|
size_t result = 0;
|
|
|
|
|
|
|
|
if (const TypeId* ty = get_if<TypeId>(&bci))
|
|
|
|
result = std::hash<TypeId>()(*ty);
|
|
|
|
else if (const TypePackId* tp = get_if<TypePackId>(&bci))
|
|
|
|
result = std::hash<TypePackId>()(*tp);
|
|
|
|
else if (Constraint const* const* c = get_if<const Constraint*>(&bci))
|
|
|
|
result = std::hash<const Constraint*>()(*c);
|
|
|
|
else
|
|
|
|
LUAU_ASSERT(!"Should be unreachable");
|
|
|
|
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2022-07-28 20:41:13 -07:00
|
|
|
[[maybe_unused]] static void dumpBindings(NotNull<Scope> scope, ToStringOptions& opts)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
|
|
|
for (const auto& [k, v] : scope->bindings)
|
|
|
|
{
|
2022-11-11 00:04:44 +02:00
|
|
|
auto d = toString(v.typeId, opts);
|
|
|
|
printf("\t%s : %s\n", k.c_str(), d.c_str());
|
2022-06-03 13:32:20 -07:00
|
|
|
}
|
|
|
|
|
2022-07-28 20:41:13 -07:00
|
|
|
for (NotNull<Scope> child : scope->children)
|
2022-06-03 13:32:20 -07:00
|
|
|
dumpBindings(child, opts);
|
|
|
|
}
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
static std::pair<std::vector<TypeId>, std::vector<TypePackId>> saturateArguments(TypeArena* arena, NotNull<BuiltinTypes> builtinTypes,
|
2022-09-08 14:44:50 -07:00
|
|
|
const TypeFun& fn, const std::vector<TypeId>& rawTypeArguments, const std::vector<TypePackId>& rawPackArguments)
|
2022-08-04 14:27:28 -07:00
|
|
|
{
|
|
|
|
std::vector<TypeId> saturatedTypeArguments;
|
|
|
|
std::vector<TypeId> extraTypes;
|
|
|
|
std::vector<TypePackId> saturatedPackArguments;
|
|
|
|
|
|
|
|
for (size_t i = 0; i < rawTypeArguments.size(); ++i)
|
|
|
|
{
|
|
|
|
TypeId ty = rawTypeArguments[i];
|
|
|
|
|
|
|
|
if (i < fn.typeParams.size())
|
|
|
|
saturatedTypeArguments.push_back(ty);
|
|
|
|
else
|
|
|
|
extraTypes.push_back(ty);
|
|
|
|
}
|
|
|
|
|
|
|
|
// If we collected extra types, put them in a type pack now. This case is
|
|
|
|
// mutually exclusive with the type pack -> type conversion we do below:
|
|
|
|
// extraTypes will only have elements in it if we have more types than we
|
|
|
|
// have parameter slots for them to go into.
|
2023-05-19 11:59:59 -07:00
|
|
|
if (!extraTypes.empty() && !fn.typePackParams.empty())
|
2022-08-04 14:27:28 -07:00
|
|
|
{
|
|
|
|
saturatedPackArguments.push_back(arena->addTypePack(extraTypes));
|
|
|
|
}
|
|
|
|
|
|
|
|
for (size_t i = 0; i < rawPackArguments.size(); ++i)
|
|
|
|
{
|
|
|
|
TypePackId tp = rawPackArguments[i];
|
|
|
|
|
|
|
|
// If we are short on regular type saturatedTypeArguments and we have a single
|
|
|
|
// element type pack, we can decompose that to the type it contains and
|
|
|
|
// use that as a type parameter.
|
|
|
|
if (saturatedTypeArguments.size() < fn.typeParams.size() && size(tp) == 1 && finite(tp) && first(tp) && saturatedPackArguments.empty())
|
|
|
|
{
|
|
|
|
saturatedTypeArguments.push_back(*first(tp));
|
|
|
|
}
|
2023-05-19 11:59:59 -07:00
|
|
|
else if (saturatedPackArguments.size() < fn.typePackParams.size())
|
2022-08-04 14:27:28 -07:00
|
|
|
{
|
|
|
|
saturatedPackArguments.push_back(tp);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t typesProvided = saturatedTypeArguments.size();
|
|
|
|
size_t typesRequired = fn.typeParams.size();
|
|
|
|
|
|
|
|
size_t packsProvided = saturatedPackArguments.size();
|
|
|
|
size_t packsRequired = fn.typePackParams.size();
|
|
|
|
|
|
|
|
// Extra types should be accumulated in extraTypes, not saturatedTypeArguments. Extra
|
|
|
|
// packs will be accumulated in saturatedPackArguments, so we don't have an
|
|
|
|
// assertion for that.
|
|
|
|
LUAU_ASSERT(typesProvided <= typesRequired);
|
|
|
|
|
|
|
|
// If we didn't provide enough types, but we did provide a type pack, we
|
|
|
|
// don't want to use defaults. The rationale for this is that if the user
|
|
|
|
// provides a pack but doesn't provide enough types, we want to report an
|
|
|
|
// error, rather than simply using the default saturatedTypeArguments, if they exist. If
|
|
|
|
// they did provide enough types, but not enough packs, we of course want to
|
|
|
|
// use the default packs.
|
|
|
|
bool needsDefaults = (typesProvided < typesRequired && packsProvided == 0) || (typesProvided == typesRequired && packsProvided < packsRequired);
|
|
|
|
|
|
|
|
if (needsDefaults)
|
|
|
|
{
|
|
|
|
// Default types can reference earlier types. It's legal to write
|
|
|
|
// something like
|
|
|
|
// type T<A, B = A> = (A, B) -> number
|
|
|
|
// and we need to respect that. We use an ApplyTypeFunction for this.
|
|
|
|
ApplyTypeFunction atf{arena};
|
|
|
|
|
|
|
|
for (size_t i = 0; i < typesProvided; ++i)
|
|
|
|
atf.typeArguments[fn.typeParams[i].ty] = saturatedTypeArguments[i];
|
|
|
|
|
|
|
|
for (size_t i = typesProvided; i < typesRequired; ++i)
|
|
|
|
{
|
|
|
|
TypeId defaultTy = fn.typeParams[i].defaultValue.value_or(nullptr);
|
|
|
|
|
|
|
|
// We will fill this in with the error type later.
|
|
|
|
if (!defaultTy)
|
|
|
|
break;
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
TypeId instantiatedDefault = atf.substitute(defaultTy).value_or(builtinTypes->errorRecoveryType());
|
2022-08-04 14:27:28 -07:00
|
|
|
atf.typeArguments[fn.typeParams[i].ty] = instantiatedDefault;
|
|
|
|
saturatedTypeArguments.push_back(instantiatedDefault);
|
|
|
|
}
|
|
|
|
|
|
|
|
for (size_t i = 0; i < packsProvided; ++i)
|
|
|
|
{
|
|
|
|
atf.typePackArguments[fn.typePackParams[i].tp] = saturatedPackArguments[i];
|
|
|
|
}
|
|
|
|
|
|
|
|
for (size_t i = packsProvided; i < packsRequired; ++i)
|
|
|
|
{
|
|
|
|
TypePackId defaultTp = fn.typePackParams[i].defaultValue.value_or(nullptr);
|
|
|
|
|
|
|
|
// We will fill this in with the error type pack later.
|
|
|
|
if (!defaultTp)
|
|
|
|
break;
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
TypePackId instantiatedDefault = atf.substitute(defaultTp).value_or(builtinTypes->errorRecoveryTypePack());
|
2022-08-04 14:27:28 -07:00
|
|
|
atf.typePackArguments[fn.typePackParams[i].tp] = instantiatedDefault;
|
|
|
|
saturatedPackArguments.push_back(instantiatedDefault);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// If we didn't create an extra type pack from overflowing parameter packs,
|
|
|
|
// and we're still missing a type pack, plug in an empty type pack as the
|
|
|
|
// value of the empty packs.
|
|
|
|
if (extraTypes.empty() && saturatedPackArguments.size() + 1 == fn.typePackParams.size())
|
|
|
|
{
|
|
|
|
saturatedPackArguments.push_back(arena->addTypePack({}));
|
|
|
|
}
|
|
|
|
|
|
|
|
// We need to have _something_ when we substitute the generic saturatedTypeArguments,
|
|
|
|
// even if they're missing, so we use the error type as a filler.
|
|
|
|
for (size_t i = saturatedTypeArguments.size(); i < typesRequired; ++i)
|
|
|
|
{
|
2023-01-03 19:33:19 +02:00
|
|
|
saturatedTypeArguments.push_back(builtinTypes->errorRecoveryType());
|
2022-08-04 14:27:28 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
for (size_t i = saturatedPackArguments.size(); i < packsRequired; ++i)
|
|
|
|
{
|
2023-01-03 19:33:19 +02:00
|
|
|
saturatedPackArguments.push_back(builtinTypes->errorRecoveryTypePack());
|
2022-08-04 14:27:28 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// At this point, these two conditions should be true. If they aren't we
|
|
|
|
// will run into access violations.
|
|
|
|
LUAU_ASSERT(saturatedTypeArguments.size() == fn.typeParams.size());
|
|
|
|
LUAU_ASSERT(saturatedPackArguments.size() == fn.typePackParams.size());
|
|
|
|
|
|
|
|
return {saturatedTypeArguments, saturatedPackArguments};
|
|
|
|
}
|
|
|
|
|
|
|
|
bool InstantiationSignature::operator==(const InstantiationSignature& rhs) const
|
|
|
|
{
|
|
|
|
return fn == rhs.fn && arguments == rhs.arguments && packArguments == rhs.packArguments;
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t HashInstantiationSignature::operator()(const InstantiationSignature& signature) const
|
|
|
|
{
|
|
|
|
size_t hash = std::hash<TypeId>{}(signature.fn.type);
|
|
|
|
for (const GenericTypeDefinition& p : signature.fn.typeParams)
|
|
|
|
{
|
|
|
|
hash ^= (std::hash<TypeId>{}(p.ty) << 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
for (const GenericTypePackDefinition& p : signature.fn.typePackParams)
|
|
|
|
{
|
|
|
|
hash ^= (std::hash<TypePackId>{}(p.tp) << 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
for (const TypeId a : signature.arguments)
|
|
|
|
{
|
|
|
|
hash ^= (std::hash<TypeId>{}(a) << 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
for (const TypePackId a : signature.packArguments)
|
|
|
|
{
|
|
|
|
hash ^= (std::hash<TypePackId>{}(a) << 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
return hash;
|
|
|
|
}
|
|
|
|
|
2022-06-03 13:32:20 -07:00
|
|
|
void dump(ConstraintSolver* cs, ToStringOptions& opts)
|
|
|
|
{
|
|
|
|
printf("constraints:\n");
|
2022-08-25 13:55:08 -07:00
|
|
|
for (NotNull<const Constraint> c : cs->unsolvedConstraints)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2022-08-25 13:55:08 -07:00
|
|
|
auto it = cs->blockedConstraints.find(c);
|
|
|
|
int blockCount = it == cs->blockedConstraints.end() ? 0 : int(it->second);
|
|
|
|
printf("\t%d\t%s\n", blockCount, toString(*c, opts).c_str());
|
2022-06-03 13:32:20 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-05-12 15:15:01 +03:00
|
|
|
struct InstantiationQueuer : TypeOnceVisitor
|
|
|
|
{
|
|
|
|
ConstraintSolver* solver;
|
|
|
|
NotNull<Scope> scope;
|
|
|
|
Location location;
|
|
|
|
|
|
|
|
explicit InstantiationQueuer(NotNull<Scope> scope, const Location& location, ConstraintSolver* solver)
|
|
|
|
: solver(solver)
|
|
|
|
, scope(scope)
|
|
|
|
, location(location)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
bool visit(TypeId ty, const PendingExpansionType& petv) override
|
|
|
|
{
|
|
|
|
solver->pushConstraint(scope, location, TypeAliasExpansionConstraint{ty});
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool visit(TypeId ty, const TypeFamilyInstanceType& tfit) override
|
|
|
|
{
|
|
|
|
solver->pushConstraint(scope, location, ReduceConstraint{ty});
|
|
|
|
return true;
|
|
|
|
}
|
2023-05-25 23:46:51 +03:00
|
|
|
|
|
|
|
bool visit(TypeId ty, const ClassType& ctv) override
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|
2023-05-12 15:15:01 +03:00
|
|
|
};
|
|
|
|
|
2022-11-11 00:04:44 +02:00
|
|
|
ConstraintSolver::ConstraintSolver(NotNull<Normalizer> normalizer, NotNull<Scope> rootScope, std::vector<NotNull<Constraint>> constraints,
|
2023-03-31 15:21:14 +03:00
|
|
|
ModuleName moduleName, NotNull<ModuleResolver> moduleResolver, std::vector<RequireCycle> requireCycles, DcrLogger* logger)
|
2022-10-06 16:55:58 -07:00
|
|
|
: arena(normalizer->arena)
|
2023-01-03 19:33:19 +02:00
|
|
|
, builtinTypes(normalizer->builtinTypes)
|
2022-10-06 16:55:58 -07:00
|
|
|
, normalizer(normalizer)
|
2022-11-11 00:04:44 +02:00
|
|
|
, constraints(std::move(constraints))
|
2022-06-03 13:32:20 -07:00
|
|
|
, rootScope(rootScope)
|
2022-09-01 16:00:14 -07:00
|
|
|
, currentModuleName(std::move(moduleName))
|
|
|
|
, moduleResolver(moduleResolver)
|
|
|
|
, requireCycles(requireCycles)
|
2022-09-08 14:44:50 -07:00
|
|
|
, logger(logger)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2022-09-01 16:00:14 -07:00
|
|
|
opts.exhaustive = true;
|
|
|
|
|
2022-11-11 00:04:44 +02:00
|
|
|
for (NotNull<Constraint> c : this->constraints)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2022-06-16 17:54:42 -07:00
|
|
|
unsolvedConstraints.push_back(c);
|
2022-06-03 13:32:20 -07:00
|
|
|
|
2022-06-16 17:54:42 -07:00
|
|
|
for (NotNull<const Constraint> dep : c->dependencies)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
|
|
|
block(dep, c);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-10-06 16:55:58 -07:00
|
|
|
void ConstraintSolver::randomize(unsigned seed)
|
|
|
|
{
|
2022-10-14 01:59:53 +03:00
|
|
|
if (unsolvedConstraints.empty())
|
|
|
|
return;
|
|
|
|
|
|
|
|
unsigned int rng = seed;
|
|
|
|
|
|
|
|
for (size_t i = unsolvedConstraints.size() - 1; i > 0; --i)
|
|
|
|
{
|
|
|
|
// Fisher-Yates shuffle
|
|
|
|
size_t j = rng % (i + 1);
|
|
|
|
|
|
|
|
std::swap(unsolvedConstraints[i], unsolvedConstraints[j]);
|
|
|
|
|
|
|
|
// LCG RNG, constants from Numerical Recipes
|
|
|
|
// This may occasionally result in skewed shuffles due to distribution properties, but this is a debugging tool so it should be good enough
|
|
|
|
rng = rng * 1664525 + 1013904223;
|
|
|
|
}
|
2022-10-06 16:55:58 -07:00
|
|
|
}
|
|
|
|
|
2022-06-03 13:32:20 -07:00
|
|
|
void ConstraintSolver::run()
|
|
|
|
{
|
2022-10-14 01:59:53 +03:00
|
|
|
if (isDone())
|
2022-06-03 13:32:20 -07:00
|
|
|
return;
|
|
|
|
|
|
|
|
if (FFlag::DebugLuauLogSolver)
|
|
|
|
{
|
|
|
|
printf("Starting solver\n");
|
|
|
|
dump(this, opts);
|
2022-11-11 00:04:44 +02:00
|
|
|
printf("Bindings:\n");
|
|
|
|
dumpBindings(rootScope, opts);
|
2022-06-03 13:32:20 -07:00
|
|
|
}
|
|
|
|
|
2023-03-03 15:45:38 +02:00
|
|
|
if (logger)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2022-09-08 14:44:50 -07:00
|
|
|
logger->captureInitialSolverState(rootScope, unsolvedConstraints);
|
2022-06-16 17:54:42 -07:00
|
|
|
}
|
2022-06-03 13:32:20 -07:00
|
|
|
|
2022-06-16 17:54:42 -07:00
|
|
|
auto runSolverPass = [&](bool force) {
|
|
|
|
bool progress = false;
|
2022-06-03 13:32:20 -07:00
|
|
|
|
2022-06-16 17:54:42 -07:00
|
|
|
size_t i = 0;
|
|
|
|
while (i < unsolvedConstraints.size())
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2022-06-16 17:54:42 -07:00
|
|
|
NotNull<const Constraint> c = unsolvedConstraints[i];
|
|
|
|
if (!force && isBlocked(c))
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2022-06-16 17:54:42 -07:00
|
|
|
++i;
|
2022-06-03 13:32:20 -07:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
2022-06-16 17:54:42 -07:00
|
|
|
std::string saveMe = FFlag::DebugLuauLogSolver ? toString(*c, opts) : std::string{};
|
2022-09-08 14:44:50 -07:00
|
|
|
StepSnapshot snapshot;
|
2022-06-03 13:32:20 -07:00
|
|
|
|
2023-03-03 15:45:38 +02:00
|
|
|
if (logger)
|
2022-06-16 17:54:42 -07:00
|
|
|
{
|
2022-09-08 14:44:50 -07:00
|
|
|
snapshot = logger->prepareStepSnapshot(rootScope, c, force, unsolvedConstraints);
|
2022-06-16 17:54:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
bool success = tryDispatch(c, force);
|
|
|
|
|
|
|
|
progress |= success;
|
2022-06-03 13:32:20 -07:00
|
|
|
|
|
|
|
if (success)
|
|
|
|
{
|
2022-08-25 13:55:08 -07:00
|
|
|
unblock(c);
|
2022-06-16 17:54:42 -07:00
|
|
|
unsolvedConstraints.erase(unsolvedConstraints.begin() + i);
|
|
|
|
|
2023-03-03 15:45:38 +02:00
|
|
|
if (logger)
|
2022-06-16 17:54:42 -07:00
|
|
|
{
|
2022-09-08 14:44:50 -07:00
|
|
|
logger->commitStepSnapshot(snapshot);
|
2022-06-16 17:54:42 -07:00
|
|
|
}
|
|
|
|
|
2022-06-03 13:32:20 -07:00
|
|
|
if (FFlag::DebugLuauLogSolver)
|
|
|
|
{
|
2022-06-16 17:54:42 -07:00
|
|
|
if (force)
|
|
|
|
printf("Force ");
|
2022-06-03 13:32:20 -07:00
|
|
|
printf("Dispatched\n\t%s\n", saveMe.c_str());
|
2023-02-24 10:24:22 -08:00
|
|
|
|
|
|
|
if (force)
|
|
|
|
{
|
|
|
|
printf("Blocked on:\n");
|
|
|
|
|
|
|
|
for (const auto& [bci, cv] : blocked)
|
|
|
|
{
|
|
|
|
if (end(cv) == std::find(begin(cv), end(cv), c))
|
|
|
|
continue;
|
|
|
|
|
|
|
|
if (auto bty = get_if<TypeId>(&bci))
|
|
|
|
printf("\tType %s\n", toString(*bty, opts).c_str());
|
|
|
|
else if (auto btp = get_if<TypePackId>(&bci))
|
|
|
|
printf("\tPack %s\n", toString(*btp, opts).c_str());
|
|
|
|
else if (auto cc = get_if<const Constraint*>(&bci))
|
|
|
|
printf("\tCons %s\n", toString(**cc, opts).c_str());
|
|
|
|
else
|
|
|
|
LUAU_ASSERT(!"Unreachable??");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-06-03 13:32:20 -07:00
|
|
|
dump(this, opts);
|
|
|
|
}
|
|
|
|
}
|
2022-06-16 17:54:42 -07:00
|
|
|
else
|
|
|
|
++i;
|
|
|
|
|
|
|
|
if (force && success)
|
|
|
|
return true;
|
2022-06-03 13:32:20 -07:00
|
|
|
}
|
2022-06-16 17:54:42 -07:00
|
|
|
|
|
|
|
return progress;
|
|
|
|
};
|
|
|
|
|
|
|
|
bool progress = false;
|
|
|
|
do
|
|
|
|
{
|
|
|
|
progress = runSolverPass(false);
|
|
|
|
if (!progress)
|
|
|
|
progress |= runSolverPass(true);
|
2022-06-03 13:32:20 -07:00
|
|
|
} while (progress);
|
|
|
|
|
2022-10-14 01:59:53 +03:00
|
|
|
finalizeModule();
|
|
|
|
|
2022-06-03 13:32:20 -07:00
|
|
|
if (FFlag::DebugLuauLogSolver)
|
2022-06-16 17:54:42 -07:00
|
|
|
{
|
2022-06-03 13:32:20 -07:00
|
|
|
dumpBindings(rootScope, opts);
|
2022-06-16 17:54:42 -07:00
|
|
|
}
|
2022-06-03 13:32:20 -07:00
|
|
|
|
2023-03-03 15:45:38 +02:00
|
|
|
if (logger)
|
2022-06-16 17:54:42 -07:00
|
|
|
{
|
2022-09-08 14:44:50 -07:00
|
|
|
logger->captureFinalSolverState(rootScope, unsolvedConstraints);
|
2022-06-16 17:54:42 -07:00
|
|
|
}
|
2022-06-03 13:32:20 -07:00
|
|
|
}
|
|
|
|
|
2022-10-14 01:59:53 +03:00
|
|
|
bool ConstraintSolver::isDone()
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
|
|
|
return unsolvedConstraints.empty();
|
|
|
|
}
|
|
|
|
|
2022-10-14 01:59:53 +03:00
|
|
|
void ConstraintSolver::finalizeModule()
|
|
|
|
{
|
2023-01-03 19:33:19 +02:00
|
|
|
Anyification a{arena, rootScope, builtinTypes, &iceReporter, builtinTypes->anyType, builtinTypes->anyTypePack};
|
2022-10-14 01:59:53 +03:00
|
|
|
std::optional<TypePackId> returnType = a.substitute(rootScope->returnType);
|
|
|
|
if (!returnType)
|
|
|
|
{
|
|
|
|
reportError(CodeTooComplex{}, Location{});
|
2023-01-03 19:33:19 +02:00
|
|
|
rootScope->returnType = builtinTypes->errorTypePack;
|
2022-10-14 01:59:53 +03:00
|
|
|
}
|
|
|
|
else
|
2023-05-19 11:59:59 -07:00
|
|
|
{
|
|
|
|
rootScope->returnType = anyifyModuleReturnTypePackGenerics(*returnType);
|
|
|
|
}
|
2022-10-14 01:59:53 +03:00
|
|
|
}
|
|
|
|
|
2022-06-16 17:54:42 -07:00
|
|
|
bool ConstraintSolver::tryDispatch(NotNull<const Constraint> constraint, bool force)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2022-06-16 17:54:42 -07:00
|
|
|
if (!force && isBlocked(constraint))
|
2022-06-03 13:32:20 -07:00
|
|
|
return false;
|
|
|
|
|
|
|
|
bool success = false;
|
|
|
|
|
|
|
|
if (auto sc = get<SubtypeConstraint>(*constraint))
|
2022-06-16 17:54:42 -07:00
|
|
|
success = tryDispatch(*sc, constraint, force);
|
2022-06-03 13:32:20 -07:00
|
|
|
else if (auto psc = get<PackSubtypeConstraint>(*constraint))
|
2022-06-16 17:54:42 -07:00
|
|
|
success = tryDispatch(*psc, constraint, force);
|
2022-06-03 13:32:20 -07:00
|
|
|
else if (auto gc = get<GeneralizationConstraint>(*constraint))
|
2022-06-16 17:54:42 -07:00
|
|
|
success = tryDispatch(*gc, constraint, force);
|
2022-06-03 13:32:20 -07:00
|
|
|
else if (auto ic = get<InstantiationConstraint>(*constraint))
|
2022-06-16 17:54:42 -07:00
|
|
|
success = tryDispatch(*ic, constraint, force);
|
2022-06-30 16:29:02 -07:00
|
|
|
else if (auto uc = get<UnaryConstraint>(*constraint))
|
|
|
|
success = tryDispatch(*uc, constraint, force);
|
|
|
|
else if (auto bc = get<BinaryConstraint>(*constraint))
|
|
|
|
success = tryDispatch(*bc, constraint, force);
|
2022-09-01 16:00:14 -07:00
|
|
|
else if (auto ic = get<IterableConstraint>(*constraint))
|
|
|
|
success = tryDispatch(*ic, constraint, force);
|
2022-06-23 18:44:07 -07:00
|
|
|
else if (auto nc = get<NameConstraint>(*constraint))
|
|
|
|
success = tryDispatch(*nc, constraint);
|
2022-08-04 14:27:28 -07:00
|
|
|
else if (auto taec = get<TypeAliasExpansionConstraint>(*constraint))
|
|
|
|
success = tryDispatch(*taec, constraint);
|
2022-09-01 16:00:14 -07:00
|
|
|
else if (auto fcc = get<FunctionCallConstraint>(*constraint))
|
|
|
|
success = tryDispatch(*fcc, constraint);
|
2022-09-23 11:32:10 -07:00
|
|
|
else if (auto fcc = get<PrimitiveTypeConstraint>(*constraint))
|
|
|
|
success = tryDispatch(*fcc, constraint);
|
|
|
|
else if (auto hpc = get<HasPropConstraint>(*constraint))
|
|
|
|
success = tryDispatch(*hpc, constraint);
|
2022-11-18 10:45:14 -08:00
|
|
|
else if (auto spc = get<SetPropConstraint>(*constraint))
|
2023-01-20 14:02:39 +02:00
|
|
|
success = tryDispatch(*spc, constraint, force);
|
2023-02-24 10:24:22 -08:00
|
|
|
else if (auto spc = get<SetIndexerConstraint>(*constraint))
|
|
|
|
success = tryDispatch(*spc, constraint, force);
|
2022-11-04 10:02:37 -07:00
|
|
|
else if (auto sottc = get<SingletonOrTopTypeConstraint>(*constraint))
|
|
|
|
success = tryDispatch(*sottc, constraint);
|
2023-02-24 10:24:22 -08:00
|
|
|
else if (auto uc = get<UnpackConstraint>(*constraint))
|
|
|
|
success = tryDispatch(*uc, constraint);
|
2023-05-19 11:59:59 -07:00
|
|
|
else if (auto rc = get<RefineConstraint>(*constraint))
|
|
|
|
success = tryDispatch(*rc, constraint, force);
|
2023-05-12 15:15:01 +03:00
|
|
|
else if (auto rc = get<ReduceConstraint>(*constraint))
|
|
|
|
success = tryDispatch(*rc, constraint, force);
|
|
|
|
else if (auto rpc = get<ReducePackConstraint>(*constraint))
|
|
|
|
success = tryDispatch(*rpc, constraint, force);
|
2022-06-03 13:32:20 -07:00
|
|
|
else
|
2022-09-23 11:32:10 -07:00
|
|
|
LUAU_ASSERT(false);
|
2022-06-03 13:32:20 -07:00
|
|
|
|
|
|
|
if (success)
|
|
|
|
unblock(constraint);
|
|
|
|
|
|
|
|
return success;
|
|
|
|
}
|
|
|
|
|
2022-06-16 17:54:42 -07:00
|
|
|
bool ConstraintSolver::tryDispatch(const SubtypeConstraint& c, NotNull<const Constraint> constraint, bool force)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2022-06-16 17:54:42 -07:00
|
|
|
if (isBlocked(c.subType))
|
|
|
|
return block(c.subType, constraint);
|
|
|
|
else if (isBlocked(c.superType))
|
|
|
|
return block(c.superType, constraint);
|
|
|
|
|
2023-03-31 15:21:14 +03:00
|
|
|
return tryUnify(constraint, c.subType, c.superType);
|
2022-06-03 13:32:20 -07:00
|
|
|
}
|
|
|
|
|
2022-06-16 17:54:42 -07:00
|
|
|
bool ConstraintSolver::tryDispatch(const PackSubtypeConstraint& c, NotNull<const Constraint> constraint, bool force)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2022-09-01 16:00:14 -07:00
|
|
|
if (isBlocked(c.subPack))
|
|
|
|
return block(c.subPack, constraint);
|
|
|
|
else if (isBlocked(c.superPack))
|
|
|
|
return block(c.superPack, constraint);
|
|
|
|
|
2023-03-31 15:21:14 +03:00
|
|
|
return tryUnify(constraint, c.subPack, c.superPack);
|
2022-06-03 13:32:20 -07:00
|
|
|
}
|
|
|
|
|
2022-06-16 17:54:42 -07:00
|
|
|
bool ConstraintSolver::tryDispatch(const GeneralizationConstraint& c, NotNull<const Constraint> constraint, bool force)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2023-04-28 14:55:55 +03:00
|
|
|
TypeId generalizedType = follow(c.generalizedType);
|
|
|
|
|
2022-06-16 17:54:42 -07:00
|
|
|
if (isBlocked(c.sourceType))
|
|
|
|
return block(c.sourceType, constraint);
|
2023-04-28 14:55:55 +03:00
|
|
|
else if (get<PendingExpansionType>(generalizedType))
|
|
|
|
return block(generalizedType, constraint);
|
2022-06-03 13:32:20 -07:00
|
|
|
|
2023-05-12 15:15:01 +03:00
|
|
|
std::optional<QuantifierResult> generalized = quantify(arena, c.sourceType, constraint->scope);
|
2023-03-10 11:20:04 -08:00
|
|
|
if (generalized)
|
|
|
|
{
|
2023-04-28 14:55:55 +03:00
|
|
|
if (get<BlockedType>(generalizedType))
|
2023-05-12 15:15:01 +03:00
|
|
|
asMutable(generalizedType)->ty.emplace<BoundType>(generalized->result);
|
2023-03-10 11:20:04 -08:00
|
|
|
else
|
2023-05-12 15:15:01 +03:00
|
|
|
unify(generalizedType, generalized->result, constraint->scope);
|
|
|
|
|
|
|
|
for (auto [free, gen] : generalized->insertedGenerics.pairings)
|
|
|
|
unify(free, gen, constraint->scope);
|
|
|
|
|
|
|
|
for (auto [free, gen] : generalized->insertedGenericPacks.pairings)
|
|
|
|
unify(free, gen, constraint->scope);
|
2023-03-10 11:20:04 -08:00
|
|
|
}
|
2022-06-16 17:54:42 -07:00
|
|
|
else
|
2023-03-10 11:20:04 -08:00
|
|
|
{
|
|
|
|
reportError(CodeTooComplex{}, constraint->location);
|
|
|
|
asMutable(c.generalizedType)->ty.emplace<BoundType>(builtinTypes->errorRecoveryType());
|
|
|
|
}
|
2022-06-16 17:54:42 -07:00
|
|
|
|
|
|
|
unblock(c.generalizedType);
|
|
|
|
unblock(c.sourceType);
|
2022-06-03 13:32:20 -07:00
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2022-06-16 17:54:42 -07:00
|
|
|
bool ConstraintSolver::tryDispatch(const InstantiationConstraint& c, NotNull<const Constraint> constraint, bool force)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2022-06-16 17:54:42 -07:00
|
|
|
if (isBlocked(c.superType))
|
|
|
|
return block(c.superType, constraint);
|
2022-06-03 13:32:20 -07:00
|
|
|
|
2023-05-19 11:59:59 -07:00
|
|
|
if (!blockOnPendingTypes(c.superType, constraint))
|
2023-05-12 15:15:01 +03:00
|
|
|
return false;
|
|
|
|
|
2022-09-29 15:11:54 -07:00
|
|
|
Instantiation inst(TxnLog::empty(), arena, TypeLevel{}, constraint->scope);
|
2022-06-03 13:32:20 -07:00
|
|
|
|
|
|
|
std::optional<TypeId> instantiated = inst.substitute(c.superType);
|
|
|
|
|
2023-04-28 14:55:55 +03:00
|
|
|
LUAU_ASSERT(get<BlockedType>(c.subType));
|
2023-05-19 11:59:59 -07:00
|
|
|
|
|
|
|
if (!instantiated.has_value())
|
|
|
|
{
|
|
|
|
reportError(UnificationTooComplex{}, constraint->location);
|
|
|
|
|
|
|
|
asMutable(c.subType)->ty.emplace<BoundType>(errorRecoveryType());
|
|
|
|
unblock(c.subType);
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2023-04-28 14:55:55 +03:00
|
|
|
asMutable(c.subType)->ty.emplace<BoundType>(*instantiated);
|
2022-06-30 16:29:02 -07:00
|
|
|
|
2023-05-12 15:15:01 +03:00
|
|
|
InstantiationQueuer queuer{constraint->scope, constraint->location, this};
|
|
|
|
queuer.traverse(c.subType);
|
|
|
|
|
2022-06-03 13:32:20 -07:00
|
|
|
unblock(c.subType);
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2022-06-30 16:29:02 -07:00
|
|
|
bool ConstraintSolver::tryDispatch(const UnaryConstraint& c, NotNull<const Constraint> constraint, bool force)
|
|
|
|
{
|
|
|
|
TypeId operandType = follow(c.operandType);
|
|
|
|
|
|
|
|
if (isBlocked(operandType))
|
|
|
|
return block(operandType, constraint);
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
if (get<FreeType>(operandType))
|
2022-06-30 16:29:02 -07:00
|
|
|
return block(operandType, constraint);
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
LUAU_ASSERT(get<BlockedType>(c.resultType));
|
2022-06-30 16:29:02 -07:00
|
|
|
|
2022-09-23 11:32:10 -07:00
|
|
|
switch (c.op)
|
2022-06-30 16:29:02 -07:00
|
|
|
{
|
2022-10-14 01:59:53 +03:00
|
|
|
case AstExprUnary::Not:
|
|
|
|
{
|
2023-01-03 19:33:19 +02:00
|
|
|
asMutable(c.resultType)->ty.emplace<BoundType>(builtinTypes->booleanType);
|
2023-03-24 10:34:14 -07:00
|
|
|
|
|
|
|
unblock(c.resultType);
|
2022-10-14 01:59:53 +03:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
case AstExprUnary::Len:
|
|
|
|
{
|
2022-10-28 01:22:49 +03:00
|
|
|
// __len must return a number.
|
2023-01-03 19:33:19 +02:00
|
|
|
asMutable(c.resultType)->ty.emplace<BoundType>(builtinTypes->numberType);
|
2023-03-24 10:34:14 -07:00
|
|
|
|
|
|
|
unblock(c.resultType);
|
2022-10-14 01:59:53 +03:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
case AstExprUnary::Minus:
|
|
|
|
{
|
2023-02-03 14:34:12 +02:00
|
|
|
if (isNumber(operandType) || get<AnyType>(operandType) || get<ErrorType>(operandType) || get<NeverType>(operandType))
|
2022-09-23 11:32:10 -07:00
|
|
|
{
|
2023-01-03 19:33:19 +02:00
|
|
|
asMutable(c.resultType)->ty.emplace<BoundType>(c.operandType);
|
2022-09-23 11:32:10 -07:00
|
|
|
}
|
2023-01-03 19:33:19 +02:00
|
|
|
else if (std::optional<TypeId> mm = findMetatableEntry(builtinTypes, errors, operandType, "__unm", constraint->location))
|
2022-10-28 01:22:49 +03:00
|
|
|
{
|
2023-02-17 16:53:37 +02:00
|
|
|
TypeId mmTy = follow(*mm);
|
2022-10-28 01:22:49 +03:00
|
|
|
|
2023-02-17 16:53:37 +02:00
|
|
|
if (get<FreeType>(mmTy) && !force)
|
|
|
|
return block(mmTy, constraint);
|
2022-10-28 01:22:49 +03:00
|
|
|
|
2023-02-17 16:53:37 +02:00
|
|
|
TypePackId argPack = arena->addTypePack(TypePack{{operandType}, {}});
|
|
|
|
TypePackId retPack = arena->addTypePack(BlockedTypePack{});
|
2022-10-28 01:22:49 +03:00
|
|
|
|
2023-02-17 16:53:37 +02:00
|
|
|
asMutable(c.resultType)->ty.emplace<FreeType>(constraint->scope);
|
2022-10-28 01:22:49 +03:00
|
|
|
|
2023-02-17 16:53:37 +02:00
|
|
|
pushConstraint(constraint->scope, constraint->location, PackSubtypeConstraint{retPack, arena->addTypePack(TypePack{{c.resultType}})});
|
2022-10-28 01:22:49 +03:00
|
|
|
|
2023-02-17 16:53:37 +02:00
|
|
|
pushConstraint(constraint->scope, constraint->location, FunctionCallConstraint{mmTy, argPack, retPack, nullptr});
|
2022-10-28 01:22:49 +03:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2023-01-03 19:33:19 +02:00
|
|
|
asMutable(c.resultType)->ty.emplace<BoundType>(builtinTypes->errorRecoveryType());
|
2022-10-28 01:22:49 +03:00
|
|
|
}
|
|
|
|
|
2023-03-24 10:34:14 -07:00
|
|
|
unblock(c.resultType);
|
2022-10-28 01:22:49 +03:00
|
|
|
return true;
|
2022-10-14 01:59:53 +03:00
|
|
|
}
|
2022-06-30 16:29:02 -07:00
|
|
|
}
|
|
|
|
|
2022-10-28 01:22:49 +03:00
|
|
|
LUAU_ASSERT(false);
|
2022-06-30 16:29:02 -07:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool ConstraintSolver::tryDispatch(const BinaryConstraint& c, NotNull<const Constraint> constraint, bool force)
|
|
|
|
{
|
|
|
|
TypeId leftType = follow(c.leftType);
|
|
|
|
TypeId rightType = follow(c.rightType);
|
2022-08-25 13:55:08 -07:00
|
|
|
TypeId resultType = follow(c.resultType);
|
2022-06-30 16:29:02 -07:00
|
|
|
|
2023-04-28 14:55:55 +03:00
|
|
|
LUAU_ASSERT(get<BlockedType>(resultType));
|
|
|
|
|
2022-10-21 10:33:43 -07:00
|
|
|
bool isLogical = c.op == AstExprBinary::Op::And || c.op == AstExprBinary::Op::Or;
|
|
|
|
|
|
|
|
/* Compound assignments create constraints of the form
|
|
|
|
*
|
|
|
|
* A <: Binary<op, A, B>
|
|
|
|
*
|
|
|
|
* This constraint is the one that is meant to unblock A, so it doesn't
|
|
|
|
* make any sense to stop and wait for someone else to do it.
|
|
|
|
*/
|
|
|
|
|
2023-04-14 15:05:27 +03:00
|
|
|
// If any is present, the expression must evaluate to any as well.
|
|
|
|
bool leftAny = get<AnyType>(leftType) || get<ErrorType>(leftType);
|
|
|
|
bool rightAny = get<AnyType>(rightType) || get<ErrorType>(rightType);
|
|
|
|
bool anyPresent = leftAny || rightAny;
|
|
|
|
|
2022-10-21 10:33:43 -07:00
|
|
|
if (isBlocked(leftType) && leftType != resultType)
|
|
|
|
return block(c.leftType, constraint);
|
|
|
|
|
|
|
|
if (isBlocked(rightType) && rightType != resultType)
|
|
|
|
return block(c.rightType, constraint);
|
|
|
|
|
|
|
|
if (!force)
|
2022-06-30 16:29:02 -07:00
|
|
|
{
|
2022-10-21 10:33:43 -07:00
|
|
|
// Logical expressions may proceed if the LHS is free.
|
2023-04-14 15:05:27 +03:00
|
|
|
if (hasTypeInIntersection<FreeType>(leftType) && !isLogical)
|
2022-10-21 10:33:43 -07:00
|
|
|
return block(leftType, constraint);
|
2022-06-30 16:29:02 -07:00
|
|
|
}
|
|
|
|
|
2022-10-21 10:33:43 -07:00
|
|
|
// Logical expressions may proceed if the LHS is free.
|
2023-04-14 15:05:27 +03:00
|
|
|
if (isBlocked(leftType) || (hasTypeInIntersection<FreeType>(leftType) && !isLogical))
|
2022-06-30 16:29:02 -07:00
|
|
|
{
|
2023-01-03 19:33:19 +02:00
|
|
|
asMutable(resultType)->ty.emplace<BoundType>(errorRecoveryType());
|
2022-10-21 10:33:43 -07:00
|
|
|
unblock(resultType);
|
2022-06-30 16:29:02 -07:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2022-10-21 10:33:43 -07:00
|
|
|
// Metatables go first, even if there is primitive behavior.
|
|
|
|
if (auto it = kBinaryOpMetamethods.find(c.op); it != kBinaryOpMetamethods.end())
|
2022-08-25 13:55:08 -07:00
|
|
|
{
|
2022-10-21 10:33:43 -07:00
|
|
|
// Metatables are not the same. The metamethod will not be invoked.
|
|
|
|
if ((c.op == AstExprBinary::Op::CompareEq || c.op == AstExprBinary::Op::CompareNe) &&
|
2023-01-03 19:33:19 +02:00
|
|
|
getMetatable(leftType, builtinTypes) != getMetatable(rightType, builtinTypes))
|
2022-10-21 10:33:43 -07:00
|
|
|
{
|
|
|
|
// TODO: Boolean singleton false? The result is _always_ boolean false.
|
2023-01-03 19:33:19 +02:00
|
|
|
asMutable(resultType)->ty.emplace<BoundType>(builtinTypes->booleanType);
|
2022-10-21 10:33:43 -07:00
|
|
|
unblock(resultType);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::optional<TypeId> mm;
|
|
|
|
|
|
|
|
// The LHS metatable takes priority over the RHS metatable, where
|
|
|
|
// present.
|
2023-01-03 19:33:19 +02:00
|
|
|
if (std::optional<TypeId> leftMm = findMetatableEntry(builtinTypes, errors, leftType, it->second, constraint->location))
|
2022-10-21 10:33:43 -07:00
|
|
|
mm = leftMm;
|
2023-01-03 19:33:19 +02:00
|
|
|
else if (std::optional<TypeId> rightMm = findMetatableEntry(builtinTypes, errors, rightType, it->second, constraint->location))
|
2022-10-21 10:33:43 -07:00
|
|
|
mm = rightMm;
|
|
|
|
|
|
|
|
if (mm)
|
|
|
|
{
|
2022-12-09 10:07:25 -08:00
|
|
|
Instantiation instantiation{TxnLog::empty(), arena, TypeLevel{}, constraint->scope};
|
|
|
|
std::optional<TypeId> instantiatedMm = instantiation.substitute(*mm);
|
|
|
|
if (!instantiatedMm)
|
|
|
|
{
|
|
|
|
reportError(CodeTooComplex{}, constraint->location);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2022-10-21 10:33:43 -07:00
|
|
|
// TODO: Is a table with __call legal here?
|
|
|
|
// TODO: Overloads
|
2023-01-03 19:33:19 +02:00
|
|
|
if (const FunctionType* ftv = get<FunctionType>(follow(*instantiatedMm)))
|
2022-10-21 10:33:43 -07:00
|
|
|
{
|
|
|
|
TypePackId inferredArgs;
|
|
|
|
// For >= and > we invoke __lt and __le respectively with
|
|
|
|
// swapped argument ordering.
|
|
|
|
if (c.op == AstExprBinary::Op::CompareGe || c.op == AstExprBinary::Op::CompareGt)
|
|
|
|
{
|
|
|
|
inferredArgs = arena->addTypePack({rightType, leftType});
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
inferredArgs = arena->addTypePack({leftType, rightType});
|
|
|
|
}
|
|
|
|
|
|
|
|
unify(inferredArgs, ftv->argTypes, constraint->scope);
|
|
|
|
|
|
|
|
TypeId mmResult;
|
|
|
|
|
|
|
|
// Comparison operations always evaluate to a boolean,
|
|
|
|
// regardless of what the metamethod returns.
|
|
|
|
switch (c.op)
|
|
|
|
{
|
|
|
|
case AstExprBinary::Op::CompareEq:
|
|
|
|
case AstExprBinary::Op::CompareNe:
|
|
|
|
case AstExprBinary::Op::CompareGe:
|
|
|
|
case AstExprBinary::Op::CompareGt:
|
|
|
|
case AstExprBinary::Op::CompareLe:
|
|
|
|
case AstExprBinary::Op::CompareLt:
|
2023-01-03 19:33:19 +02:00
|
|
|
mmResult = builtinTypes->booleanType;
|
2022-10-21 10:33:43 -07:00
|
|
|
break;
|
|
|
|
default:
|
2023-05-25 23:46:51 +03:00
|
|
|
if (get<NeverType>(leftType) || get<NeverType>(rightType))
|
|
|
|
mmResult = builtinTypes->neverType;
|
|
|
|
else
|
|
|
|
mmResult = first(ftv->retTypes).value_or(errorRecoveryType());
|
2022-10-21 10:33:43 -07:00
|
|
|
}
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
asMutable(resultType)->ty.emplace<BoundType>(mmResult);
|
2022-10-21 10:33:43 -07:00
|
|
|
unblock(resultType);
|
2022-12-09 10:07:25 -08:00
|
|
|
|
2023-01-06 18:07:19 +02:00
|
|
|
(*c.astOriginalCallTypes)[c.astFragment] = *mm;
|
|
|
|
(*c.astOverloadResolvedTypes)[c.astFragment] = *instantiatedMm;
|
2022-10-21 10:33:43 -07:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// If there's no metamethod available, fall back to primitive behavior.
|
2022-08-25 13:55:08 -07:00
|
|
|
}
|
|
|
|
|
2022-10-21 10:33:43 -07:00
|
|
|
switch (c.op)
|
2022-08-25 13:55:08 -07:00
|
|
|
{
|
2022-10-21 10:33:43 -07:00
|
|
|
// For arithmetic operators, if the LHS is a number, the RHS must be a
|
|
|
|
// number as well. The result will also be a number.
|
|
|
|
case AstExprBinary::Op::Add:
|
|
|
|
case AstExprBinary::Op::Sub:
|
|
|
|
case AstExprBinary::Op::Mul:
|
|
|
|
case AstExprBinary::Op::Div:
|
|
|
|
case AstExprBinary::Op::Pow:
|
|
|
|
case AstExprBinary::Op::Mod:
|
2023-05-19 11:59:59 -07:00
|
|
|
{
|
|
|
|
const NormalizedType* normLeftTy = normalizer->normalize(leftType);
|
2023-04-14 15:05:27 +03:00
|
|
|
if (hasTypeInIntersection<FreeType>(leftType) && force)
|
|
|
|
asMutable(leftType)->ty.emplace<BoundType>(anyPresent ? builtinTypes->anyType : builtinTypes->numberType);
|
2023-05-19 11:59:59 -07:00
|
|
|
if (normLeftTy && normLeftTy->isNumber())
|
2022-10-21 10:33:43 -07:00
|
|
|
{
|
|
|
|
unify(leftType, rightType, constraint->scope);
|
2023-01-03 19:33:19 +02:00
|
|
|
asMutable(resultType)->ty.emplace<BoundType>(anyPresent ? builtinTypes->anyType : leftType);
|
2022-10-21 10:33:43 -07:00
|
|
|
unblock(resultType);
|
|
|
|
return true;
|
|
|
|
}
|
2023-05-25 23:46:51 +03:00
|
|
|
else if (get<NeverType>(leftType) || get<NeverType>(rightType))
|
|
|
|
{
|
|
|
|
unify(leftType, rightType, constraint->scope);
|
|
|
|
asMutable(resultType)->ty.emplace<BoundType>(builtinTypes->neverType);
|
|
|
|
unblock(resultType);
|
|
|
|
return true;
|
|
|
|
}
|
2022-10-21 10:33:43 -07:00
|
|
|
|
|
|
|
break;
|
2023-05-19 11:59:59 -07:00
|
|
|
}
|
2022-10-21 10:33:43 -07:00
|
|
|
// For concatenation, if the LHS is a string, the RHS must be a string as
|
|
|
|
// well. The result will also be a string.
|
|
|
|
case AstExprBinary::Op::Concat:
|
2023-04-14 15:05:27 +03:00
|
|
|
if (hasTypeInIntersection<FreeType>(leftType) && force)
|
|
|
|
asMutable(leftType)->ty.emplace<BoundType>(anyPresent ? builtinTypes->anyType : builtinTypes->stringType);
|
2022-10-21 10:33:43 -07:00
|
|
|
if (isString(leftType))
|
|
|
|
{
|
|
|
|
unify(leftType, rightType, constraint->scope);
|
2023-01-03 19:33:19 +02:00
|
|
|
asMutable(resultType)->ty.emplace<BoundType>(anyPresent ? builtinTypes->anyType : leftType);
|
2022-10-21 10:33:43 -07:00
|
|
|
unblock(resultType);
|
|
|
|
return true;
|
|
|
|
}
|
2023-05-25 23:46:51 +03:00
|
|
|
else if (get<NeverType>(leftType) || get<NeverType>(rightType))
|
|
|
|
{
|
|
|
|
unify(leftType, rightType, constraint->scope);
|
|
|
|
asMutable(resultType)->ty.emplace<BoundType>(builtinTypes->neverType);
|
|
|
|
unblock(resultType);
|
|
|
|
return true;
|
|
|
|
}
|
2022-10-21 10:33:43 -07:00
|
|
|
|
|
|
|
break;
|
|
|
|
// Inexact comparisons require that the types be both numbers or both
|
|
|
|
// strings, and evaluate to a boolean.
|
|
|
|
case AstExprBinary::Op::CompareGe:
|
|
|
|
case AstExprBinary::Op::CompareGt:
|
|
|
|
case AstExprBinary::Op::CompareLe:
|
|
|
|
case AstExprBinary::Op::CompareLt:
|
2023-05-25 23:46:51 +03:00
|
|
|
if ((isNumber(leftType) && isNumber(rightType)) || (isString(leftType) && isString(rightType)) || get<NeverType>(leftType) ||
|
|
|
|
get<NeverType>(rightType))
|
2022-10-21 10:33:43 -07:00
|
|
|
{
|
2023-01-03 19:33:19 +02:00
|
|
|
asMutable(resultType)->ty.emplace<BoundType>(builtinTypes->booleanType);
|
2022-10-21 10:33:43 -07:00
|
|
|
unblock(resultType);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
break;
|
|
|
|
// == and ~= always evaluate to a boolean, and impose no other constraints
|
|
|
|
// on their parameters.
|
|
|
|
case AstExprBinary::Op::CompareEq:
|
|
|
|
case AstExprBinary::Op::CompareNe:
|
2023-01-03 19:33:19 +02:00
|
|
|
asMutable(resultType)->ty.emplace<BoundType>(builtinTypes->booleanType);
|
2022-10-21 10:33:43 -07:00
|
|
|
unblock(resultType);
|
2022-08-25 13:55:08 -07:00
|
|
|
return true;
|
2022-10-21 10:33:43 -07:00
|
|
|
// And evalutes to a boolean if the LHS is falsey, and the RHS type if LHS is
|
|
|
|
// truthy.
|
|
|
|
case AstExprBinary::Op::And:
|
2022-11-11 00:04:44 +02:00
|
|
|
{
|
2023-05-19 11:59:59 -07:00
|
|
|
TypeId leftFilteredTy = simplifyIntersection(builtinTypes, arena, leftType, builtinTypes->falsyType).result;
|
2022-11-11 00:04:44 +02:00
|
|
|
|
2023-05-19 11:59:59 -07:00
|
|
|
asMutable(resultType)->ty.emplace<BoundType>(simplifyUnion(builtinTypes, arena, rightType, leftFilteredTy).result);
|
2022-10-21 10:33:43 -07:00
|
|
|
unblock(resultType);
|
|
|
|
return true;
|
2022-11-11 00:04:44 +02:00
|
|
|
}
|
2022-10-21 10:33:43 -07:00
|
|
|
// Or evaluates to the LHS type if the LHS is truthy, and the RHS type if
|
|
|
|
// LHS is falsey.
|
|
|
|
case AstExprBinary::Op::Or:
|
2022-11-11 00:04:44 +02:00
|
|
|
{
|
2023-05-19 11:59:59 -07:00
|
|
|
TypeId leftFilteredTy = simplifyIntersection(builtinTypes, arena, leftType, builtinTypes->truthyType).result;
|
2022-11-11 00:04:44 +02:00
|
|
|
|
2023-05-19 11:59:59 -07:00
|
|
|
asMutable(resultType)->ty.emplace<BoundType>(simplifyUnion(builtinTypes, arena, rightType, leftFilteredTy).result);
|
2022-10-21 10:33:43 -07:00
|
|
|
unblock(resultType);
|
|
|
|
return true;
|
2022-11-11 00:04:44 +02:00
|
|
|
}
|
2022-10-21 10:33:43 -07:00
|
|
|
default:
|
|
|
|
iceReporter.ice("Unhandled AstExprBinary::Op for binary operation", constraint->location);
|
|
|
|
break;
|
2022-08-25 13:55:08 -07:00
|
|
|
}
|
2022-06-30 16:29:02 -07:00
|
|
|
|
2022-10-21 10:33:43 -07:00
|
|
|
// We failed to either evaluate a metamethod or invoke primitive behavior.
|
|
|
|
unify(leftType, errorRecoveryType(), constraint->scope);
|
|
|
|
unify(rightType, errorRecoveryType(), constraint->scope);
|
2023-01-03 19:33:19 +02:00
|
|
|
asMutable(resultType)->ty.emplace<BoundType>(errorRecoveryType());
|
2022-10-21 10:33:43 -07:00
|
|
|
unblock(resultType);
|
2022-06-30 16:29:02 -07:00
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2022-09-01 16:00:14 -07:00
|
|
|
bool ConstraintSolver::tryDispatch(const IterableConstraint& c, NotNull<const Constraint> constraint, bool force)
|
|
|
|
{
|
|
|
|
/*
|
|
|
|
* for .. in loops can play out in a bunch of different ways depending on
|
|
|
|
* the shape of iteratee.
|
|
|
|
*
|
|
|
|
* iteratee might be:
|
|
|
|
* * (nextFn)
|
|
|
|
* * (nextFn, table)
|
|
|
|
* * (nextFn, table, firstIndex)
|
|
|
|
* * table with a metatable and __index
|
|
|
|
* * table with a metatable and __call but no __index (if the metatable has
|
|
|
|
* both, __index takes precedence)
|
|
|
|
* * table with an indexer but no __index or __call (or no metatable)
|
|
|
|
*
|
|
|
|
* To dispatch this constraint, we need first to know enough about iteratee
|
|
|
|
* to figure out which of the above shapes we are actually working with.
|
|
|
|
*
|
|
|
|
* If `force` is true and we still do not know, we must flag a warning. Type
|
|
|
|
* families are the fix for this.
|
|
|
|
*
|
|
|
|
* Since we need to know all of this stuff about the types of the iteratee,
|
|
|
|
* we have no choice but for ConstraintSolver to also be the thing that
|
|
|
|
* applies constraints to the types of the iterators.
|
|
|
|
*/
|
|
|
|
|
|
|
|
auto block_ = [&](auto&& t) {
|
|
|
|
if (force)
|
|
|
|
{
|
|
|
|
// If we haven't figured out the type of the iteratee by now,
|
|
|
|
// there's nothing we can do.
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
block(t, constraint);
|
|
|
|
return false;
|
|
|
|
};
|
|
|
|
|
|
|
|
auto [iteratorTypes, iteratorTail] = flatten(c.iterator);
|
2023-03-24 10:34:14 -07:00
|
|
|
if (iteratorTail && isBlocked(*iteratorTail))
|
2022-09-01 16:00:14 -07:00
|
|
|
return block_(*iteratorTail);
|
|
|
|
|
2022-09-15 15:13:58 -07:00
|
|
|
{
|
|
|
|
bool blocked = false;
|
|
|
|
for (TypeId t : iteratorTypes)
|
|
|
|
{
|
|
|
|
if (isBlocked(t))
|
|
|
|
{
|
|
|
|
block(t, constraint);
|
|
|
|
blocked = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (blocked)
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2022-09-01 16:00:14 -07:00
|
|
|
if (0 == iteratorTypes.size())
|
|
|
|
{
|
2023-01-03 19:33:19 +02:00
|
|
|
Anyification anyify{arena, constraint->scope, builtinTypes, &iceReporter, errorRecoveryType(), errorRecoveryTypePack()};
|
2022-09-01 16:00:14 -07:00
|
|
|
std::optional<TypePackId> anyified = anyify.substitute(c.variables);
|
|
|
|
LUAU_ASSERT(anyified);
|
|
|
|
unify(*anyified, c.variables, constraint->scope);
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
TypeId nextTy = follow(iteratorTypes[0]);
|
2023-01-03 19:33:19 +02:00
|
|
|
if (get<FreeType>(nextTy))
|
2022-09-01 16:00:14 -07:00
|
|
|
return block_(nextTy);
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
if (get<FunctionType>(nextTy))
|
2022-09-01 16:00:14 -07:00
|
|
|
{
|
2023-01-03 19:33:19 +02:00
|
|
|
TypeId tableTy = builtinTypes->nilType;
|
2022-09-01 16:00:14 -07:00
|
|
|
if (iteratorTypes.size() >= 2)
|
|
|
|
tableTy = iteratorTypes[1];
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
TypeId firstIndexTy = builtinTypes->nilType;
|
2022-09-01 16:00:14 -07:00
|
|
|
if (iteratorTypes.size() >= 3)
|
|
|
|
firstIndexTy = iteratorTypes[2];
|
|
|
|
|
|
|
|
return tryDispatchIterableFunction(nextTy, tableTy, firstIndexTy, c, constraint, force);
|
|
|
|
}
|
|
|
|
|
|
|
|
else
|
|
|
|
return tryDispatchIterableTable(iteratorTypes[0], c, constraint, force);
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2022-06-23 18:44:07 -07:00
|
|
|
bool ConstraintSolver::tryDispatch(const NameConstraint& c, NotNull<const Constraint> constraint)
|
|
|
|
{
|
|
|
|
if (isBlocked(c.namedType))
|
|
|
|
return block(c.namedType, constraint);
|
|
|
|
|
|
|
|
TypeId target = follow(c.namedType);
|
2022-08-11 13:42:54 -07:00
|
|
|
|
2022-09-01 16:00:14 -07:00
|
|
|
if (target->persistent || target->owningArena != arena)
|
2022-08-11 13:42:54 -07:00
|
|
|
return true;
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
if (TableType* ttv = getMutable<TableType>(target))
|
2023-01-27 13:28:45 -08:00
|
|
|
{
|
|
|
|
if (c.synthetic && !ttv->name)
|
|
|
|
ttv->syntheticName = c.name;
|
|
|
|
else
|
2023-02-17 16:53:37 +02:00
|
|
|
{
|
2023-01-27 13:28:45 -08:00
|
|
|
ttv->name = c.name;
|
2023-02-17 16:53:37 +02:00
|
|
|
ttv->instantiatedTypeParams = c.typeParameters;
|
|
|
|
ttv->instantiatedTypePackParams = c.typePackParameters;
|
|
|
|
}
|
2023-01-27 13:28:45 -08:00
|
|
|
}
|
2023-01-03 19:33:19 +02:00
|
|
|
else if (MetatableType* mtv = getMutable<MetatableType>(target))
|
2022-06-23 18:44:07 -07:00
|
|
|
mtv->syntheticName = c.name;
|
2023-01-03 19:33:19 +02:00
|
|
|
else if (get<IntersectionType>(target) || get<UnionType>(target))
|
2022-10-28 01:22:49 +03:00
|
|
|
{
|
|
|
|
// nothing (yet)
|
|
|
|
}
|
2022-06-23 18:44:07 -07:00
|
|
|
else
|
|
|
|
return block(c.namedType, constraint);
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
struct InfiniteTypeFinder : TypeOnceVisitor
|
2022-08-04 14:27:28 -07:00
|
|
|
{
|
|
|
|
ConstraintSolver* solver;
|
|
|
|
const InstantiationSignature& signature;
|
2022-09-01 16:00:14 -07:00
|
|
|
NotNull<Scope> scope;
|
2022-08-04 14:27:28 -07:00
|
|
|
bool foundInfiniteType = false;
|
|
|
|
|
2022-09-01 16:00:14 -07:00
|
|
|
explicit InfiniteTypeFinder(ConstraintSolver* solver, const InstantiationSignature& signature, NotNull<Scope> scope)
|
2022-08-04 14:27:28 -07:00
|
|
|
: solver(solver)
|
|
|
|
, signature(signature)
|
2022-09-01 16:00:14 -07:00
|
|
|
, scope(scope)
|
2022-08-04 14:27:28 -07:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
bool visit(TypeId ty, const PendingExpansionType& petv) override
|
2022-08-04 14:27:28 -07:00
|
|
|
{
|
2022-09-01 16:00:14 -07:00
|
|
|
std::optional<TypeFun> tf =
|
|
|
|
(petv.prefix) ? scope->lookupImportedType(petv.prefix->value, petv.name.value) : scope->lookupType(petv.name.value);
|
|
|
|
|
|
|
|
if (!tf.has_value())
|
|
|
|
return true;
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
auto [typeArguments, packArguments] = saturateArguments(solver->arena, solver->builtinTypes, *tf, petv.typeArguments, petv.packArguments);
|
2022-08-04 14:27:28 -07:00
|
|
|
|
2022-09-01 16:00:14 -07:00
|
|
|
if (follow(tf->type) == follow(signature.fn.type) && (signature.arguments != typeArguments || signature.packArguments != packArguments))
|
2022-08-04 14:27:28 -07:00
|
|
|
{
|
|
|
|
foundInfiniteType = true;
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
bool ConstraintSolver::tryDispatch(const TypeAliasExpansionConstraint& c, NotNull<const Constraint> constraint)
|
|
|
|
{
|
2023-01-03 19:33:19 +02:00
|
|
|
const PendingExpansionType* petv = get<PendingExpansionType>(follow(c.target));
|
2022-08-04 14:27:28 -07:00
|
|
|
if (!petv)
|
|
|
|
{
|
|
|
|
unblock(c.target);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
auto bindResult = [this, &c](TypeId result) {
|
2023-04-28 14:55:55 +03:00
|
|
|
LUAU_ASSERT(get<PendingExpansionType>(c.target));
|
2023-01-03 19:33:19 +02:00
|
|
|
asMutable(c.target)->ty.emplace<BoundType>(result);
|
2022-08-04 14:27:28 -07:00
|
|
|
unblock(c.target);
|
|
|
|
};
|
|
|
|
|
2022-09-01 16:00:14 -07:00
|
|
|
std::optional<TypeFun> tf = (petv->prefix) ? constraint->scope->lookupImportedType(petv->prefix->value, petv->name.value)
|
|
|
|
: constraint->scope->lookupType(petv->name.value);
|
|
|
|
|
|
|
|
if (!tf.has_value())
|
|
|
|
{
|
|
|
|
reportError(UnknownSymbol{petv->name.value, UnknownSymbol::Context::Type}, constraint->location);
|
2022-09-15 15:13:58 -07:00
|
|
|
bindResult(errorRecoveryType());
|
2022-09-01 16:00:14 -07:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2022-08-04 14:27:28 -07:00
|
|
|
// If there are no parameters to the type function we can just use the type
|
|
|
|
// directly.
|
2022-09-01 16:00:14 -07:00
|
|
|
if (tf->typeParams.empty() && tf->typePackParams.empty())
|
2022-08-04 14:27:28 -07:00
|
|
|
{
|
2022-09-01 16:00:14 -07:00
|
|
|
bindResult(tf->type);
|
2022-08-04 14:27:28 -07:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
auto [typeArguments, packArguments] = saturateArguments(arena, builtinTypes, *tf, petv->typeArguments, petv->packArguments);
|
2022-08-04 14:27:28 -07:00
|
|
|
|
2022-09-01 16:00:14 -07:00
|
|
|
bool sameTypes = std::equal(typeArguments.begin(), typeArguments.end(), tf->typeParams.begin(), tf->typeParams.end(), [](auto&& itp, auto&& p) {
|
|
|
|
return itp == p.ty;
|
|
|
|
});
|
2022-08-04 14:27:28 -07:00
|
|
|
|
2022-09-01 16:00:14 -07:00
|
|
|
bool samePacks =
|
|
|
|
std::equal(packArguments.begin(), packArguments.end(), tf->typePackParams.begin(), tf->typePackParams.end(), [](auto&& itp, auto&& p) {
|
2022-08-04 14:27:28 -07:00
|
|
|
return itp == p.tp;
|
|
|
|
});
|
|
|
|
|
|
|
|
// If we're instantiating the type with its generic saturatedTypeArguments we are
|
|
|
|
// performing the identity substitution. We can just short-circuit and bind
|
|
|
|
// to the TypeFun's type.
|
|
|
|
if (sameTypes && samePacks)
|
|
|
|
{
|
2022-09-01 16:00:14 -07:00
|
|
|
bindResult(tf->type);
|
2022-08-04 14:27:28 -07:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
InstantiationSignature signature{
|
2022-09-01 16:00:14 -07:00
|
|
|
*tf,
|
2022-08-04 14:27:28 -07:00
|
|
|
typeArguments,
|
|
|
|
packArguments,
|
|
|
|
};
|
|
|
|
|
|
|
|
// If we use the same signature, we don't need to bother trying to
|
|
|
|
// instantiate the alias again, since the instantiation should be
|
|
|
|
// deterministic.
|
|
|
|
if (TypeId* cached = instantiatedAliases.find(signature))
|
|
|
|
{
|
|
|
|
bindResult(*cached);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
// In order to prevent infinite types from being expanded and causing us to
|
|
|
|
// cycle infinitely, we need to scan the type function for cases where we
|
|
|
|
// expand the same alias with different type saturatedTypeArguments. See
|
|
|
|
// https://github.com/Roblox/luau/pull/68 for the RFC responsible for this.
|
|
|
|
// This is a little nicer than using a recursion limit because we can catch
|
|
|
|
// the infinite expansion before actually trying to expand it.
|
2022-09-01 16:00:14 -07:00
|
|
|
InfiniteTypeFinder itf{this, signature, constraint->scope};
|
|
|
|
itf.traverse(tf->type);
|
2022-08-04 14:27:28 -07:00
|
|
|
|
|
|
|
if (itf.foundInfiniteType)
|
|
|
|
{
|
|
|
|
// TODO (CLI-56761): Report an error.
|
2022-09-15 15:13:58 -07:00
|
|
|
bindResult(errorRecoveryType());
|
2023-02-17 16:53:37 +02:00
|
|
|
reportError(GenericError{"Recursive type being used with different parameters"}, constraint->location);
|
2022-08-04 14:27:28 -07:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
ApplyTypeFunction applyTypeFunction{arena};
|
|
|
|
for (size_t i = 0; i < typeArguments.size(); ++i)
|
|
|
|
{
|
2022-09-01 16:00:14 -07:00
|
|
|
applyTypeFunction.typeArguments[tf->typeParams[i].ty] = typeArguments[i];
|
2022-08-04 14:27:28 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
for (size_t i = 0; i < packArguments.size(); ++i)
|
|
|
|
{
|
2022-09-01 16:00:14 -07:00
|
|
|
applyTypeFunction.typePackArguments[tf->typePackParams[i].tp] = packArguments[i];
|
2022-08-04 14:27:28 -07:00
|
|
|
}
|
|
|
|
|
2022-09-01 16:00:14 -07:00
|
|
|
std::optional<TypeId> maybeInstantiated = applyTypeFunction.substitute(tf->type);
|
2022-08-04 14:27:28 -07:00
|
|
|
// Note that ApplyTypeFunction::encounteredForwardedType is never set in
|
|
|
|
// DCR, because we do not use free types for forward-declared generic
|
|
|
|
// aliases.
|
|
|
|
|
|
|
|
if (!maybeInstantiated.has_value())
|
|
|
|
{
|
|
|
|
// TODO (CLI-56761): Report an error.
|
2022-09-15 15:13:58 -07:00
|
|
|
bindResult(errorRecoveryType());
|
2022-08-04 14:27:28 -07:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
TypeId instantiated = *maybeInstantiated;
|
|
|
|
TypeId target = follow(instantiated);
|
2022-08-11 13:42:54 -07:00
|
|
|
|
2023-01-20 14:02:39 +02:00
|
|
|
// The application is not recursive, so we need to queue up application of
|
|
|
|
// any child type function instantiations within the result in order for it
|
|
|
|
// to be complete.
|
|
|
|
InstantiationQueuer queuer{constraint->scope, constraint->location, this};
|
|
|
|
queuer.traverse(target);
|
|
|
|
|
2023-03-31 15:21:14 +03:00
|
|
|
if (target->persistent || target->owningArena != arena)
|
2023-01-20 14:02:39 +02:00
|
|
|
{
|
|
|
|
bindResult(target);
|
2022-08-11 13:42:54 -07:00
|
|
|
return true;
|
2023-01-20 14:02:39 +02:00
|
|
|
}
|
2022-08-11 13:42:54 -07:00
|
|
|
|
2022-08-04 14:27:28 -07:00
|
|
|
// Type function application will happily give us the exact same type if
|
|
|
|
// there are e.g. generic saturatedTypeArguments that go unused.
|
2022-09-01 16:00:14 -07:00
|
|
|
bool needsClone = follow(tf->type) == target;
|
2022-08-04 14:27:28 -07:00
|
|
|
// Only tables have the properties we're trying to set.
|
2023-01-03 19:33:19 +02:00
|
|
|
TableType* ttv = getMutableTableType(target);
|
2022-08-04 14:27:28 -07:00
|
|
|
|
|
|
|
if (ttv)
|
|
|
|
{
|
|
|
|
if (needsClone)
|
|
|
|
{
|
|
|
|
// Substitution::clone is a shallow clone. If this is a
|
|
|
|
// metatable type, we want to mutate its table, so we need to
|
|
|
|
// explicitly clone that table as well. If we don't, we will
|
|
|
|
// mutate another module's type surface and cause a
|
|
|
|
// use-after-free.
|
2023-01-03 19:33:19 +02:00
|
|
|
if (get<MetatableType>(target))
|
2022-08-04 14:27:28 -07:00
|
|
|
{
|
|
|
|
instantiated = applyTypeFunction.clone(target);
|
2023-01-03 19:33:19 +02:00
|
|
|
MetatableType* mtv = getMutable<MetatableType>(instantiated);
|
2022-08-04 14:27:28 -07:00
|
|
|
mtv->table = applyTypeFunction.clone(mtv->table);
|
2023-01-03 19:33:19 +02:00
|
|
|
ttv = getMutable<TableType>(mtv->table);
|
2022-08-04 14:27:28 -07:00
|
|
|
}
|
2023-01-03 19:33:19 +02:00
|
|
|
else if (get<TableType>(target))
|
2022-08-04 14:27:28 -07:00
|
|
|
{
|
|
|
|
instantiated = applyTypeFunction.clone(target);
|
2023-01-03 19:33:19 +02:00
|
|
|
ttv = getMutable<TableType>(instantiated);
|
2022-08-04 14:27:28 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
target = follow(instantiated);
|
|
|
|
}
|
|
|
|
|
|
|
|
ttv->instantiatedTypeParams = typeArguments;
|
|
|
|
ttv->instantiatedTypePackParams = packArguments;
|
|
|
|
// TODO: Fill in definitionModuleName.
|
|
|
|
}
|
|
|
|
|
|
|
|
bindResult(target);
|
|
|
|
|
|
|
|
instantiatedAliases[signature] = target;
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2022-09-01 16:00:14 -07:00
|
|
|
bool ConstraintSolver::tryDispatch(const FunctionCallConstraint& c, NotNull<const Constraint> constraint)
|
|
|
|
{
|
|
|
|
TypeId fn = follow(c.fn);
|
2023-02-17 16:53:37 +02:00
|
|
|
TypePackId argsPack = follow(c.argsPack);
|
2022-09-01 16:00:14 -07:00
|
|
|
TypePackId result = follow(c.result);
|
|
|
|
|
2023-02-17 16:53:37 +02:00
|
|
|
if (isBlocked(fn))
|
2022-09-01 16:00:14 -07:00
|
|
|
{
|
|
|
|
return block(c.fn, constraint);
|
|
|
|
}
|
|
|
|
|
2023-02-10 10:50:54 -08:00
|
|
|
auto collapse = [](const auto* t) -> std::optional<TypeId> {
|
|
|
|
auto it = begin(t);
|
|
|
|
auto endIt = end(t);
|
|
|
|
|
|
|
|
LUAU_ASSERT(it != endIt);
|
|
|
|
TypeId fst = follow(*it);
|
|
|
|
while (it != endIt)
|
|
|
|
{
|
|
|
|
if (follow(*it) != fst)
|
|
|
|
return std::nullopt;
|
|
|
|
++it;
|
|
|
|
}
|
|
|
|
|
|
|
|
return fst;
|
|
|
|
};
|
|
|
|
|
|
|
|
// Sometimes the `fn` type is a union/intersection, but whose constituents are all the same pointer.
|
|
|
|
if (auto ut = get<UnionType>(fn))
|
|
|
|
fn = collapse(ut).value_or(fn);
|
|
|
|
else if (auto it = get<IntersectionType>(fn))
|
|
|
|
fn = collapse(it).value_or(fn);
|
|
|
|
|
2023-05-05 12:57:12 -07:00
|
|
|
if (c.callSite)
|
|
|
|
(*c.astOriginalCallTypes)[c.callSite] = fn;
|
|
|
|
|
2022-10-21 10:33:43 -07:00
|
|
|
// We don't support magic __call metamethods.
|
2023-01-03 19:33:19 +02:00
|
|
|
if (std::optional<TypeId> callMm = findMetatableEntry(builtinTypes, errors, fn, "__call", constraint->location))
|
2022-10-21 10:33:43 -07:00
|
|
|
{
|
|
|
|
std::vector<TypeId> args{fn};
|
|
|
|
|
|
|
|
for (TypeId arg : c.argsPack)
|
|
|
|
args.push_back(arg);
|
|
|
|
|
2023-02-17 16:53:37 +02:00
|
|
|
argsPack = arena->addTypePack(TypePack{args, {}});
|
|
|
|
fn = *callMm;
|
2022-10-21 10:33:43 -07:00
|
|
|
asMutable(c.result)->ty.emplace<FreeTypePack>(constraint->scope);
|
|
|
|
}
|
2023-02-10 10:50:54 -08:00
|
|
|
else
|
|
|
|
{
|
|
|
|
const FunctionType* ftv = get<FunctionType>(fn);
|
|
|
|
bool usedMagic = false;
|
|
|
|
|
|
|
|
if (ftv)
|
|
|
|
{
|
|
|
|
if (ftv->dcrMagicFunction)
|
|
|
|
usedMagic = ftv->dcrMagicFunction(MagicFunctionCallContext{NotNull(this), c.callSite, c.argsPack, result});
|
2022-10-21 10:33:43 -07:00
|
|
|
|
2023-02-10 10:50:54 -08:00
|
|
|
if (ftv->dcrMagicRefinement)
|
|
|
|
ftv->dcrMagicRefinement(MagicRefinementContext{constraint->scope, c.callSite, c.discriminantTypes});
|
|
|
|
}
|
2022-09-01 16:00:14 -07:00
|
|
|
|
2023-02-17 16:53:37 +02:00
|
|
|
if (!usedMagic)
|
2023-02-10 10:50:54 -08:00
|
|
|
asMutable(c.result)->ty.emplace<FreeTypePack>(constraint->scope);
|
2022-09-01 16:00:14 -07:00
|
|
|
}
|
|
|
|
|
2023-02-10 10:50:54 -08:00
|
|
|
for (std::optional<TypeId> ty : c.discriminantTypes)
|
2022-09-01 16:00:14 -07:00
|
|
|
{
|
2023-02-10 10:50:54 -08:00
|
|
|
if (!ty || !isBlocked(*ty))
|
|
|
|
continue;
|
|
|
|
|
|
|
|
// We use `any` here because the discriminant type may be pointed at by both branches,
|
|
|
|
// where the discriminant type is not negated, and the other where it is negated, i.e.
|
|
|
|
// `unknown ~ unknown` and `~unknown ~ never`, so `T & unknown ~ T` and `T & ~unknown ~ never`
|
|
|
|
// v.s.
|
|
|
|
// `any ~ any` and `~any ~ any`, so `T & any ~ T` and `T & ~any ~ T`
|
2022-09-15 15:13:58 -07:00
|
|
|
//
|
2023-02-10 10:50:54 -08:00
|
|
|
// In practice, users cannot negate `any`, so this is an implementation detail we can always change.
|
|
|
|
*asMutable(follow(*ty)) = BoundType{builtinTypes->anyType};
|
2022-09-15 15:13:58 -07:00
|
|
|
}
|
2023-02-10 10:50:54 -08:00
|
|
|
|
2023-02-17 16:53:37 +02:00
|
|
|
TypeId inferredTy = arena->addType(FunctionType{TypeLevel{}, constraint->scope.get(), argsPack, c.result});
|
|
|
|
|
2023-05-05 12:57:12 -07:00
|
|
|
const NormalizedType* normFn = normalizer->normalize(fn);
|
|
|
|
if (!normFn)
|
|
|
|
{
|
|
|
|
reportError(UnificationTooComplex{}, constraint->location);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: It would be nice to not need to convert the normalized type back to
|
|
|
|
// an intersection and flatten it.
|
|
|
|
TypeId normFnTy = normalizer->typeFromNormal(*normFn);
|
|
|
|
std::vector<TypeId> overloads = flattenIntersection(normFnTy);
|
2023-03-24 10:34:14 -07:00
|
|
|
|
|
|
|
Instantiation inst(TxnLog::empty(), arena, TypeLevel{}, constraint->scope);
|
|
|
|
|
2023-05-05 12:57:12 -07:00
|
|
|
std::vector<TypeId> arityMatchingOverloads;
|
2023-06-02 11:17:31 -07:00
|
|
|
std::optional<TxnLog> bestOverloadLog;
|
2023-05-05 12:57:12 -07:00
|
|
|
|
2023-03-24 10:34:14 -07:00
|
|
|
for (TypeId overload : overloads)
|
|
|
|
{
|
|
|
|
overload = follow(overload);
|
2023-02-24 10:24:22 -08:00
|
|
|
|
2023-03-24 10:34:14 -07:00
|
|
|
std::optional<TypeId> instantiated = inst.substitute(overload);
|
2023-05-19 11:59:59 -07:00
|
|
|
|
|
|
|
if (!instantiated.has_value())
|
|
|
|
{
|
|
|
|
reportError(UnificationTooComplex{}, constraint->location);
|
|
|
|
return true;
|
|
|
|
}
|
2023-02-24 10:24:22 -08:00
|
|
|
|
2023-05-25 23:46:51 +03:00
|
|
|
Unifier u{normalizer, constraint->scope, Location{}, Covariant};
|
2023-05-12 15:15:01 +03:00
|
|
|
u.enableScopeTests();
|
2023-02-24 10:24:22 -08:00
|
|
|
|
2023-03-24 10:34:14 -07:00
|
|
|
u.tryUnify(*instantiated, inferredTy, /* isFunctionCall */ true);
|
2023-02-24 10:24:22 -08:00
|
|
|
|
2023-03-24 10:34:14 -07:00
|
|
|
if (!u.blockedTypes.empty() || !u.blockedTypePacks.empty())
|
|
|
|
{
|
|
|
|
for (TypeId bt : u.blockedTypes)
|
|
|
|
block(bt, constraint);
|
|
|
|
for (TypePackId btp : u.blockedTypePacks)
|
|
|
|
block(btp, constraint);
|
|
|
|
return false;
|
|
|
|
}
|
2023-02-24 10:24:22 -08:00
|
|
|
|
2023-03-24 10:34:14 -07:00
|
|
|
if (const auto& e = hasUnificationTooComplex(u.errors))
|
|
|
|
reportError(*e);
|
2023-02-10 10:50:54 -08:00
|
|
|
|
2023-06-02 11:17:31 -07:00
|
|
|
const auto& e = hasCountMismatch(u.errors);
|
|
|
|
bool areArgumentsCompatible = (!e || get<CountMismatch>(*e)->context != CountMismatch::Context::Arg) && get<FunctionType>(*instantiated);
|
|
|
|
if (areArgumentsCompatible)
|
2023-05-05 12:57:12 -07:00
|
|
|
arityMatchingOverloads.push_back(*instantiated);
|
|
|
|
|
2023-03-24 10:34:14 -07:00
|
|
|
if (u.errors.empty())
|
|
|
|
{
|
2023-05-05 12:57:12 -07:00
|
|
|
if (c.callSite)
|
|
|
|
(*c.astOverloadResolvedTypes)[c.callSite] = *instantiated;
|
|
|
|
|
2023-06-02 11:17:31 -07:00
|
|
|
// This overload has no errors, so override the bestOverloadLog and use this one.
|
|
|
|
bestOverloadLog = std::move(u.log);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
else if (areArgumentsCompatible && !bestOverloadLog)
|
|
|
|
{
|
|
|
|
// This overload is erroneous. Replace its inferences with `any` iff there isn't already a TxnLog.
|
|
|
|
bestOverloadLog = std::move(u.log);
|
2023-03-24 10:34:14 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-05-05 12:57:12 -07:00
|
|
|
if (arityMatchingOverloads.size() == 1 && c.callSite)
|
|
|
|
{
|
|
|
|
// In the name of better error messages in the type checker, we provide
|
|
|
|
// it with an instantiated function signature that matched arity, but
|
|
|
|
// not the requisite subtyping requirements. This makes errors better in
|
|
|
|
// cases where only one overload fit from an arity perspective.
|
|
|
|
(*c.astOverloadResolvedTypes)[c.callSite] = arityMatchingOverloads.at(0);
|
|
|
|
}
|
|
|
|
|
2023-06-02 11:17:31 -07:00
|
|
|
// We didn't find any overload that were a viable candidate, so replace the inferences with `any`.
|
|
|
|
if (!bestOverloadLog)
|
|
|
|
{
|
|
|
|
Unifier u{normalizer, constraint->scope, Location{}, Covariant};
|
|
|
|
u.enableScopeTests();
|
|
|
|
|
|
|
|
u.tryUnify(inferredTy, builtinTypes->anyType);
|
|
|
|
u.tryUnify(fn, builtinTypes->anyType);
|
2023-03-24 10:34:14 -07:00
|
|
|
|
2023-06-02 11:17:31 -07:00
|
|
|
bestOverloadLog = std::move(u.log);
|
|
|
|
}
|
2023-03-24 10:34:14 -07:00
|
|
|
|
2023-06-02 11:17:31 -07:00
|
|
|
const auto [changedTypes, changedPacks] = bestOverloadLog->getChanges();
|
|
|
|
bestOverloadLog->commit();
|
2023-03-24 10:34:14 -07:00
|
|
|
|
|
|
|
unblock(changedTypes);
|
|
|
|
unblock(changedPacks);
|
2023-02-10 10:50:54 -08:00
|
|
|
unblock(c.result);
|
2023-05-12 15:15:01 +03:00
|
|
|
|
|
|
|
InstantiationQueuer queuer{constraint->scope, constraint->location, this};
|
|
|
|
queuer.traverse(fn);
|
|
|
|
queuer.traverse(inferredTy);
|
|
|
|
|
2022-09-01 16:00:14 -07:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2022-09-23 11:32:10 -07:00
|
|
|
bool ConstraintSolver::tryDispatch(const PrimitiveTypeConstraint& c, NotNull<const Constraint> constraint)
|
|
|
|
{
|
|
|
|
TypeId expectedType = follow(c.expectedType);
|
2023-01-03 19:33:19 +02:00
|
|
|
if (isBlocked(expectedType) || get<PendingExpansionType>(expectedType))
|
2022-09-23 11:32:10 -07:00
|
|
|
return block(expectedType, constraint);
|
|
|
|
|
2023-04-28 14:55:55 +03:00
|
|
|
LUAU_ASSERT(get<BlockedType>(c.resultType));
|
|
|
|
|
2022-09-23 11:32:10 -07:00
|
|
|
TypeId bindTo = maybeSingleton(expectedType) ? c.singletonType : c.multitonType;
|
2023-01-03 19:33:19 +02:00
|
|
|
asMutable(c.resultType)->ty.emplace<BoundType>(bindTo);
|
2023-03-24 10:34:14 -07:00
|
|
|
unblock(c.resultType);
|
2022-09-23 11:32:10 -07:00
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool ConstraintSolver::tryDispatch(const HasPropConstraint& c, NotNull<const Constraint> constraint)
|
|
|
|
{
|
|
|
|
TypeId subjectType = follow(c.subjectType);
|
|
|
|
|
2023-04-28 14:55:55 +03:00
|
|
|
LUAU_ASSERT(get<BlockedType>(c.resultType));
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
if (isBlocked(subjectType) || get<PendingExpansionType>(subjectType))
|
2022-09-23 11:32:10 -07:00
|
|
|
return block(subjectType, constraint);
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
if (get<FreeType>(subjectType))
|
2022-12-02 12:46:05 +02:00
|
|
|
{
|
2023-01-03 19:33:19 +02:00
|
|
|
TableType& ttv = asMutable(subjectType)->ty.emplace<TableType>(TableState::Free, TypeLevel{}, constraint->scope);
|
2022-12-02 12:46:05 +02:00
|
|
|
ttv.props[c.prop] = Property{c.resultType};
|
2023-01-03 19:33:19 +02:00
|
|
|
asMutable(c.resultType)->ty.emplace<FreeType>(constraint->scope);
|
2022-12-02 12:46:05 +02:00
|
|
|
unblock(c.resultType);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2023-05-19 11:59:59 -07:00
|
|
|
auto [blocked, result] = lookupTableProp(subjectType, c.prop, c.suppressSimplification);
|
2023-03-03 15:45:38 +02:00
|
|
|
if (!blocked.empty())
|
2022-12-02 12:46:05 +02:00
|
|
|
{
|
2023-03-03 15:45:38 +02:00
|
|
|
for (TypeId blocked : blocked)
|
|
|
|
block(blocked, constraint);
|
2022-09-23 11:32:10 -07:00
|
|
|
|
2022-11-18 10:45:14 -08:00
|
|
|
return false;
|
|
|
|
}
|
2022-09-23 11:32:10 -07:00
|
|
|
|
2023-03-17 16:59:30 +02:00
|
|
|
asMutable(c.resultType)->ty.emplace<BoundType>(result.value_or(builtinTypes->anyType));
|
2023-03-03 15:45:38 +02:00
|
|
|
unblock(c.resultType);
|
2022-11-18 10:45:14 -08:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
static bool isUnsealedTable(TypeId ty)
|
|
|
|
{
|
|
|
|
ty = follow(ty);
|
2023-01-03 19:33:19 +02:00
|
|
|
const TableType* ttv = get<TableType>(ty);
|
2022-11-18 10:45:14 -08:00
|
|
|
return ttv && ttv->state == TableState::Unsealed;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
2023-03-24 10:34:14 -07:00
|
|
|
* Given a path into a set of nested unsealed tables `ty`, insert a new property `replaceTy` as the leaf-most property.
|
2022-11-18 10:45:14 -08:00
|
|
|
*
|
2023-03-24 10:34:14 -07:00
|
|
|
* Fails and does nothing if every table along the way is not unsealed.
|
2022-11-18 10:45:14 -08:00
|
|
|
*
|
2023-03-24 10:34:14 -07:00
|
|
|
* Mutates the innermost table type in-place.
|
2022-11-18 10:45:14 -08:00
|
|
|
*/
|
2023-03-24 10:34:14 -07:00
|
|
|
static void updateTheTableType(
|
|
|
|
NotNull<BuiltinTypes> builtinTypes, NotNull<TypeArena> arena, TypeId ty, const std::vector<std::string>& path, TypeId replaceTy)
|
2022-11-18 10:45:14 -08:00
|
|
|
{
|
|
|
|
if (path.empty())
|
2023-03-24 10:34:14 -07:00
|
|
|
return;
|
2022-11-18 10:45:14 -08:00
|
|
|
|
|
|
|
// First walk the path and ensure that it's unsealed tables all the way
|
|
|
|
// to the end.
|
|
|
|
{
|
|
|
|
TypeId t = ty;
|
|
|
|
for (size_t i = 0; i < path.size() - 1; ++i)
|
2022-09-23 11:32:10 -07:00
|
|
|
{
|
2022-11-18 10:45:14 -08:00
|
|
|
if (!isUnsealedTable(t))
|
2023-03-24 10:34:14 -07:00
|
|
|
return;
|
2022-11-18 10:45:14 -08:00
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
const TableType* tbl = get<TableType>(t);
|
2022-11-18 10:45:14 -08:00
|
|
|
auto it = tbl->props.find(path[i]);
|
|
|
|
if (it == tbl->props.end())
|
2023-03-24 10:34:14 -07:00
|
|
|
return;
|
2022-11-18 10:45:14 -08:00
|
|
|
|
2023-04-28 14:55:55 +03:00
|
|
|
t = follow(it->second.type());
|
2022-09-23 11:32:10 -07:00
|
|
|
}
|
|
|
|
|
2022-11-18 10:45:14 -08:00
|
|
|
// The last path segment should not be a property of the table at all.
|
|
|
|
// We are not changing property types. We are only admitting this one
|
|
|
|
// new property to be appended.
|
|
|
|
if (!isUnsealedTable(t))
|
2023-03-24 10:34:14 -07:00
|
|
|
return;
|
2023-01-03 19:33:19 +02:00
|
|
|
const TableType* tbl = get<TableType>(t);
|
2022-11-18 10:45:14 -08:00
|
|
|
if (0 != tbl->props.count(path.back()))
|
2023-03-24 10:34:14 -07:00
|
|
|
return;
|
2022-11-18 10:45:14 -08:00
|
|
|
}
|
2022-09-23 11:32:10 -07:00
|
|
|
|
2023-03-24 10:34:14 -07:00
|
|
|
TypeId t = ty;
|
|
|
|
ErrorVec dummy;
|
2022-11-18 10:45:14 -08:00
|
|
|
|
|
|
|
for (size_t i = 0; i < path.size() - 1; ++i)
|
2022-09-23 11:32:10 -07:00
|
|
|
{
|
2023-03-24 10:34:14 -07:00
|
|
|
auto propTy = findTablePropertyRespectingMeta(builtinTypes, dummy, t, path[i], Location{});
|
|
|
|
dummy.clear();
|
2022-11-18 10:45:14 -08:00
|
|
|
|
2023-03-24 10:34:14 -07:00
|
|
|
if (!propTy)
|
|
|
|
return;
|
2022-11-18 10:45:14 -08:00
|
|
|
|
2023-03-24 10:34:14 -07:00
|
|
|
t = *propTy;
|
2022-09-23 11:32:10 -07:00
|
|
|
}
|
|
|
|
|
2023-03-24 10:34:14 -07:00
|
|
|
const std::string& lastSegment = path.back();
|
|
|
|
|
|
|
|
t = follow(t);
|
|
|
|
TableType* tt = getMutable<TableType>(t);
|
|
|
|
if (auto mt = get<MetatableType>(t))
|
|
|
|
tt = getMutable<TableType>(mt->table);
|
2022-11-18 10:45:14 -08:00
|
|
|
|
2023-03-24 10:34:14 -07:00
|
|
|
if (!tt)
|
|
|
|
return;
|
|
|
|
|
2023-04-28 14:55:55 +03:00
|
|
|
tt->props[lastSegment].setType(replaceTy);
|
2022-11-18 10:45:14 -08:00
|
|
|
}
|
|
|
|
|
2023-01-20 14:02:39 +02:00
|
|
|
bool ConstraintSolver::tryDispatch(const SetPropConstraint& c, NotNull<const Constraint> constraint, bool force)
|
2022-11-18 10:45:14 -08:00
|
|
|
{
|
|
|
|
TypeId subjectType = follow(c.subjectType);
|
|
|
|
|
|
|
|
if (isBlocked(subjectType))
|
|
|
|
return block(subjectType, constraint);
|
|
|
|
|
2023-01-20 14:02:39 +02:00
|
|
|
if (!force && get<FreeType>(subjectType))
|
|
|
|
return block(subjectType, constraint);
|
|
|
|
|
2022-11-18 10:45:14 -08:00
|
|
|
std::optional<TypeId> existingPropType = subjectType;
|
|
|
|
for (const std::string& segment : c.path)
|
|
|
|
{
|
2023-03-03 15:45:38 +02:00
|
|
|
if (!existingPropType)
|
2022-11-18 10:45:14 -08:00
|
|
|
break;
|
2023-03-03 15:45:38 +02:00
|
|
|
|
|
|
|
auto [blocked, result] = lookupTableProp(*existingPropType, segment);
|
|
|
|
if (!blocked.empty())
|
|
|
|
{
|
|
|
|
for (TypeId blocked : blocked)
|
|
|
|
block(blocked, constraint);
|
|
|
|
return false;
|
2022-11-18 10:45:14 -08:00
|
|
|
}
|
2023-03-03 15:45:38 +02:00
|
|
|
|
|
|
|
existingPropType = result;
|
2022-09-23 11:32:10 -07:00
|
|
|
}
|
2022-11-18 10:45:14 -08:00
|
|
|
|
|
|
|
auto bind = [](TypeId a, TypeId b) {
|
2023-01-03 19:33:19 +02:00
|
|
|
asMutable(a)->ty.emplace<BoundType>(b);
|
2022-11-18 10:45:14 -08:00
|
|
|
};
|
|
|
|
|
|
|
|
if (existingPropType)
|
2022-09-23 11:32:10 -07:00
|
|
|
{
|
2023-02-24 10:24:22 -08:00
|
|
|
if (!isBlocked(c.propType))
|
|
|
|
unify(c.propType, *existingPropType, constraint->scope);
|
2022-11-18 10:45:14 -08:00
|
|
|
bind(c.resultType, c.subjectType);
|
2023-03-24 10:34:14 -07:00
|
|
|
unblock(c.resultType);
|
2022-11-18 10:45:14 -08:00
|
|
|
return true;
|
|
|
|
}
|
2022-09-23 11:32:10 -07:00
|
|
|
|
2023-03-03 15:45:38 +02:00
|
|
|
if (auto mt = get<MetatableType>(subjectType))
|
|
|
|
subjectType = follow(mt->table);
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
if (get<FreeType>(subjectType))
|
2022-11-18 10:45:14 -08:00
|
|
|
{
|
|
|
|
TypeId ty = arena->freshType(constraint->scope);
|
|
|
|
|
|
|
|
// Mint a chain of free tables per c.path
|
|
|
|
for (auto it = rbegin(c.path); it != rend(c.path); ++it)
|
|
|
|
{
|
2023-01-03 19:33:19 +02:00
|
|
|
TableType t{TableState::Free, TypeLevel{}, constraint->scope};
|
2022-11-18 10:45:14 -08:00
|
|
|
t.props[*it] = {ty};
|
|
|
|
|
|
|
|
ty = arena->addType(std::move(t));
|
|
|
|
}
|
|
|
|
|
|
|
|
LUAU_ASSERT(ty);
|
|
|
|
|
|
|
|
bind(subjectType, ty);
|
2023-02-24 10:24:22 -08:00
|
|
|
if (follow(c.resultType) != follow(ty))
|
|
|
|
bind(c.resultType, ty);
|
2023-03-24 10:34:14 -07:00
|
|
|
unblock(subjectType);
|
|
|
|
unblock(c.resultType);
|
2022-11-18 10:45:14 -08:00
|
|
|
return true;
|
|
|
|
}
|
2023-01-03 19:33:19 +02:00
|
|
|
else if (auto ttv = getMutable<TableType>(subjectType))
|
2022-11-18 10:45:14 -08:00
|
|
|
{
|
|
|
|
if (ttv->state == TableState::Free)
|
|
|
|
{
|
2023-02-24 10:24:22 -08:00
|
|
|
LUAU_ASSERT(!subjectType->persistent);
|
|
|
|
|
2022-11-18 10:45:14 -08:00
|
|
|
ttv->props[c.path[0]] = Property{c.propType};
|
|
|
|
bind(c.resultType, c.subjectType);
|
2023-03-24 10:34:14 -07:00
|
|
|
unblock(c.resultType);
|
2022-11-18 10:45:14 -08:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
else if (ttv->state == TableState::Unsealed)
|
|
|
|
{
|
2023-02-24 10:24:22 -08:00
|
|
|
LUAU_ASSERT(!subjectType->persistent);
|
|
|
|
|
2023-03-24 10:34:14 -07:00
|
|
|
updateTheTableType(builtinTypes, NotNull{arena}, subjectType, c.path, c.propType);
|
|
|
|
bind(c.resultType, c.subjectType);
|
|
|
|
unblock(subjectType);
|
|
|
|
unblock(c.resultType);
|
2022-11-18 10:45:14 -08:00
|
|
|
return true;
|
|
|
|
}
|
2022-09-23 11:32:10 -07:00
|
|
|
else
|
2022-11-18 10:45:14 -08:00
|
|
|
{
|
|
|
|
bind(c.resultType, subjectType);
|
2023-03-24 10:34:14 -07:00
|
|
|
unblock(c.resultType);
|
2022-11-18 10:45:14 -08:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
}
|
2023-03-10 11:20:04 -08:00
|
|
|
else
|
2023-01-20 14:02:39 +02:00
|
|
|
{
|
2023-03-10 11:20:04 -08:00
|
|
|
// Other kinds of types don't change shape when properties are assigned
|
|
|
|
// to them. (if they allow properties at all!)
|
2023-01-20 14:02:39 +02:00
|
|
|
bind(c.resultType, subjectType);
|
2023-03-24 10:34:14 -07:00
|
|
|
unblock(c.resultType);
|
2023-01-20 14:02:39 +02:00
|
|
|
return true;
|
|
|
|
}
|
2023-02-24 10:24:22 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
bool ConstraintSolver::tryDispatch(const SetIndexerConstraint& c, NotNull<const Constraint> constraint, bool force)
|
|
|
|
{
|
|
|
|
TypeId subjectType = follow(c.subjectType);
|
|
|
|
if (isBlocked(subjectType))
|
|
|
|
return block(subjectType, constraint);
|
|
|
|
|
|
|
|
if (auto ft = get<FreeType>(subjectType))
|
2022-11-18 10:45:14 -08:00
|
|
|
{
|
2023-02-24 10:24:22 -08:00
|
|
|
Scope* scope = ft->scope;
|
|
|
|
TableType* tt = &asMutable(subjectType)->ty.emplace<TableType>(TableState::Free, TypeLevel{}, scope);
|
|
|
|
tt->indexer = TableIndexer{c.indexType, c.propType};
|
|
|
|
|
|
|
|
asMutable(c.resultType)->ty.emplace<BoundType>(subjectType);
|
|
|
|
asMutable(c.propType)->ty.emplace<FreeType>(scope);
|
|
|
|
unblock(c.propType);
|
|
|
|
unblock(c.resultType);
|
|
|
|
|
2022-11-18 10:45:14 -08:00
|
|
|
return true;
|
2022-09-23 11:32:10 -07:00
|
|
|
}
|
2023-02-24 10:24:22 -08:00
|
|
|
else if (auto tt = get<TableType>(subjectType))
|
|
|
|
{
|
|
|
|
if (tt->indexer)
|
|
|
|
{
|
|
|
|
// TODO This probably has to be invariant.
|
|
|
|
unify(c.indexType, tt->indexer->indexType, constraint->scope);
|
|
|
|
asMutable(c.propType)->ty.emplace<BoundType>(tt->indexer->indexResultType);
|
|
|
|
asMutable(c.resultType)->ty.emplace<BoundType>(subjectType);
|
|
|
|
unblock(c.propType);
|
|
|
|
unblock(c.resultType);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
else if (tt->state == TableState::Free || tt->state == TableState::Unsealed)
|
|
|
|
{
|
2023-05-12 15:15:01 +03:00
|
|
|
TypeId promotedIndexTy = arena->freshType(tt->scope);
|
|
|
|
unify(c.indexType, promotedIndexTy, constraint->scope);
|
|
|
|
|
2023-02-24 10:24:22 -08:00
|
|
|
auto mtt = getMutable<TableType>(subjectType);
|
2023-05-12 15:15:01 +03:00
|
|
|
mtt->indexer = TableIndexer{promotedIndexTy, c.propType};
|
2023-02-24 10:24:22 -08:00
|
|
|
asMutable(c.propType)->ty.emplace<FreeType>(tt->scope);
|
|
|
|
asMutable(c.resultType)->ty.emplace<BoundType>(subjectType);
|
|
|
|
unblock(c.propType);
|
|
|
|
unblock(c.resultType);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
// Do not augment sealed or generic tables that lack indexers
|
|
|
|
}
|
2022-09-23 11:32:10 -07:00
|
|
|
|
2023-02-24 10:24:22 -08:00
|
|
|
asMutable(c.propType)->ty.emplace<BoundType>(builtinTypes->errorRecoveryType());
|
|
|
|
asMutable(c.resultType)->ty.emplace<BoundType>(builtinTypes->errorRecoveryType());
|
|
|
|
unblock(c.propType);
|
|
|
|
unblock(c.resultType);
|
2022-09-23 11:32:10 -07:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2022-11-04 10:02:37 -07:00
|
|
|
bool ConstraintSolver::tryDispatch(const SingletonOrTopTypeConstraint& c, NotNull<const Constraint> constraint)
|
2022-10-21 10:33:43 -07:00
|
|
|
{
|
2022-11-04 10:02:37 -07:00
|
|
|
if (isBlocked(c.discriminantType))
|
|
|
|
return false;
|
2022-10-21 10:33:43 -07:00
|
|
|
|
2022-11-04 10:02:37 -07:00
|
|
|
TypeId followed = follow(c.discriminantType);
|
2022-10-21 10:33:43 -07:00
|
|
|
|
2022-11-04 10:02:37 -07:00
|
|
|
// `nil` is a singleton type too! There's only one value of type `nil`.
|
2023-01-03 19:33:19 +02:00
|
|
|
if (c.negated && (get<SingletonType>(followed) || isNil(followed)))
|
|
|
|
*asMutable(c.resultType) = NegationType{c.discriminantType};
|
|
|
|
else if (!c.negated && get<SingletonType>(followed))
|
|
|
|
*asMutable(c.resultType) = BoundType{c.discriminantType};
|
2022-11-04 10:02:37 -07:00
|
|
|
else
|
2023-05-19 11:59:59 -07:00
|
|
|
*asMutable(c.resultType) = BoundType{builtinTypes->anyType};
|
|
|
|
|
|
|
|
unblock(c.resultType);
|
2022-10-21 10:33:43 -07:00
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2023-02-24 10:24:22 -08:00
|
|
|
bool ConstraintSolver::tryDispatch(const UnpackConstraint& c, NotNull<const Constraint> constraint)
|
|
|
|
{
|
|
|
|
TypePackId sourcePack = follow(c.sourcePack);
|
|
|
|
TypePackId resultPack = follow(c.resultPack);
|
|
|
|
|
|
|
|
if (isBlocked(sourcePack))
|
|
|
|
return block(sourcePack, constraint);
|
|
|
|
|
|
|
|
if (isBlocked(resultPack))
|
|
|
|
{
|
|
|
|
asMutable(resultPack)->ty.emplace<BoundTypePack>(sourcePack);
|
|
|
|
unblock(resultPack);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
TypePack srcPack = extendTypePack(*arena, builtinTypes, sourcePack, size(resultPack));
|
|
|
|
|
|
|
|
auto destIter = begin(resultPack);
|
|
|
|
auto destEnd = end(resultPack);
|
|
|
|
|
|
|
|
size_t i = 0;
|
|
|
|
while (destIter != destEnd)
|
|
|
|
{
|
|
|
|
if (i >= srcPack.head.size())
|
|
|
|
break;
|
|
|
|
TypeId srcTy = follow(srcPack.head[i]);
|
|
|
|
|
|
|
|
if (isBlocked(*destIter))
|
|
|
|
{
|
|
|
|
if (follow(srcTy) == *destIter)
|
|
|
|
{
|
|
|
|
// Cyclic type dependency. (????)
|
|
|
|
asMutable(*destIter)->ty.emplace<FreeType>(constraint->scope);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
asMutable(*destIter)->ty.emplace<BoundType>(srcTy);
|
|
|
|
unblock(*destIter);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
unify(*destIter, srcTy, constraint->scope);
|
|
|
|
|
|
|
|
++destIter;
|
|
|
|
++i;
|
|
|
|
}
|
|
|
|
|
|
|
|
// We know that resultPack does not have a tail, but we don't know if
|
|
|
|
// sourcePack is long enough to fill every value. Replace every remaining
|
|
|
|
// result TypeId with the error recovery type.
|
|
|
|
|
|
|
|
while (destIter != destEnd)
|
|
|
|
{
|
|
|
|
if (isBlocked(*destIter))
|
|
|
|
{
|
|
|
|
asMutable(*destIter)->ty.emplace<BoundType>(builtinTypes->errorRecoveryType());
|
|
|
|
unblock(*destIter);
|
|
|
|
}
|
|
|
|
|
|
|
|
++destIter;
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2023-05-19 11:59:59 -07:00
|
|
|
namespace
|
|
|
|
{
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Search for types that prevent us from being ready to dispatch a particular
|
|
|
|
* RefineConstraint.
|
|
|
|
*/
|
|
|
|
struct FindRefineConstraintBlockers : TypeOnceVisitor
|
|
|
|
{
|
|
|
|
std::unordered_set<TypeId> found;
|
|
|
|
bool visit(TypeId ty, const BlockedType&) override
|
|
|
|
{
|
|
|
|
found.insert(ty);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool visit(TypeId ty, const PendingExpansionType&) override
|
|
|
|
{
|
|
|
|
found.insert(ty);
|
|
|
|
return false;
|
|
|
|
}
|
2023-05-25 23:46:51 +03:00
|
|
|
|
|
|
|
bool visit(TypeId ty, const ClassType&) override
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|
2023-05-19 11:59:59 -07:00
|
|
|
};
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
static bool isNegatedAny(TypeId ty)
|
|
|
|
{
|
|
|
|
ty = follow(ty);
|
|
|
|
const NegationType* nt = get<NegationType>(ty);
|
|
|
|
if (!nt)
|
|
|
|
return false;
|
|
|
|
TypeId negatedTy = follow(nt->ty);
|
|
|
|
return bool(get<AnyType>(negatedTy));
|
|
|
|
}
|
|
|
|
|
|
|
|
bool ConstraintSolver::tryDispatch(const RefineConstraint& c, NotNull<const Constraint> constraint, bool force)
|
|
|
|
{
|
|
|
|
if (isBlocked(c.discriminant))
|
|
|
|
return block(c.discriminant, constraint);
|
|
|
|
|
|
|
|
FindRefineConstraintBlockers fbt;
|
|
|
|
fbt.traverse(c.discriminant);
|
|
|
|
|
|
|
|
if (!fbt.found.empty())
|
|
|
|
{
|
|
|
|
bool foundOne = false;
|
|
|
|
|
|
|
|
for (TypeId blocked : fbt.found)
|
|
|
|
{
|
|
|
|
if (blocked == c.type)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
block(blocked, constraint);
|
|
|
|
foundOne = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (foundOne)
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* HACK: Refinements sometimes produce a type T & ~any under the assumption
|
|
|
|
* that ~any is the same as any. This is so so weird, but refinements needs
|
|
|
|
* some way to say "I may refine this, but I'm not sure."
|
|
|
|
*
|
|
|
|
* It does this by refining on a blocked type and deferring the decision
|
|
|
|
* until it is unblocked.
|
|
|
|
*
|
|
|
|
* Refinements also get negated, so we wind up with types like T & ~*blocked*
|
|
|
|
*
|
|
|
|
* We need to treat T & ~any as T in this case.
|
|
|
|
*/
|
|
|
|
|
|
|
|
if (c.mode == RefineConstraint::Intersection && isNegatedAny(c.discriminant))
|
|
|
|
{
|
|
|
|
asMutable(c.resultType)->ty.emplace<BoundType>(c.type);
|
|
|
|
unblock(c.resultType);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
const TypeId type = follow(c.type);
|
|
|
|
|
|
|
|
LUAU_ASSERT(get<BlockedType>(c.resultType));
|
|
|
|
|
|
|
|
if (type == c.resultType)
|
|
|
|
{
|
|
|
|
/*
|
|
|
|
* Sometimes, we get a constraint of the form
|
|
|
|
*
|
|
|
|
* *blocked-N* ~ refine *blocked-N* & U
|
|
|
|
*
|
|
|
|
* The constraint essentially states that a particular type is a
|
|
|
|
* refinement of itself. This is weird and I think vacuous.
|
|
|
|
*
|
|
|
|
* I *believe* it is safe to replace the result with a fresh type that
|
|
|
|
* is constrained by U. We effect this by minting a fresh type for the
|
|
|
|
* result when U = any, else we bind the result to whatever discriminant
|
|
|
|
* was offered.
|
|
|
|
*/
|
|
|
|
if (get<AnyType>(follow(c.discriminant)))
|
|
|
|
asMutable(c.resultType)->ty.emplace<FreeType>(constraint->scope);
|
|
|
|
else
|
|
|
|
asMutable(c.resultType)->ty.emplace<BoundType>(c.discriminant);
|
|
|
|
|
|
|
|
unblock(c.resultType);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
auto [result, blockedTypes] = c.mode == RefineConstraint::Intersection ? simplifyIntersection(builtinTypes, NotNull{arena}, type, c.discriminant)
|
|
|
|
: simplifyUnion(builtinTypes, NotNull{arena}, type, c.discriminant);
|
|
|
|
|
|
|
|
if (!force && !blockedTypes.empty())
|
|
|
|
return block(blockedTypes, constraint);
|
|
|
|
|
|
|
|
asMutable(c.resultType)->ty.emplace<BoundType>(result);
|
|
|
|
|
|
|
|
unblock(c.resultType);
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2023-05-12 15:15:01 +03:00
|
|
|
bool ConstraintSolver::tryDispatch(const ReduceConstraint& c, NotNull<const Constraint> constraint, bool force)
|
|
|
|
{
|
|
|
|
TypeId ty = follow(c.ty);
|
2023-05-19 11:59:59 -07:00
|
|
|
FamilyGraphReductionResult result =
|
|
|
|
reduceFamilies(ty, constraint->location, NotNull{arena}, builtinTypes, constraint->scope, normalizer, nullptr, force);
|
2023-05-12 15:15:01 +03:00
|
|
|
|
|
|
|
for (TypeId r : result.reducedTypes)
|
|
|
|
unblock(r);
|
|
|
|
|
|
|
|
for (TypePackId r : result.reducedPacks)
|
|
|
|
unblock(r);
|
|
|
|
|
|
|
|
if (force)
|
|
|
|
return true;
|
|
|
|
|
|
|
|
for (TypeId b : result.blockedTypes)
|
|
|
|
block(b, constraint);
|
|
|
|
|
|
|
|
for (TypePackId b : result.blockedPacks)
|
|
|
|
block(b, constraint);
|
|
|
|
|
|
|
|
return result.blockedTypes.empty() && result.blockedPacks.empty();
|
|
|
|
}
|
|
|
|
|
|
|
|
bool ConstraintSolver::tryDispatch(const ReducePackConstraint& c, NotNull<const Constraint> constraint, bool force)
|
|
|
|
{
|
|
|
|
TypePackId tp = follow(c.tp);
|
2023-05-19 11:59:59 -07:00
|
|
|
FamilyGraphReductionResult result =
|
|
|
|
reduceFamilies(tp, constraint->location, NotNull{arena}, builtinTypes, constraint->scope, normalizer, nullptr, force);
|
2023-05-12 15:15:01 +03:00
|
|
|
|
|
|
|
for (TypeId r : result.reducedTypes)
|
|
|
|
unblock(r);
|
|
|
|
|
|
|
|
for (TypePackId r : result.reducedPacks)
|
|
|
|
unblock(r);
|
|
|
|
|
|
|
|
if (force)
|
|
|
|
return true;
|
|
|
|
|
|
|
|
for (TypeId b : result.blockedTypes)
|
|
|
|
block(b, constraint);
|
|
|
|
|
|
|
|
for (TypePackId b : result.blockedPacks)
|
|
|
|
block(b, constraint);
|
|
|
|
|
|
|
|
return result.blockedTypes.empty() && result.blockedPacks.empty();
|
|
|
|
}
|
|
|
|
|
2022-09-01 16:00:14 -07:00
|
|
|
bool ConstraintSolver::tryDispatchIterableTable(TypeId iteratorTy, const IterableConstraint& c, NotNull<const Constraint> constraint, bool force)
|
|
|
|
{
|
|
|
|
auto block_ = [&](auto&& t) {
|
|
|
|
if (force)
|
|
|
|
{
|
|
|
|
// TODO: I believe it is the case that, if we are asked to force
|
|
|
|
// this constraint, then we can do nothing but fail. I'd like to
|
|
|
|
// find a code sample that gets here.
|
2022-09-23 11:32:10 -07:00
|
|
|
LUAU_ASSERT(false);
|
2022-09-01 16:00:14 -07:00
|
|
|
}
|
|
|
|
else
|
|
|
|
block(t, constraint);
|
|
|
|
return false;
|
|
|
|
};
|
|
|
|
|
|
|
|
// We may have to block here if we don't know what the iteratee type is,
|
|
|
|
// if it's a free table, if we don't know it has a metatable, and so on.
|
|
|
|
iteratorTy = follow(iteratorTy);
|
2023-01-03 19:33:19 +02:00
|
|
|
if (get<FreeType>(iteratorTy))
|
2022-09-01 16:00:14 -07:00
|
|
|
return block_(iteratorTy);
|
|
|
|
|
|
|
|
auto anyify = [&](auto ty) {
|
2023-01-03 19:33:19 +02:00
|
|
|
Anyification anyify{arena, constraint->scope, builtinTypes, &iceReporter, builtinTypes->anyType, builtinTypes->anyTypePack};
|
2022-09-01 16:00:14 -07:00
|
|
|
std::optional anyified = anyify.substitute(ty);
|
|
|
|
if (!anyified)
|
|
|
|
reportError(CodeTooComplex{}, constraint->location);
|
|
|
|
else
|
|
|
|
unify(*anyified, ty, constraint->scope);
|
|
|
|
};
|
|
|
|
|
|
|
|
auto errorify = [&](auto ty) {
|
2023-01-03 19:33:19 +02:00
|
|
|
Anyification anyify{arena, constraint->scope, builtinTypes, &iceReporter, errorRecoveryType(), errorRecoveryTypePack()};
|
2022-09-01 16:00:14 -07:00
|
|
|
std::optional errorified = anyify.substitute(ty);
|
|
|
|
if (!errorified)
|
|
|
|
reportError(CodeTooComplex{}, constraint->location);
|
|
|
|
else
|
|
|
|
unify(*errorified, ty, constraint->scope);
|
|
|
|
};
|
|
|
|
|
2023-05-25 23:46:51 +03:00
|
|
|
auto neverify = [&](auto ty) {
|
|
|
|
Anyification anyify{arena, constraint->scope, builtinTypes, &iceReporter, builtinTypes->neverType, builtinTypes->neverTypePack};
|
|
|
|
std::optional neverified = anyify.substitute(ty);
|
|
|
|
if (!neverified)
|
|
|
|
reportError(CodeTooComplex{}, constraint->location);
|
|
|
|
else
|
|
|
|
unify(*neverified, ty, constraint->scope);
|
|
|
|
};
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
if (get<AnyType>(iteratorTy))
|
2022-09-01 16:00:14 -07:00
|
|
|
{
|
|
|
|
anyify(c.variables);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
if (get<ErrorType>(iteratorTy))
|
2022-09-01 16:00:14 -07:00
|
|
|
{
|
|
|
|
errorify(c.variables);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2023-05-25 23:46:51 +03:00
|
|
|
if (get<NeverType>(iteratorTy))
|
|
|
|
{
|
|
|
|
neverify(c.variables);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2022-09-01 16:00:14 -07:00
|
|
|
// Irksome: I don't think we have any way to guarantee that this table
|
|
|
|
// type never has a metatable.
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
if (auto iteratorTable = get<TableType>(iteratorTy))
|
2022-09-01 16:00:14 -07:00
|
|
|
{
|
2023-03-31 15:21:14 +03:00
|
|
|
/*
|
|
|
|
* We try not to dispatch IterableConstraints over free tables because
|
|
|
|
* it's possible that there are other constraints on the table that will
|
|
|
|
* clarify what we should do.
|
|
|
|
*
|
|
|
|
* We should eventually introduce a type family to talk about iteration.
|
|
|
|
*/
|
|
|
|
if (iteratorTable->state == TableState::Free && !force)
|
|
|
|
return block(iteratorTy, constraint);
|
2022-09-01 16:00:14 -07:00
|
|
|
|
|
|
|
if (iteratorTable->indexer)
|
|
|
|
{
|
|
|
|
TypePackId expectedVariablePack = arena->addTypePack({iteratorTable->indexer->indexType, iteratorTable->indexer->indexResultType});
|
|
|
|
unify(c.variables, expectedVariablePack, constraint->scope);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
errorify(c.variables);
|
|
|
|
}
|
2023-01-03 19:33:19 +02:00
|
|
|
else if (std::optional<TypeId> iterFn = findMetatableEntry(builtinTypes, errors, iteratorTy, "__iter", Location{}))
|
2022-09-29 15:11:54 -07:00
|
|
|
{
|
|
|
|
if (isBlocked(*iterFn))
|
|
|
|
{
|
|
|
|
return block(*iterFn, constraint);
|
|
|
|
}
|
|
|
|
|
|
|
|
Instantiation instantiation(TxnLog::empty(), arena, TypeLevel{}, constraint->scope);
|
|
|
|
|
|
|
|
if (std::optional<TypeId> instantiatedIterFn = instantiation.substitute(*iterFn))
|
|
|
|
{
|
2023-01-03 19:33:19 +02:00
|
|
|
if (auto iterFtv = get<FunctionType>(*instantiatedIterFn))
|
2022-09-29 15:11:54 -07:00
|
|
|
{
|
|
|
|
TypePackId expectedIterArgs = arena->addTypePack({iteratorTy});
|
|
|
|
unify(iterFtv->argTypes, expectedIterArgs, constraint->scope);
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
TypePack iterRets = extendTypePack(*arena, builtinTypes, iterFtv->retTypes, 2);
|
2022-09-29 15:11:54 -07:00
|
|
|
|
2022-12-02 12:46:05 +02:00
|
|
|
if (iterRets.head.size() < 1)
|
2022-09-29 15:11:54 -07:00
|
|
|
{
|
|
|
|
// We've done what we can; this will get reported as an
|
|
|
|
// error by the type checker.
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2022-12-02 12:46:05 +02:00
|
|
|
TypeId nextFn = iterRets.head[0];
|
|
|
|
TypeId table = iterRets.head.size() == 2 ? iterRets.head[1] : arena->freshType(constraint->scope);
|
2022-09-29 15:11:54 -07:00
|
|
|
|
|
|
|
if (std::optional<TypeId> instantiatedNextFn = instantiation.substitute(nextFn))
|
|
|
|
{
|
|
|
|
const TypeId firstIndex = arena->freshType(constraint->scope);
|
|
|
|
|
|
|
|
// nextTy : (iteratorTy, indexTy?) -> (indexTy, valueTailTy...)
|
2023-01-03 19:33:19 +02:00
|
|
|
const TypePackId nextArgPack = arena->addTypePack({table, arena->addType(UnionType{{firstIndex, builtinTypes->nilType}})});
|
2022-09-29 15:11:54 -07:00
|
|
|
const TypePackId valueTailTy = arena->addTypePack(FreeTypePack{constraint->scope});
|
|
|
|
const TypePackId nextRetPack = arena->addTypePack(TypePack{{firstIndex}, valueTailTy});
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
const TypeId expectedNextTy = arena->addType(FunctionType{nextArgPack, nextRetPack});
|
2022-09-29 15:11:54 -07:00
|
|
|
unify(*instantiatedNextFn, expectedNextTy, constraint->scope);
|
|
|
|
|
|
|
|
pushConstraint(constraint->scope, constraint->location, PackSubtypeConstraint{c.variables, nextRetPack});
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
reportError(UnificationTooComplex{}, constraint->location);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
// TODO: Support __call and function overloads (what does an overload even mean for this?)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
reportError(UnificationTooComplex{}, constraint->location);
|
|
|
|
}
|
|
|
|
}
|
2023-01-03 19:33:19 +02:00
|
|
|
else if (auto iteratorMetatable = get<MetatableType>(iteratorTy))
|
2022-09-01 16:00:14 -07:00
|
|
|
{
|
|
|
|
TypeId metaTy = follow(iteratorMetatable->metatable);
|
2023-01-03 19:33:19 +02:00
|
|
|
if (get<FreeType>(metaTy))
|
2022-09-01 16:00:14 -07:00
|
|
|
return block_(metaTy);
|
|
|
|
|
2022-09-23 11:32:10 -07:00
|
|
|
LUAU_ASSERT(false);
|
2022-09-01 16:00:14 -07:00
|
|
|
}
|
|
|
|
else
|
|
|
|
errorify(c.variables);
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool ConstraintSolver::tryDispatchIterableFunction(
|
|
|
|
TypeId nextTy, TypeId tableTy, TypeId firstIndexTy, const IterableConstraint& c, NotNull<const Constraint> constraint, bool force)
|
|
|
|
{
|
|
|
|
// We need to know whether or not this type is nil or not.
|
|
|
|
// If we don't know, block and reschedule ourselves.
|
|
|
|
firstIndexTy = follow(firstIndexTy);
|
2023-01-03 19:33:19 +02:00
|
|
|
if (get<FreeType>(firstIndexTy))
|
2022-09-01 16:00:14 -07:00
|
|
|
{
|
|
|
|
if (force)
|
2022-09-23 11:32:10 -07:00
|
|
|
LUAU_ASSERT(false);
|
2022-09-01 16:00:14 -07:00
|
|
|
else
|
|
|
|
block(firstIndexTy, constraint);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2023-03-17 16:59:30 +02:00
|
|
|
TypeId firstIndex;
|
|
|
|
TypeId retIndex;
|
|
|
|
if (isNil(firstIndexTy) || isOptional(firstIndexTy))
|
|
|
|
{
|
|
|
|
firstIndex = arena->addType(UnionType{{arena->freshType(constraint->scope), builtinTypes->nilType}});
|
|
|
|
retIndex = firstIndex;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
firstIndex = firstIndexTy;
|
|
|
|
retIndex = arena->addType(UnionType{{firstIndexTy, builtinTypes->nilType}});
|
|
|
|
}
|
2022-09-01 16:00:14 -07:00
|
|
|
|
2023-01-27 13:28:45 -08:00
|
|
|
// nextTy : (tableTy, indexTy?) -> (indexTy?, valueTailTy...)
|
2023-03-17 16:59:30 +02:00
|
|
|
const TypePackId nextArgPack = arena->addTypePack({tableTy, firstIndex});
|
2022-09-01 16:00:14 -07:00
|
|
|
const TypePackId valueTailTy = arena->addTypePack(FreeTypePack{constraint->scope});
|
2023-03-17 16:59:30 +02:00
|
|
|
const TypePackId nextRetPack = arena->addTypePack(TypePack{{retIndex}, valueTailTy});
|
2022-09-01 16:00:14 -07:00
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
const TypeId expectedNextTy = arena->addType(FunctionType{TypeLevel{}, constraint->scope, nextArgPack, nextRetPack});
|
2023-05-25 23:46:51 +03:00
|
|
|
ErrorVec errors = unify(nextTy, expectedNextTy, constraint->scope);
|
|
|
|
|
|
|
|
// if there are no errors from unifying the two, we can pass forward the expected type as our selected resolution.
|
|
|
|
if (errors.empty())
|
|
|
|
(*c.astOverloadResolvedTypes)[c.nextAstFragment] = expectedNextTy;
|
2022-09-01 16:00:14 -07:00
|
|
|
|
2023-01-27 13:28:45 -08:00
|
|
|
auto it = begin(nextRetPack);
|
|
|
|
std::vector<TypeId> modifiedNextRetHead;
|
|
|
|
|
|
|
|
// The first value is never nil in the context of the loop, even if it's nil
|
|
|
|
// in the next function's return type, because the loop will not advance if
|
|
|
|
// it's nil.
|
|
|
|
if (it != end(nextRetPack))
|
|
|
|
{
|
|
|
|
TypeId firstRet = *it;
|
|
|
|
TypeId modifiedFirstRet = stripNil(builtinTypes, *arena, firstRet);
|
|
|
|
modifiedNextRetHead.push_back(modifiedFirstRet);
|
|
|
|
++it;
|
|
|
|
}
|
|
|
|
|
|
|
|
for (; it != end(nextRetPack); ++it)
|
|
|
|
modifiedNextRetHead.push_back(*it);
|
|
|
|
|
|
|
|
TypePackId modifiedNextRetPack = arena->addTypePack(std::move(modifiedNextRetHead), it.tail());
|
2023-03-17 16:59:30 +02:00
|
|
|
auto psc = pushConstraint(constraint->scope, constraint->location, PackSubtypeConstraint{c.variables, modifiedNextRetPack});
|
|
|
|
inheritBlocks(constraint, psc);
|
2022-09-01 16:00:14 -07:00
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2023-05-19 11:59:59 -07:00
|
|
|
std::pair<std::vector<TypeId>, std::optional<TypeId>> ConstraintSolver::lookupTableProp(
|
|
|
|
TypeId subjectType, const std::string& propName, bool suppressSimplification)
|
2022-11-18 10:45:14 -08:00
|
|
|
{
|
2023-02-24 10:24:22 -08:00
|
|
|
std::unordered_set<TypeId> seen;
|
2023-05-19 11:59:59 -07:00
|
|
|
return lookupTableProp(subjectType, propName, suppressSimplification, seen);
|
2023-02-24 10:24:22 -08:00
|
|
|
}
|
|
|
|
|
2023-05-19 11:59:59 -07:00
|
|
|
std::pair<std::vector<TypeId>, std::optional<TypeId>> ConstraintSolver::lookupTableProp(
|
|
|
|
TypeId subjectType, const std::string& propName, bool suppressSimplification, std::unordered_set<TypeId>& seen)
|
2023-02-24 10:24:22 -08:00
|
|
|
{
|
|
|
|
if (!seen.insert(subjectType).second)
|
2023-03-03 15:45:38 +02:00
|
|
|
return {};
|
2022-11-18 10:45:14 -08:00
|
|
|
|
2023-03-03 15:45:38 +02:00
|
|
|
subjectType = follow(subjectType);
|
2022-11-18 10:45:14 -08:00
|
|
|
|
2023-03-03 15:45:38 +02:00
|
|
|
if (isBlocked(subjectType))
|
|
|
|
return {{subjectType}, std::nullopt};
|
|
|
|
else if (get<AnyType>(subjectType) || get<NeverType>(subjectType))
|
2023-02-24 10:24:22 -08:00
|
|
|
{
|
2023-03-03 15:45:38 +02:00
|
|
|
return {{}, subjectType};
|
2023-02-24 10:24:22 -08:00
|
|
|
}
|
|
|
|
else if (auto ttv = getMutable<TableType>(subjectType))
|
2022-11-18 10:45:14 -08:00
|
|
|
{
|
|
|
|
if (auto prop = ttv->props.find(propName); prop != ttv->props.end())
|
2023-05-25 23:46:51 +03:00
|
|
|
return {{}, FFlag::DebugLuauReadWriteProperties ? prop->second.readType() : prop->second.type()};
|
2022-11-18 10:45:14 -08:00
|
|
|
else if (ttv->indexer && maybeString(ttv->indexer->indexType))
|
2023-03-03 15:45:38 +02:00
|
|
|
return {{}, ttv->indexer->indexResultType};
|
2023-02-24 10:24:22 -08:00
|
|
|
else if (ttv->state == TableState::Free)
|
|
|
|
{
|
2023-03-03 15:45:38 +02:00
|
|
|
TypeId result = arena->freshType(ttv->scope);
|
|
|
|
ttv->props[propName] = Property{result};
|
|
|
|
return {{}, result};
|
2023-02-24 10:24:22 -08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
else if (auto mt = get<MetatableType>(subjectType))
|
|
|
|
{
|
2023-05-19 11:59:59 -07:00
|
|
|
auto [blocked, result] = lookupTableProp(mt->table, propName, suppressSimplification, seen);
|
2023-03-03 15:45:38 +02:00
|
|
|
if (!blocked.empty() || result)
|
|
|
|
return {blocked, result};
|
2023-02-24 10:24:22 -08:00
|
|
|
|
|
|
|
TypeId mtt = follow(mt->metatable);
|
|
|
|
|
|
|
|
if (get<BlockedType>(mtt))
|
2023-03-03 15:45:38 +02:00
|
|
|
return {{mtt}, std::nullopt};
|
2023-02-24 10:24:22 -08:00
|
|
|
else if (auto metatable = get<TableType>(mtt))
|
|
|
|
{
|
|
|
|
auto indexProp = metatable->props.find("__index");
|
|
|
|
if (indexProp == metatable->props.end())
|
2023-03-03 15:45:38 +02:00
|
|
|
return {{}, result};
|
2023-02-24 10:24:22 -08:00
|
|
|
|
|
|
|
// TODO: __index can be an overloaded function.
|
|
|
|
|
2023-04-28 14:55:55 +03:00
|
|
|
TypeId indexType = follow(indexProp->second.type());
|
2023-02-24 10:24:22 -08:00
|
|
|
|
|
|
|
if (auto ft = get<FunctionType>(indexType))
|
2023-03-17 16:59:30 +02:00
|
|
|
{
|
|
|
|
TypePack rets = extendTypePack(*arena, builtinTypes, ft->retTypes, 1);
|
|
|
|
if (1 == rets.head.size())
|
|
|
|
return {{}, rets.head[0]};
|
|
|
|
else
|
|
|
|
{
|
|
|
|
// This should probably be an error: We need the first result of the MT.__index method,
|
|
|
|
// but it returns 0 values. See CLI-68672
|
|
|
|
return {{}, builtinTypes->nilType};
|
|
|
|
}
|
|
|
|
}
|
2023-03-03 15:45:38 +02:00
|
|
|
else
|
2023-05-19 11:59:59 -07:00
|
|
|
return lookupTableProp(indexType, propName, suppressSimplification, seen);
|
2023-02-24 10:24:22 -08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
else if (auto ct = get<ClassType>(subjectType))
|
|
|
|
{
|
2023-03-03 15:45:38 +02:00
|
|
|
if (auto p = lookupClassProp(ct, propName))
|
2023-04-28 14:55:55 +03:00
|
|
|
return {{}, p->type()};
|
2023-05-19 11:59:59 -07:00
|
|
|
if (ct->indexer)
|
|
|
|
{
|
|
|
|
return {{}, ct->indexer->indexResultType};
|
|
|
|
}
|
2023-02-24 10:24:22 -08:00
|
|
|
}
|
|
|
|
else if (auto pt = get<PrimitiveType>(subjectType); pt && pt->metatable)
|
|
|
|
{
|
|
|
|
const TableType* metatable = get<TableType>(follow(*pt->metatable));
|
|
|
|
LUAU_ASSERT(metatable);
|
|
|
|
|
|
|
|
auto indexProp = metatable->props.find("__index");
|
|
|
|
if (indexProp == metatable->props.end())
|
2023-03-03 15:45:38 +02:00
|
|
|
return {{}, std::nullopt};
|
2023-02-24 10:24:22 -08:00
|
|
|
|
2023-05-19 11:59:59 -07:00
|
|
|
return lookupTableProp(indexProp->second.type(), propName, suppressSimplification, seen);
|
2022-11-18 10:45:14 -08:00
|
|
|
}
|
2023-03-03 15:45:38 +02:00
|
|
|
else if (auto ft = get<FreeType>(subjectType))
|
|
|
|
{
|
|
|
|
Scope* scope = ft->scope;
|
|
|
|
|
|
|
|
TableType* tt = &asMutable(subjectType)->ty.emplace<TableType>();
|
|
|
|
tt->state = TableState::Free;
|
|
|
|
tt->scope = scope;
|
|
|
|
TypeId propType = arena->freshType(scope);
|
|
|
|
tt->props[propName] = Property{propType};
|
|
|
|
|
|
|
|
return {{}, propType};
|
|
|
|
}
|
2023-01-03 19:33:19 +02:00
|
|
|
else if (auto utv = get<UnionType>(subjectType))
|
2022-11-18 10:45:14 -08:00
|
|
|
{
|
2023-03-03 15:45:38 +02:00
|
|
|
std::vector<TypeId> blocked;
|
2023-03-31 15:21:14 +03:00
|
|
|
std::set<TypeId> options;
|
2022-11-18 10:45:14 -08:00
|
|
|
|
2023-03-03 15:45:38 +02:00
|
|
|
for (TypeId ty : utv)
|
|
|
|
{
|
2023-05-19 11:59:59 -07:00
|
|
|
auto [innerBlocked, innerResult] = lookupTableProp(ty, propName, suppressSimplification, seen);
|
2023-03-03 15:45:38 +02:00
|
|
|
blocked.insert(blocked.end(), innerBlocked.begin(), innerBlocked.end());
|
|
|
|
if (innerResult)
|
2023-03-31 15:21:14 +03:00
|
|
|
options.insert(*innerResult);
|
2023-03-03 15:45:38 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
if (!blocked.empty())
|
|
|
|
return {blocked, std::nullopt};
|
2023-01-27 13:28:45 -08:00
|
|
|
|
2023-03-03 15:45:38 +02:00
|
|
|
if (options.empty())
|
|
|
|
return {{}, std::nullopt};
|
|
|
|
else if (options.size() == 1)
|
2023-03-31 15:21:14 +03:00
|
|
|
return {{}, *begin(options)};
|
2023-05-19 11:59:59 -07:00
|
|
|
else if (options.size() == 2 && !suppressSimplification)
|
|
|
|
{
|
|
|
|
TypeId one = *begin(options);
|
|
|
|
TypeId two = *(++begin(options));
|
|
|
|
return {{}, simplifyUnion(builtinTypes, arena, one, two).result};
|
|
|
|
}
|
2023-03-03 15:45:38 +02:00
|
|
|
else
|
2023-03-31 15:21:14 +03:00
|
|
|
return {{}, arena->addType(UnionType{std::vector<TypeId>(begin(options), end(options))})};
|
2022-11-18 10:45:14 -08:00
|
|
|
}
|
2023-01-03 19:33:19 +02:00
|
|
|
else if (auto itv = get<IntersectionType>(subjectType))
|
2022-11-18 10:45:14 -08:00
|
|
|
{
|
2023-03-03 15:45:38 +02:00
|
|
|
std::vector<TypeId> blocked;
|
2023-03-31 15:21:14 +03:00
|
|
|
std::set<TypeId> options;
|
2022-11-18 10:45:14 -08:00
|
|
|
|
2023-03-03 15:45:38 +02:00
|
|
|
for (TypeId ty : itv)
|
|
|
|
{
|
2023-05-19 11:59:59 -07:00
|
|
|
auto [innerBlocked, innerResult] = lookupTableProp(ty, propName, suppressSimplification, seen);
|
2023-03-03 15:45:38 +02:00
|
|
|
blocked.insert(blocked.end(), innerBlocked.begin(), innerBlocked.end());
|
|
|
|
if (innerResult)
|
2023-03-31 15:21:14 +03:00
|
|
|
options.insert(*innerResult);
|
2023-03-03 15:45:38 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
if (!blocked.empty())
|
|
|
|
return {blocked, std::nullopt};
|
2023-01-27 13:28:45 -08:00
|
|
|
|
2023-03-03 15:45:38 +02:00
|
|
|
if (options.empty())
|
|
|
|
return {{}, std::nullopt};
|
|
|
|
else if (options.size() == 1)
|
2023-03-31 15:21:14 +03:00
|
|
|
return {{}, *begin(options)};
|
2023-05-19 11:59:59 -07:00
|
|
|
else if (options.size() == 2 && !suppressSimplification)
|
|
|
|
{
|
|
|
|
TypeId one = *begin(options);
|
|
|
|
TypeId two = *(++begin(options));
|
|
|
|
return {{}, simplifyIntersection(builtinTypes, arena, one, two).result};
|
|
|
|
}
|
2023-03-03 15:45:38 +02:00
|
|
|
else
|
2023-03-31 15:21:14 +03:00
|
|
|
return {{}, arena->addType(IntersectionType{std::vector<TypeId>(begin(options), end(options))})};
|
2022-11-18 10:45:14 -08:00
|
|
|
}
|
|
|
|
|
2023-03-03 15:45:38 +02:00
|
|
|
return {{}, std::nullopt};
|
2022-11-18 10:45:14 -08:00
|
|
|
}
|
|
|
|
|
2023-03-31 15:21:14 +03:00
|
|
|
static TypeId getErrorType(NotNull<BuiltinTypes> builtinTypes, TypeId)
|
|
|
|
{
|
|
|
|
return builtinTypes->errorRecoveryType();
|
|
|
|
}
|
|
|
|
|
|
|
|
static TypePackId getErrorType(NotNull<BuiltinTypes> builtinTypes, TypePackId)
|
|
|
|
{
|
|
|
|
return builtinTypes->errorRecoveryTypePack();
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename TID>
|
|
|
|
bool ConstraintSolver::tryUnify(NotNull<const Constraint> constraint, TID subTy, TID superTy)
|
|
|
|
{
|
2023-05-25 23:46:51 +03:00
|
|
|
Unifier u{normalizer, constraint->scope, constraint->location, Covariant};
|
2023-05-12 15:15:01 +03:00
|
|
|
u.enableScopeTests();
|
2023-03-31 15:21:14 +03:00
|
|
|
|
|
|
|
u.tryUnify(subTy, superTy);
|
|
|
|
|
|
|
|
if (!u.blockedTypes.empty() || !u.blockedTypePacks.empty())
|
|
|
|
{
|
|
|
|
for (TypeId bt : u.blockedTypes)
|
|
|
|
block(bt, constraint);
|
|
|
|
for (TypePackId btp : u.blockedTypePacks)
|
|
|
|
block(btp, constraint);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (const auto& e = hasUnificationTooComplex(u.errors))
|
|
|
|
reportError(*e);
|
|
|
|
|
|
|
|
if (!u.errors.empty())
|
|
|
|
{
|
|
|
|
TID errorType = getErrorType(builtinTypes, TID{});
|
|
|
|
u.tryUnify(subTy, errorType);
|
|
|
|
u.tryUnify(superTy, errorType);
|
|
|
|
}
|
|
|
|
|
|
|
|
const auto [changedTypes, changedPacks] = u.log.getChanges();
|
|
|
|
|
|
|
|
u.log.commit();
|
|
|
|
|
|
|
|
unblock(changedTypes);
|
|
|
|
unblock(changedPacks);
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2022-06-16 17:54:42 -07:00
|
|
|
void ConstraintSolver::block_(BlockedConstraintId target, NotNull<const Constraint> constraint)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
|
|
|
blocked[target].push_back(constraint);
|
|
|
|
|
|
|
|
auto& count = blockedConstraints[constraint];
|
|
|
|
count += 1;
|
|
|
|
}
|
|
|
|
|
2022-06-16 17:54:42 -07:00
|
|
|
void ConstraintSolver::block(NotNull<const Constraint> target, NotNull<const Constraint> constraint)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2023-03-03 15:45:38 +02:00
|
|
|
if (logger)
|
2022-09-08 14:44:50 -07:00
|
|
|
logger->pushBlock(constraint, target);
|
|
|
|
|
2022-08-25 13:55:08 -07:00
|
|
|
if (FFlag::DebugLuauLogSolver)
|
2022-09-01 16:00:14 -07:00
|
|
|
printf("block Constraint %s on\t%s\n", toString(*target, opts).c_str(), toString(*constraint, opts).c_str());
|
2022-09-08 14:44:50 -07:00
|
|
|
|
2023-02-24 10:24:22 -08:00
|
|
|
block_(target.get(), constraint);
|
2022-06-03 13:32:20 -07:00
|
|
|
}
|
|
|
|
|
2022-06-16 17:54:42 -07:00
|
|
|
bool ConstraintSolver::block(TypeId target, NotNull<const Constraint> constraint)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2023-03-03 15:45:38 +02:00
|
|
|
if (logger)
|
2022-09-08 14:44:50 -07:00
|
|
|
logger->pushBlock(constraint, target);
|
|
|
|
|
2022-08-25 13:55:08 -07:00
|
|
|
if (FFlag::DebugLuauLogSolver)
|
2022-09-01 16:00:14 -07:00
|
|
|
printf("block TypeId %s on\t%s\n", toString(target, opts).c_str(), toString(*constraint, opts).c_str());
|
2022-09-08 14:44:50 -07:00
|
|
|
|
2023-02-24 10:24:22 -08:00
|
|
|
block_(follow(target), constraint);
|
2022-06-16 17:54:42 -07:00
|
|
|
return false;
|
2022-06-03 13:32:20 -07:00
|
|
|
}
|
|
|
|
|
2022-06-16 17:54:42 -07:00
|
|
|
bool ConstraintSolver::block(TypePackId target, NotNull<const Constraint> constraint)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2023-03-03 15:45:38 +02:00
|
|
|
if (logger)
|
2022-09-08 14:44:50 -07:00
|
|
|
logger->pushBlock(constraint, target);
|
|
|
|
|
2022-08-25 13:55:08 -07:00
|
|
|
if (FFlag::DebugLuauLogSolver)
|
2022-09-01 16:00:14 -07:00
|
|
|
printf("block TypeId %s on\t%s\n", toString(target, opts).c_str(), toString(*constraint, opts).c_str());
|
2022-09-08 14:44:50 -07:00
|
|
|
|
2022-06-03 13:32:20 -07:00
|
|
|
block_(target, constraint);
|
2022-06-16 17:54:42 -07:00
|
|
|
return false;
|
2022-06-03 13:32:20 -07:00
|
|
|
}
|
|
|
|
|
2023-03-17 16:59:30 +02:00
|
|
|
void ConstraintSolver::inheritBlocks(NotNull<const Constraint> source, NotNull<const Constraint> addition)
|
|
|
|
{
|
|
|
|
// Anything that is blocked on this constraint must also be blocked on our
|
|
|
|
// synthesized constraints.
|
|
|
|
auto blockedIt = blocked.find(source.get());
|
|
|
|
if (blockedIt != blocked.end())
|
|
|
|
{
|
|
|
|
for (const auto& blockedConstraint : blockedIt->second)
|
|
|
|
{
|
|
|
|
block(addition, blockedConstraint);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
struct Blocker : TypeOnceVisitor
|
2022-09-23 11:32:10 -07:00
|
|
|
{
|
|
|
|
NotNull<ConstraintSolver> solver;
|
|
|
|
NotNull<const Constraint> constraint;
|
|
|
|
|
|
|
|
bool blocked = false;
|
|
|
|
|
|
|
|
explicit Blocker(NotNull<ConstraintSolver> solver, NotNull<const Constraint> constraint)
|
|
|
|
: solver(solver)
|
|
|
|
, constraint(constraint)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2023-05-25 23:46:51 +03:00
|
|
|
bool visit(TypeId ty, const PendingExpansionType&) override
|
2022-09-23 11:32:10 -07:00
|
|
|
{
|
|
|
|
blocked = true;
|
|
|
|
solver->block(ty, constraint);
|
|
|
|
return false;
|
|
|
|
}
|
2023-05-25 23:46:51 +03:00
|
|
|
|
|
|
|
bool visit(TypeId ty, const ClassType&) override
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|
2022-09-23 11:32:10 -07:00
|
|
|
};
|
|
|
|
|
2023-05-19 11:59:59 -07:00
|
|
|
bool ConstraintSolver::blockOnPendingTypes(TypeId target, NotNull<const Constraint> constraint)
|
2022-09-23 11:32:10 -07:00
|
|
|
{
|
|
|
|
Blocker blocker{NotNull{this}, constraint};
|
|
|
|
blocker.traverse(target);
|
|
|
|
return !blocker.blocked;
|
|
|
|
}
|
|
|
|
|
2023-05-19 11:59:59 -07:00
|
|
|
bool ConstraintSolver::blockOnPendingTypes(TypePackId pack, NotNull<const Constraint> constraint)
|
2022-09-23 11:32:10 -07:00
|
|
|
{
|
|
|
|
Blocker blocker{NotNull{this}, constraint};
|
|
|
|
blocker.traverse(pack);
|
|
|
|
return !blocker.blocked;
|
|
|
|
}
|
|
|
|
|
2022-06-03 13:32:20 -07:00
|
|
|
void ConstraintSolver::unblock_(BlockedConstraintId progressed)
|
|
|
|
{
|
|
|
|
auto it = blocked.find(progressed);
|
|
|
|
if (it == blocked.end())
|
|
|
|
return;
|
|
|
|
|
|
|
|
// unblocked should contain a value always, because of the above check
|
2022-06-16 17:54:42 -07:00
|
|
|
for (NotNull<const Constraint> unblockedConstraint : it->second)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
|
|
|
auto& count = blockedConstraints[unblockedConstraint];
|
2022-08-25 13:55:08 -07:00
|
|
|
if (FFlag::DebugLuauLogSolver)
|
2022-09-01 16:00:14 -07:00
|
|
|
printf("Unblocking count=%d\t%s\n", int(count), toString(*unblockedConstraint, opts).c_str());
|
2022-08-25 13:55:08 -07:00
|
|
|
|
2022-06-03 13:32:20 -07:00
|
|
|
// This assertion being hit indicates that `blocked` and
|
|
|
|
// `blockedConstraints` desynchronized at some point. This is problematic
|
|
|
|
// because we rely on this count being correct to skip over blocked
|
|
|
|
// constraints.
|
|
|
|
LUAU_ASSERT(count > 0);
|
|
|
|
count -= 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
blocked.erase(it);
|
|
|
|
}
|
|
|
|
|
2022-06-16 17:54:42 -07:00
|
|
|
void ConstraintSolver::unblock(NotNull<const Constraint> progressed)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2023-03-03 15:45:38 +02:00
|
|
|
if (logger)
|
2022-09-08 14:44:50 -07:00
|
|
|
logger->popBlock(progressed);
|
|
|
|
|
2023-02-24 10:24:22 -08:00
|
|
|
return unblock_(progressed.get());
|
2022-06-03 13:32:20 -07:00
|
|
|
}
|
|
|
|
|
2023-05-12 15:15:01 +03:00
|
|
|
void ConstraintSolver::unblock(TypeId ty)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2023-05-05 12:57:12 -07:00
|
|
|
DenseHashSet<TypeId> seen{nullptr};
|
2022-09-08 14:44:50 -07:00
|
|
|
|
2023-05-12 15:15:01 +03:00
|
|
|
TypeId progressed = ty;
|
2023-05-05 12:57:12 -07:00
|
|
|
while (true)
|
|
|
|
{
|
|
|
|
if (seen.find(progressed))
|
|
|
|
iceReporter.ice("ConstraintSolver::unblock encountered a self-bound type!");
|
|
|
|
seen.insert(progressed);
|
2023-02-24 10:24:22 -08:00
|
|
|
|
2023-05-05 12:57:12 -07:00
|
|
|
if (logger)
|
|
|
|
logger->popBlock(progressed);
|
|
|
|
|
|
|
|
unblock_(progressed);
|
|
|
|
|
|
|
|
if (auto bt = get<BoundType>(progressed))
|
|
|
|
progressed = bt->boundTo;
|
|
|
|
else
|
|
|
|
break;
|
|
|
|
}
|
2022-06-03 13:32:20 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
void ConstraintSolver::unblock(TypePackId progressed)
|
|
|
|
{
|
2023-03-03 15:45:38 +02:00
|
|
|
if (logger)
|
2022-09-08 14:44:50 -07:00
|
|
|
logger->popBlock(progressed);
|
|
|
|
|
2022-06-03 13:32:20 -07:00
|
|
|
return unblock_(progressed);
|
|
|
|
}
|
|
|
|
|
2022-08-25 13:55:08 -07:00
|
|
|
void ConstraintSolver::unblock(const std::vector<TypeId>& types)
|
|
|
|
{
|
|
|
|
for (TypeId t : types)
|
|
|
|
unblock(t);
|
|
|
|
}
|
|
|
|
|
|
|
|
void ConstraintSolver::unblock(const std::vector<TypePackId>& packs)
|
|
|
|
{
|
|
|
|
for (TypePackId t : packs)
|
|
|
|
unblock(t);
|
|
|
|
}
|
|
|
|
|
2022-06-16 17:54:42 -07:00
|
|
|
bool ConstraintSolver::isBlocked(TypeId ty)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2023-01-03 19:33:19 +02:00
|
|
|
return nullptr != get<BlockedType>(follow(ty)) || nullptr != get<PendingExpansionType>(follow(ty));
|
2022-06-03 13:32:20 -07:00
|
|
|
}
|
|
|
|
|
2022-09-01 16:00:14 -07:00
|
|
|
bool ConstraintSolver::isBlocked(TypePackId tp)
|
|
|
|
{
|
|
|
|
return nullptr != get<BlockedTypePack>(follow(tp));
|
|
|
|
}
|
|
|
|
|
2022-06-16 17:54:42 -07:00
|
|
|
bool ConstraintSolver::isBlocked(NotNull<const Constraint> constraint)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2022-06-16 17:54:42 -07:00
|
|
|
auto blockedIt = blockedConstraints.find(constraint);
|
|
|
|
return blockedIt != blockedConstraints.end() && blockedIt->second > 0;
|
2022-06-03 13:32:20 -07:00
|
|
|
}
|
|
|
|
|
2023-05-25 23:46:51 +03:00
|
|
|
ErrorVec ConstraintSolver::unify(TypeId subType, TypeId superType, NotNull<Scope> scope)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
2023-05-25 23:46:51 +03:00
|
|
|
Unifier u{normalizer, scope, Location{}, Covariant};
|
2023-05-12 15:15:01 +03:00
|
|
|
u.enableScopeTests();
|
2022-06-03 13:32:20 -07:00
|
|
|
|
|
|
|
u.tryUnify(subType, superType);
|
2022-08-25 13:55:08 -07:00
|
|
|
|
2022-09-01 16:00:14 -07:00
|
|
|
if (!u.errors.empty())
|
|
|
|
{
|
2022-09-15 15:13:58 -07:00
|
|
|
TypeId errorType = errorRecoveryType();
|
2022-09-01 16:00:14 -07:00
|
|
|
u.tryUnify(subType, errorType);
|
|
|
|
u.tryUnify(superType, errorType);
|
|
|
|
}
|
|
|
|
|
2022-08-25 13:55:08 -07:00
|
|
|
const auto [changedTypes, changedPacks] = u.log.getChanges();
|
|
|
|
|
2022-06-03 13:32:20 -07:00
|
|
|
u.log.commit();
|
2022-08-25 13:55:08 -07:00
|
|
|
|
|
|
|
unblock(changedTypes);
|
|
|
|
unblock(changedPacks);
|
2023-05-25 23:46:51 +03:00
|
|
|
|
|
|
|
return std::move(u.errors);
|
2022-06-03 13:32:20 -07:00
|
|
|
}
|
|
|
|
|
2023-05-25 23:46:51 +03:00
|
|
|
ErrorVec ConstraintSolver::unify(TypePackId subPack, TypePackId superPack, NotNull<Scope> scope)
|
2022-06-03 13:32:20 -07:00
|
|
|
{
|
|
|
|
UnifierSharedState sharedState{&iceReporter};
|
2023-05-25 23:46:51 +03:00
|
|
|
Unifier u{normalizer, scope, Location{}, Covariant};
|
2023-05-12 15:15:01 +03:00
|
|
|
u.enableScopeTests();
|
2022-06-03 13:32:20 -07:00
|
|
|
|
|
|
|
u.tryUnify(subPack, superPack);
|
2022-08-25 13:55:08 -07:00
|
|
|
|
|
|
|
const auto [changedTypes, changedPacks] = u.log.getChanges();
|
|
|
|
|
2022-06-03 13:32:20 -07:00
|
|
|
u.log.commit();
|
2022-08-25 13:55:08 -07:00
|
|
|
|
|
|
|
unblock(changedTypes);
|
|
|
|
unblock(changedPacks);
|
2023-05-25 23:46:51 +03:00
|
|
|
|
|
|
|
return std::move(u.errors);
|
2022-06-03 13:32:20 -07:00
|
|
|
}
|
|
|
|
|
2023-02-17 16:53:37 +02:00
|
|
|
NotNull<Constraint> ConstraintSolver::pushConstraint(NotNull<Scope> scope, const Location& location, ConstraintV cv)
|
2022-08-04 14:27:28 -07:00
|
|
|
{
|
2022-09-01 16:00:14 -07:00
|
|
|
std::unique_ptr<Constraint> c = std::make_unique<Constraint>(scope, location, std::move(cv));
|
2022-08-04 14:27:28 -07:00
|
|
|
NotNull<Constraint> borrow = NotNull(c.get());
|
|
|
|
solverConstraints.push_back(std::move(c));
|
|
|
|
unsolvedConstraints.push_back(borrow);
|
2023-02-17 16:53:37 +02:00
|
|
|
|
|
|
|
return borrow;
|
2022-08-04 14:27:28 -07:00
|
|
|
}
|
|
|
|
|
2022-09-01 16:00:14 -07:00
|
|
|
TypeId ConstraintSolver::resolveModule(const ModuleInfo& info, const Location& location)
|
|
|
|
{
|
|
|
|
if (info.name.empty())
|
|
|
|
{
|
|
|
|
reportError(UnknownRequire{}, location);
|
2022-09-15 15:13:58 -07:00
|
|
|
return errorRecoveryType();
|
2022-09-01 16:00:14 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
for (const auto& [location, path] : requireCycles)
|
|
|
|
{
|
2023-05-25 23:46:51 +03:00
|
|
|
if (!path.empty() && path.front() == info.name)
|
2023-01-03 19:33:19 +02:00
|
|
|
return builtinTypes->anyType;
|
2022-09-01 16:00:14 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
ModulePtr module = moduleResolver->getModule(info.name);
|
|
|
|
if (!module)
|
|
|
|
{
|
|
|
|
if (!moduleResolver->moduleExists(info.name) && !info.optional)
|
2023-04-21 14:41:03 -07:00
|
|
|
reportError(UnknownRequire{moduleResolver->getHumanReadableModuleName(info.name)}, location);
|
2022-09-01 16:00:14 -07:00
|
|
|
|
2022-09-15 15:13:58 -07:00
|
|
|
return errorRecoveryType();
|
2022-09-01 16:00:14 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
if (module->type != SourceCode::Type::Module)
|
|
|
|
{
|
2023-04-21 14:41:03 -07:00
|
|
|
reportError(IllegalRequire{module->humanReadableName, "Module is not a ModuleScript. It cannot be required."}, location);
|
2022-09-15 15:13:58 -07:00
|
|
|
return errorRecoveryType();
|
2022-09-01 16:00:14 -07:00
|
|
|
}
|
|
|
|
|
2023-02-10 10:50:54 -08:00
|
|
|
TypePackId modulePack = module->returnType;
|
2022-09-01 16:00:14 -07:00
|
|
|
if (get<Unifiable::Error>(modulePack))
|
2022-09-15 15:13:58 -07:00
|
|
|
return errorRecoveryType();
|
2022-09-01 16:00:14 -07:00
|
|
|
|
|
|
|
std::optional<TypeId> moduleType = first(modulePack);
|
|
|
|
if (!moduleType)
|
|
|
|
{
|
2023-04-21 14:41:03 -07:00
|
|
|
reportError(IllegalRequire{module->humanReadableName, "Module does not return exactly 1 value. It cannot be required."}, location);
|
2022-09-15 15:13:58 -07:00
|
|
|
return errorRecoveryType();
|
2022-09-01 16:00:14 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
return *moduleType;
|
|
|
|
}
|
|
|
|
|
2022-08-18 14:04:33 -07:00
|
|
|
void ConstraintSolver::reportError(TypeErrorData&& data, const Location& location)
|
|
|
|
{
|
|
|
|
errors.emplace_back(location, std::move(data));
|
2022-09-01 16:00:14 -07:00
|
|
|
errors.back().moduleName = currentModuleName;
|
2022-08-18 14:04:33 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
void ConstraintSolver::reportError(TypeError e)
|
|
|
|
{
|
|
|
|
errors.emplace_back(std::move(e));
|
2022-09-01 16:00:14 -07:00
|
|
|
errors.back().moduleName = currentModuleName;
|
2022-08-18 14:04:33 -07:00
|
|
|
}
|
|
|
|
|
2022-09-15 15:13:58 -07:00
|
|
|
TypeId ConstraintSolver::errorRecoveryType() const
|
|
|
|
{
|
2023-01-03 19:33:19 +02:00
|
|
|
return builtinTypes->errorRecoveryType();
|
2022-09-15 15:13:58 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
TypePackId ConstraintSolver::errorRecoveryTypePack() const
|
|
|
|
{
|
2023-01-03 19:33:19 +02:00
|
|
|
return builtinTypes->errorRecoveryTypePack();
|
2022-09-15 15:13:58 -07:00
|
|
|
}
|
|
|
|
|
2022-10-21 10:33:43 -07:00
|
|
|
TypeId ConstraintSolver::unionOfTypes(TypeId a, TypeId b, NotNull<Scope> scope, bool unifyFreeTypes)
|
|
|
|
{
|
|
|
|
a = follow(a);
|
|
|
|
b = follow(b);
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
if (unifyFreeTypes && (get<FreeType>(a) || get<FreeType>(b)))
|
2022-10-21 10:33:43 -07:00
|
|
|
{
|
2023-05-25 23:46:51 +03:00
|
|
|
Unifier u{normalizer, scope, Location{}, Covariant};
|
2023-05-12 15:15:01 +03:00
|
|
|
u.enableScopeTests();
|
2022-10-21 10:33:43 -07:00
|
|
|
u.tryUnify(b, a);
|
|
|
|
|
|
|
|
if (u.errors.empty())
|
|
|
|
{
|
|
|
|
u.log.commit();
|
|
|
|
return a;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2023-01-03 19:33:19 +02:00
|
|
|
return builtinTypes->errorRecoveryType(builtinTypes->anyType);
|
2022-10-21 10:33:43 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (*a == *b)
|
|
|
|
return a;
|
|
|
|
|
|
|
|
std::vector<TypeId> types = reduceUnion({a, b});
|
|
|
|
if (types.empty())
|
2023-01-03 19:33:19 +02:00
|
|
|
return builtinTypes->neverType;
|
2022-10-21 10:33:43 -07:00
|
|
|
|
|
|
|
if (types.size() == 1)
|
|
|
|
return types[0];
|
|
|
|
|
2023-01-03 19:33:19 +02:00
|
|
|
return arena->addType(UnionType{types});
|
2022-10-21 10:33:43 -07:00
|
|
|
}
|
|
|
|
|
2023-05-19 11:59:59 -07:00
|
|
|
TypePackId ConstraintSolver::anyifyModuleReturnTypePackGenerics(TypePackId tp)
|
|
|
|
{
|
|
|
|
tp = follow(tp);
|
|
|
|
|
|
|
|
if (const VariadicTypePack* vtp = get<VariadicTypePack>(tp))
|
|
|
|
{
|
|
|
|
TypeId ty = follow(vtp->ty);
|
|
|
|
return get<GenericType>(ty) ? builtinTypes->anyTypePack : tp;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!get<TypePack>(follow(tp)))
|
|
|
|
return tp;
|
|
|
|
|
|
|
|
std::vector<TypeId> resultTypes;
|
|
|
|
std::optional<TypePackId> resultTail;
|
|
|
|
|
|
|
|
TypePackIterator it = begin(tp);
|
|
|
|
|
|
|
|
for (TypePackIterator e = end(tp); it != e; ++it)
|
|
|
|
{
|
|
|
|
TypeId ty = follow(*it);
|
|
|
|
resultTypes.push_back(get<GenericType>(ty) ? builtinTypes->anyType : ty);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (std::optional<TypePackId> tail = it.tail())
|
|
|
|
resultTail = anyifyModuleReturnTypePackGenerics(*tail);
|
|
|
|
|
|
|
|
return arena->addTypePack(resultTypes, resultTail);
|
|
|
|
}
|
|
|
|
|
2022-06-03 13:32:20 -07:00
|
|
|
} // namespace Luau
|