From 9f260c1442d5a59c31adad370ca2931519bfecfd Mon Sep 17 00:00:00 2001
From: Hongbo Wu <hongbo@omnivore.app>
Date: Thu, 8 Dec 2022 18:20:58 +0800
Subject: [PATCH 1/4] Create a label for new group if not exists

---
 .../src/resolvers/recommendations/index.ts    |  4 +++
 packages/api/src/services/labels.ts           | 27 +++++++++++++++++++
 2 files changed, 31 insertions(+)

diff --git a/packages/api/src/resolvers/recommendations/index.ts b/packages/api/src/resolvers/recommendations/index.ts
index 8dd3b1d09..2f2e617b7 100644
--- a/packages/api/src/resolvers/recommendations/index.ts
+++ b/packages/api/src/resolvers/recommendations/index.ts
@@ -38,6 +38,7 @@ import { In } from 'typeorm'
 import { getPageByParam } from '../../elastic/pages'
 import { enqueueRecommendation } from '../../utils/createTask'
 import { env } from '../../env'
+import { createLabel } from '../../services/labels'
 
 export const createGroupResolver = authorized<
   CreateGroupSuccess,
@@ -71,6 +72,9 @@ export const createGroupResolver = authorized<
       expiresInDays: input.expiresInDays,
     })
 
+    // create a new label for the group
+    await createLabel(userData, { name: input.name })
+
     const inviteUrl = getInviteUrl(invite)
     const user = userDataToUser(userData)
 
diff --git a/packages/api/src/services/labels.ts b/packages/api/src/services/labels.ts
index cb8432b06..6d9ed14ef 100644
--- a/packages/api/src/services/labels.ts
+++ b/packages/api/src/services/labels.ts
@@ -76,3 +76,30 @@ export const getLabelsByIds = async (
     select: ['id', 'name', 'color', 'description', 'createdAt'],
   })
 }
+
+export const createLabel = async (
+  user: User,
+  label: {
+    name: string
+    color?: string
+    description?: string
+  }
+): Promise<Label> => {
+  const existingLabel = await getRepository(Label).findOneBy({
+    user: { id: user.id },
+    name: ILike(label.name),
+  })
+
+  if (existingLabel) {
+    return existingLabel
+  }
+
+  // create a new label and assign a random color if not provided
+  label.color =
+    label.color || `#${Math.floor(Math.random() * 16777215).toString(16)}`
+
+  return getRepository(Label).save({
+    ...label,
+    user: { id: user.id },
+  })
+}

From b942303b18119f61eb014582697ff2e4efb20e40 Mon Sep 17 00:00:00 2001
From: Hongbo Wu <hongbo@omnivore.app>
Date: Thu, 8 Dec 2022 18:35:22 +0800
Subject: [PATCH 2/4] Create a rule after creating or joining a group

---
 .../src/resolvers/recommendations/index.ts    | 39 ++++++++++++++++++-
 packages/api/src/services/rules.ts            | 28 +++++++++++++
 2 files changed, 66 insertions(+), 1 deletion(-)
 create mode 100644 packages/api/src/services/rules.ts

diff --git a/packages/api/src/resolvers/recommendations/index.ts b/packages/api/src/resolvers/recommendations/index.ts
index 2f2e617b7..4b97cd652 100644
--- a/packages/api/src/resolvers/recommendations/index.ts
+++ b/packages/api/src/resolvers/recommendations/index.ts
@@ -39,6 +39,8 @@ import { getPageByParam } from '../../elastic/pages'
 import { enqueueRecommendation } from '../../utils/createTask'
 import { env } from '../../env'
 import { createLabel } from '../../services/labels'
+import { createRule } from '../../services/rules'
+import { RuleActionType } from '../../entity/rule'
 
 export const createGroupResolver = authorized<
   CreateGroupSuccess,
@@ -73,7 +75,23 @@ export const createGroupResolver = authorized<
     })
 
     // create a new label for the group
-    await createLabel(userData, { name: input.name })
+    const label = await createLabel(userData, { name: input.name })
+
+    // create a rule to add the label to all pages in the group
+    await createRule(userData, {
+      name: input.name,
+      actions: [
+        {
+          type: RuleActionType.AddLabel,
+          params: [label.id],
+        },
+        {
+          type: RuleActionType.SendNotification,
+          params: [input.name],
+        },
+      ],
+      filter: `recommendedBy:${group.name}`,
+    })
 
     const inviteUrl = getInviteUrl(invite)
     const user = userDataToUser(userData)
@@ -267,6 +285,25 @@ export const joinGroupResolver = authorized<
 
     const group = await joinGroup(user, inviteCode)
 
