Impressia/Vernissage/EnvironmentObjects/Client+Statuses.swift

//
//  https://mczachurski.dev
//  Copyright © 2023 Marcin Czachurski and the repository contributors.
//  Licensed under the MIT License.
//

import Foundation
import PixelfedKit

/// Pixelfed 'Statuses'.
extension Client {
    public class Statuses: BaseClient {
        
        public func status(withId statusId: String) async throws -> Status {
            return try await pixelfedClient.status(statusId: statusId)
        }
        
        func favourite(statusId: String) async throws -> Status? {
            return try await pixelfedClient.favourite(statusId: statusId)
        }
        
        func unfavourite(statusId: String) async throws -> Status? {
            return try await pixelfedClient.unfavourite(statusId: statusId)
        }
        
        func pin(statusId: String) async throws -> Status? {
            return try await pixelfedClient.pin(statusId: statusId)
        }
        
        func unpin(statusId: String) async throws -> Status? {
            return try await pixelfedClient.unpin(statusId: statusId)
        }
        
        func boost(statusId: String) async throws -> Status? {
            return try await pixelfedClient.boost(statusId: statusId)
        }
        
        func unboost(statusId: String) async throws -> Status? {
            return try await pixelfedClient.unboost(statusId: statusId)
        }
        
        func bookmark(statusId: String) async throws -> Status? {
            return try await pixelfedClient.bookmark(statusId: statusId)
        }
        
        func unbookmark(statusId: String) async throws -> Status? {
            return try await pixelfedClient.unbookmark(statusId: statusId)
        }
        
        func new(status: Pixelfed.Statuses.Components) async throws -> Status? {
            return try await pixelfedClient.new(statusComponents: status)
        }
        
        func delete(statusId: String) async throws {
            try await pixelfedClient.delete(statusId: statusId)
        }
        
        func comments(to statusId: String) async throws -> [CommentModel] {
            var commentViewModels: [CommentModel] = []
            
            try await self.getCommentDescendants(to: statusId, showDivider: true, to: &commentViewModels)
            
            return commentViewModels
        }
        
        public func favouritedBy(statusId: String, limit: Int, page: Int) async throws -> [Account] {
            return try await pixelfedClient.favouritedBy(for: statusId, limit: limit, page: page)
        }
        
        public func rebloggedBy(statusId: String, limit: Int, page: Int) async throws -> [Account] {
            return try await pixelfedClient.rebloggedBy(for: statusId, limit: limit, page: page)
        }
        
        private func getCommentDescendants(to statusId: String, showDivider: Bool, to commentViewModels: inout [CommentModel]) async throws {
            let context = try await pixelfedClient.getContext(for: statusId)
            
            let descendants = context.descendants.toStatusModels()
            for status in descendants {
                commentViewModels.append(CommentModel(status: status, showDivider: showDivider))
                
                if status.repliesCount > 0 {
                    try await self.getCommentDescendants(to: status.id, showDivider: false, to: &commentViewModels)
                }
            }
        }
    }
}