Вы не можете выбрать более 25 тем Темы должны начинаться с буквы или цифры, могут содержать дефисы(-) и должны содержать не более 35 символов.
 
 
 

164 строки
5.2 KiB

  1. //
  2. // mdlUser.swift
  3. //
  4. //
  5. // Created by Michiel Carman on 4/11/2024.
  6. //
  7. import Fluent
  8. import Vapor
  9. import Foundation
  10. import FluentKit
  11. final class mdlAssociate: Model, Content {
  12. static let schema = "associate"
  13. @ID(custom: "id", generatedBy: .database) var id: Int?
  14. @Field(key: "firstName") var firstname: String
  15. @Field(key: "lastName") var lastname: String
  16. @Field(key: "userName") var username: String
  17. @Field(key: "password") var password: String
  18. @Siblings(through: mdlXAssocDept.self, from: \.$associate, to: \.$department) var departments: [mdlDepartment]
  19. @Timestamp(key: "createDate", on: .create) var createdate: Date?
  20. @Field(key: "createUserId") var createuserid: Int?
  21. @Timestamp(key: "updateDate", on: .update) var updatedate: Date?
  22. @Field(key: "updateUserId") var updateuserid: Int?
  23. init() { }
  24. init(id: Int? = nil,
  25. firstname: String,
  26. lastname: String,
  27. username: String,
  28. password: String ,
  29. createuserid: Int? = nil,
  30. updateuserid: Int? = nil
  31. ) {
  32. self.id = id
  33. self.firstname = firstname
  34. self.lastname = lastname
  35. self.username = username
  36. self.password = password
  37. self.createuserid = createuserid
  38. self.updateuserid = updateuserid
  39. }
  40. }
  41. extension mdlAssociate{
  42. struct Create: AsyncMigration {
  43. func prepare(on database: Database) async throws {
  44. try await database.schema("associate")
  45. .field("id", .int, .identifier(auto: true))
  46. .field("firstName", .string)
  47. .field("lastName", .string)
  48. // .field("userName", .string)
  49. // .field("password", .string)
  50. .field("createDate", .datetime)
  51. .field("createUserId", .int)
  52. .field("updateDate", .datetime)
  53. .field("updateUserId", .int)
  54. .create()
  55. }
  56. func revert(on database: Database) async throws {
  57. try await database.schema("associate").delete()
  58. }
  59. }
  60. struct Mod1: AsyncMigration{
  61. func prepare(on database: Database) async throws {
  62. try await database.schema("associate")
  63. .field("userName", .string).unique(on: "userName")
  64. .field("password", .string)
  65. .update()
  66. }
  67. func revert(on database: Database) async throws {
  68. try await database.schema("associate")
  69. .deleteField("userName")
  70. .deleteField("password")
  71. .update()
  72. }
  73. }
  74. public static func ObjQuery(req: Request) async throws -> QueryBuilder<mdlAssociate>{
  75. let query = mdlAssociate.query(on: req.db)
  76. .with(\.$departments){
  77. dept in dept.with(\.$documents){
  78. doc in doc
  79. .with(\.$doctype)
  80. .with(\.$parent)
  81. .with(\.$subsequent)
  82. }
  83. }
  84. if let x = req.parameters.get("x"){
  85. let decryptedString = try CustomCrypto.DecryptString(input: x)
  86. let array = decryptedString.components(separatedBy: ":")
  87. ///This call is the standard get by id call
  88. if (array.count == 1){
  89. let id = Int(array[0])!
  90. return query.filter(\.$id == id)
  91. }
  92. ///Optional else if if there are multiple parameters in the call
  93. ///You must include an else for each call that has parameters
  94. ///Need to check the endpoint to get the right filter statements
  95. ///else if(x.count == 2){
  96. /// let username = x[0]
  97. /// let password = x[1]
  98. /// return query.filter(\.$username == username).filter(\.$password == password)
  99. ///}
  100. else{
  101. throw Abort(.badRequest)
  102. }
  103. }
  104. else{
  105. return query
  106. }
  107. }
  108. }
  109. extension mdlAssociate{
  110. struct CreateUser: Content{
  111. var username: String
  112. var password: String
  113. var confirmpassword: String
  114. var firstname: String
  115. var lastname: String
  116. }
  117. struct Output: Content{
  118. var id: Int?
  119. var firstname: String
  120. var lastname: String
  121. }
  122. }
  123. extension mdlAssociate.CreateUser: Validatable{
  124. static func validations(_ validations: inout Validations){
  125. validations.add("firstname", as: String.self, is: !.empty)
  126. validations.add("lastname", as: String.self, is: !.empty)
  127. validations.add("username", as: String.self, is: !.empty)
  128. validations.add("password", as: String.self, is: .count(8...))
  129. }
  130. }
  131. extension mdlAssociate: ModelAuthenticatable{
  132. static let usernameKey = \mdlAssociate.$username
  133. static var passwordHashKey = \mdlAssociate.$password
  134. func verify(password: String) throws -> Bool {
  135. try Bcrypt.verify(password, created: self.password)
  136. }
  137. }
  138. extension mdlAssociate{
  139. func generateToken() async throws -> mdlToken {
  140. try .init(
  141. value: [UInt8].random(count: 16).base64,
  142. userid: self.requireID()
  143. )
  144. }
  145. }