sing-box-for-apple/ApplicationLibrary/Views/Profile/NewProfileMenuView.swift

import Foundation
import Libbox
import Library
import SwiftUI
import UniformTypeIdentifiers

@MainActor
public struct NewProfileMenuView: View {
    @EnvironmentObject private var environments: ExtensionEnvironments
    @Environment(\.dismiss) private var dismiss
    @State private var alert: AlertState?
    @State private var importRequest: NewProfileView.ImportRequest?
    @State private var localImportRequest: NewProfileView.LocalImportRequest?
    #if os(tvOS)
        @State private var importCompleted = false
    #else
        @State private var showFileImporter = false
        @State private var showQRScanner = false
    #endif
    #if os(macOS)
        @State private var showNewProfile = false
    #endif

    public init() {}

    public var body: some View {
        #if os(macOS)
            macOSBody
        #else
            otherBody
        #endif
    }

    #if os(macOS)
        private var macOSBody: some View {
            VStack(alignment: .leading, spacing: 0) {
                Text("New Profile")
                    .font(.headline)
                    .padding(.horizontal, 20)
                    .padding(.top, 20)
                    .padding(.bottom, 12)

                menuContent
            }
            .safeAreaInset(edge: .bottom) {
                VStack(spacing: 0) {
                    Divider()
                    HStack {
                        Spacer()
                        Button("Cancel") {
                            dismiss()
                        }
                    }
                    .padding()
                    .background(Color(NSColor.controlBackgroundColor))
                }
            }
            .alert($alert)
            .fileImporter(
                isPresented: $showFileImporter,
                allowedContentTypes: [.profile, .json],
                allowsMultipleSelection: false
            ) { result in
                handleFileImport(result)
            }
            .sheet(isPresented: $showNewProfile) {
                NewProfileView(onSuccess: { profile in
                    await SharedPreferences.selectedProfileID.set(profile.mustID)
                    dismiss()
                })
                .environmentObject(environments)
            }
            .sheet(item: $localImportRequest) { request in
                NewProfileView(localImportRequest: request, onSuccess: { profile in
                    await SharedPreferences.selectedProfileID.set(profile.mustID)
                    dismiss()
                })
                .environmentObject(environments)
            }
            .sheet(isPresented: $showQRScanner) {
                QRScannerView { result in
                    handleQRScanResult(result)
                }
                .frame(minWidth: 500, minHeight: 400)
            }
            .sheet(item: $importRequest) { request in
                NewProfileView(request, onSuccess: { profile in
                    await SharedPreferences.selectedProfileID.set(profile.mustID)
                    dismiss()
                })
                .environmentObject(environments)
            }
        }
    #endif

    private var otherBody: some View {
        Group {
            if let request = importRequest {
                NewProfileView(request, onSuccess: { profile in
                    await SharedPreferences.selectedProfileID.set(profile.mustID)
                    dismiss()
                })
                .environmentObject(environments)
            } else if let request = localImportRequest {
                NewProfileView(localImportRequest: request, onSuccess: { profile in
                    await SharedPreferences.selectedProfileID.set(profile.mustID)
                    dismiss()
                })
                .environmentObject(environments)
            } else {
                menuContent
            }
        }
        .navigationTitle("New Profile")
        #if os(iOS)
            .navigationBarTitleDisplayMode(.inline)
        #endif
            .alert($alert)
        #if os(tvOS)
            .onChange(of: importCompleted) { newValue in
                if newValue {
                    dismiss()
                }
            }
        #else
            .fileImporter(
                    isPresented: $showFileImporter,
                    allowedContentTypes: [.profile, .json],
                    allowsMultipleSelection: false
                ) { result in
                    handleFileImport(result)
                }
        #endif
        #if !os(tvOS)
        .sheet(isPresented: $showQRScanner) {
            QRScannerView { result in
                handleQRScanResult(result)
            }
        }
        #endif
    }

