simplex-chat/apps/ios/Shared/Views/Badges/BadgesRedeemCodeView.swift

//
//  BadgesRedeemCodeView.swift
//  SimpleX (iOS)
//
//  Created by spaced4ndy on 28.07.2026.
//  Copyright © 2026 SimpleX Chat. All rights reserved.
//

import SwiftUI
import SimpleXChat
import CodeScanner

private let badgeCodePrefix = "SB"
private let badgeCodeBodyLength = 20
private let badgeCodeGroupLength = 5

// Regroups what was typed; validity and the folding of ambiguous characters are core's alone.
private func formatBadgeCodeInput(_ s: String) -> String {
    var normalized = ""
    for c in s.prefix(256).uppercased() {
        guard c.isLetter || c.isNumber else { continue }
        normalized.append(c)
        if normalized.count == badgeCodePrefix.count + badgeCodeBodyLength { break }
    }
    guard normalized.hasPrefix(badgeCodePrefix) else { return normalized }
    var groups = [badgeCodePrefix]
    var i = normalized.index(normalized.startIndex, offsetBy: badgeCodePrefix.count)
    while i < normalized.endIndex {
        let j = normalized.index(i, offsetBy: badgeCodeGroupLength, limitedBy: normalized.endIndex) ?? normalized.endIndex
        groups.append(String(normalized[i..<j]))
        i = j
    }
    return groups.joined(separator: "-")
}

enum BadgeRedeemOutcome {
    case redeemed
    case refused(message: String)
    case cancelled
}

// sets the badge before returning, so a caller that dismisses on .redeemed lands on Your Badge
// instead of showing the switch from Support. .cancelled when the user cancels the retry alert.
func redeemBadgeCode(_ user: User, _ code: String) async -> BadgeRedeemOutcome {
    do {
        guard let redeemed = try await apiRedeemBadgeCode(user.userId, code) else { return .cancelled }
        return await MainActor.run { () -> BadgeRedeemOutcome in
            BadgeModel.shared.set(userId: user.userId, badgeState: redeemed.badgeState)
            ChatModel.shared.updateUser(redeemed.user)
            if let badgeState = redeemed.badgeState, !badgeState.shown {
                // a replay adds no purchase; a fresh code's badge can be retired on arrival
                return .refused(message: redeemed.newBadge
                    ? NSLocalizedString("The code was accepted, but the badge it grants has already ended.", comment: "alert message")
                    : NSLocalizedString("This code has already been used.", comment: "alert message")
                )
            }
            UserDefaults.standard.set(true, forKey: DEFAULT_SUPPORTER_BANNER_SHOWN)
            return .redeemed
        }
    } catch let error {
        logger.error("apiRedeemBadgeCode: \(responseError(error))")
        return .refused(message: redeemErrorText(error, purchase: false))
    }
}

func showCannotRedeemAlert(_ message: String) {
    showAlert(NSLocalizedString("Cannot redeem code", comment: "alert title"), message: message)
}

struct BadgesRedeemCodeView: View {
    @EnvironmentObject var theme: AppTheme
    @EnvironmentObject var chatModel: ChatModel
    @Environment(\.dismiss) var dismiss: DismissAction
    @State private var code = ""
    @State private var canonicalCode: String? = nil
    @State private var submitting = false
    @State private var showQRCodeScanner = true

    var body: some View {
        ZStack {
            GeometryReader { g in
                ScrollView {
                    VStack(alignment: .center, spacing: 16) {
                        entryContent(g)
                    }
                    .padding(.horizontal, 25)
                    .padding(.top, 8)
                    .padding(.bottom, 20)
                    .frame(minHeight: g.size.height)
                }
            }

            if submitting {
                ZStack {
                    Circle()
                        .fill(.white)
                        .opacity(0.7)
                        .frame(width: 56, height: 56)
                    ProgressView().scaleEffect(2)
                }
            }
        }
        .frame(maxHeight: .infinity)
        .navigationBarTitleDisplayMode(.inline)
    }

    @ViewBuilder
    private func entryContent(_ g: GeometryProxy) -> some View {
        Text("Redeem code")
            .font(.largeTitle)
            .bold()
            .foregroundColor(theme.colors.primary)
            .multilineTextAlignment(.center)
            .fixedSize(horizontal: false, vertical: true)

        Text("Paste the code you received.")
            .font(.body)
            .multilineTextAlignment(.center)
            .fixedSize(horizontal: false, vertical: true)

        codeField()

        pasteButton()

        ScannerInView(
            showQRCodeScanner: $showQRCodeScanner,
            scannerPaused: $submitting,
            processQRCode: processQRCode,
            scanMode: .oncePerCode,
            placeholderBackground: Color(.tertiarySystemFill)
        )
        .fixedSize(horizontal: false, vertical: true)
        .padding(.top, 12)

        Spacer(minLength: 0)

        VStack(spacing: 10) {
            submitButton()
                .padding(.vertical, 10)
            Color.clear
                .frame(height: 22)
        }
        .padding(.bottom, g.safeAreaInsets.bottom == 0 ? 20 : 0)
    }

