element-x-ios/ElementX/Sources/Other/SwiftUI/Views/ToolbarButton.swift

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

import Compound
import SwiftUI

struct ToolbarButton: View {
    enum Role {
        static let cancel = Role.cancel(title: L10n.actionCancel)
        static let done = Role.confirm(title: L10n.actionDone)
        static let save = Role.confirm(title: L10n.actionSave)
        static let close = Role.cancel(title: L10n.actionClose)
        
        case cancel(title: String)
        case confirm(title: String)
        case destructive(title: String)
        
        var title: String {
            switch self {
            case .cancel(let title), .confirm(let title), .destructive(let title):
                title
            }
        }
        
        @ViewBuilder
        var icon: some View {
            switch self {
            case .cancel:
                CompoundIcon(\.close)
                    .foregroundStyle(.compound.iconPrimary)
            case .confirm:
                CompoundIcon(\.check)
                    .foregroundStyle(.compound.iconOnSolidPrimary)
            case .destructive:
                CompoundIcon(\.delete)
                    .foregroundStyle(.compound.iconOnSolidPrimary)
            }
        }
        
        var tint: Color? {
            switch self {
            case .cancel:
                nil
            case .confirm:
                .compound.bgAccentRest
            case .destructive:
                .compound.bgCriticalPrimary
            }
        }
    }
    
    let role: Role
    let action: () -> Void
    
    var body: some View {
        if #available(iOS 26, *) {
            if let tint = role.tint {
                glassButton
                    .tint(tint)
                    .backportButtonStyleGlassProminent()
            } else {
                glassButton
            }
        } else {
            flatButton
        }
    }
    
    private var glassButton: some View {
        Button(action: action) {
            role.icon
                .accessibilityLabel(role.title)
        }
    }
    
    private var flatButton: some View {
        Button(role.title, action: action)
    }
}

struct ToolbarButton_Previews: PreviewProvider, TestablePreview {
    static var previews: some View {
        ElementNavigationStack {
            Color.clear
                .toolbar {
                    ToolbarItem(placement: .confirmationAction) {
                        ToolbarButton(role: .done) { }
                    }
                    ToolbarItem(placement: .cancellationAction) {
                        ToolbarButton(role: .cancel) { }
                    }
                    ToolbarItem(placement: .primaryAction) {
                        ToolbarButton(role: .destructive(title: L10n.actionRemove)) { }
                    }
                }
        }
    }
}