GetGoing/ProjectClock/AddAlarmViewController.swift

//
//  AddAlarmViewController.swift
//  ProjectClock
//
//  Created by Hykilpikonna on 1/8/21.
// and Dallon :)

import UIKit

class AddAlarmViewController: EndEditingOnReturn
{
    // Editing variables
    var alarmCell: AlarmTableCell? = nil
    var editMode: Bool { alarmCell != nil }
    
    override func viewDidLoad()
    {
        // End edit on return
        alarmNameTextField.delegate = self
        
        // Load alarm to edit if in edit mode
        if let alarmCell = alarmCell, let alarm = alarmCell.alarm
        {
            // Toggle editing mode
            viewTitle.text = "Edit Alarm"
            
            // Set all the original values to be edited
            let (y,m,d) = Date().getYMD()
            timePicker.date = Date.create(y, m, d, alarm.hour, alarm.minute)
            
            // Toggle proper repeats
            repeatWeekdaysSwitch.isOn = alarm.repeats[1...5].allSatisfy { $0 }
            repeatWeekendsSwitch.isOn = alarm.repeats[0] && alarm.repeats[6]
            
            alarmNameTextField.text = alarm.text
            updateETA()
            
            // Sets the WVM
            wvmPicker.selectRow(alarm.wakeMethod.index, inComponent: 0, animated: true)
            
            // Sets alarm tone
            ringtonePicker.selectRow(ringtones.firstIndex { $0.tone == alarm.alarmTone }!, inComponent: 0, animated: true)
        }
    }
    
    // UI: Make scroll view scrollable
    @IBOutlet weak var scrollView: UIScrollView!
    @IBOutlet weak var scrollViewInner: UIView!
    override func viewDidLayoutSubviews()
    {
        scrollView.addSubview(scrollViewInner)
        scrollView.contentSize = scrollViewInner.frame.size
    }
    
    // Pickers
    @IBOutlet weak var timePicker: UIDatePicker!
    @IBOutlet weak var wvmPicker: UIPickerView!
    @IBOutlet weak var ringtonePicker: UIPickerView!
    
    // UI Elements
    @IBOutlet weak var repeatWeekdaysSwitch: UISwitch!
    @IBOutlet weak var repeatWeekendsSwitch: UISwitch!
    @IBOutlet weak var alarmNameTextField: UITextField!
    @IBOutlet weak var timeTillAlarmLabel: UILabel!
    @IBOutlet weak var viewTitle: UILabel!
    
    /**
     Removes the currently selcted alarm.
     Returns the removed Alarm object.
     */
    @discardableResult
    func removeCurrentAlarm() -> Alarm?
    {
        guard let alarm = alarmCell?.alarm else { return nil }
        
        // Removes the alarm from stored alarms
        let alarms = Alarms.fromLocal()
        alarms.list = alarms.list.filter { $0 != alarmCell?.alarm }
        alarms.localSave()
        
        return alarm
    }
    
    /**
     Called when the time for the alarm is changed.
     Sets the time away at the top of the View.
     */
    @IBAction func alarmTimeUpdated(_ sender: Any) { updateETA() }
    
    /**
     Called when the user clicks the remove button and brings them back to the home page
     */
    @IBAction func cancelAlarmButton(_ sender: Any) {
        if editMode {
            removeCurrentAlarm()
        }
        
        self.dismiss(animated: true, completion: nil)
        //might need to reset all UI elements
    }
    
    /**
     Called when the user clicks Add Alarm
     */
    @IBAction func addAlarmButton(_ sender: Any)
    {
        var oldAlarm: Alarm? = nil
        let alarm = createAlarm()
        let alarms = Alarms.fromLocal()
        
        // Check if editing alarm
        if (editMode)
        {
            oldAlarm = removeCurrentAlarm()
        }
        // Check for existing alarm
        else
        {
            if (alarms.list.contains { $0 == alarm })
            {
                msg("Sorry", "An identical or similar alarm already exists, please try again")
                return
            }
        }
        
        // Add the alarm to the list and save the list
        Alarms.fromLocal().apply { $0.list.append(alarm) }.localSave();
        
        //Schedules notification for the alarm
        if editMode
        {
            Notification.removeNotification(alarm: oldAlarm!)
        }
        Notification.scheduleNotification(alarm: alarm)
        
        // Dismiss this view
        self.dismiss(animated: true, completion: nil)
    }
    
    /**
     Create alarm, but it doesn't add the alarm to the list
     */
    func createAlarm() -> Alarm
    {
        let (h, m, _) = timePicker.date.getHMS()
        
        // Create the alarm
        let alarm = Alarm(hour: h, minute: m,
                          text: alarmNameTextField.text ?? "Alarm",
                          wakeMethod: wvms[wvmPicker.selectedRow(inComponent: 0)],
                          lastActivate: Date(), alarmTone: ringtones[ringtonePicker.selectedRow(inComponent: 0)].tone, toneName: ringtones[ringtonePicker.selectedRow(inComponent: 0)].name)
        
        // Set alarm.repeats to correspond with what the user selects
        (0...6).forEach { alarm.repeats[$0] = false }
        if repeatWeekdaysSwitch.isOn { (1...5).forEach { alarm.repeats[$0] = true } }
        if repeatWeekendsSwitch.isOn { [0, 6].forEach { alarm.repeats[$0] = true } }
        
        return alarm
    }
    
    /**
     Dynamically the ETA label for the alarm
     */
    func updateETA() {
        let timeTill = createAlarm().nextActivate!.timeIntervalSince(Date()).str()
        timeTillAlarmLabel.text = "Going off in \(timeTill)"
    }
}

class WVMDataSource: UIPickerView, UIPickerViewDelegate, UIPickerViewDataSource
{
    required init?(coder: NSCoder)
    {
        super.init(coder: coder)
        delegate = self
        dataSource = self
    }
    
    func numberOfComponents(in pickerView: UIPickerView) -> Int
    {
        return 1
    }
    
    func pickerView(_ v: UIPickerView, numberOfRowsInComponent: Int) -> Int
    {
        return wvms.count
    }
    
    func pickerView(_ v: UIPickerView, titleForRow r: Int, forComponent: Int) -> String?
    {
        return wvms[r].name + " - " + wvms[r].desc
    }
}


class RingtonesDataSource: UIPickerView, UIPickerViewDelegate, UIPickerViewDataSource
{
    required init?(coder: NSCoder)
    {
        super.init(coder: coder)
        delegate = self
        dataSource = self
    }
    
    func numberOfComponents(in pickerView: UIPickerView) -> Int
    {
        return 1
    }
    
    func pickerView(_ v: UIPickerView, numberOfRowsInComponent: Int) -> Int
    {
        return ringtones.count
    }

    func pickerView(_ v: UIPickerView, titleForRow r: Int, forComponent: Int) -> String?
    {
        return ringtones[r].name
        
    }
}