Parse CSV QualifiedNames instead of just Identifier for ADTs

bmeck · bmeck · commit 30474144cbf2 · 2022-11-08T10:22:33.000-06:00
diff --git a/src/include/souffle/io/ReadStream.h b/src/include/souffle/io/ReadStream.h
@@ -149,7 +149,7 @@ class ReadStream : public SerialisationStream<false> {
 
         // Consume initial character
         consumeChar(source, '$', pos);
-        std::string constructor = readIdentifier(source, pos);
+        std::string constructor = readQualifiedName(source, pos);
 
         json11::Json branchInfo = [&]() -> json11::Json {
             for (auto branch : branches.array_items()) {
@@ -250,7 +250,7 @@ class ReadStream : public SerialisationStream<false> {
      * Consume preceding whitespace.
      * TODO (darth_tytus): use std::string_view?
      */
-    std::string readIdentifier(const std::string& source, std::size_t& pos) {
+    std::string readQualifiedName(const std::string& source, std::size_t& pos) {
         consumeWhiteSpace(source, pos);
         if (pos >= source.length()) {
             throw std::invalid_argument("Unexpected end of input");
@@ -259,7 +259,7 @@ class ReadStream : public SerialisationStream<false> {
         const std::size_t bgn = pos;
         while (pos < source.length()) {
             unsigned char ch = static_cast<unsigned char>(source[pos]);
-            bool valid = std::isalnum(ch) || ch == '_' || ch == '?';
+            bool valid = std::isalnum(ch) || ch == '_' || ch == '?' || ch == '.';
             if (!valid) break;
             ++pos;
         }
diff --git a/tests/syntactic/CMakeLists.txt b/tests/syntactic/CMakeLists.txt
@@ -64,6 +64,7 @@ positive_test(type)
 positive_test(union_comp_type)
 positive_test(dot_identifiers)
 positive_test(input_adt_names1)
+positive_test(input_adt_names2)
 positive_test(input_directive_rfc4180)
 if (NOT MSVC)
 # does not pass with Visual Studio pre-processor because it preserves all whitespaces
diff --git a/tests/syntactic/input_adt_names2/B.csv b/tests/syntactic/input_adt_names2/B.csv
@@ -0,0 +1,2 @@
+$MyInstance.A
+$MyInstance.A2(123)
diff --git a/tests/syntactic/input_adt_names2/facts/A.facts b/tests/syntactic/input_adt_names2/facts/A.facts
@@ -0,0 +1,2 @@
+$MyInstance.A
+$MyInstance.A2(123)
diff --git a/tests/syntactic/input_adt_names2/input_adt_names2.dl b/tests/syntactic/input_adt_names2/input_adt_names2.dl
@@ -0,0 +1,11 @@
+.comp Component {
+  .type ADT = A {} | A2 { n: number }
+}
+.init MyInstance = Component
+.decl A(value: MyInstance.ADT)
+.decl B(value: MyInstance.ADT)
+
+B(v) :- A(v).
+
+.input A
+.output B
diff --git a/tests/syntactic/input_adt_names2/input_adt_names2.err b/tests/syntactic/input_adt_names2/input_adt_names2.err
diff --git a/tests/syntactic/input_adt_names2/input_adt_names2.out b/tests/syntactic/input_adt_names2/input_adt_names2.out

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+$MyInstance.A`
	`2`	`+$MyInstance.A2(123)`