    private func codeField() -> some View {
        TextField(text: $code) { Text(verbatim: "SB-XXXXX-XXXXX-XXXXX-XXXXX") }
            .font(.body.monospaced())
            .multilineTextAlignment(.center)
            .autocorrectionDisabled(true)
            .textInputAutocapitalization(.characters)
            .disabled(submitting)
            .padding(EdgeInsets(top: 14, leading: 12, bottom: 14, trailing: 12))
            .background(Color(.tertiarySystemFill))
            .cornerRadius(10.0)
            .onChange(of: code) { applyCodeInput($0) }
    }

    private func pasteButton() -> some View {
        Button {
            if let pasted = UIPasteboard.general.string { applyCodeInput(pasted) }
        } label: {
            Text("Paste")
                .font(.body)
                .fontWeight(.medium)
                .foregroundColor(theme.colors.primary)
        }
        .disabled(submitting)
    }

    // a changed field re-enters through onChange with the formatted text, which is when it is parsed
    private func applyCodeInput(_ s: String) {
        let formatted = formatBadgeCodeInput(s)
        if formatted != code {
            code = formatted
        } else {
            canonicalCode = parseBadgeCode(formatted)
        }
    }

    private func processQRCode(_ resp: Result<ScanResult, ScanError>) {
        switch resp {
        case let .success(r):
            let formatted = formatBadgeCodeInput(r.string)
            if let canonical = parseBadgeCode(formatted) {
                // set here rather than through onChange, which runs after redeem() reads it
                code = formatted
                canonicalCode = canonical
                redeem()
            } else {
                showAlert(
                    NSLocalizedString("Invalid QR code", comment: "alert title"),
                    message: NSLocalizedString("The code you scanned is not a badge code.", comment: "alert message")
                )
            }
        case let .failure(e):
            logger.error("processQRCode QR code error: \(e.localizedDescription)")
            showAlert(
                NSLocalizedString("Invalid QR code", comment: "alert title"),
                message: String.localizedStringWithFormat(NSLocalizedString("Error scanning code: %@", comment: "alert message"), e.localizedDescription)
            )
        }
    }

    private func submitButton() -> some View {
        let disabled = canonicalCode == nil || submitting
        return Button {
            redeem()
        } label: {
            Text("Redeem")
        }
        .buttonStyle(OnboardingButtonStyle(isDisabled: disabled))
        .disabled(disabled)
    }

    private func redeem() {
        guard let sending = canonicalCode, let user = chatModel.currentUser else { return }
        submitting = true
        Task {
            let outcome = await redeemBadgeCode(user, sending)
            await MainActor.run {
                submitting = false
                switch outcome {
                case .redeemed: dismiss()
                case let .refused(message): showCannotRedeemAlert(message)
                case .cancelled: break
                }
            }
        }
    }
}

private weak var badgeLinkSheet: UIViewController?
// the open screen's own state, not a copy of its step, so the two cannot disagree
private weak var badgeLinkState: BadgeLinkState?

func isBadgeLinkIssuing() -> Bool {
    badgeLinkSheet?.presentingViewController != nil && badgeLinkState?.step == .issuing
}

func openBadgeLink(_ codeText: String) {
    guard let code = parseBadgeCode(codeText) else {
        return showCannotRedeemAlert(NSLocalizedString("This code is not valid.", comment: "alert message"))
    }
    let state = BadgeLinkState()
    badgeLinkSheet = showAppSheet {
        NavigationView {
            BadgesRedeemLinkView(code: code, state: state)
                .modifier(ThemedBackground())
        }
    }
    badgeLinkState = state
}

enum BadgeLinkStep {
    case confirming
    case issuing
    case redeemed
    case viewingBadge
}

final class BadgeLinkState: ObservableObject {
    @Published var step = BadgeLinkStep.confirming
}

// Any web page can send a badge link, and a profile holds one badge at a time,
// so this screen asks before redeeming, names the profile, and offers nothing but the redemption.
// It is a code redemption with the code hidden, so it inherits the redeem screen's behaviour: an
// interrupted request is not resumed, and the code stays on the page that issued the link.
struct BadgesRedeemLinkView: View {
    @EnvironmentObject var theme: AppTheme
    @EnvironmentObject var chatModel: ChatModel
    @ObservedObject private var badgeModel = BadgeModel.shared
    let code: String
    @ObservedObject var state: BadgeLinkState

    var body: some View {
        switch state.step {
        case .confirming: if profileHasBadge { badgeHeld() } else { confirming() }
        case .issuing: beingIssued()
        case .redeemed, .viewingBadge: BadgesView(showsAsSheet: true)
        }
    }

    // read for the profile the screen names, when it renders, so the two cannot disagree;
    // core refuses a code while the profile shows a badge
    private var profileHasBadge: Bool {
        badgeModel.userId == chatModel.currentUser?.userId && badgeModel.badgeState?.shown == true
    }