+    // create a new label for the group
+    const label = await createLabel(user, { name: group.name })
+
+    // create a rule to add the label to all pages in the group
+    await createRule(user, {
+      name: group.name,
+      actions: [
+        {
+          type: RuleActionType.AddLabel,
+          params: [label.id],
+        },
+        {
+          type: RuleActionType.SendNotification,
+          params: [group.name],
+        },
+      ],
+      filter: `recommendedBy:${group.name}`,
+    })
+
     return {
       group,
     }
diff --git a/packages/api/src/services/rules.ts b/packages/api/src/services/rules.ts
new file mode 100644
index 000000000..b87bd6396
--- /dev/null
+++ b/packages/api/src/services/rules.ts
@@ -0,0 +1,28 @@
+import { User } from '../entity/user'
+import { Rule, RuleAction } from '../entity/rule'
+import { getRepository } from '../entity/utils'
+import { ILike } from 'typeorm'
+
+export const createRule = async (
+  user: User,
+  rule: {
+    name: string
+    description?: string
+    actions: RuleAction[]
+    filter: string
+  }
+): Promise<Rule> => {
+  const existingRule = await getRepository(Rule).findOneBy({
+    user: { id: user.id },
+    name: ILike(rule.name),
+  })
+
+  if (existingRule) {
+    return existingRule
+  }
+
+  return getRepository(Rule).save({
+    ...rule,
+    user: { id: user.id },
+  })
+}

From 04cb360c2fb3e76cc83107c6380010d8d6742552 Mon Sep 17 00:00:00 2001
From: Hongbo Wu <hongbo@omnivore.app>
Date: Thu, 8 Dec 2022 18:50:36 +0800
Subject: [PATCH 3/4] Fix some issue

---
 .../src/resolvers/recommendations/index.ts    | 42 ++-----------------
 packages/api/src/services/groups.ts           | 27 ++++++++++++
 packages/api/src/services/labels.ts           |  6 +--
 packages/api/src/services/rules.ts            |  7 ++--
 4 files changed, 36 insertions(+), 46 deletions(-)

diff --git a/packages/api/src/resolvers/recommendations/index.ts b/packages/api/src/resolvers/recommendations/index.ts
index 4b97cd652..bb1a48aac 100644
--- a/packages/api/src/resolvers/recommendations/index.ts
+++ b/packages/api/src/resolvers/recommendations/index.ts
@@ -25,6 +25,7 @@ import {
 } from '../../generated/graphql'
 import {
   createGroup,
+  createLabelAndRuleForGroup,
   getInviteUrl,
   getRecommendationGroups,
   joinGroup,
@@ -38,9 +39,6 @@ import { In } from 'typeorm'
 import { getPageByParam } from '../../elastic/pages'
 import { enqueueRecommendation } from '../../utils/createTask'
 import { env } from '../../env'
-import { createLabel } from '../../services/labels'
-import { createRule } from '../../services/rules'
-import { RuleActionType } from '../../entity/rule'
 
 export const createGroupResolver = authorized<
   CreateGroupSuccess,
@@ -74,24 +72,7 @@ export const createGroupResolver = authorized<
       expiresInDays: input.expiresInDays,
     })
 
-    // create a new label for the group
-    const label = await createLabel(userData, { name: input.name })
-
-    // create a rule to add the label to all pages in the group
-    await createRule(userData, {
-      name: input.name,
-      actions: [
-        {
-          type: RuleActionType.AddLabel,
-          params: [label.id],
-        },
-        {
-          type: RuleActionType.SendNotification,
-          params: [input.name],
-        },
-      ],
-      filter: `recommendedBy:${group.name}`,
-    })
+    await createLabelAndRuleForGroup(uid, group.name)
 
     const inviteUrl = getInviteUrl(invite)
     const user = userDataToUser(userData)
@@ -285,24 +266,7 @@ export const joinGroupResolver = authorized<
 
     const group = await joinGroup(user, inviteCode)
 
