feat: add api subject recs

drizzle/orm.ts

@@ -13,6 +13,7 @@ export type ISubjectRelation = typeof schema.chiiSubjectRelations.$inferSelect;
 export type ISubjectTopic = typeof schema.chiiSubjectTopics.$inferSelect;
 export type ISubjectPost = typeof schema.chiiSubjectPosts.$inferSelect;
 export type ISubjectEpStatus = typeof schema.chiiEpStatus.$inferSelect;
+export type ISubjectRec = typeof schema.chiiSubjectRec.$inferSelect;
 
 export type IEpisode = typeof schema.chiiEpisodes.$inferSelect;
 

drizzle/schema.ts

@@ -1008,15 +1008,15 @@ export const chiiSubjectPosts = mysqlTable(
 export const chiiSubjectRec = mysqlTable(
   'chii_subject_rec',
   {
-    subjectId: mediumint('subject_id').notNull(),
-    recSubjectId: mediumint('rec_subject_id').notNull(),
-    mioSim: float('mio_sim').notNull(),
-    mioCount: mediumint('mio_count').notNull(),
+    subjectID: mediumint('subject_id').notNull(),
+    recSubjectID: mediumint('rec_subject_id').notNull(),
+    sim: float('mio_sim').notNull(),
+    count: mediumint('mio_count').notNull(),
   },
   (table) => {
     return {
-      subjectId: index('subject_id').on(table.subjectId),
-      mioCount: index('mio_count').on(table.mioCount),
+      subjectId: index('subject_id').on(table.subjectID),
+      mioCount: index('mio_count').on(table.count),
     };
   },
 );

lib/types/res.ts

@@ -433,6 +433,16 @@ export const SubjectStaff = t.Object(
   { $id: 'SubjectStaff' },
 );
 
+export type ISubjectRec = Static<typeof SubjectRec>;
+export const SubjectRec = t.Object(
+  {
+    subject: t.Ref(SlimSubject),
+    sim: t.Number(),
+    count: t.Integer(),
+  },
+  { $id: 'SubjectRec', title: 'SubjectRec' },
+);
+
 export type ICharacterRelation = Static<typeof CharacterRelation>;
 export const CharacterRelation = t.Object(
   {

routes/private/routes/subject.test.ts

@@ -62,6 +62,17 @@ describe('subject', () => {
     expect(res.json()).toMatchSnapshot();
   });
 
+  test('should get subject recs', async () => {
+    const app = createTestServer();
+    await app.register(setup);
+    const res = await app.inject({
+      method: 'get',
+      url: '/subjects/12/recs',
+      query: { limit: '2', offset: '0' },
+    });
+    expect(res.json()).toMatchSnapshot();
+  });
+
   test('should get subject comments', async () => {
     const app = createTestServer();
     await app.register(setup);

routes/private/routes/subject.ts

@@ -56,6 +56,14 @@
   };
 }
 
+function toSubjectRec(subject: orm.ISubject, rec: orm.ISubjectRec): res.ISubjectRec {
+  return {
+    subject: convert.toSlimSubject(subject),
+    sim: rec.sim,
+    count: rec.count,
+  };
+}
+
 // eslint-disable-next-line @typescript-eslint/require-await
 export async function setup(app: App) {
   app.get(
@@ -381,6 +389,67 @@
     },
   );
 
+  app.get(
+    '/subjects/:subjectID/recs',
+    {
+      schema: {
+        summary: '获取条目的推荐',
+        operationId: 'getSubjectRecs',
+        tags: [Tag.Subject],
+        security: [{ [Security.CookiesSession]: [], [Security.HTTPBearer]: [] }],
+        params: t.Object({
+          subjectID: t.Integer(),
+        }),
+        querystring: t.Object({
+          limit: t.Optional(
+            t.Integer({ default: 10, minimum: 1, maximum: 10, description: 'max 10' }),
+          ),
+          offset: t.Optional(t.Integer({ default: 0, minimum: 0, description: 'min 0' })),
+        }),
+        response: {
+          200: res.Paged(t.Ref(res.SubjectRec)),
+        },
+      },
+    },
+    async ({ auth, params: { subjectID }, query: { limit = 10, offset = 0 } }) => {
+      const subject = await fetcher.fetchSlimSubjectByID(subjectID, auth.allowNsfw);
+      if (!subject) {
+        throw new NotFoundError(`subject ${subjectID}`);
+      }
+      const condition = op.and(
+        op.eq(schema.chiiSubjectRec.subjectID, subjectID),
+        op.ne(schema.chiiSubjects.ban, 1),
+        auth.allowNsfw ? undefined : op.eq(schema.chiiSubjects.nsfw, false),
+      );
+      const [{ count = 0 } = {}] = await db
+        .select({ count: op.count() })
+        .from(schema.chiiSubjectRec)
+        .innerJoin(
+          schema.chiiSubjects,
+          op.eq(schema.chiiSubjectRec.recSubjectID, schema.chiiSubjects.id),
+        )
+        .where(condition)
+        .execute();
+      const data = await db
+        .select()
+        .from(schema.chiiSubjectRec)
+        .innerJoin(
+          schema.chiiSubjects,
+          op.eq(schema.chiiSubjectRec.recSubjectID, schema.chiiSubjects.id),
+        )
+        .where(condition)
+        .orderBy(op.asc(schema.chiiSubjectRec.count))
+        .limit(limit)
+        .offset(offset)
+        .execute();
+      const recs = data.map((d) => toSubjectRec(d.chii_subjects, d.chii_subject_rec));
+      return {
+        data: recs,
+        total: count,
+      };
+    },
+  );
+
   app.get(
     '/subjects/:subjectID/comments',
     {

routes/res.ts

@@ -38,6 +38,7 @@ export function addSchemas(app: App) {
   app.addSchema(res.SubjectImages);
   app.addSchema(res.SubjectPlatform);
   app.addSchema(res.SubjectRating);
+  app.addSchema(res.SubjectRec);
   app.addSchema(res.SubjectRelation);
   app.addSchema(res.SubjectRelationType);
   app.addSchema(res.SubjectStaff);