    // the screen cannot tell a repeated link for the badge it shows from a code for another badge,
    // so it asserts neither
    private func badgeHeld() -> some View {
        linkStep(
            "Profile already has a badge",
            primary: ("View your badge", { state.step = .viewingBadge }),
            textButton: ("Cancel", { closeIfShowing() })
        ) {
            linkText(NSLocalizedString("A badge can be added to another profile, or here once this badge ends.", comment: "badge link, profile has a badge"))
            linkText(NSLocalizedString("Its code is on the page you bought it on, under Show code.", comment: "badge link, profile has a badge"))
        }
    }

    private func confirming() -> some View {
        linkStep(
            "Add badge to your profile?",
            primary: ("Add badge", { redeemFromLink() }),
            textButton: ("Cancel", { closeIfShowing() })
        ) {
            linkText(String.localizedStringWithFormat(NSLocalizedString("The badge will be added to the profile %@.", comment: "badge link confirmation"), chatModel.currentUser?.displayName ?? ""))
        }
    }

    // leaving does not cancel: the badge is still added after the screen closes
    private func beingIssued() -> some View {
        linkStep("Badge is being issued", textButton: ("Dismiss", { closeIfShowing() })) {
            Spacer()
            ProgressView().scaleEffect(2)
        }
    }

    // with no primary, the text button takes its place, so a button does not move between steps
    private func linkStep<Content: View>(
        _ title: LocalizedStringKey,
        primary: (label: LocalizedStringKey, action: () -> Void)? = nil,
        textButton: (label: LocalizedStringKey, action: () -> Void),
        @ViewBuilder content: () -> Content
    ) -> some View {
        let textButtonLabel = Text(textButton.label)
            .font(.body)
            .fontWeight(.medium)
            .foregroundColor(theme.colors.primary)
        return VStack(alignment: .center, spacing: 16) {
            Text(title)
                .font(.largeTitle)
                .bold()
                .foregroundColor(theme.colors.primary)
                .multilineTextAlignment(.center)
                .fixedSize(horizontal: false, vertical: true)

            content()

            Spacer()

            VStack(spacing: 10) {
                if let primary {
                    Button {
                        primary.action()
                    } label: {
                        Text(primary.label)
                    }
                    .buttonStyle(OnboardingButtonStyle(isDisabled: false))
                    .padding(.vertical, 10)

                    Button {
                        textButton.action()
                    } label: {
                        textButtonLabel
                    }
                    .frame(height: 22)
                } else {
                    Button {
                        textButton.action()
                    } label: {
                        // the inset OnboardingButtonStyle gives a primary
                        textButtonLabel.padding()
                    }
                    .padding(.vertical, 10)
                    Color.clear
                        .frame(height: 22)
                }
            }
        }
        .padding(.horizontal, 25)
        .padding(.top, 48)
        .padding(.bottom, 20)
        .frame(maxHeight: .infinity)
        .navigationBarTitleDisplayMode(.inline)
    }

    private func linkText(_ text: String) -> some View {
        Text(text)
            .font(.body)
            .multilineTextAlignment(.center)
            .fixedSize(horizontal: false, vertical: true)
    }

    private func redeemFromLink() {
        // a second tap before the screen changes must not send the code again
        guard state.step == .confirming else { return }
        guard let user = chatModel.currentUser else { return closeIfShowing() }
        state.step = .issuing
        Task {
            let outcome = await redeemBadgeCode(user, code)
            await MainActor.run {
                switch outcome {
                case .redeemed:
                    if !isShowing {
                        showAlert(
                            NSLocalizedString("Badge added", comment: "alert title"),
                            message: String.localizedStringWithFormat(NSLocalizedString("The badge was added to the profile %@.", comment: "alert message"), user.displayName)
                        )
                    }
                    // a covered screen is not left on the spinner; a closed one is not written to
                    if isOpen { state.step = .redeemed }
                case let .refused(message): closeIfShowing { showCannotRedeemAlert(message) }
                case .cancelled: closeIfShowing()
                }
            }
        }
    }

    private var isOpen: Bool {
        badgeLinkState === state && badgeLinkSheet?.presentingViewController != nil
    }

    // an alert over this screen still counts: dismissing from the presenter dismisses it too
    private var isShowing: Bool {
        guard isOpen, let sheet = badgeLinkSheet else { return false }
        return sheet.presentedViewController == nil || sheet.presentedViewController is UIAlertController
    }

    // An outcome can arrive after this screen was closed or covered: it must not close another screen,
    // and a covered one goes back to asking, so it is never left locked on the spinner.
    private func closeIfShowing(then: @escaping () -> Void = {}) {
        if isShowing, let presenter = badgeLinkSheet?.presentingViewController {
            presenter.dismiss(animated: true, completion: then)
        } else {
            if isOpen { state.step = .confirming }
            then()
        }
    }
}

struct BadgesRedeemCodeView_Previews: PreviewProvider {
    static var previews: some View {
        NavigationView {
            BadgesRedeemCodeView()
        }
    }
}