omnivore/apple/OmnivoreKit/Sources/Views/SnoozeView.swift

import Models
import SwiftUI

public struct SnoozeView: View {
  @Binding var snoozePresented: Bool
  @Binding var itemToSnoozeID: String?
  let snoozeAction: (SnoozeActionParams) -> Void

  public init(
    snoozePresented: Binding<Bool>,
    itemToSnoozeID: Binding<String?>,
    snoozeAction: @escaping (SnoozeActionParams) -> Void
  ) {
    self._snoozePresented = snoozePresented
    self._itemToSnoozeID = itemToSnoozeID
    self.snoozeAction = snoozeAction
  }

  public var body: some View {
    VStack {
      Spacer()

      HStack {
        SnoozeIconButtonView(snooze: Snooze.currentValues[0], action: { snoozeItem($0) })
        SnoozeIconButtonView(snooze: Snooze.currentValues[1], action: { snoozeItem($0) })
      }

      Spacer(minLength: 32)

      HStack {
        SnoozeIconButtonView(snooze: Snooze.currentValues[2], action: { snoozeItem($0) })
        SnoozeIconButtonView(snooze: Snooze.currentValues[3], action: { snoozeItem($0) })
      }
      Spacer()
    }.padding(EdgeInsets(top: 8, leading: 16, bottom: 8, trailing: 16))
  }

  private func snoozeItem(_ snooze: Snooze) {
    if let itemID = itemToSnoozeID {
      withAnimation(.linear(duration: 0.4)) {
        snoozeAction(
          SnoozeActionParams(
            feedItemId: itemID,
            snoozeUntilDate: snooze.until,
            successMessage: "Snoozed until \(snooze.untilStr)"
          )
        )
      }
    }
    itemToSnoozeID = nil
    snoozePresented = false
  }
}

public struct SnoozeActionParams {
  public let feedItemId: String
  public let snoozeUntilDate: Date
  public let successMessage: String?
}

private struct SnoozeIconButtonView: View {
  let snooze: Snooze
  let action: (_ snooze: Snooze) -> Void

  var body: some View {
    Button(
      action: { action(snooze) },
      label: {
        VStack(alignment: .center, spacing: 8) {
          snooze.icon
            .font(.appTitle)
            .foregroundColor(.appYellow48)
          Text(snooze.title)
            .font(.appBody)
            .foregroundColor(.appGrayText)
          Text(snooze.untilStr)
            .font(.appCaption)
            .foregroundColor(.appGrayText)
        }
        .frame(
          maxWidth: .infinity,
          maxHeight: .infinity
        )
      }
    )
    .frame(height: 100)
  }
}

struct Snooze {
  let until: Date
  let icon: Image
  let title: String
  let untilStr: String

  init(until: Date, icon: Image, title: String, needsDay: Bool) {
    self.until = until
    self.icon = icon
    self.title = title
    let formatter = DateFormatter()
    formatter.dateFormat = needsDay ? "EEE h:mm a" : "h:mm a"
    self.untilStr = formatter.string(from: until)
  }

  static var currentValues: [Snooze] {
    calculateValues(for: Date(), calendar: Calendar.current)
  }

  static func calculateValues(for now: Date, calendar: Calendar) -> [Snooze] {
    var res: [Snooze] = []
    let components = calendar.dateComponents([.year, .month, .day, .hour, .timeZone, .weekday], from: now)

    var tonightComponent = components
    tonightComponent.hour = 20

    var thisMorningComponent = components
    thisMorningComponent.hour = 8

    let tonight = calendar.date(from: tonightComponent)!
    let thisMorning = calendar.date(from: thisMorningComponent)!

    let tomorrowMorning = Calendar.current.date(byAdding: DateComponents(day: 1), to: thisMorning)

    // Add either tonight or tomorrow night
    if now < tonight {
      res.append(Snooze(until: tonight, icon: .moonStars, title: "Tonight", needsDay: false))
    } else {
      let tomorrowNight = Calendar.current.date(byAdding: DateComponents(day: 1), to: tonight)!
      res.append(Snooze(until: tomorrowNight, icon: .moonStars, title: "Tomorrow night", needsDay: false))
    }

    if let tomorrowMorning = tomorrowMorning {
      res.append(Snooze(until: tomorrowMorning, icon: .sunHorizon, title: "Tomorrow morning", needsDay: false))
    }

    if let weekday = components.weekday {
      // Add this or next weekend
      if weekday < 5 {
        let thisWeekend = Calendar.current.date(byAdding: DateComponents(day: 7 - weekday), to: thisMorning)
        res.append(Snooze(until: thisWeekend!, icon: .mountains, title: "This weekend", needsDay: true))
      } else {
        let nextWeekend = Calendar.current.date(byAdding: DateComponents(day: 7 - (weekday - 5)), to: thisMorning)!
        res.append(Snooze(until: nextWeekend, icon: .mountains, title: "Next weekend", needsDay: true))
      }
      let nextWeek = Calendar.current.date(byAdding: DateComponents(day: weekday + 5), to: thisMorning)!
      res.append(Snooze(until: nextWeek, icon: .chartLineUp, title: "Next week", needsDay: true))
    }

    return Array(res.sorted(by: { $0.until > $1.until }).reversed())
  }
}