Adobe-Downloader/Adobe Downloader/Views/AppCardView.swift

//
//  Adobe Downloader
//
//  Created by X1a0He on 2024/10/30.
//

import SwiftUI
import Combine

private enum AppCardConstants {
    static let cardWidth: CGFloat = 250
    static let cardHeight: CGFloat = 200
    static let iconSize: CGFloat = 64
    static let cornerRadius: CGFloat = 10
    static let buttonHeight: CGFloat = 32
    static let titleFontSize: CGFloat = 16
    static let buttonFontSize: CGFloat = 14
    
    static let shadowOpacity: Double = 0.05
    static let shadowRadius: CGFloat = 2
    static let strokeOpacity: Double = 0.1
    static let strokeWidth: CGFloat = 2
    static let backgroundOpacity: Double = 0.05
}

final class IconCache {
    static let shared = IconCache()
    private var cache = NSCache<NSString, NSImage>()
    
    func getIcon(for url: String) -> NSImage? {
        cache.object(forKey: url as NSString)
    }
    
    func setIcon(_ image: NSImage, for url: String) {
        cache.setObject(image, forKey: url as NSString)
    }
}

@MainActor
final class AppCardViewModel: ObservableObject {
    @Published var iconImage: NSImage?
    @Published var showError = false
    @Published var errorMessage = ""
    @Published var showVersionPicker = false
    @Published var selectedVersion = ""
    @Published var showLanguagePicker = false
    @Published var selectedLanguage = ""
    @Published var showExistingFileAlert = false
    @Published var existingFilePath: URL?
    @Published var pendingVersion = ""
    @Published var pendingLanguage = ""
    @Published var showRedownloadConfirm = false
    
    let sap: Sap
    weak var networkManager: NetworkManager?
    
    @Published var isDownloading = false
    private let userDefaults = UserDefaults.standard
    
    private var useDefaultDirectory: Bool {
        StorageData.shared.useDefaultDirectory
    }
    
    private var defaultDirectory: String {
        StorageData.shared.defaultDirectory
    }
    
    private var cancellables = Set<AnyCancellable>()
    
    init(sap: Sap, networkManager: NetworkManager?) {
        self.sap = sap
        self.networkManager = networkManager
        
        Task { @MainActor in
            setupObservers()
        }
    }
    
    @MainActor
    private func setupObservers() {
        networkManager?.$downloadTasks
            .receive(on: RunLoop.main)
            .sink { [weak self] tasks in
                guard let self = self else { return }
                let hasActiveTask = tasks.contains { 
                    $0.sapCode == self.sap.sapCode && self.isTaskActive($0.status)
                }
                
                if hasActiveTask != self.isDownloading {
                    self.isDownloading = hasActiveTask
                    self.objectWillChange.send()
                }
            }
            .store(in: &cancellables)
        
        networkManager?.objectWillChange
            .receive(on: RunLoop.main)
            .sink { [weak self] _ in
                self?.updateDownloadingStatus()
            }
            .store(in: &cancellables)
    }
    
    private func isTaskActive(_ status: DownloadStatus) -> Bool {
        switch status {
        case .downloading, .preparing, .waiting, .retrying:
            return true
        case .paused:
            return false
        case .completed, .failed:
            return false
        }
    }
    
    @MainActor
    func updateDownloadingStatus() {
        guard let networkManager = networkManager else {
            self.isDownloading = false
            return
        }
        
        let hasActiveTask = networkManager.downloadTasks.contains { 
            $0.sapCode == sap.sapCode && isTaskActive($0.status)
        }
        
        if hasActiveTask != self.isDownloading {
            self.isDownloading = hasActiveTask
            self.objectWillChange.send()
        }
    }
    
