chore: wip

Author: chrisbbreuer

storage/framework/core/orm/src/generated/User.ts

@@ -1,5 +1,6 @@
 import type { ColumnType, Generated, Insertable, Selectable, Updateable } from 'kysely'
-import { type Err, type Result, err, handleError, ok } from '@stacksjs/error-handling'
+import type { Result } from '@stacksjs/error-handling'
+import { err, handleError, ok } from '@stacksjs/error-handling'
 import { db } from '@stacksjs/query-builder'
 
 // import { Kysely, MysqlDialect, PostgresDialect } from 'kysely'
@@ -41,10 +42,89 @@ export class UserModel {
     this.userData = userData
   }
 
+  // Method to find a user by ID
+  static async find(id: number): Promise<UserModel | null> {
+    const userData = await db.selectFrom('users')
+      .where('id', '=', id)
+      .selectAll()
+      .executeTakeFirst()
+
+    if (!userData)
+      return null
+
+    return new UserModel(userData)
+  }
+
+  // Method to get a user by criteria
+  static async get(criteria: Partial<UserType>): Promise<UserModel[]> {
+    if (criteria.id !== undefined) {
+      const users = await db.selectFrom('users')
+        .where('id', '=', criteria.id)
+        .selectAll()
+        .execute()
+      return users.map(user => new UserModel(user))
+    }
+
+    return []
+  }
+
+  // Method to get all users
+  static async all(): Promise<UserModel[]> {
+    const users = await db.selectFrom('users')
+      .selectAll()
+      .execute()
+
+    return users.map(user => new UserModel(user))
+  }
+
+  // Method to create a new user
+  static async create(newUser: NewUser): Promise<UserModel> {
+    const user = await db.insertInto('users')
+      .values(newUser)
+      .returningAll()
+      .executeTakeFirstOrThrow()
+
+    return new UserModel(user)
+  }
+
+  // Method to update a user
+  static async update(id: number, userUpdate: UserUpdate): Promise<UserModel> {
+    const user = await db.updateTable('users')
+      .set(userUpdate)
+      .where('id', '=', id)
+      .returningAll()
+      .executeTakeFirstOrThrow()
+
+    return new UserModel(user)
+  }
+
+  // Method to remove a user
+  static async remove(id: number): Promise<UserModel> {
+    const user = await db.deleteFrom('users')
+      .where('id', '=', id)
+      .returningAll()
+      .executeTakeFirstOrThrow()
+
+    return new UserModel(user)
+  }
+
+  // Method to find a user by email
+  static async findByEmail(email: string): Promise<UserModel | null> {
+    const userData = await db.selectFrom('users')
+      .where('email', '=', email)
+      .selectAll()
+      .executeTakeFirst()
+
+    if (!userData)
+      return null
+
+    return new UserModel(userData)
+  }
+
   async where(criteria: Partial<UserType>) {
     let query = db.selectFrom('users')
     if (criteria.id)
-      query = query.where('id', '=', criteria.id) // Kysely is immutable, you must re-assign!
+      query = query.where('id', '=', criteria.id) // Kysely is immutable, we must re-assign
     if (criteria.email)
       query = query.where('email', '=', criteria.email)
     if (criteria.name !== undefined) {
@@ -65,6 +145,29 @@ export class UserModel {
     return await query.selectAll().execute()
   }
 
+  async first() {
+    return await db.selectFrom('users')
+      .selectAll()
+      .executeTakeFirst()
+  }
+
+  async last() {
+    return await db.selectFrom('users')
+      .selectAll()
+      .orderBy('id', 'desc')
+      .executeTakeFirst()
+  }
+
+  // Method to get the user instance itself
+  self() {
+    return this
+  }
+
+  // Method to get the user instance data
+  get() {
+    return this.userData
+  }
+
   // Method to update the user instance
   async update(userUpdate: UserUpdate): Promise<Result<UserType, Error>> {
     if (this.userData.id === undefined)
@@ -137,6 +240,8 @@ export class UserModel {
   }
 }
 
+const Model = UserModel
+
 // starting here, ORM functions
 export async function find(id: number): Promise<UserModel | null> {
   const userData = await db.selectFrom('users')
@@ -154,7 +259,7 @@ export async function get(criteria: Partial<UserType>) {
   let query = db.selectFrom('users')
 
   if (criteria.id)
-    query = query.where('id', '=', criteria.id) // Kysely is immutable, you must re-assign!
+    query = query.where('id', '=', criteria.id) // Kysely is immutable, we must re-assign
 
   if (criteria.email)
     query = query.where('email', '=', criteria.email)
@@ -195,6 +300,26 @@ export async function create(newUser: NewUser) {
     .executeTakeFirstOrThrow()
 }
 
+export async function getOne(criteria: Partial<UserType>) {
+  return await db.selectFrom('users')
+    .where('id', '=', criteria.id)
+    .selectAll()
+    .executeTakeFirst()
+}
+
+export async function first() {
+  return await db.selectFrom('users')
+    .selectAll()
+    .executeTakeFirst()
+}
+
+export async function last() {
+  return await db.selectFrom('users')
+    .selectAll()
+    .orderBy('id', 'desc')
+    .executeTakeFirst()
+}
+
 export async function update(id: number, userUpdate: UserUpdate) {
   return await db.updateTable('users')
     .set(userUpdate)
@@ -216,6 +341,38 @@ export async function findByEmail(email: string) {
     .executeTakeFirst()
 }
 
+export async function where(criteria: Partial<UserType>) {
+  let query = db.selectFrom('users')
+
+  if (criteria.id)
+    query = query.where('id', '=', criteria.id) // Kysely is immutable, we must re-assign
+
+  if (criteria.email)
+    query = query.where('email', '=', criteria.email)
+
+  if (criteria.name !== undefined) {
+    query = query.where(
+      'name',
+      criteria.name === null ? 'is' : '=',
+      criteria.name,
+    )
+  }
+
+  if (criteria.password)
+    query = query.where('password', '=', criteria.password)
+
+  if (criteria.created_at)
+    query = query.where('created_at', '=', criteria.created_at)
+
+  if (criteria.updated_at)
+    query = query.where('updated_at', '=', criteria.updated_at)
+
+  if (criteria.deleted_at)
+    query = query.where('deleted_at', '=', criteria.deleted_at)
+
+  return await query.selectAll().execute()
+}
+
 export const User = {
   find,
   get,
@@ -224,4 +381,8 @@ export const User = {
   update,
   remove,
   findByEmail,
+  Model,
+  first,
+  last,
+  where,
 }