feat(ai): serve premium insights to entitled readers

Public Yohaku reads used a hard premium block. Reuse the TTS entitlement check so owners and active members can load cached insights.

Author: Innei

apps/core/src/modules/ai/ai-insights/ai-insights.adapter.ts

@@ -7,6 +7,7 @@ import { CollectionRefTypes } from '~/constants/db.constant'
 import { DatabaseService } from '~/processors/database/database.service'
 
 import { ConfigsService } from '../../configs/configs.service'
+import { EntitlementService } from '../../membership/entitlement.service'
 import { AI_PROMPTS } from '../ai.prompts'
 import { AiService } from '../ai.service'
 import { isGlobalArticleVisible } from '../ai-article-visibility.util'
@@ -54,6 +55,7 @@ export class AiInsightsAdapter implements MultilangAdapter<
     private readonly configService: ConfigsService,
     private readonly aiService: AiService,
     private readonly eventEmitter: EventEmitter2,
+    private readonly entitlementService: EntitlementService,
   ) {}
 
   toInsightsDoc(row: AiInsightsRow | null): AIInsightsModel | null {
@@ -75,7 +77,11 @@ export class AiInsightsAdapter implements MultilangAdapter<
 
   async resolveArticleDetailed(
     articleId: string,
-    options?: { blockPremium?: boolean },
+    options?: {
+      blockPremium?: boolean
+      isOwner?: boolean
+      readerId?: string
+    },
   ): Promise<{
     article: ArticleForInsights
     sourceLang: string
@@ -97,7 +103,12 @@ export class AiInsightsAdapter implements MultilangAdapter<
     if (
       options?.blockPremium &&
       article.type === CollectionRefTypes.Post &&
-      (article.document as { isPremium?: boolean | null }).isPremium
+      (await this.entitlementService.isPremiumLocked({
+        isPremium: (article.document as { isPremium?: boolean | null })
+          .isPremium,
+        isOwner: Boolean(options.isOwner),
+        readerId: options.readerId,
+      }))
     ) {
       throw createAppException(AppErrorCode.POST_HIDDEN_OR_ENCRYPTED)
     }

apps/core/src/modules/ai/ai-insights/ai-insights.controller.ts

@@ -12,7 +12,9 @@ import type { FastifyReply } from 'fastify'
 
 import { ApiController } from '~/common/decorators/api-controller.decorator'
 import { Auth } from '~/common/decorators/auth.decorator'
+import { CurrentReaderId } from '~/common/decorators/current-user.decorator'
 import { HTTPDecorators } from '~/common/decorators/http.decorator'
+import { HasAdminAccess } from '~/common/decorators/role.decorator'
 import { AppErrorCode, createAppException } from '~/common/errors'
 import { withMeta } from '~/common/response/envelope.types'
 import { MetaObjectBuilder } from '~/common/response/meta-builder'
@@ -118,10 +120,14 @@ export class AiInsightsController {
   getArticleInsights(
     @Param() params: EntityIdDto,
     @Query() query: GetInsightsQueryDto,
+    @HasAdminAccess() isOwner?: boolean,
+    @CurrentReaderId() readerId?: string,
   ) {
     return this.service.getOrGenerateInsightsForArticle(params.id, {
       lang: query.lang ? parseLanguageCode(query.lang) : DEFAULT_SUMMARY_LANG,
       onlyDb: query.onlyDb,
+      isOwner: Boolean(isOwner),
+      readerId,
     })
   }
 
@@ -131,6 +137,8 @@ export class AiInsightsController {
     @Param() params: EntityIdDto,
     @Query() query: GetInsightsStreamQueryDto,
     @Res() reply: FastifyReply,
+    @HasAdminAccess() isOwner?: boolean,
+    @CurrentReaderId() readerId?: string,
   ) {
     initSse(reply)
     let closed = false
@@ -144,6 +152,8 @@ export class AiInsightsController {
           lang: query.lang
             ? parseLanguageCode(query.lang)
             : DEFAULT_SUMMARY_LANG,
+          isOwner: Boolean(isOwner),
+          readerId,
         },
       )
       let sentToken = false

apps/core/src/modules/ai/ai-insights/ai-insights.service.ts

@@ -140,7 +140,7 @@ export class AiInsightsService implements OnModuleInit {
 
   async streamInsightsForArticle(
     articleId: string,
-    options: { lang: string },
+    options: { lang: string; isOwner?: boolean; readerId?: string },
   ): Promise<{
     events: AsyncIterable<AiStreamEvent>
     result: Promise<AIInsightsModel>
@@ -151,7 +151,11 @@ export class AiInsightsService implements OnModuleInit {
     }
     const { article, sourceLang } = await this.adapter.resolveArticleDetailed(
       articleId,
-      { blockPremium: true },
+      {
+        blockPremium: true,
+        isOwner: options.isOwner,
+        readerId: options.readerId,
+      },
     )
     const lang = options.lang || sourceLang
     const existing = await this.findValidInsights(articleId, lang, article.text)
@@ -169,11 +173,20 @@ export class AiInsightsService implements OnModuleInit {
 
   async getOrGenerateInsightsForArticle(
     articleId: string,
-    options: { lang: string; onlyDb?: boolean },
+    options: {
+      lang: string
+      onlyDb?: boolean
+      isOwner?: boolean
+      readerId?: string
+    },
   ): Promise<AIInsightsModel | null> {
     const { article, sourceLang } = await this.adapter.resolveArticleDetailed(
       articleId,
-      { blockPremium: true },
+      {
+        blockPremium: true,
+        isOwner: options.isOwner,
+        readerId: options.readerId,
+      },
     )
     const lang = options.lang || sourceLang
     const existing = await this.findValidInsights(articleId, lang, article.text)

apps/core/test/src/modules/ai/ai-insights-translation.service.spec.ts

@@ -41,6 +41,7 @@ const createService = () => {
     configService as any,
     aiService as any,
     eventEmitter as any,
+    { isPremiumLocked: vi.fn(async () => false) } as any,
   )
   const multilang = new MultilangGenerationService(
     aiInFlightService as any,

apps/core/test/src/modules/ai/ai-insights.faux.e2e.spec.ts

@@ -119,6 +119,7 @@ function createService(runtime: PiRuntimeAdapter) {
     configService as any,
     aiService as any,
     eventEmitter as any,
+    { isPremiumLocked: vi.fn(async () => false) } as any,
   )
   const multilang = new MultilangGenerationService(
     aiInFlightService as any,

apps/core/test/src/modules/ai/ai-insights.service.spec.ts

@@ -56,6 +56,15 @@ const createService = () => {
     createInsightsTranslationTask: vi.fn(),
   }
   const eventEmitter = { emit: vi.fn() }
+  const entitlementService = {
+    isPremiumLocked: vi.fn(
+      async (input: {
+        isPremium?: boolean | null
+        isOwner: boolean
+        readerId?: string
+      }) => Boolean(input.isPremium) && !input.isOwner && !input.readerId,
+    ),
+  }
   const generationMetrics = {
     attachLatest: vi.fn(async (_type: string, items: unknown[]) =>
       items.map((item) => ({
@@ -72,6 +81,7 @@ const createService = () => {
     configService as any,
     aiService as any,
     eventEmitter as any,
+    entitlementService as any,
   )
   const multilang = new MultilangGenerationService(
     aiInFlightService as any,
@@ -94,6 +104,7 @@ const createService = () => {
     aiTaskService,
     configService,
     databaseService,
+    entitlementService,
     eventEmitter,
     generationMetrics,
     repository,
@@ -270,6 +281,58 @@ describe('AiInsightsService', () => {
     ).rejects.toThrow(AppException)
   })
 
+  it('serves cached insights for a premium post to an entitled reader', async () => {
+    const { databaseService, repository, service } = createService()
+    databaseService.findGlobalById.mockResolvedValue({
+      type: CollectionRefTypes.Post,
+      document: {
+        id: 'post-1',
+        title: 'Premium Post',
+        text: 'Premium text',
+        isPublished: true,
+        isPremium: true,
+      },
+    })
+    repository.findByRefAndLang.mockResolvedValue({
+      ...row,
+      hash: 'mismatch-so-onlyDb-returns-null',
+    } as any)
+
+    await expect(
+      service.getOrGenerateInsightsForArticle('post-1', {
+        lang: 'zh',
+        onlyDb: true,
+        readerId: 'reader-1',
+      }),
+    ).resolves.toBeNull()
+  })
+
+  it('serves cached insights for a premium post to the owner', async () => {
+    const { databaseService, repository, service } = createService()
+    databaseService.findGlobalById.mockResolvedValue({
+      type: CollectionRefTypes.Post,
+      document: {
+        id: 'post-1',
+        title: 'Premium Post',
+        text: 'Premium text',
+        isPublished: true,
+        isPremium: true,
+      },
+    })
+    repository.findByRefAndLang.mockResolvedValue({
+      ...row,
+      hash: 'mismatch-so-onlyDb-returns-null',
+    } as any)
+
+    await expect(
+      service.getOrGenerateInsightsForArticle('post-1', {
+        lang: 'zh',
+        onlyDb: true,
+        isOwner: true,
+      }),
+    ).resolves.toBeNull()
+  })
+
   it('does not block background insight regeneration for a premium post', async () => {
     const {
       aiTaskService,