    func getDestinationURL(version: String, language: String) async throws -> URL {
        let platform = sap.versions[version]?.apPlatform ?? "unknown"
        let installerName = sap.sapCode == "APRO" 
            ? "Adobe Downloader \(sap.sapCode)_\(version)_\(platform).dmg"
            : "Adobe Downloader \(sap.sapCode)_\(version)-\(language)-\(platform)"

        if useDefaultDirectory && !defaultDirectory.isEmpty {
            return URL(fileURLWithPath: defaultDirectory)
                .appendingPathComponent(installerName)
        }
        
        return try await withCheckedThrowingContinuation { continuation in
            DispatchQueue.main.async {
                let panel = NSOpenPanel()
                panel.title = "选择保存位置"
                panel.canCreateDirectories = true
                panel.canChooseDirectories = true
                panel.canChooseFiles = false
                panel.directoryURL = FileManager.default.urls(for: .downloadsDirectory, in: .userDomainMask).first
                
                if panel.runModal() == .OK, let selectedURL = panel.url {
                    continuation.resume(returning: selectedURL.appendingPathComponent(installerName))
                } else {
                    continuation.resume(throwing: NetworkError.cancelled)
                }
            }
        }
    }

    func handleError(_ error: Error) {
        Task { @MainActor in
            if case NetworkError.cancelled = error { return }
            errorMessage = error.localizedDescription
            showError = true
        }
    }

    func loadIcon() {
        if let bestIcon = sap.getBestIcon(),
           let iconURL = URL(string: bestIcon.url) {
            
            if let cachedImage = IconCache.shared.getIcon(for: bestIcon.url) {
                self.iconImage = cachedImage
                return
            }
            
            Task {
                do {
                    var request = URLRequest(url: iconURL)
                    request.timeoutInterval = 10
                    
                    let (data, response) = try await URLSession.shared.data(for: request)
                    
                    guard let httpResponse = response as? HTTPURLResponse,
                          (200...299).contains(httpResponse.statusCode) else {
                        throw URLError(.badServerResponse)
                    }
                    
                    await MainActor.run {
                        if let image = NSImage(data: data) {
                            IconCache.shared.setIcon(image, for: bestIcon.url)
                            self.iconImage = image
                        }
                    }
                } catch {
                    await MainActor.run {
                        if let localImage = NSImage(named: sap.sapCode) {
                            self.iconImage = localImage
                        }
                    }
                }
            }
        } else {
            if let localImage = NSImage(named: sap.sapCode) {
                self.iconImage = localImage
            }
        }
    }

    func handleDownloadRequest(_ version: String, useDefaultLanguage: Bool, defaultLanguage: String) async {
        await MainActor.run {
            if useDefaultLanguage {
                Task {
                    await checkAndStartDownload(version: version, language: defaultLanguage)
                }
            } else {
                selectedVersion = version
                showLanguagePicker = true
            }
        }
    }
    
    func checkAndStartDownload(version: String, language: String) async {
        if let networkManager = networkManager {
            if let existingPath = networkManager.isVersionDownloaded(sap: sap, version: version, language: language) {
                await MainActor.run {
                    existingFilePath = existingPath
                    pendingVersion = version
                    pendingLanguage = language
                    showExistingFileAlert = true
                }
            } else {
                do {
                    let destinationURL = try await getDestinationURL(version: version, language: language)
                    try await networkManager.startDownload(
                        sap: sap,
                        selectedVersion: version,
                        language: language,
                        destinationURL: destinationURL
                    )
                } catch {
                    handleError(error)
                }
            }
        }
    }

