WIP: [internal] Query/AggregateQuery to proto representation 2

packages/firestore/src/api.ts

@@ -219,6 +219,10 @@ export {
  */
 export { isBase64Available as _isBase64Available } from './platform/base64';
 export { DatabaseId as _DatabaseId } from './core/database_info';
+export {
+  queryToProtoQueryTarget as _queryToQueryTargetProto,
+  aggregationQueryToProtoRunAggregationQueryRequest as _aggregationQueryToProtoRunAggregationQueryRequest
+} from './remote/serializer';
 export {
   cast as _cast,
   validateIsNotUsedTogether as _validateIsNotUsedTogether

packages/firestore/src/remote/serializer.ts

@@ -15,7 +15,7 @@
  * limitations under the License.
  */
 
-import { Aggregate } from '../core/aggregate';
+import { Aggregate, AggregateImpl } from '../core/aggregate';
 import { Bound } from '../core/bound';
 import { DatabaseId } from '../core/database_info';
 import {
@@ -32,6 +32,7 @@
   newQuery,
   newQueryForPath,
   Query,
+  queryToAggregateTarget,
   queryToTarget
 } from '../core/query';
 import { SnapshotVersion } from '../core/snapshot_version';
@@ -93,17 +94,21 @@
 import { ByteString } from '../util/byte_string';
 import { Code, FirestoreError } from '../util/error';
 import { isNullOrUndefined } from '../util/types';
-
+import { Query as ApiQuery } from '../lite-api/reference';
 import { ExistenceFilter } from './existence_filter';
 import { Serializer } from './number_serializer';
 import { mapCodeFromRpcCode } from './rpc_error';
 import {
   DocumentWatchChange,
   ExistenceFilterChange,
   WatchChange,
   WatchTargetChange,
   WatchTargetChangeState
 } from './watch_change';
+import { AggregateSpec } from '../lite-api/aggregate_types';
+import { mapToArray } from '../util/obj';
+import {ensureFirestoreConfigured, Firestore} from "../api/database";
+import {cast} from "../util/input_validation";
 
 const DIRECTIONS = (() => {
   const dirs: { [dir: string]: ProtoOrderDirection } = {};
@@ -900,6 +905,49 @@
   return { queryTarget, parent };
 }
 
+export function queryToProtoQueryTarget(
+  query: ApiQuery
+): { queryTarget: ProtoQueryTarget; parent: ResourcePath } | null {
+  const firestore = cast(query.firestore, Firestore);
+  const client = ensureFirestoreConfigured(firestore);
+  const serializer = client._onlineComponents?.datastore.serializer;
+  if (serializer === undefined) {
+    return null;
+  }
+  return toQueryTarget(serializer!, queryToTarget(query._query));
+}
+
+export function aggregationQueryToProtoRunAggregationQueryRequest<
+  AggregateSpecType extends AggregateSpec
+>(
+  query: ApiQuery,
+  aggregateSpec: AggregateSpecType
+): {
+  request: ProtoRunAggregationQueryRequest;
+  aliasMap: Record<string, string>;
+  parent: ResourcePath;
+} | null {
+  const aggregates = mapToArray(aggregateSpec, (aggregate, alias) => {
+    return new AggregateImpl(
+      alias,
+      aggregate.aggregateType,
+      aggregate._internalFieldPath
+    );
+  });
+  const firestore = cast(query.firestore, Firestore);
+  const client = ensureFirestoreConfigured(firestore);
+  const serializer = client._onlineComponents?.datastore.serializer;
+  if (serializer === undefined) {
+    return null;
+  }
+
+  return toRunAggregationQueryRequest(
+    serializer!,
+    queryToAggregateTarget(query._query),
+    aggregates
+  );
+}
+
 export function toRunAggregationQueryRequest(
   serializer: JsonProtoSerializer,
   target: Target,