-    // create a new label for the group
-    const label = await createLabel(user, { name: group.name })
-
-    // create a rule to add the label to all pages in the group
-    await createRule(user, {
-      name: group.name,
-      actions: [
-        {
-          type: RuleActionType.AddLabel,
-          params: [label.id],
-        },
-        {
-          type: RuleActionType.SendNotification,
-          params: [group.name],
-        },
-      ],
-      filter: `recommendedBy:${group.name}`,
-    })
+    await createLabelAndRuleForGroup(user.id, group.name)
 
     return {
       group,
diff --git a/packages/api/src/services/groups.ts b/packages/api/src/services/groups.ts
index 2cebfaa91..716ccdddc 100644
--- a/packages/api/src/services/groups.ts
+++ b/packages/api/src/services/groups.ts
@@ -8,6 +8,9 @@ import { RecommendationGroup, User as GraphqlUser } from '../generated/graphql'
 import { getRepository } from '../entity/utils'
 import { homePageURL } from '../env'
 import { userDataToUser } from '../utils/helpers'
+import { createLabel } from './labels'
+import { createRule } from './rules'
+import { RuleActionType } from '../entity/rule'
 
 export const createGroup = async (input: {
   admin: User
@@ -200,3 +203,27 @@ export const leaveGroup = async (
     return true
   })
 }
+
+export const createLabelAndRuleForGroup = async (
+  userId: string,
+  groupName: string
+) => {
+  // create a new label for the group
+  const label = await createLabel(userId, { name: groupName })
+
+  // create a rule to add the label to all pages in the group
+  await createRule(userId, {
+    name: groupName,
+    actions: [
+      {
+        type: RuleActionType.AddLabel,
+        params: [label.id],
+      },
+      {
+        type: RuleActionType.SendNotification,
+        params: [groupName],
+      },
+    ],
+    filter: `recommendedBy:"${groupName}"`,
+  })
+}
diff --git a/packages/api/src/services/labels.ts b/packages/api/src/services/labels.ts
index 6d9ed14ef..d83287ec4 100644
--- a/packages/api/src/services/labels.ts
+++ b/packages/api/src/services/labels.ts
@@ -78,7 +78,7 @@ export const getLabelsByIds = async (
 }
 
 export const createLabel = async (
-  user: User,
+  userId: string,
   label: {
     name: string
     color?: string
@@ -86,7 +86,7 @@ export const createLabel = async (
   }
 ): Promise<Label> => {
   const existingLabel = await getRepository(Label).findOneBy({
-    user: { id: user.id },
+    user: { id: userId },
     name: ILike(label.name),
   })
 
@@ -100,6 +100,6 @@ export const createLabel = async (
 
   return getRepository(Label).save({
     ...label,
-    user: { id: user.id },
+    user: { id: userId },
   })
 }
diff --git a/packages/api/src/services/rules.ts b/packages/api/src/services/rules.ts
index b87bd6396..c4fa6e2a3 100644
--- a/packages/api/src/services/rules.ts
+++ b/packages/api/src/services/rules.ts
@@ -1,10 +1,9 @@
-import { User } from '../entity/user'
 import { Rule, RuleAction } from '../entity/rule'
 import { getRepository } from '../entity/utils'
 import { ILike } from 'typeorm'
 
 export const createRule = async (
-  user: User,
+  userId: string,
   rule: {
     name: string
     description?: string
@@ -13,7 +12,7 @@ export const createRule = async (
   }
 ): Promise<Rule> => {
   const existingRule = await getRepository(Rule).findOneBy({
-    user: { id: user.id },
+    user: { id: userId },
     name: ILike(rule.name),
   })
 
@@ -23,6 +22,6 @@ export const createRule = async (
 
   return getRepository(Rule).save({
     ...rule,
-    user: { id: user.id },
+    user: { id: userId },
   })
 }

From b8e7b1580524c376c3e716ad33bf9a1229890a50 Mon Sep 17 00:00:00 2001
From: Hongbo Wu <hongbo@omnivore.app>
Date: Thu, 8 Dec 2022 19:01:54 +0800
Subject: [PATCH 4/4] Fix generating colors with less than 6 digits

---
 packages/api/src/services/labels.ts |  4 ++--
 packages/api/src/utils/helpers.ts   | 10 ++++++++++
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/packages/api/src/services/labels.ts b/packages/api/src/services/labels.ts
index d83287ec4..b20baeee2 100644
--- a/packages/api/src/services/labels.ts
+++ b/packages/api/src/services/labels.ts
@@ -6,6 +6,7 @@ import { addLabelInPage } from '../elastic/labels'
 import { getRepository } from '../entity/utils'
 import { Link } from '../entity/link'
 import DataLoader from 'dataloader'
+import { generateRandomColor } from '../utils/helpers'
 
 const batchGetLabelsFromLinkIds = async (
   linkIds: readonly string[]
@@ -95,8 +96,7 @@ export const createLabel = async (
   }
 
   // create a new label and assign a random color if not provided
-  label.color =
-    label.color || `#${Math.floor(Math.random() * 16777215).toString(16)}`
+  label.color = label.color || generateRandomColor()
 
   return getRepository(Label).save({
     ...label,
diff --git a/packages/api/src/utils/helpers.ts b/packages/api/src/utils/helpers.ts
index a3137fbfb..45fefe83c 100644
--- a/packages/api/src/utils/helpers.ts
+++ b/packages/api/src/utils/helpers.ts
@@ -273,3 +273,13 @@ export const wordsCount = (text: string, isHtml = true): number => {
 export const isBase64Image = (str: string): boolean => {
   return str.startsWith('data:image/')
 }
+
+export const generateRandomColor = (): string => {
+  return (
+    '#' +
+    Math.floor(Math.random() * 16777215)
+      .toString(16)
+      .padStart(6, '0')
+      .toUpperCase()
+  )
+}