vOOice/VoiceInk/Services/DictionaryImportExportService.swift

import Foundation
import AppKit
import UniformTypeIdentifiers
import SwiftData

struct DictionaryExportData: Codable {
    let version: String
    let vocabularyWords: [String]
    let wordReplacements: [String: String]
    let exportDate: Date
}

class DictionaryImportExportService {
    static let shared = DictionaryImportExportService()

    private init() {}

    func exportDictionary(from context: ModelContext) {
        // Fetch vocabulary words from SwiftData
        var dictionaryWords: [String] = []
        let vocabularyDescriptor = FetchDescriptor<VocabularyWord>(sortBy: [SortDescriptor(\VocabularyWord.word)])
        if let items = try? context.fetch(vocabularyDescriptor) {
            dictionaryWords = items.map { $0.word }
        }

        // Fetch word replacements from SwiftData
        var wordReplacements: [String: String] = [:]
        let replacementsDescriptor = FetchDescriptor<WordReplacement>()
        if let replacements = try? context.fetch(replacementsDescriptor) {
            wordReplacements = Dictionary(uniqueKeysWithValues: replacements.map { ($0.originalText, $0.replacementText) })
        }

        let version = Bundle.main.object(forInfoDictionaryKey: "CFBundleShortVersionString") as? String ?? "1.0.0"

        let exportData = DictionaryExportData(
            version: version,
            vocabularyWords: dictionaryWords,
            wordReplacements: wordReplacements,
            exportDate: Date()
        )

        let encoder = JSONEncoder()
        encoder.outputFormatting = .prettyPrinted
        encoder.dateEncodingStrategy = .iso8601

        do {
            let jsonData = try encoder.encode(exportData)

            let savePanel = NSSavePanel()
            savePanel.allowedContentTypes = [UTType.json]
            savePanel.nameFieldStringValue = "VoiceInk_Dictionary.json"
            savePanel.title = "Export Dictionary Data"
            savePanel.message = "Choose a location to save your vocabulary and word replacements."

            DispatchQueue.main.async {
                if savePanel.runModal() == .OK {
                    if let url = savePanel.url {
                        do {
                            try jsonData.write(to: url)
                            self.showAlert(title: "Export Successful", message: "Dictionary data exported successfully to \(url.lastPathComponent).")
                        } catch {
                            self.showAlert(title: "Export Error", message: "Could not save dictionary data: \(error.localizedDescription)")
                        }
                    }
                } else {
                    self.showAlert(title: "Export Canceled", message: "Export operation was canceled.")
                }
            }
        } catch {
            self.showAlert(title: "Export Error", message: "Could not encode dictionary data: \(error.localizedDescription)")
        }
    }

    func importDictionary(into context: ModelContext) {
        let openPanel = NSOpenPanel()
        openPanel.allowedContentTypes = [UTType.json]
        openPanel.canChooseFiles = true
        openPanel.canChooseDirectories = false
        openPanel.allowsMultipleSelection = false
        openPanel.title = "Import Dictionary Data"
        openPanel.message = "Choose a dictionary file to import. New items will be added, existing items will be kept."

        DispatchQueue.main.async {
            if openPanel.runModal() == .OK {
                guard let url = openPanel.url else {
                    self.showAlert(title: "Import Error", message: "Could not get the file URL.")
                    return
                }

                do {
                    let jsonData = try Data(contentsOf: url)
                    let decoder = JSONDecoder()
                    decoder.dateDecodingStrategy = .iso8601
                    let importedData = try decoder.decode(DictionaryExportData.self, from: jsonData)

                    // Fetch existing vocabulary words from SwiftData
                    let vocabularyDescriptor = FetchDescriptor<VocabularyWord>()
                    let existingItems = (try? context.fetch(vocabularyDescriptor)) ?? []
                    let existingWordsLower = Set(existingItems.map { $0.word.lowercased() })
                    let originalExistingCount = existingItems.count
                    var newWordsAdded = 0

                    // Import vocabulary words
                    for importedWord in importedData.vocabularyWords {
                        if !existingWordsLower.contains(importedWord.lowercased()) {
                            let newWord = VocabularyWord(word: importedWord)
                            context.insert(newWord)
                            newWordsAdded += 1
                        }
                    }

                    // Fetch existing word replacements from SwiftData
                    let replacementsDescriptor = FetchDescriptor<WordReplacement>()
                    let existingReplacements = (try? context.fetch(replacementsDescriptor)) ?? []
                    var addedCount = 0
                    var updatedCount = 0

                    // Import word replacements
                    for (importedKey, importedReplacement) in importedData.wordReplacements {
                        let normalizedImportedKey = self.normalizeReplacementKey(importedKey)
                        let importedWords = self.extractWords(from: normalizedImportedKey)

                        // Check for conflicts and update existing replacements
                        var hasConflict = false
                        for existingReplacement in existingReplacements {
                            var existingWords = self.extractWords(from: existingReplacement.originalText)
                            var modified = false

                            for importedWord in importedWords {
                                if let index = existingWords.firstIndex(where: { $0.lowercased() == importedWord.lowercased() }) {
                                    existingWords.remove(at: index)
                                    modified = true
                                    hasConflict = true
                                }
                            }

                            if modified {
                                if existingWords.isEmpty {
                                    context.delete(existingReplacement)
                                } else {
                                    existingReplacement.originalText = existingWords.joined(separator: ", ")
                                }
                                updatedCount += 1
                            }
                        }

                        // Add new replacement
                        let newReplacement = WordReplacement(originalText: normalizedImportedKey, replacementText: importedReplacement)
                        context.insert(newReplacement)
                        addedCount += 1
                    }

                    // Save all changes
                    try context.save()

                    var message = "Dictionary data imported successfully from \(url.lastPathComponent).\n\n"
                    message += "Vocabulary Words: \(newWordsAdded) added, \(originalExistingCount) kept\n"
                    message += "Word Replacements: \(addedCount) added, \(updatedCount) updated"

                    self.showAlert(title: "Import Successful", message: message)

                } catch {
                    self.showAlert(title: "Import Error", message: "Error importing dictionary data: \(error.localizedDescription). The file might be corrupted or not in the correct format.")
                }
            } else {
                self.showAlert(title: "Import Canceled", message: "Import operation was canceled.")
            }
        }
    }

    private func normalizeReplacementKey(_ key: String) -> String {
        let words = extractWords(from: key)
        return words.joined(separator: ", ")
    }

    private func extractWords(from key: String) -> [String] {
        return key
            .split(separator: ",")
            .map { $0.trimmingCharacters(in: .whitespacesAndNewlines) }
            .filter { !$0.isEmpty }
    }

    private func showAlert(title: String, message: String) {
        DispatchQueue.main.async {
            let alert = NSAlert()
            alert.messageText = title
            alert.informativeText = message
            alert.alertStyle = .informational
            alert.addButton(withTitle: "OK")
            alert.runModal()
        }
    }
}