//
//  PersistenceADD.swift
//  Feels
//
//  Created by Trey Tartt on 2/17/22.
//

import CoreData

extension PersistenceController {
    public func add(mood: Mood, forDate date: Date, entryType: EntryType) {
        if let existingEntry = getEntry(byDate: date) {
            viewContext.delete(existingEntry)
            try? viewContext.save()
        }
        
        let newItem = MoodEntry(context: viewContext)
        newItem.timestamp = Date()
        newItem.moodValue = Int16(mood.rawValue)
        newItem.forDate = date
        newItem.weekDay = Int16(Calendar.current.component(.weekday, from: date))
        newItem.canEdit = true
        newItem.canDelete = true
        newItem.entryType = Int16(entryType.rawValue)
        
        EventLogger.log(event: "add_entry", withData: ["entry_type": entryType.rawValue])
        
        saveAndRunDataListerners()
    }
    
    func fillInMissingDates() {
        let currentOnboarding = UserDefaultsStore.getOnboarding()
        let endDate = ShowBasedOnVoteLogics.getLastDateVoteShouldExist(onboardingData: currentOnboarding)
        
        let fetchRequest = NSFetchRequest<MoodEntry>(entityName: "MoodEntry")
        fetchRequest.sortDescriptors = [NSSortDescriptor(key: "forDate", ascending: false)]
        let entries = try! viewContext.fetch(fetchRequest)
        
        if let firstEntry = entries.last?.forDate {
            let allDates: [Date] = Date.dates(from: firstEntry, toDate: endDate, includingToDate: true).map({
                let zeroDate = Calendar.current.date(bySettingHour: 0, minute: 0, second: 0, of: $0)!
                return zeroDate
            })
                    
            let existingEntries: [Date] = entries.compactMap({
                if let date = $0.forDate {
                    let zeroDate = Calendar.current.date(bySettingHour: 0, minute: 0, second: 0, of: date)!
                    return zeroDate
                }
                return nil
            })
            
            let allDatesSet = Set(allDates)
            let existingEntriesSet = Set(existingEntries)
            let missing = Array(allDatesSet.subtracting(existingEntriesSet)).sorted(by: >)
            for date in missing {
                // add 12 hours, if you enter a things right at 12:00.00 it wont show .... mabye
                // due to utc offset?
                let adjustedDate = Calendar.current.date(byAdding: .hour, value: 12, to: date)!
                add(mood: .missing, forDate: adjustedDate, entryType: .filledInMissing)
            }
            
            if !missing.isEmpty {
                EventLogger.log(event: "filled_in_missing_entries", withData: ["count": missing.count])
            }
        }
    }
    
    func removeNoForDates() {
        let fetchRequest = NSFetchRequest<MoodEntry>(entityName: "MoodEntry")
        fetchRequest.sortDescriptors = [NSSortDescriptor(key: "forDate", ascending: false)]
        let entries = try! viewContext.fetch(fetchRequest)
        
        for entry in entries {
            guard let _ = entry.forDate else {
                viewContext.delete(entry)
                try? viewContext.save()
                return
            }
        }
        EventLogger.log(event: "removed_entry_no_for_date", withData: ["count": entries.count])
    }
}