yattee/tvOS/AccountSelectionView.swift

import Defaults
import Foundation
import SwiftUI

struct AccountSelectionView: View {
    var showHeader = true

    @EnvironmentObject<InstancesModel> private var instancesModel
    @EnvironmentObject<AccountsModel> private var accountsModel

    @Default(.accounts) private var accounts
    @Default(.instances) private var instances

    var body: some View {
        Section(header: Text(showHeader ? "Current Account" : "")) {
            Button(accountButtonTitle(account: accountsModel.current)) {
                if let account = nextAccount {
                    accountsModel.setCurrent(account)
                }
            }
            .disabled(instances.isEmpty)
            .contextMenu {
                ForEach(allAccounts) { account in
                    Button(accountButtonTitle(account: account)) {
                        accountsModel.setCurrent(account)
                    }
                }

                Button("Cancel", role: .cancel) {}
            }
        }
        .id(UUID())
    }

    var allAccounts: [Instance.Account] {
        accounts + instances.map(\.anonymousAccount)
    }

    private var nextAccount: Instance.Account? {
        allAccounts.next(after: accountsModel.current)
    }

    func accountButtonTitle(account: Instance.Account! = nil) -> String {
        guard account != nil else {
            return "Not selected"
        }

        return instances.count > 1 ? "\(account.description) — \(account.instance.shortDescription)" : account.description
    }
}