    private var menuContent: some View {
        FormView {
            Section {
                #if os(tvOS)
                    FormNavigationLink {
                        ImportProfileView(onComplete: {
                            importCompleted = true
                        })
                        .environmentObject(environments)
                    } label: {
                        Label("Import from iPhone or iPad", systemImage: "iphone.and.arrow.forward")
                    }
                #endif

                #if !os(tvOS)
                    FormButton {
                        showFileImporter = true
                    } label: {
                        Label("Import from File", systemImage: "doc.badge.plus")
                    }
                #endif

                #if !os(tvOS)
                    FormButton {
                        showQRScanner = true
                    } label: {
                        Label("Scan QR Code", systemImage: "qrcode.viewfinder")
                    }
                #endif

                #if os(macOS)
                    FormButton {
                        showNewProfile = true
                    } label: {
                        Label("Create Manually", systemImage: "square.and.pencil")
                    }
                #else
                    FormNavigationLink {
                        NewProfileView(onSuccess: { profile in
                            await SharedPreferences.selectedProfileID.set(profile.mustID)
                            dismiss()
                        })
                        .environmentObject(environments)
                    } label: {
                        Label("Create Manually", systemImage: "square.and.pencil")
                    }
                #endif
            }
        }
    }

    #if !os(tvOS)
        private func handleFileImport(_ result: Result<[URL], Error>) {
            do {
                let urls = try result.get()
                guard let url = urls.first else { return }

                if url.pathExtension.lowercased() == "json" {
                    let fileName = url.deletingPathExtension().lastPathComponent
                    localImportRequest = NewProfileView.LocalImportRequest(name: fileName, fileURL: url)
                } else {
                    _ = url.startAccessingSecurityScopedResource()
                    defer { url.stopAccessingSecurityScopedResource() }

                    let content = try LibboxProfileContent.from(Data(contentsOf: url))

                    alert = AlertState(
                        title: String(localized: "Import Profile"),
                        message: String(localized: "Are you sure to import profile \(content.name)?"),
                        primaryButton: .default(String(localized: "Import")) {
                            Task {
                                do {
                                    try await content.importProfile()
                                    environments.profileUpdate.send()
                                    dismiss()
                                } catch {
                                    alert = AlertState(error: error)
                                }
                            }
                        },
                        secondaryButton: .cancel()
                    )
                }
            } catch {
                alert = AlertState(error: error)
            }
        }
    #endif

    #if !os(tvOS)
        private func handleQRScanResult(_ result: QRScanResult) {
            switch result {
            case let .qrCode(string, _):
                handleQRCodeString(string)
            case let .qrsData(data):
                handleQRSData(data)
            }
        }

        private func handleQRCodeString(_ string: String) {
            var error: NSError?
            let remoteProfile = LibboxParseRemoteProfileImportLink(string, &error)
            if let error {
                alert = AlertState(error: error)
                return
            }
            guard let remoteProfile else {
                alert = AlertState(errorMessage: String(localized: "The QR code does not contain a valid profile import link."))
                return
            }
            importRequest = NewProfileView.ImportRequest(name: remoteProfile.name, url: remoteProfile.url)
        }

        private func handleQRSData(_ data: Data) {
            do {
                let (actualData, _, _) = try BinaryMeta.readFileHeaderMeta(buffer: data)
                let content = try LibboxProfileContent.from(actualData)
                alert = AlertState(
                    title: String(localized: "Import Profile"),
                    message: String(localized: "Are you sure to import profile \(content.name)?"),
                    primaryButton: .default(String(localized: "Import")) {
                        Task {
                            do {
                                try await content.importProfile()
                                environments.profileUpdate.send()
                                dismiss()
                            } catch {
                                alert = AlertState(error: error)
                            }
                        }
                    },
                    secondaryButton: .cancel()
                )
            } catch {
                alert = AlertState(error: error)
            }
        }
    #endif
}