feat: support variadic functions and variadic function types

sdkrystian · sdkrystian · commit d456f87e5ddd · 2024-05-27T14:14:33.000-04:00
diff --git a/include/mrdocs/Metadata/Type.hpp b/include/mrdocs/Metadata/Type.hpp
@@ -196,6 +196,7 @@ struct FunctionTypeInfo
     QualifierKind CVQualifiers = QualifierKind::None;
     ReferenceKind RefQualifier = ReferenceKind::None;
     NoexceptInfo ExceptionSpec;
+    bool IsVariadic = false;
 
     TypeInfo* innerType() const noexcept override
     {
diff --git a/mrdocs.rnc b/mrdocs.rnc
@@ -432,12 +432,13 @@ grammar
                 "pointer" |
                 "member-pointer" |
                 "array" |
-                "function" |
-                "pack"
+                "function"
             } ?,
             ID ?,
             Name ?,
             attribute parent { text } ?,
+            attribute is-pack { "1" } ?,
+            attribute is-variadic { "1" } ?,
             attribute cv-qualifiers { "const"|"volatile"|"const volatile" } ?,
             attribute ref-qualifier { "&"|"&&" } ?,
             attribute exception-spec { text } ?,
diff --git a/share/mrdocs/addons/generator/asciidoc/partials/declarator-after.adoc.hbs b/share/mrdocs/addons/generator/asciidoc/partials/declarator-after.adoc.hbs
@@ -7,7 +7,7 @@
 {{else if (eq kind "function")~}}
     ({{#each param-types~}}
         {{~>declarator nolink=../nolink~}}{{~#unless @last}}, {{/unless~}}
-    {{/each~}})
+    {{/each~}}{{#if is-variadic}}{{#if param-types}}, {{/if}}...{{/if}})
     {{~#if cv-qualifiers}} {{cv-qualifiers}}{{/if~}}
     {{#if (eq ref-qualifier "lvalue")}} &{{else if (eq ref-qualifier "rvalue")}} &&{{/if~}}
     {{#if exception-spec}} {{exception-spec}}{{/if~}}
diff --git a/share/mrdocs/addons/generator/asciidoc/partials/signature/function.adoc.hbs b/share/mrdocs/addons/generator/asciidoc/partials/signature/function.adoc.hbs
@@ -15,7 +15,7 @@
     {{/unless}}{{>declarator type decl-name=name~}}
     {{#if default}} = {{default}}{{/if~}}
     {{#unless @last}},{{/unless~}}
-{{/each~}})
+{{/each~}}{{#if symbol.isVariadic}}{{#if symbol.params}}, {{/if}}...{{/if}})
 {{~#if symbol.isConst}} const{{/if~}}
 {{#if symbol.isVolatile}} volatile{{/if~}}
 {{#if symbol.refQualifier}} {{symbol.refQualifier}}{{/if~}}
diff --git a/src/lib/AST/ASTVisitor.cpp b/src/lib/AST/ASTVisitor.cpp
@@ -3537,6 +3537,7 @@ class TypeInfoBuilder
             FPT->getRefQualifier());
         I->CVQualifiers = convertToQualifierKind(
             FPT->getMethodQuals().getFastQualifiers());
+        I->IsVariadic = FPT->isVariadic();
         getASTVisitor().buildNoexceptInfo(I->ExceptionSpec, FPT);
         *std::exchange(Inner, &I->ReturnType) = std::move(I);
     }
diff --git a/src/lib/AST/AnyBlock.hpp b/src/lib/AST/AnyBlock.hpp
@@ -886,6 +886,11 @@ class TypeInfoBlock
                 return Error("wrong TypeInfo kind");
             return decodeRecord(R, static_cast<
                 FunctionTypeInfo&>(*I_).ExceptionSpec, Blob);
+        case TYPEINFO_IS_VARIADIC:
+            if(! I_->isFunction())
+                return Error("wrong TypeInfo kind");
+            return decodeRecord(R, static_cast<
+                FunctionTypeInfo&>(*I_).IsVariadic, Blob);
         default:
             return AnyBlock::parseRecord(R, ID, Blob);
         }
diff --git a/src/lib/AST/BitcodeIDs.hpp b/src/lib/AST/BitcodeIDs.hpp
@@ -114,6 +114,7 @@ enum RecordID
     TYPEINFO_CVQUAL,
     TYPEINFO_NOEXCEPT,
     TYPEINFO_REFQUAL,
+    TYPEINFO_IS_VARIADIC,
     BASE_ACCESS,
     BASE_IS_VIRTUAL,
     FIELD_ATTRIBUTES,
diff --git a/src/lib/AST/BitcodeWriter.cpp b/src/lib/AST/BitcodeWriter.cpp
@@ -335,6 +335,7 @@ RecordIDNameMap = []()
         {TYPEINFO_CVQUAL, {"TypeinfoCV", &Integer32Abbrev}},
         {TYPEINFO_NOEXCEPT, {"TypeinfoNoexcept", &NoexceptAbbrev}},
         {TYPEINFO_REFQUAL, {"TypeinfoRefqual", &Integer32Abbrev}},
+        {TYPEINFO_IS_VARIADIC, {"TypeinfoIsVariadic", &BoolAbbrev}},
         {TYPEDEF_IS_USING, {"IsUsing", &BoolAbbrev}},
         {VARIABLE_BITS, {"Bits", &Integer32ArrayAbbrev}},
         {USING_SYMBOLS, {"UsingSymbols", &SymbolIDsAbbrev}},
@@ -437,7 +438,7 @@ RecordsByBlock{
     // TypeInfo
     {BI_TYPEINFO_BLOCK_ID,
         {TYPEINFO_KIND, TYPEINFO_IS_PACK, TYPEINFO_CVQUAL,
-        TYPEINFO_NOEXCEPT, TYPEINFO_REFQUAL}},
+        TYPEINFO_NOEXCEPT, TYPEINFO_REFQUAL, TYPEINFO_IS_VARIADIC}},
     {BI_TYPEINFO_PARENT_BLOCK_ID,
         {}},
     {BI_TYPEINFO_CHILD_BLOCK_ID,
@@ -1048,6 +1049,7 @@ emitBlock(
 
             emitRecord(t.RefQualifier, TYPEINFO_REFQUAL);
             emitRecord(t.ExceptionSpec, TYPEINFO_NOEXCEPT);
+            emitRecord(t.IsVariadic, TYPEINFO_IS_VARIADIC);
         }
 
         if constexpr(T::isNamed())
diff --git a/src/lib/Gen/xml/CXXTags.hpp b/src/lib/Gen/xml/CXXTags.hpp
@@ -252,6 +252,9 @@ writeType(
 
             if constexpr(T::isFunction())
             {
+                if(t.IsVariadic)
+                    attrs.push({"is-variadic", "1"});
+
                 if(t.RefQualifier != ReferenceKind::None)
                     attrs.push({"ref-qualifier", toString(t.RefQualifier)});
 
diff --git a/src/lib/Metadata/DomMetadata.cpp b/src/lib/Metadata/DomMetadata.cpp
@@ -500,6 +500,7 @@ domCreate(
                 toString(t.ExceptionSpec));
             entries.emplace_back("ref-qualifier",
                 toString(t.RefQualifier));
+            entries.emplace_back("is-variadic", t.IsVariadic);
         }
     });
     return dom::Object(std::move(entries));
diff --git a/src/lib/Metadata/Type.cpp b/src/lib/Metadata/Type.cpp
@@ -217,6 +217,14 @@ operator()(
                 writeFullType(**first, write);
             }
         }
+
+        if(t.IsVariadic)
+        {
+            if(! t.ParamTypes.empty())
+                write(", ");
+            write("...");
+        }
+
         write(')');
 
         if(t.CVQualifiers != QualifierKind::None)
diff --git a/test-files/old-tests/variadic-function.cpp b/test-files/old-tests/variadic-function.cpp
@@ -0,0 +1,10 @@
+using T = void(...);
+
+using U = void(int, ...);
+
+void f(...);
+
+void g(int, ...);
+
+template<typename A = void(...), typename B = void(int, ...)>
+struct C;
diff --git a/test-files/old-tests/variadic-function.xml b/test-files/old-tests/variadic-function.xml
@@ -0,0 +1,37 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<mrdocs xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+       xsi:noNamespaceSchemaLocation="https://github.com/cppalliance/mrdocs/raw/develop/mrdocs.rnc">
+<namespace id="//////////////////////////8=">
+  <alias name="T" id="PvNOREc+JWxdkVgvSz+jC45IbGQ=">
+    <file path="variadic-function.cpp" line="1" class="def"/>
+    <type class="function" is-variadic="1">
+      <return-type name="void"/>
+    </type>
+  </alias>
+  <alias name="U" id="h6hupdyuT6SfsCGFl6UUFGtvrcc=">
+    <file path="variadic-function.cpp" line="3" class="def"/>
+    <type class="function" is-variadic="1">
+      <return-type name="void"/>
+      <param-type name="int"/>
+    </type>
+  </alias>
+  <function name="f" id="CMaAySU4txrIlGl6edqWDSabBRA=">
+    <file path="variadic-function.cpp" line="5"/>
+    <attr id="is-variadic"/>
+  </function>
+  <function name="g" id="lldWvRg9y+NwRnDpK3x2NF9+e08=">
+    <file path="variadic-function.cpp" line="7"/>
+    <attr id="is-variadic"/>
+    <param>
+      <type name="int"/>
+    </param>
+  </function>
+  <template>
+    <tparam name="A" class="type" default="void(...)"/>
+    <tparam name="B" class="type" default="void(int, ...)"/>
+    <struct name="C" id="8nYw0KA23eAszqSOs0wOXAD1DYQ=">
+      <file path="variadic-function.cpp" line="10"/>
+    </struct>
+  </template>
+</namespace>
+</mrdocs>

Original file line number	Diff line number	Diff line change
`@@ -196,6 +196,7 @@ struct FunctionTypeInfo`
`196`	`196`	`QualifierKind CVQualifiers = QualifierKind::None;`
`197`	`197`	`ReferenceKind RefQualifier = ReferenceKind::None;`
`198`	`198`	`NoexceptInfo ExceptionSpec;`
	`199`	`+ bool IsVariadic = false;`
`199`	`200`
`200`	`201`	`TypeInfo* innerType() const noexcept override`
`201`	`202`	`{`
Original file line number	Diff line number	Diff line change
`@@ -3537,6 +3537,7 @@ class TypeInfoBuilder`
`3537`	`3537`	`FPT->getRefQualifier());`
`3538`	`3538`	`I->CVQualifiers = convertToQualifierKind(`
`3539`	`3539`	`FPT->getMethodQuals().getFastQualifiers());`
	`3540`	`+ I->IsVariadic = FPT->isVariadic();`
`3540`	`3541`	`getASTVisitor().buildNoexceptInfo(I->ExceptionSpec, FPT);`
`3541`	`3542`	`*std::exchange(Inner, &I->ReturnType) = std::move(I);`
`3542`	`3543`	`}`
Original file line number	Diff line number	Diff line change
`@@ -252,6 +252,9 @@ writeType(`
`252`	`252`
`253`	`253`	`if constexpr(T::isFunction())`
`254`	`254`	`{`
	`255`	`+ if(t.IsVariadic)`
	`256`	`+ attrs.push({"is-variadic", "1"});`
	`257`	`+`
`255`	`258`	`if(t.RefQualifier != ReferenceKind::None)`
`256`	`259`	`attrs.push({"ref-qualifier", toString(t.RefQualifier)});`
`257`	`260`
Original file line number	Diff line number	Diff line change
`@@ -500,6 +500,7 @@ domCreate(`
`500`	`500`	`toString(t.ExceptionSpec));`
`501`	`501`	`entries.emplace_back("ref-qualifier",`
`502`	`502`	`toString(t.RefQualifier));`
	`503`	`+ entries.emplace_back("is-variadic", t.IsVariadic);`
`503`	`504`	`}`
`504`	`505`	`});`
`505`	`506`	`return dom::Object(std::move(entries));`
Original file line number	Diff line number	Diff line change
`@@ -217,6 +217,14 @@ operator()(`
`217`	`217`	`writeFullType(**first, write);`
`218`	`218`	`}`
`219`	`219`	`}`
	`220`	`+`
	`221`	`+ if(t.IsVariadic)`
	`222`	`+ {`
	`223`	`+ if(! t.ParamTypes.empty())`
	`224`	`+ write(", ");`
	`225`	`+ write("...");`
	`226`	`+ }`
	`227`	`+`
`220`	`228`	`write(')');`
`221`	`229`
`222`	`230`	`if(t.CVQualifiers != QualifierKind::None)`