Update porfolio (#14)

Author: adborbas

Sources/Grodt/Application/routes.swift

@@ -84,6 +84,11 @@ func routes(_ app: Application) async throws {
             .daily()
             .at(3, 0)
         app.queues.add(LoggingJobEventDelegate(logger: app.logger))
+        
+        let userTokenCleanerJob = UserTokenClearUpJob(userTokenClearing: UserTokenClearer(database: app.db))
+        app.queues.schedule(userTokenCleanerJob)
+            .hourly()
+        
         try app.queues.startScheduledJobs()
         try app.queues.startInProcessJobs()
     }

Sources/Grodt/BusinessLogic/UserTokenClearUp/UserTokenClearUpJob.swift

@@ -0,0 +1,14 @@
+import Vapor
+import Queues
+
+struct UserTokenClearUpJob: AsyncScheduledJob, @unchecked Sendable {
+    private let userTokenClearer: UserTokenClearing
+    
+    init(userTokenClearing: UserTokenClearing) {
+        self.userTokenClearer = userTokenClearing
+    }
+    
+    func run(context: Queues.QueueContext) async throws {
+        try await userTokenClearer.clearExpiredTokens()
+    }
+}

Sources/Grodt/BusinessLogic/UserTokenClearUp/UserTokenClearing.swift

@@ -0,0 +1,22 @@
+import Fluent
+import Foundation
+
+protocol UserTokenClearing {
+    func clearExpiredTokens() async throws
+}
+
+class UserTokenClearer: UserTokenClearing {
+    private let database: Database
+    
+    init(database: Database) {
+        self.database = database
+    }
+    
+    func clearExpiredTokens() async throws {
+        let allTokens = try await UserToken.query(on: database).all()
+        let expiredTokens = allTokens.filter { !$0.isValid }
+        for token in expiredTokens {
+            try await token.delete(on: database)
+        }
+    }
+}

Sources/Grodt/Controllers/PortfoliosController.swift

@@ -26,6 +26,7 @@ struct PortfoliosController: RouteCollection {
 
         portfolios.group(":id") { portfolio in
             portfolio.get(use: portfolioDetail)
+            portfolio.put(use: update)
             portfolio.delete(use: delete)
 
             portfolio.group("historicalPerformance") { pref in
@@ -77,6 +78,31 @@ struct PortfoliosController: RouteCollection {
         return try await dataMapper.portfolio(from: portfolio)
     }
 
+    func update(req: Request) async throws -> PortfolioDTO {
+            let id = try req.requiredID()
+            guard let userID = req.auth.get(User.self)?.id else {
+                throw Abort(.badRequest)
+            }
+            
+            let updateDTO = try req.content.decode(UpdatePortfolioRequestDTO.self)
+            
+            guard var portfolio = try await portfolioRepository.portfolio(for: userID, with: id) else {
+                throw Abort(.notFound)
+            }
+            
+            portfolio.name = updateDTO.name
+            
+            if portfolio.currency.code != updateDTO.currency {
+                guard let newCurrency = try await currencyRepository.currency(for: updateDTO.currency) else {
+                    throw Abort(.badRequest)
+                }
+                portfolio.currency = newCurrency
+            }
+            
+            let updatedPortfolio = try await portfolioRepository.update(portfolio)
+            return try await dataMapper.portfolio(from: updatedPortfolio)
+        }
+    
     func delete(req: Request) async throws -> HTTPStatus {
         let id = try req.requiredID()
 

Sources/Grodt/DTOs/UpdatePortfolioRequestDTO.swift

@@ -0,0 +1,6 @@
+import Foundation
+
+struct UpdatePortfolioRequestDTO: Codable {
+    let name: String
+    let currency: String
+}

Sources/Grodt/Persistency/Repositories/PortfolioRepository.swift

@@ -5,7 +5,9 @@ protocol PortfolioRepository {
     func allPortfolios(for userID: User.IDValue) async throws -> [Portfolio]
     func portfolio(for userID: User.IDValue, with id: Portfolio.IDValue) async throws -> Portfolio?
     func create(_ portfolio: Portfolio) async throws -> Portfolio
+    func update(_ portfolio: Portfolio) async throws -> Portfolio
     func delete(for userID: User.IDValue, with id: Portfolio.IDValue) async throws
+    
     func historicalPerformance(with id: Portfolio.IDValue) async throws -> HistoricalPortfolioPerformance
     func updateHistoricalPerformance(_ historicalPerformance: HistoricalPortfolioPerformance) async throws
     func createHistoricalPerformance(_ historicalPerformance: HistoricalPortfolioPerformance) async throws
@@ -54,6 +56,20 @@ class PostgresPortfolioRepository: PortfolioRepository {
         return portfolioWithTransactions
     }
 
+    func update(_ portfolio: Portfolio) async throws -> Portfolio {
+        try await portfolio.save(on: database)
+        
+        guard let updatedPortfolio = try await Portfolio.query(on: database)
+            .filter(\Portfolio.$id == portfolio.id!)
+            .with(\.$transactions)
+            .with(\.$historicalPerformance)
+            .first() else {
+            throw FluentError.noResults
+        }
+        
+        return updatedPortfolio
+    }
+    
     func delete(for userID: User.IDValue, with id: Portfolio.IDValue) async throws {
         guard let portfolio = try await self.portfolio(for: userID, with: id) else {
             throw FluentError.noResults