From d57816704b433c719346251e056a51642238ef69 Mon Sep 17 00:00:00 2001
From: William Chong <me@williamchong.cloud>
Date: Fri, 17 Feb 2023 14:43:18 +0800
Subject: [PATCH 1/2] =?UTF-8?q?=E2=9C=A8=20[Trending]=20Limit=20number=20o?=
 =?UTF-8?q?f=20work=20from=20same=20owner?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/pages/writing-nft/index.vue | 55 +++++++++++++++++++++++++++------
 1 file changed, 46 insertions(+), 9 deletions(-)

diff --git a/src/pages/writing-nft/index.vue b/src/pages/writing-nft/index.vue
index 9fb171acf..e34daa0e4 100644
--- a/src/pages/writing-nft/index.vue
+++ b/src/pages/writing-nft/index.vue
@@ -90,12 +90,19 @@
 <script>
 import { LIKECOIN_NFT_CAMPAIGN_ITEMS } from '~/constant';
 
-import { getLatestNFTClasses, getTopNFTClasses } from '~/util/api';
+import {
+  getLatestNFTClasses,
+  getTopNFTClasses,
+  getISCNRecord,
+} from '~/util/api';
 import { logTrackerEvent } from '~/util/EventLogger';
 
 import navigationListenerMixin from '~/mixins/navigation-listener';
 import walletMixin from '~/mixins/wallet';
 
+const MAX_NFT_CLASS_COUNT_PER_OWNER = 2;
+const NFT_CLASS_DISPLAY_COUNT = 10;
+
 export default {
   name: 'WritingNFTPage',
   layout: 'default',
@@ -213,17 +220,47 @@ export default {
       this.$axios.$get(getTopNFTClasses()),
       this.$axios.$get(getLatestNFTClasses()),
     ]);
