48 lines
1.3 KiB
Swift
48 lines
1.3 KiB
Swift
import Fluent
|
|
import Vapor
|
|
|
|
struct VendorDB {
|
|
func create(_ model: Vendor.Create, on db: any Database) async throws -> Vendor.DTO {
|
|
let model = model.toModel()
|
|
try await model.save(on: db)
|
|
return model.toDTO()
|
|
}
|
|
|
|
func fetchAll(withBranches: Bool? = nil, on db: any Database) async throws -> [Vendor.DTO] {
|
|
var query = Vendor.query(on: db).sort(\.$name, .ascending)
|
|
if withBranches == true {
|
|
query = query.with(\.$branches)
|
|
}
|
|
return try await query.all().map { $0.toDTO(includeBranches: withBranches) }
|
|
}
|
|
|
|
func get(id: Vendor.IDValue, withBranches: Bool? = nil, on db: any Database) async throws -> Vendor.DTO? {
|
|
var query = Vendor.query(on: db).filter(\.$id == id)
|
|
|
|
if withBranches == true {
|
|
query = query.with(\.$branches)
|
|
}
|
|
return try await query.first().map { $0.toDTO(includeBranches: withBranches) }
|
|
}
|
|
|
|
func update(
|
|
id: Vendor.IDValue,
|
|
with updates: Vendor.Update,
|
|
on db: any Database
|
|
) async throws -> Vendor.DTO {
|
|
guard let vendor = try await Vendor.find(id, on: db) else {
|
|
throw Abort(.notFound)
|
|
}
|
|
vendor.applyUpdates(updates)
|
|
return vendor.toDTO()
|
|
}
|
|
|
|
func delete(id: Vendor.IDValue, on db: any Database) async throws {
|
|
guard let vendor = try await Vendor.find(id, on: db) else {
|
|
throw Abort(.notFound)
|
|
}
|
|
try await vendor.delete(on: db)
|
|
}
|
|
|
|
}
|