AddSiteController.swift

import UIKit
import SwiftUI
import WordPressAuthenticator

/// Manages the site creation flows.
struct AddSiteController {
    let viewController: UIViewController
    let source: String

    func showSiteCreationScreen(selection: AddSiteMenuViewModel.Selection) {
        switch selection {
        case .dotCom: showDotComSiteCreationScreen()
        case .selfHosted: showSelfHostedSiteLoginScreen()
        }
    }

    func showDotComSiteCreationScreen() {
        if let account = try? WPAccount.lookupDefaultWordPressComAccount(in: ContextManager.shared.mainContext),
           account.needsEmailVerification {
            VerifyEmailModal.present(on: viewController)
            return
        }

        JetpackFeaturesRemovalCoordinator.presentSiteCreationOverlayIfNeeded(in: viewController, source: source, onDidDismiss: { [weak viewController] in
            guard JetpackFeaturesRemovalCoordinator.siteCreationPhase() != .two else {
                return
            }
            // Display site creation flow if not in phase two
            let wizardLauncher = SiteCreationWizardLauncher()
            guard let wizard = wizardLauncher.ui else {
                return
            }
            RootViewCoordinator.shared.isSiteCreationActive = true
            viewController?.present(wizard, animated: true)
            SiteCreationAnalyticsHelper.trackSiteCreationAccessed(source: source)
        })
    }

    func showSelfHostedSiteLoginScreen() {
        guard FeatureFlag.authenticateUsingApplicationPassword.enabled else {
            WordPressAuthenticator.showLoginForSelfHostedSite(viewController)
            return
        }
        showApplicationPasswordAuthenticationForSelfHostedSite()
    }

    private func showApplicationPasswordAuthenticationForSelfHostedSite() {
        guard let window = viewController.view.window else {
            return wpAssertionFailure("window missing")
        }
        let client = SelfHostedSiteAuthenticator(session: URLSession(configuration: .ephemeral))
        let view = LoginWithUrlView(client: client, anchor: window) { [weak viewController] credentials in
            viewController?.dismiss(animated: true)
            WordPressAuthenticator.shared.delegate!.sync(credentials: .init(wporg: credentials)) {
                NotificationCenter.default.post(name: Foundation.Notification.Name(rawValue: WordPressAuthenticator.WPSigninDidFinishNotification), object: nil)
            }
        }.toolbar {
            ToolbarItem(placement: .cancellationAction) {
                Button(SharedStrings.Button.cancel) { [weak viewController] in
                    viewController?.dismiss(animated: true)
                }
            }
        }
        let hostVC = UIHostingController(rootView: view)
        let navigationVC = UINavigationController(rootViewController: hostVC)
        navigationVC.modalPresentationStyle = .formSheet
        viewController.present(navigationVC, animated: true)
    }
}

struct AddSiteMenuViewModel {
    let actions: [Action]

    enum Selection: String {
        case dotCom
        case selfHosted
    }

    struct Action: Identifiable {
        let id = UUID()
        let title: String
        let handler: () -> Void

        var uiAction: UIAction {
            UIAction(title: title, handler: { _ in handler() })
        }
    }

    init(context: ContextManager = .shared, onSelection: @escaping (Selection) -> Void) {
        let defaultAccount = try? WPAccount.lookupDefaultWordPressComAccount(in: context.mainContext)
        let canAddSelfHostedSite = FeatureFlag.selfHostedSites.enabled

        var actions: [Action] = []
        if defaultAccount != nil {
            actions.append(Action(title: Strings.createDotComSite) {
                onSelection(.dotCom)
            })
        }

        if canAddSelfHostedSite {
            actions.append(Action(title: Strings.addSelfHostedSite) {
                onSelection(.selfHosted)
            })
        }
        self.actions = actions
    }
}

extension AddSiteMenuViewModel {
    func makeBarButtonItem() -> UIBarButtonItem? {
        let actions = self.actions
        guard !actions.isEmpty else {
            return nil
        }
        let item = UIBarButtonItem(systemItem: .add)
        item.setAdaptiveActions(actions.map(\.uiAction))
        item.accessibilityIdentifier = "add-site-button"
        return item
    }
}

private enum Strings {
    static let createDotComSite = NSLocalizedString("button.createDotCoSite", value: "Create WordPress.com site", comment: "Create WordPress.com site button")
    static let addSelfHostedSite = NSLocalizedString("button.addSelfHostedSite", value: "Add self-hosted site", comment: "Add self-hosted site button")
}