From 931c80dcefe03b9a41b2318869a3ccde8cef593d Mon Sep 17 00:00:00 2001
From: Yifan Yang <yifayan@gmail.com>
Date: Fri, 3 Dec 2021 17:32:49 -0800
Subject: [PATCH] [bundle-size][firestore] Refine bundle definition (#5773)

---
 .../bundle-definitions/firestore-lite.json    | 102 +++++++++++++-
 .../bundle-definitions/firestore.json         | 132 +++++++++++++++++-
 2 files changed, 228 insertions(+), 6 deletions(-)

diff --git a/repo-scripts/size-analysis/bundle-definitions/firestore-lite.json b/repo-scripts/size-analysis/bundle-definitions/firestore-lite.json
index 78168f958d1..0313cc7af77 100644
--- a/repo-scripts/size-analysis/bundle-definitions/firestore-lite.json
+++ b/repo-scripts/size-analysis/bundle-definitions/firestore-lite.json
@@ -22,10 +22,7 @@
             "path": "firestore/lite",
             "imports": [
               "getFirestore",
-              "collection",
-              "getDocs",
               "doc",
-              "setDoc",
               "getDoc"
             ]
           }
@@ -33,6 +30,38 @@
       }
     ]
   },
+  {
+    "name": "Write data",
+    "dependencies": [
+      {
+        "packageName": "firebase",
+        "versionOrTag": "latest",
+        "imports": [
+          {
+            "path": "app",
+            "imports": [
+              "initializeApp"
+            ]
+          }
+        ]
+      },
+      {
+        "packageName": "firebase",
+        "versionOrTag": "latest",
+        "imports": [
+          {
+            "path": "firestore/lite",
+            "imports": [
+              "getFirestore",
+              "collection",
+              "doc",
+              "setDoc"
+            ]
+          }
+        ]
+      }
+    ]
+  },
   {
     "name": "Query",
     "dependencies": [
@@ -66,5 +95,72 @@
         ]
       }
     ]
+  },
+  {
+    "name": "Query Cursors",
+    "dependencies": [
+      {
+        "packageName": "firebase",
+        "versionOrTag": "latest",
+        "imports": [
+          {
+            "path": "app",
+            "imports": [
+              "initializeApp"
+            ]
+          }
+        ]
+      },
+      {
+        "packageName": "firebase",
+        "versionOrTag": "latest",
+        "imports": [
+          {
+            "path": "firestore/lite",
+            "imports": [
+              "getFirestore",
+              "collection",
+              "doc",
+              "getDocs",
+              "query",
+              "orderBy",
+              "startAt",
+              "endBefore"
+            ]
+          }
+        ]
+      }
+    ]
+  },
+  {
+    "name": "Transaction",
+    "dependencies": [
+      {
+        "packageName": "firebase",
+        "versionOrTag": "latest",
+        "imports": [
+          {
+            "path": "app",
+            "imports": [
+              "initializeApp"
+            ]
+          }
+        ]
+      },
+      {
+        "packageName": "firebase",
+        "versionOrTag": "latest",
+        "imports": [
+          {
+            "path": "firestore/lite",
+            "imports": [
+              "getFirestore",
+              "doc",
+              "runTransaction"
+            ]
+          }
+        ]
+      }
+    ]
   }
 ]
diff --git a/repo-scripts/size-analysis/bundle-definitions/firestore.json b/repo-scripts/size-analysis/bundle-definitions/firestore.json
index 87129d83b86..c55219b9ffa 100644
--- a/repo-scripts/size-analysis/bundle-definitions/firestore.json
+++ b/repo-scripts/size-analysis/bundle-definitions/firestore.json
@@ -23,10 +23,39 @@
             "imports": [
               "getFirestore",
               "doc",
-              "getDoc",
-              "setDoc",
+              "getDoc"
+            ]
+          }
+        ]
+      }
+    ]
+  },
+  {
+    "name": "Write data",
+    "dependencies": [
+      {
+        "packageName": "firebase",
+        "versionOrTag": "latest",
+        "imports": [
+          {
+            "path": "app",
+            "imports": [
+              "initializeApp"
+            ]
+          }
+        ]
+      },
+      {
+        "packageName": "firebase",
+        "versionOrTag": "latest",
+        "imports": [
+          {
+            "path": "firestore",
+            "imports": [
+              "getFirestore",
               "collection",
-              "getDocs"
+              "doc",
+              "setDoc"
             ]
           }
         ]
@@ -98,5 +127,102 @@
         ]
       }
     ]
+  },
+  {
+    "name": "Query Cursors",
+    "dependencies": [
+      {
+        "packageName": "firebase",
+        "versionOrTag": "latest",
+        "imports": [
+          {
+            "path": "app",
+            "imports": [
+              "initializeApp"
+            ]
+          }
+        ]
+      },
+      {
+        "packageName": "firebase",
+        "versionOrTag": "latest",
+        "imports": [
+          {
+            "path": "firestore",
+            "imports": [
+              "getFirestore",
+              "collection",
+              "doc",
+              "getDocs",
+              "query",
+              "orderBy",
+              "startAt",
+              "endBefore"
+            ]
+          }
+        ]
+      }
+    ]
+  },
+  {
+    "name": "Persistence",
+    "dependencies": [
+      {
+        "packageName": "firebase",
+        "versionOrTag": "latest",
+        "imports": [
+          {
+            "path": "app",
+            "imports": [
+              "initializeApp"
+            ]
+          }
+        ]
+      },
+      {
+        "packageName": "firebase",
+        "versionOrTag": "latest",
+        "imports": [
+          {
+            "path": "firestore",
+            "imports": [
+              "getFirestore",
+              "enableMultiTabIndexedDbPersistence"
+            ]
+          }
+        ]
+      }
+    ]
+  },
+  {
+    "name": "Transaction",
+    "dependencies": [
+      {
+        "packageName": "firebase",
+        "versionOrTag": "latest",
+        "imports": [
+          {
+            "path": "app",
+            "imports": [
+              "initializeApp"
+            ]
+          }
+        ]
+      },
+      {
+        "packageName": "firebase",
+        "versionOrTag": "latest",
+        "imports": [
+          {
+            "path": "firestore",
+            "imports": [
+              "getFirestore",
+              "doc",
+              "runTransaction"
+            ]
+          }
+        ]
+      }
+    ]
   }
 ]