extract SFINAE constraints

#feat

Author: alandefreitas

include/mrdocs/Metadata/Type.hpp

@@ -106,6 +106,18 @@ struct TypeInfo
     */
     bool IsPackExpansion = false;
 
+    /** The constraints associated with the type
+
+        This represents the constraints associated with the type,
+        such as SFINAE constraints.
+
+        For instance, if SFINAE detection is enabled, the
+        expression `std::enable_if_t<std::is_integral_v<T>, T>`
+        will have type `T` (NamedType) and constraints
+        `{std::is_integral_v<T>}`.
+     */
+    std::vector<ExprInfo> Constraints;
+
     constexpr virtual ~TypeInfo() = default;
 
     constexpr bool isNamed()           const noexcept { return Kind == TypeKind::Named; }
@@ -123,14 +135,17 @@ struct TypeInfo
         The inner type is the type which is modified
         by a specifier (e.g. "int" in "pointer to int".
     */
-    virtual TypeInfo* innerType() const noexcept
+    virtual
+    TypeInfo*
+    innerType() const noexcept
     {
         return nullptr;
     }
 
     /** Return the symbol named by this type.
     */
-    SymbolID namedSymbol() const noexcept;
+    SymbolID
+    namedSymbol() const noexcept;
 
 protected:
     constexpr

src/lib/AST/ASTVisitor.cpp

@@ -1884,25 +1884,32 @@ extractSFINAEInfo(QualType const T)
     // Find the parameter that represents the SFINAE result
     auto const Args = templateInfo->Arguments;
     MRDOCS_SYMBOL_TRACE(Args, context_);
-    auto const param_arg = tryGetTemplateArgument(
+    auto const resultType = tryGetTemplateArgument(
         SFINAEControl->Parameters, Args, SFINAEControl->ParamIdx);
-    MRDOCS_CHECK_OR(param_arg, std::nullopt);
-    MRDOCS_SYMBOL_TRACE(*param_arg, context_);
+    MRDOCS_CHECK_OR(resultType, std::nullopt);
+    MRDOCS_SYMBOL_TRACE(*resultType, context_);
 
     // Create a vector of template arguments that represent the
     // controlling parameters of the SFINAE template
-    std::vector<TemplateArgument> ControllingArgs;
+    SFINAEInfo Result;
+    Result.Type = resultType->getAsType();
     for (std::size_t I = 0; I < Args.size(); ++I)
     {
         if (SFINAEControl->ControllingParams[I])
         {
             MRDOCS_SYMBOL_TRACE(Args[I], context_);
-            ControllingArgs.emplace_back(Args[I]);
+            Expr* E = Args[I].getAsExpr();
+            if (!E)
+            {
+                continue;
+            }
+            Result.Constraints.emplace_back();
+            populate(Result.Constraints.back(), E);
         }
     }
 
     // Return the main type and controlling types
-    return SFINAEInfo{param_arg->getAsType(), std::move(ControllingArgs)};
+    return Result;
 }
 
 std::optional<ASTVisitor::SFINAEControlParams>
@@ -1963,16 +1970,16 @@ getSFINAEControlParams(
 
         // Find the index of the parameter that represents the SFINAE result
         // in the underlying template arguments
-        auto param_arg = tryGetTemplateArgument(
+        auto resultType = tryGetTemplateArgument(
             sfinaeControl->Parameters,
             underlyingTemplateInfo->Arguments,
             sfinaeControl->ParamIdx);
-        MRDOCS_CHECK_OR(param_arg, std::nullopt);
-        MRDOCS_SYMBOL_TRACE(*param_arg, context_);
+        MRDOCS_CHECK_OR(resultType, std::nullopt);
+        MRDOCS_SYMBOL_TRACE(*resultType, context_);
 
         // Find the index of the parameter that represents the SFINAE result
         // in the primary template arguments
-        unsigned ParamIdx = FindParam(ATD->getInjectedTemplateArgs(context_), *param_arg);
+        unsigned ParamIdx = FindParam(ATD->getInjectedTemplateArgs(context_), *resultType);
 
         // Return the controlling parameters with values corresponding to
         // the primary template arguments
@@ -2060,11 +2067,11 @@ getSFINAEControlParams(
                 }
 
                 auto [template_params, controlling_params, param_idx] = *sfinae_result;
-                auto param_arg = tryGetTemplateArgument(
+                auto resultType = tryGetTemplateArgument(
                     template_params, sfinae_info->Arguments, param_idx);
-                if(! param_arg)
+                if(! resultType)
                     return true;
-                auto CurrentTypeFromBase = param_arg->getAsType();
+                auto CurrentTypeFromBase = resultType->getAsType();
                 if (CurrentType.isNull())
                 {
                     CurrentType = CurrentTypeFromBase;

src/lib/AST/ASTVisitor.hpp

@@ -734,7 +734,7 @@ class ASTVisitor
         QualType Type;
 
         // The template arguments used in the SFINAE context.
-        std::vector<TemplateArgument> Arguments;
+        std::vector<ExprInfo> Constraints;
     };
 
     /* Get the underlying type result of a SFINAE type

src/lib/AST/TerminalTypeVisitor.hpp

@@ -87,6 +87,10 @@ class TerminalTypeVisitor
     // The optional NestedNameSpecifier.
     const NestedNameSpecifier* NNS_ = nullptr;
 
+protected:
+    // Constraints associated with the type (e.g., SFINAE)
+    std::vector<ExprInfo> Constraints;
+
 public:
     /** Constructor for TerminalTypeVisitor.
 
@@ -571,6 +575,7 @@ class TerminalTypeVisitor
         if (auto SFINAE = getASTVisitor().extractSFINAEInfo(T))
         {
             NNS_ = nullptr;
+            Constraints = SFINAE->Constraints;
             return getDerived().Visit(SFINAE->Type);
         }
 
@@ -606,6 +611,7 @@ class TerminalTypeVisitor
         if (auto SFINAE = getASTVisitor().extractSFINAEInfo(T))
         {
             NNS_ = nullptr;
+            Constraints = SFINAE->Constraints;
             return getDerived().Visit(SFINAE->Type);
         }
 

src/lib/AST/TypeInfoBuilder.cpp

@@ -102,6 +102,7 @@ buildDecltype(
     getASTVisitor().populate(
         I->Operand, T->getUnderlyingExpr());
     I->CVQualifiers = toQualifierKind(quals);
+    I->Constraints = this->Constraints;
     *Inner = std::move(I);
     Result->IsPackExpansion = pack;
 }
@@ -129,6 +130,7 @@ buildAuto(
         // Constraint->Prefix = getASTVisitor().buildNameInfo(
         //     cast<Decl>(CD->getDeclContext()));
     }
+    I->Constraints = this->Constraints;
     *Inner = std::move(I);
     Result->IsPackExpansion = pack;
 }
@@ -145,12 +147,12 @@ buildTerminal(
 {
     auto TI = std::make_unique<NamedTypeInfo>();
     TI->CVQualifiers = toQualifierKind(quals);
-
-    auto Name = std::make_unique<NameInfo>();
-    Name->Name = getASTVisitor().toString(T);
-    Name->Prefix = getASTVisitor().toNameInfo(NNS);
-    TI->Name = std::move(Name);
+    TI->Name = std::make_unique<NameInfo>();
+    TI->Name->Name = getASTVisitor().toString(T);
+    TI->Name->Prefix = getASTVisitor().toNameInfo(NNS);
+    TI->Constraints = this->Constraints;
     *Inner = std::move(TI);
+    Result->Constraints = this->Constraints;
     Result->IsPackExpansion = pack;
 }
 
@@ -187,7 +189,9 @@ buildTerminal(
         Name->Prefix = getASTVisitor().toNameInfo(NNS);
         I->Name = std::move(Name);
     }
+    I->Constraints = this->Constraints;
     *Inner = std::move(I);
+    Result->Constraints = this->Constraints;
     Result->IsPackExpansion = pack;
 }
 
@@ -240,7 +244,9 @@ buildTerminal(
         getASTVisitor().populate(Name->TemplateArgs, *TArgs);
         TI->Name = std::move(Name);
     }
+    TI->Constraints = this->Constraints;
     *Inner = std::move(TI);
+    Result->Constraints = this->Constraints;
     Result->IsPackExpansion = pack;
 }
 

src/lib/Metadata/Type.cpp

@@ -83,12 +83,15 @@ SymbolID
 TypeInfo::
 namedSymbol() const noexcept
 {
-    if(! isNamed())
+    if (!isNamed())
+    {
         return SymbolID::invalid;
-    const auto* NT = static_cast<
-        const NamedTypeInfo*>(this);
-    if(! NT->Name)
+    }
+    auto const* NT = static_cast<NamedTypeInfo const*>(this);
+    if (!NT->Name)
+    {
         return SymbolID::invalid;
+    }
     return NT->Name->id;
 }