import { sql } from 'drizzle-orm'
import { threadSubscriptions } from '../db/schema'

export function ensureThreadSubscription(uploadId: string, userId: number, db: Pick<ReturnType<typeof useDb>, 'insert'> = useDb()) {
  return db.insert(threadSubscriptions).values({ uploadId, userId })
    .onDuplicateKeyUpdate({ set: { userId: sql`${threadSubscriptions.userId}` } })
}