    func createCompletedTask(_ path: URL) async {
        guard let networkManager = networkManager,
              let productInfo = sap.versions[pendingVersion] else { return }

        let existingTask = networkManager.downloadTasks.first { task in
            return task.sapCode == sap.sapCode && 
                   task.version == pendingVersion && 
                   task.language == pendingLanguage &&
                   task.directory == path
        }
        
        if existingTask != nil {
            return
        }

        var productsToDownload: [ProductsToDownload] = []
        let mainProduct = ProductsToDownload(
            sapCode: sap.sapCode,
            version: pendingVersion,
            buildGuid: productInfo.buildGuid
        )
        productsToDownload.append(mainProduct)

        for dependency in productInfo.dependencies {
            if let dependencyVersions = networkManager.saps[dependency.sapCode]?.versions {
                let sortedVersions = dependencyVersions.sorted { first, second in
                    first.value.productVersion.compare(second.value.productVersion, options: .numeric) == .orderedDescending
                }
                
                var buildGuid = ""
                for (_, versionInfo) in sortedVersions where versionInfo.baseVersion == dependency.version {
                    if StorageData.shared.allowedPlatform.contains(versionInfo.apPlatform) {
                        buildGuid = versionInfo.buildGuid
                        break
                    }
                }
                
                if !buildGuid.isEmpty {
                    let dependencyProduct = ProductsToDownload(
                        sapCode: dependency.sapCode,
                        version: dependency.version,
                        buildGuid: buildGuid
                    )
                    productsToDownload.append(dependencyProduct)
                }
            }
        }

        let completedTask = NewDownloadTask(
            sapCode: sap.sapCode,
            version: pendingVersion,
            language: pendingLanguage,
            displayName: sap.displayName,
            directory: path,
            productsToDownload: productsToDownload,
            retryCount: 0,
            createAt: Date(),
            totalStatus: .completed(DownloadStatus.CompletionInfo(
                timestamp: Date(),
                totalTime: 0,
                totalSize: 0
            )),
            totalProgress: 1.0,
            totalDownloadedSize: 0,
            totalSize: 0,
            totalSpeed: 0,
            platform: ""
        )

        await MainActor.run {
            networkManager.downloadTasks.append(completedTask)
            networkManager.objectWillChange.send()
        }
    }
    
    var dependenciesCount: Int {
        if let firstVersion = sap.versions.first?.value {
            return firstVersion.dependencies.count
        }
        return 0
    }
    
    var hasValidIcon: Bool {
        iconImage != nil
    }
    
    var canDownload: Bool {
        !isDownloading
    }
    
    var downloadButtonTitle: String {
        isDownloading ? String(localized: "下载中") : String(localized: "下载")
    }
    
    var downloadButtonIcon: String {
        isDownloading ? "hourglass.circle.fill" : "arrow.down.circle"
    }
}

struct AppCardView: View {
    @StateObject private var viewModel: AppCardViewModel
    @EnvironmentObject private var networkManager: NetworkManager
    @StorageValue(\.useDefaultLanguage) private var useDefaultLanguage
    @StorageValue(\.defaultLanguage) private var defaultLanguage
    
    init(sap: Sap) {
        _viewModel = StateObject(wrappedValue: AppCardViewModel(sap: sap, networkManager: nil))
    }
    
    var body: some View {
        CardContainer {
            VStack {
                IconView(viewModel: viewModel)
                ProductInfoView(viewModel: viewModel)
                Spacer()
                DownloadButtonView(viewModel: viewModel)
            }
        }
        .modifier(CardModifier())
        .modifier(SheetModifier(viewModel: viewModel, networkManager: networkManager))
        .modifier(AlertModifier(viewModel: viewModel, confirmRedownload: true))
        .onAppear(perform: setupViewModel)
        .onChange(of: networkManager.downloadTasks, perform: updateDownloadStatus)
    }
    
    private func setupViewModel() {
        viewModel.networkManager = networkManager
        viewModel.updateDownloadingStatus()
    }
    
    private func updateDownloadStatus(_ _: [NewDownloadTask]) {
        viewModel.updateDownloadingStatus()
    }
}

private struct CardContainer<Content: View>: View {
    let content: Content
    
    init(@ViewBuilder content: () -> Content) {
        self.content = content()
    }
    
    var body: some View {
        content
            .padding()
            .frame(width: AppCardConstants.cardWidth, height: AppCardConstants.cardHeight)
    }
}

