fix comments

Author: milaGGL

.changeset/wicked-tomatoes-grow.md

@@ -2,4 +2,4 @@
 '@firebase/firestore': minor
 ---
 
-Expose getCountFromServer() and Lite SDK getCount()to enable Firestore count-only aggregate queries
+Added `getCountFromServer()` (`getCount()` in the Lite SDK), which fetches the number of documents in the result set without actually downloading the documents.

packages/firestore/lite/index.ts

@@ -31,6 +31,7 @@ export {
   aggregateQuerySnapshotEqual,
   getCount
 } from '../src/lite-api/aggregate';
+
 export {
   AggregateField,
   AggregateFieldType,

packages/firestore/src/api.ts

@@ -15,16 +15,19 @@
  * limitations under the License.
  */
 
+export {
+  aggregateQuerySnapshotEqual,
+  getCountFromServer
+} from './api/aggregate';
+
 export {
   AggregateField,
   AggregateFieldType,
   AggregateSpec,
   AggregateSpecData,
   AggregateQuerySnapshot,
-  aggregateQuerySnapshotEqual,
-  count,
-  getCountFromServer
-} from './api/aggregate';
+  count
+} from './lite-api/aggregate_types';
 
 export { FieldPath, documentId } from './api/field_path';
 

packages/firestore/src/api/aggregate.ts

@@ -17,23 +17,16 @@
 
 import { Query } from '../api';
 import { firestoreClientRunCountQuery } from '../core/firestore_client';
-import { AggregateField, AggregateQuerySnapshot } from '../lite-api/aggregate_types';
+import {
+  AggregateField,
+  AggregateQuerySnapshot
+} from '../lite-api/aggregate_types';
 import { cast } from '../util/input_validation';
 
 import { ensureFirestoreConfigured, Firestore } from './database';
 import { ExpUserDataWriter } from './reference_impl';
 
-export {
-  aggregateQuerySnapshotEqual
-} from '../lite-api/aggregate';
-export {
-  AggregateField,
-  AggregateFieldType,
-  AggregateSpec,
-  AggregateSpecData,
-  AggregateQuerySnapshot,
-  count
-} from '../lite-api/aggregate_types';
+export { aggregateQuerySnapshotEqual } from '../lite-api/aggregate';
 
 /**
  * Executes the query and returns the results as a `AggregateQuerySnapshot` from the

packages/firestore/src/core/count_query_runner.ts

@@ -16,7 +16,10 @@
  */
 
 import { AbstractUserDataWriter, Query } from '../api';
-import { AggregateField, AggregateQuerySnapshot } from '../lite-api/aggregate_types';
+import {
+  AggregateField,
+  AggregateQuerySnapshot
+} from '../lite-api/aggregate_types';
 import { Value } from '../protos/firestore_proto_api';
 import { Datastore, invokeRunAggregationQueryRpc } from '../remote/datastore';
 import { hardAssert } from '../util/assert';
@@ -25,7 +28,7 @@ import { hardAssert } from '../util/assert';
  * CountQueryRunner encapsulates the logic needed to run the count aggregation
  * queries.
  */
-export class CountQueryRunner<T> {
+export class CountQueryRunner {
   constructor(
     private readonly query: Query<unknown>,
     private readonly datastore: Datastore,

packages/firestore/src/core/firestore_client.ts

@@ -16,11 +16,12 @@
  */
 
 import { GetOptions } from '@firebase/firestore-types';
-import { AbstractUserDataWriter } from '../api';
+
 import {
+  AbstractUserDataWriter,
   AggregateField,
   AggregateQuerySnapshot
-} from '../api/aggregate';
+} from '../api';
 import { LoadBundleTask } from '../api/bundle';
 import {
   CredentialChangeListener,

packages/firestore/src/lite-api/aggregate.ts

@@ -16,10 +16,15 @@
  */
 
 import { deepEqual } from '@firebase/util';
+
 import { CountQueryRunner } from '../core/count_query_runner';
 import { cast } from '../util/input_validation';
 
-import { AggregateField, AggregateQuerySnapshot, AggregateSpec } from './aggregate_types';
+import {
+  AggregateField,
+  AggregateQuerySnapshot,
+  AggregateSpec
+} from './aggregate_types';
 import { getDatastore } from './components';
 import { Firestore } from './database';
 import { Query, queryEqual } from './reference';