-    [this.trendingClassIds, this.latestClassIds] = [trendingRes, latestRes].map(
-      res =>
-        (res.classes || [])
-          .filter(
-            c => c.metadata?.nft_meta_collection_id === 'likerland_writing_nft'
-          )
-          .map(c => c.id)
-          .slice(0, 10)
+    const [trendingClasses, latestClasses] = [trendingRes, latestRes].map(res =>
+      (res.classes || []).filter(
+        c => c.metadata?.nft_meta_collection_id === 'likerland_writing_nft'
+      )
     );
+    this.latestClassIds = latestClasses
+      .slice(0, NFT_CLASS_DISPLAY_COUNT)
+      .map(c => c.id);
+    this.trendingClassIds = await this.filterNFTClassesByOwner(trendingClasses);
   },
   methods: {
+    async getClassOwner(classData) {
+      try {
+        const iscnPrefix = classData.parent.iscn_id_prefix;
+        const data = await this.$axios.$get(getISCNRecord(iscnPrefix));
+        return data.owner;
+      } catch (err) {
+        console.error(`Failed to fetch owner of ${classData.id}`);
+        return null;
+      }
+    },
+    async filterNFTClassesByOwner(nftClasses) {
+      const ownerList = await Promise.all(
+        // use 3 times of display count to ensure enough classes
+        nftClasses.slice(0, NFT_CLASS_DISPLAY_COUNT * 3).map(this.getClassOwner)
+      );
+      const filteredNFTClasses = [];
+      const ownerToNFTClassCountMap = {};
+      for (let i = 0; i < ownerList.length; i += 1) {
+        const owner = ownerList[i];
+        if (!ownerToNFTClassCountMap[owner]) {
+          ownerToNFTClassCountMap[owner] = 0;
+        }
+        if (ownerToNFTClassCountMap[owner] < MAX_NFT_CLASS_COUNT_PER_OWNER) {
+          filteredNFTClasses.push(nftClasses[i].id);
+          ownerToNFTClassCountMap[owner] += 1;
+        }
+        if (filteredNFTClasses.length >= NFT_CLASS_DISPLAY_COUNT) break;
+      }
+      return filteredNFTClasses;
+    },
     handleTabClick(tab) {
       const { query } = this.$route;
       this.$router.replace({

From 25806536b903795022236f21ffc237138d9fbf9b Mon Sep 17 00:00:00 2001
From: Wei-Jun Hung <33746295+WeiJun0827@users.noreply.github.com>
Date: Fri, 17 Feb 2023 14:50:35 +0800
Subject: [PATCH 2/2] =?UTF-8?q?=E2=9C=A8=20Handle=20legacy=20followees=20(?=
 =?UTF-8?q?#936)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* ✨ Includes email followees in wallet followees

* ✨ Delete email followee when unfollow

* 🔥 Remove duplicate variable

* ✨ Move email followees to wallet followees after email verification

* 🎨 Handle with transaction

* 🎨 Format code and read DB before writing

* 🎨 Handle migration with batched write
---
 src/server/api/routes/users/v2/email.js  | 33 ++++++++++++++++---
 src/server/api/routes/users/v2/follow.js | 42 +++++++++++++++++-------
 2 files changed, 58 insertions(+), 17 deletions(-)

diff --git a/src/server/api/routes/users/v2/email.js b/src/server/api/routes/users/v2/email.js
index 4d25a04f2..76a67558a 100644
--- a/src/server/api/routes/users/v2/email.js
+++ b/src/server/api/routes/users/v2/email.js
@@ -5,6 +5,7 @@ const {
   db,
   FieldValue,
   walletUserCollection,
+  nftMintSubscriptionCollection,
 } = require('../../../../modules/firebase');
 const {
   authenticateV2Login,
@@ -93,22 +94,44 @@ router.put('/:wallet/email', async (req, res, next) => {
       res.status(400).send('MISSING_TOKEN');
       return;
     }
-    await db.runTransaction(async t => {
-      const userRef = walletUserCollection.doc(user);
+
+    const userRef = walletUserCollection.doc(user);
+    const email = await db.runTransaction(async t => {
       const userDoc = await t.get(userRef);
       if (!userDoc.exists) {
         throw new Error('USER_NOT_FOUND');
       }
-      const { emailUnconfirmed, emailVerifyToken } = userDoc.data();
+      const { emailUnconfirmed: email, emailVerifyToken } = userDoc.data();
       if (emailVerifyToken !== token) {
         throw new Error('INVALID_TOKEN');
       }
-      await t.update(userRef, {
-        email: emailUnconfirmed,
+      t.update(userRef, {
+        email,
         emailUnconfirmed: FieldValue.delete(),
         emailVerifyToken: FieldValue.delete(),
       });
+      return email;
     });
+
+    const legacyQuery = nftMintSubscriptionCollection
+      .where('subscriberEmail', '==', email)
+      .limit(499);
+    let querySnapshot = await legacyQuery.get();
+    while (querySnapshot.size) {
+      const batch = db.batch();
+      const legacyFollowees = querySnapshot.docs.map(
+        doc => doc.data().subscribedWallet
+      );
+      batch.update(userRef, {
+        followees: FieldValue.arrayUnion(...legacyFollowees),
+      });
+      querySnapshot.docs.forEach(doc => batch.delete(doc.ref));
+      // eslint-disable-next-line no-await-in-loop
+      await batch.commit();
+      // eslint-disable-next-line no-await-in-loop
+      querySnapshot = await legacyQuery.get();
+    }
+
     res.sendStatus(200);
   } catch (error) {
     switch (error.message) {
diff --git a/src/server/api/routes/users/v2/follow.js b/src/server/api/routes/users/v2/follow.js
index abc27eb94..803a04fab 100644
--- a/src/server/api/routes/users/v2/follow.js
+++ b/src/server/api/routes/users/v2/follow.js
@@ -10,6 +10,7 @@ const {
   db,
   FieldValue,
   walletUserCollection,
+  nftMintSubscriptionCollection,
 } = require('../../../../modules/firebase');
 
 const router = Router();
@@ -26,7 +27,18 @@ router.get(
         res.json({ followees: [] });
         return;
       }
-      const { followees = [] } = userDoc.data();
+      const { followees: walletFollowees = [], email } = userDoc.data();
+      const snapshot = await nftMintSubscriptionCollection
+        .where('subscriberEmail', '==', email)
+        .get();
+      const legacyFollowees = snapshot.docs.map(doc => {
+        const { subscribedWallet } = doc.data();
+        return subscribedWallet;
+      });
+      const followees = new Set([
+        ...walletFollowees,
+        ...legacyFollowees,
+      ]).values();
       res.json({ followees });
     } catch (err) {
       handleRestfulError(req, res, next, err);
@@ -58,8 +70,7 @@ router.post(
             throw new Error('EMAIL_NOT_SET_YET');
           }
         }
-        const creatorRef = walletUserCollection.doc(user);
-        await t.update(creatorRef, {
+        await t.update(userRef, {
           followees: FieldValue.arrayUnion(creator),
         });
       });
@@ -92,18 +103,25 @@ router.delete(
         res.status(400).send('INVALID_CREATOR_ADDRESS');
         return;
       }
-      await walletUserCollection.doc(user).update({
-        followees: FieldValue.arrayRemove(creator),
+      await db.runTransaction(async t => {
+        const userDoc = await t.get(walletUserCollection.doc(user));
+        const { email } = userDoc.data();
+        const snapshot = await t.get(
+          nftMintSubscriptionCollection
+            .where('subscriberEmail', '==', email)
+            .where('subscribedWallet', '==', creator)
+            .limit(1)
+        );
+        t.update(walletUserCollection.doc(user), {
+          followees: FieldValue.arrayRemove(creator),
+        });
+        if (snapshot.docs.length > 0) {
+          t.delete(snapshot.docs[0].ref);
+        }
       });
       res.sendStatus(200);
     } catch (err) {
-      switch (err.message) {
-        case 'EMAIL_NOT_SET_YET':
-          res.status(403).send('EMAIL_NOT_SET_YET');
-          break;
-        default:
-          handleRestfulError(req, res, next, err);
-      }
+      handleRestfulError(req, res, next, err);
     }
   }
 );