private struct IconView: View {
    @ObservedObject var viewModel: AppCardViewModel
    
    var body: some View {
        Group {
            if viewModel.hasValidIcon {
                Image(nsImage: viewModel.iconImage!)
                    .resizable()
                    .interpolation(.high)
                    .scaledToFit()
            } else {
                Image(systemName: "app.fill")
                    .resizable()
                    .scaledToFit()
                    .foregroundColor(.secondary)
            }
        }
        .frame(width: AppCardConstants.iconSize, height: AppCardConstants.iconSize)
        .onAppear(perform: viewModel.loadIcon)
    }
}

private struct ProductInfoView: View {
    @ObservedObject var viewModel: AppCardViewModel
    
    var body: some View {
        VStack {
            Text(viewModel.sap.displayName)
                .font(.system(size: AppCardConstants.titleFontSize))
                .fontWeight(.bold)
                .lineLimit(2)
                .multilineTextAlignment(.center)
            
            HStack(spacing: 4) {
                Text("可用版本: \(viewModel.sap.versions.count)")
                Text("|")
                Text("依赖包: \(viewModel.dependenciesCount)")
            }
            .font(.caption)
            .foregroundColor(.secondary)
            .frame(height: 20)
        }
    }
}

private struct DownloadButtonView: View {
    @ObservedObject var viewModel: AppCardViewModel
    
    var body: some View {
        Button(action: { viewModel.showVersionPicker = true }) {
            Label(viewModel.downloadButtonTitle,
                  systemImage: viewModel.downloadButtonIcon)
                .font(.system(size: AppCardConstants.buttonFontSize))
                .frame(minWidth: 0, maxWidth: .infinity)
                .frame(height: AppCardConstants.buttonHeight)
        }
        .buttonStyle(.borderedProminent)
        .tint(viewModel.isDownloading ? .gray : .blue)
        .disabled(!viewModel.canDownload)
    }
}

private struct CardModifier: ViewModifier {
    func body(content: Content) -> some View {
        content
            .background(
                RoundedRectangle(cornerRadius: AppCardConstants.cornerRadius)
                    .fill(Color.black.opacity(AppCardConstants.backgroundOpacity))
            )
            .overlay(
                RoundedRectangle(cornerRadius: AppCardConstants.cornerRadius)
                    .stroke(Color.gray.opacity(AppCardConstants.strokeOpacity), 
                           lineWidth: AppCardConstants.strokeWidth)
            )
            .shadow(
                color: Color.primary.opacity(AppCardConstants.shadowOpacity),
                radius: AppCardConstants.shadowRadius,
                x: 0,
                y: 1
            )
    }
}

private struct SheetModifier: ViewModifier {
    @ObservedObject var viewModel: AppCardViewModel
    let networkManager: NetworkManager
    @StorageValue(\.useDefaultLanguage) private var useDefaultLanguage
    @StorageValue(\.defaultLanguage) private var defaultLanguage
    
    func body(content: Content) -> some View {
        content
            .sheet(isPresented: $viewModel.showVersionPicker) {
                VersionPickerView(sap: viewModel.sap) { version in
                    Task {
                        await viewModel.handleDownloadRequest(
                            version,
                            useDefaultLanguage: useDefaultLanguage,
                            defaultLanguage: defaultLanguage
                        )
                    }
                }
                .environmentObject(networkManager)
            }
            .sheet(isPresented: $viewModel.showLanguagePicker) {
                LanguagePickerView(languages: AppStatics.supportedLanguages) { language in
                    Task {
                        await viewModel.checkAndStartDownload(
                            version: viewModel.selectedVersion,
                            language: language
                        )
                    }
                }
            }
    }
}

