How to extend/wrap drizzle with additional function?

[dimonnwc3]

I want to add a wrapper function findFirstOrThrow

async function findFirstOrThrow(params) {
  const result = await db.query.users.findFirst(params)

  if (!result) throw new Error("error")

  return result
}

it has two issues:

1. works only with users table
2. params are not typed in typescript

how do I extend or make wrapper properly, to make it work on any tables and with correct typings?

[Angelelz]

Here is one way:

type UserFindFirstInput = Parameters<typeof db['query']['users']['findFirst']>[0]

async function findFirstOrThrow(params: UserFindFirstInput) {
  return await db.query.users.findFirst(params).then(val => {
    if (!val) throw new Error("error")
    
    return val
  })
}

[brandon-pereira]

This works... unless the params are columns or with, then the resulting type is not returned. It would need to be some type of generic function that is passed into findFirst but I couldn't get that part working 🤔

Example of possible API:

type UserFindFirstInput = Parameters<typeof db['query']['users']['findFirst']>[0]

async function findFirstOrThrow<T>(params: T) {
   // this does not work! the findFirst generic is significantly different!
  return await db.query.users.findFirst<T>(params).then(val => {
    if (!val) throw new Error("error")
    
    return val
  })
}