letro-ios/ElementX/Sources/Screens/ManageRoomMemberSheet/ManageRoomMemberSheetViewModel.swift

//
// Copyright 2025 New Vector Ltd.
//
// SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-Element-Commercial
// Please see LICENSE files in the repository root for full details.
//

import Combine
import Foundation
import SwiftUI

typealias ManageRoomMemberSheetViewModelType = StateStoreViewModelV2<ManageRoomMemberSheetViewState, ManageRoomMemberSheetViewAction>

class ManageRoomMemberSheetViewModel: ManageRoomMemberSheetViewModelType, ManageRoomMemberSheetViewModelProtocol {
    private let roomProxy: JoinedRoomProxyProtocol
    private let userIndicatorController: UserIndicatorControllerProtocol
    private let analyticsService: AnalyticsService
    
    private var actionsSubject: PassthroughSubject<ManageRoomMemberSheetViewModelAction, Never> = .init()
    
    var actions: AnyPublisher<ManageRoomMemberSheetViewModelAction, Never> {
        actionsSubject.eraseToAnyPublisher()
    }
    
    init(memberDetails: ManageRoomMemberDetails,
         permissions: ManageRoomMemberPermissions,
         roomProxy: JoinedRoomProxyProtocol,
         userIndicatorController: UserIndicatorControllerProtocol,
         analyticsService: AnalyticsService,
         mediaProvider: MediaProviderProtocol) {
        self.userIndicatorController = userIndicatorController
        self.roomProxy = roomProxy
        self.analyticsService = analyticsService
        super.init(initialViewState: .init(memberDetails: memberDetails, permissions: permissions), mediaProvider: mediaProvider)
    }
    
    override func process(viewAction: ManageRoomMemberSheetViewAction) {
        switch viewAction {
        case .kick:
            displayAlert(.kick)
        case .ban:
            displayAlert(.ban)
        case .displayDetails:
            actionsSubject.send(.dismiss(shouldShowDetails: true))
        case .unban:
            displayAlert(.unban)
        }
    }
    
    private func displayAlert(_ alertType: ManageRoomMemberSheetViewAlertType) {
        let memberID = state.memberDetails.id
        let memberName = state.memberDetails.name
        
        var reason: String?
        let binding: Binding<String> = .init(get: { reason ?? "" },
                                             set: { reason = $0.isBlank ? nil : $0 })
        switch alertType {
        case .kick:
            state.bindings.alertInfo = .init(id: alertType,
                                             title: L10n.screenBottomSheetManageRoomMemberKickMemberConfirmationTitle,
                                             message: L10n.screenBottomSheetManageRoomMemberKickMemberConfirmationDescription,
                                             primaryButton: .init(title: L10n.actionCancel, role: .cancel) { },
                                             secondaryButton: .init(title: L10n.screenBottomSheetManageRoomMemberKickMemberConfirmationAction) { [weak self] in Task { await self?.kickMember(id: memberID, name: memberName, reason: reason) } },
                                             textFields: [.init(placeholder: L10n.commonReason,
                                                                text: binding,
                                                                autoCapitalization: .sentences,
                                                                autoCorrectionDisabled: false)])
        case .ban:
            state.bindings.alertInfo = .init(id: alertType,
                                             title: L10n.screenBottomSheetManageRoomMemberBanMemberConfirmationTitle,
                                             message: L10n.screenBottomSheetManageRoomMemberBanMemberConfirmationDescription,
                                             primaryButton: .init(title: L10n.actionCancel, role: .cancel) { },
                                             secondaryButton: .init(title: L10n.screenBottomSheetManageRoomMemberBanMemberConfirmationAction) { [weak self] in Task { await self?.banMember(id: memberID, name: memberName, reason: reason) } },
                                             textFields: [.init(placeholder: L10n.commonReason,
                                                                text: binding,
                                                                autoCapitalization: .sentences,
                                                                autoCorrectionDisabled: false)])
        case .unban:
            state.bindings.alertInfo = .init(id: alertType,
                                             title: L10n.screenBottomSheetManageRoomMemberUnbanMemberConfirmationTitle,
                                             message: L10n.screenBottomSheetManageRoomMemberUnbanMemberConfirmationDescription,
                                             primaryButton: .init(title: L10n.actionCancel, role: .cancel) { },
                                             secondaryButton: .init(title: L10n.screenBottomSheetManageRoomMemberUnbanMemberConfirmationAction) { [weak self] in Task { await self?.unbanMember(id: memberID, name: memberName) } })
        }
    }
    
    private func kickMember(id: String, name: String?, reason: String?) async {
        let indicatorTitle = L10n.screenBottomSheetManageRoomMemberRemovingUser(name ?? id)
        showManageMemberIndicator(title: indicatorTitle)
        
        switch await roomProxy.kickUser(id, reason: reason) {
        case .success:
            hideManageMemberIndicator(title: indicatorTitle)
            analyticsService.trackRoomModeration(action: .KickMember, role: nil)
            actionsSubject.send(.dismiss(shouldShowDetails: false))
        case .failure:
            showManageMemberFailure(title: indicatorTitle)
        }
    }
    
    private func banMember(id: String, name: String?, reason: String?) async {
        let indicatorTitle = L10n.screenBottomSheetManageRoomMemberBanningUser(name ?? id)
        showManageMemberIndicator(title: indicatorTitle)
        
        switch await roomProxy.banUser(id, reason: reason) {
        case .success:
            hideManageMemberIndicator(title: indicatorTitle)
            analyticsService.trackRoomModeration(action: .BanMember, role: nil)
            actionsSubject.send(.dismiss(shouldShowDetails: false))
        case .failure:
            showManageMemberFailure(title: indicatorTitle)
        }
    }
    
    private func unbanMember(id: String, name: String?) async {
        let indicatorTitle = L10n.screenBottomSheetManageRoomMemberUnbanningUser(name ?? id)
        showManageMemberIndicator(title: indicatorTitle)
        
        switch await roomProxy.unbanUser(id) {
        case .success:
            hideManageMemberIndicator(title: indicatorTitle)
            analyticsService.trackRoomModeration(action: .UnbanMember, role: nil)
            actionsSubject.send(.dismiss(shouldShowDetails: false))
        case .failure:
            showManageMemberFailure(title: indicatorTitle)
        }
    }
    
    private func showManageMemberIndicator(title: String) {
        userIndicatorController.submitIndicator(UserIndicator(id: title,
                                                              type: .toast(progress: .indeterminate),
                                                              title: title,
                                                              persistent: true))
    }
    
    private func hideManageMemberIndicator(title: String) {
        userIndicatorController.retractIndicatorWithId(title)
    }
    
    private func showManageMemberFailure(title: String) {
        userIndicatorController.retractIndicatorWithId(title)
        userIndicatorController.submitIndicator(UserIndicator(title: L10n.commonFailed, iconName: "xmark"))
    }
}

extension ManageRoomMemberSheetViewModel: Identifiable {
    var id: String {
        state.memberDetails.id
    }
}