From 8c84fc58b8786b99b93955f763979c17bae4a017 Mon Sep 17 00:00:00 2001
From: Hongbo Wu <hongbo@omnivore.app>
Date: Thu, 20 Jun 2024 17:36:57 +0800
Subject: [PATCH] add more subscription features to the score api payload

---
 packages/api/src/jobs/update_home.ts | 20 +++++++++++++++++++-
 packages/api/src/services/score.ts   | 10 ++++++++++
 2 files changed, 29 insertions(+), 1 deletion(-)

diff --git a/packages/api/src/jobs/update_home.ts b/packages/api/src/jobs/update_home.ts
index 661b235d0..b565ed7a2 100644
--- a/packages/api/src/jobs/update_home.ts
+++ b/packages/api/src/jobs/update_home.ts
@@ -1,7 +1,7 @@
 import client from 'prom-client'
 import { LibraryItem } from '../entity/library_item'
 import { PublicItem } from '../entity/public_item'
-import { Subscription } from '../entity/subscription'
+import { Subscription, SubscriptionType } from '../entity/subscription'
 import { User } from '../entity/user'
 import { registerMetric } from '../prometheus'
 import { redisDataSource } from '../redis_data_source'
@@ -41,6 +41,9 @@ interface Candidate {
   subscription?: {
     name: string
     type: string
+    autoAddToLibrary?: boolean | null
+    createdAt: Date
+    fetchContent?: boolean | null
   }
 }
 
@@ -102,6 +105,7 @@ const publicItemToCandidate = (item: PublicItem): Candidate => ({
   subscription: {
     name: item.source.name,
     type: item.source.type,
+    createdAt: item.source.createdAt,
   },
   score: 0,
 })
@@ -222,6 +226,20 @@ const rankCandidates = async (
         word_count: item.wordCount,
         published_at: item.publishedAt,
         subscription: item.subscription?.name,
+        inbox_folder: item.folder === 'inbox',
+        is_feed: item.subscription?.type === SubscriptionType.Rss,
+        is_newsletter: item.subscription?.type === SubscriptionType.Newsletter,
+        is_subscription: !!item.subscription,
+        item_word_count: item.wordCount,
+        subscription_count: 0,
+        subscription_auto_add_to_library: item.subscription?.autoAddToLibrary,
+        subscription_fetch_content: item.subscription?.fetchContent,
+        days_since_subscribed: item.subscription
+          ? Math.floor(
+              (Date.now() - item.subscription.createdAt.getTime()) /
+                (1000 * 60 * 60 * 24)
+            )
+          : undefined,
       } as Feature
       return acc
     }, {} as Record<string, Feature>),
diff --git a/packages/api/src/services/score.ts b/packages/api/src/services/score.ts
index c8c8c1660..25c78f5a0 100644
--- a/packages/api/src/services/score.ts
+++ b/packages/api/src/services/score.ts
@@ -6,6 +6,12 @@ export interface Feature {
   has_thumbnail: boolean
   has_site_icon: boolean
   saved_at: Date
+  item_word_count: number
+  is_subscription: boolean
+  inbox_folder: boolean
+  is_newsletter: boolean
+  is_feed: boolean
+
   site?: string
   language?: string
   author?: string
@@ -15,6 +21,10 @@ export interface Feature {
   folder?: string
   published_at?: Date
   subscription?: string
+  subscription_auto_add_to_library?: boolean
+  subscription_fetch_content?: boolean
+  days_since_subscribed?: number
+  subscription_count?: number
 }
 
 export interface ScoreApiRequestBody {