import { ServiceImpl } from '@connectrpc/connect'
import { Service } from '../../../proto/bsky_connect.js'
import { Database } from '../db/index.js'
import { TimeCidKeyset, paginate } from '../db/pagination.js'

export default (db: Database): Partial<ServiceImpl<typeof Service>> => ({
  async getActorFeeds(req) {
    const { actorDid, limit, cursor } = req

    const { ref } = db.db.dynamic
    let builder = db.db
      .selectFrom('feed_generator')
      .selectAll()
      .where('feed_generator.creator', '=', actorDid)

    const keyset = new TimeCidKeyset(
      ref('feed_generator.createdAt'),
      ref('feed_generator.cid'),
    )
    builder = paginate(builder, {
      limit,
      cursor,
      keyset,
    })
    const feeds = await builder.execute()

    return {
      uris: feeds.map((f) => f.uri),
      cursor: keyset.packFromResult(feeds),
    }
  },

  async getSuggestedFeeds(req) {
    const feeds = await db.db
      .selectFrom('suggested_feed')
      .orderBy('suggested_feed.order', 'asc')
      .if(!!req.cursor, (q) => q.where('order', '>', parseInt(req.cursor, 10)))
      .limit(req.limit || 50)
      .selectAll()
      .execute()
    return {
      uris: feeds.map((f) => f.uri),
      cursor: feeds.at(-1)?.order.toString(),
    }
  },

  async searchFeedGenerators(req) {
    const { ref } = db.db.dynamic
    const limit = req.limit
    const query = req.query.trim()
    let builder = db.db
      .selectFrom('feed_generator')
      .if(!!query, (q) => q.where('displayName', 'ilike', `%${query}%`))
      .selectAll()
    const keyset = new TimeCidKeyset(
      ref('feed_generator.createdAt'),
      ref('feed_generator.cid'),
    )
    builder = paginate(builder, { limit, keyset })
    const feeds = await builder.execute()
    return {
      uris: feeds.map((f) => f.uri),
      cursor: keyset.packFromResult(feeds),
    }
  },

  async getFeedGeneratorStatus() {
    throw new Error('unimplemented')
  },
})