fix: Add Query for collectionGroup.

mathrunet · Dec 22, 2023 · eca16ba · eca16ba
1 parent 82702bc
commit eca16ba
Showing 1 changed file with 44 additions and 2 deletions.
diff --git a/packages/masamune_builder/lib/model/collection_model_query_class.dart b/packages/masamune_builder/lib/model/collection_model_query_class.dart
@@ -497,6 +497,14 @@ List<Spec> collectionModelQueryClass(
               ..body = Code(
                   "_\$_${model.name}CollectionQuery(modelQuery.limitTo(value))"),
           ),
+          Method(
+            (m) => m
+              ..name = "collectionGroup"
+              ..lambda = true
+              ..returns = Reference("_\$_${model.name}CollectionQuery")
+              ..body = Code(
+                  "_\$_${model.name}CollectionQuery(modelQuery.collectionGroup())"),
+          ),
           Method(
             (m) => m
               ..name = "reset"
@@ -505,6 +513,18 @@ List<Spec> collectionModelQueryClass(
               ..body =
                   Code("_\$_${model.name}CollectionQuery(modelQuery.reset())"),
           ),
+          Method(
+            (m) => m
+              ..name = "uid"
+              ..type = MethodType.getter
+              ..lambda = true
+              ..returns = Reference(
+                "StringModelQuerySelector<_\$_${model.name}CollectionQuery>",
+              )
+              ..body = Code(
+                "StringModelQuerySelector<_\$_${model.name}CollectionQuery>(key: \"@uid\", toQuery: _toQuery, modelQuery: modelQuery)",
+              ),
+          ),
           ...model.parameters.map((param) {
             return Method(
               (m) => m
@@ -518,7 +538,8 @@ List<Spec> collectionModelQueryClass(
                   ),
                 )
                 ..body = Code(
-                    "${_querySelectorClass(param, "_\$_${model.name}CollectionQuery")}(key: \"${param.jsonKey}\", toQuery: _toQuery, modelQuery: modelQuery)"),
+                  "${_querySelectorClass(param, "_\$_${model.name}CollectionQuery")}(key: \"${param.jsonKey}\", toQuery: _toQuery, modelQuery: modelQuery)",
+                ),
             );
           }),
         ]),
@@ -911,6 +932,14 @@ List<Spec> collectionModelQueryClass(
                 ..returns = Reference("_\$_${model.name}MirrorCollectionQuery")
                 ..body = Code("_\$_${model.name}MirrorCollectionQuery(query)"),
             ),
+            Method(
+              (m) => m
+                ..name = "collectionGroup"
+                ..lambda = true
+                ..returns = Reference("_\$_${model.name}MirrorCollectionQuery")
+                ..body = Code(
+                    "_\$_${model.name}MirrorCollectionQuery(modelQuery.collectionGroup())"),
+            ),
             Method(
               (m) => m
                 ..name = "limitTo"
@@ -934,6 +963,18 @@ List<Spec> collectionModelQueryClass(
                 ..body = Code(
                     "_\$_${model.name}MirrorCollectionQuery(modelQuery.reset())"),
             ),
+            Method(
+              (m) => m
+                ..name = "uid"
+                ..type = MethodType.getter
+                ..lambda = true
+                ..returns = Reference(
+                  "StringModelQuerySelector<_\$_${model.name}MirrorCollectionQuery>",
+                )
+                ..body = Code(
+                  "StringModelQuerySelector<_\$_${model.name}MirrorCollectionQuery>(key: \"@uid\", toQuery: _toQuery, modelQuery: modelQuery)",
+                ),
+            ),
             ...model.parameters.map((param) {
               return Method(
                 (m) => m
@@ -947,7 +988,8 @@ List<Spec> collectionModelQueryClass(
                     ),
                   )
                   ..body = Code(
-                      "${_querySelectorClass(param, "_\$_${model.name}MirrorCollectionQuery")}(key: \"${param.jsonKey}\", toQuery: _toQuery, modelQuery: modelQuery)"),
+                    "${_querySelectorClass(param, "_\$_${model.name}MirrorCollectionQuery")}(key: \"${param.jsonKey}\", toQuery: _toQuery, modelQuery: modelQuery)",
+                  ),
               );
             }),
           ]),