#Preview {
    let networkManager = NetworkManager()
    let sap = Sap(
        hidden: false,
        displayName: "Photoshop",
        sapCode: "PHSP",
        versions: [
            "25.0.0": Sap.Versions(
                sapCode: "PHSP",
                baseVersion: "25.0.0",
                productVersion: "25.0.0",
                apPlatform: "macuniversal",
                dependencies: [
                    Sap.Versions.Dependencies(sapCode: "ACR", version: "9.6"),
                    Sap.Versions.Dependencies(sapCode: "COCM", version: "1.0"),
                    Sap.Versions.Dependencies(sapCode: "COSY", version: "2.4.1")
                ],
                buildGuid: ""
            )
        ],
        icons: [
            Sap.ProductIcon(
                size: "192x192",
                url: "https://ffc-static-cdn.oobesaas.adobe.com/icons/PHSP/25.0.0/192x192.png"
            )
        ]
    )
    
    return AppCardView(sap: sap)
        .environmentObject(networkManager)
}

struct AlertModifier: ViewModifier {
    @ObservedObject var viewModel: AppCardViewModel
    let confirmRedownload: Bool

    func body(content: Content) -> some View {
        content
            .sheet(isPresented: $viewModel.showExistingFileAlert) {
                if let path = viewModel.existingFilePath {
                    ExistingFileAlertView(
                        path: path,
                        onUseExisting: {
                            viewModel.showExistingFileAlert = false
                            if !viewModel.pendingVersion.isEmpty && !viewModel.pendingLanguage.isEmpty {
                                Task {
                                    if let networkManager = viewModel.networkManager,
                                       !networkManager.downloadTasks.contains(where: { task in
                                           task.sapCode == viewModel.sap.sapCode &&
                                           task.version == viewModel.pendingVersion &&
                                           task.language == viewModel.pendingLanguage
                                       }) {
                                        await viewModel.createCompletedTask(path)
                                    }
                                }
                            }
                        },
                        onRedownload: {
                            viewModel.showExistingFileAlert = false
                            if !viewModel.pendingVersion.isEmpty && !viewModel.pendingLanguage.isEmpty {
                                if confirmRedownload {
                                    viewModel.showRedownloadConfirm = true
                                } else {
                                    Task {
                                        await startRedownload()
                                    }
                                }
                            }
                        },
                        onCancel: {
                            viewModel.showExistingFileAlert = false
                        },
                        iconImage: viewModel.iconImage
                    )
                }
            }
            .alert("确认重新下载", isPresented: $viewModel.showRedownloadConfirm) {
                Button("取消", role: .cancel) { }
                Button("确认") {
                    Task {
                        await startRedownload()
                    }
                }
            } message: {
                Text("是否确认重新下载？这将覆盖现有的安装程序。")
            }
            .alert("下载错误", isPresented: $viewModel.showError) {
                Button("确定", role: .cancel) { }
                Button("重试") {
                    if !viewModel.selectedVersion.isEmpty {
                        Task {
                            await viewModel.checkAndStartDownload(
                                version: viewModel.selectedVersion,
                                language: viewModel.selectedLanguage
                            )
                        }
                    }
                }
            } message: {
                Text(viewModel.errorMessage)
            }
    }
    
    private func startRedownload() async {
        guard let networkManager = viewModel.networkManager else { return }
        
        do {
            networkManager.downloadTasks.removeAll { task in
                task.sapCode == viewModel.sap.sapCode &&
                task.version == viewModel.pendingVersion &&
                task.language == viewModel.pendingLanguage
            }
            
            if let existingPath = viewModel.existingFilePath {
                try? FileManager.default.removeItem(at: existingPath)
            }
            
            let destinationURL = try await viewModel.getDestinationURL(
                version: viewModel.pendingVersion,
                language: viewModel.pendingLanguage
            )
            
            try await networkManager.startDownload(
                sap: viewModel.sap,
                selectedVersion: viewModel.pendingVersion,
                language: viewModel.pendingLanguage,
                destinationURL: destinationURL
            )
        } catch {
            viewModel.handleError(error)
        }
    }
}