iphone/Maps/UI/Search/DatePickerViewController.swift - master - githubmirror/omim

iphone/Maps/UI/Search/DatePickerViewController.swift
112 строк4 KB
Alexander Boriskov
[iOS] Fixed date picker and guest picker in horizontal orientation
21 окт 2020, 17:29
21 окт 2020, 17:29e517138
import DatePicker
import UIKit

@objc protocol DatePickerViewControllerDelegate: AnyObject {
  func datePicker(_ datePicker: DatePickerViewController, didSelectStartDate startDate: Date, endDate: Date)
  func datePickerDidClick(_ datePicker: DatePickerViewController, didSelectStartDate startDate: Date?, endDate: Date?)
  func datePickerDidCancel(_ datePicker: DatePickerViewController)
}

@objc final class DatePickerViewController: UIViewController {
  private static let kHeight: CGFloat = 550
  private let transitioning = CoverVerticalModalTransitioning(presentationHeight: min(UIScreen.main.bounds.height, kHeight))

  @IBOutlet var checkInLabel: UILabel!
  @IBOutlet var startDateLabel: UILabel!
  @IBOutlet var endDateLabel: UILabel!
  @IBOutlet var numberOfDaysLabel: UILabel!
  @IBOutlet var doneButton: UIButton!
  @IBOutlet var cancelButton: UIButton!
  @IBOutlet var datePickerView: DatePickerView!

  @objc var initialCheckInDate: Date?
  @objc var initialCheckOutDate: Date?

  @objc weak var delegate: DatePickerViewControllerDelegate?

  lazy var dateFormatter: DateFormatter = {
    let formatter = DateFormatter()
    formatter.setLocalizedDateFormatFromTemplate("EEE, MMMMd")
    return formatter
  }()

  override func viewDidLoad() {
    super.viewDidLoad()
    datePickerView.delegate = self
    if let checkInDate = initialCheckInDate, let checkOutDate = initialCheckOutDate, checkOutDate > checkInDate {
      datePickerView.startDate = checkInDate
      datePickerView.endDate = checkOutDate
      doneButton.isEnabled = true
    } else {
      doneButton.isEnabled = false
    }
  }

  override var transitioningDelegate: UIViewControllerTransitioningDelegate? {
    get { return transitioning }
    set {}
  }

  override var modalPresentationStyle: UIModalPresentationStyle {
    get { return alternativeSizeClass(iPhone: .custom, iPad: .popover) }
    set {}
  }

  @IBAction func onDone(_ sender: UIButton) {
    guard let startDate = datePickerView.startDate, let endDate = datePickerView.endDate else { fatalError() }
    delegate?.datePicker(self, didSelectStartDate: startDate, endDate: endDate)
  }

  @IBAction func onCancel(_ sender: UIButton) {
    delegate?.datePickerDidCancel(self)
  }

  override func traitCollectionDidChange(_ previousTraitCollection: UITraitCollection?) {
    super.traitCollectionDidChange(previousTraitCollection)
    alternativeSizeClass(iPhone: {
      if let superview = view.superview {
        view.height = min(UIScreen.main.bounds.height, DatePickerViewController.kHeight)
        view.minY = superview.height - view.height
      }
    }, iPad: {})
  }
}

extension DatePickerViewController: DatePickerViewDelegate {
  func datePickerView(_ view: DatePickerView, didSelect date: Date) {
    defer {
      doneButton.isEnabled = view.endDate != nil
      if let startDate = view.startDate, let endDate = view.endDate {
        let days = Calendar.current.dateComponents([.day], from: startDate, to: endDate).day!
        numberOfDaysLabel.text = String(format: L("date_picker_amout_of_days"), days)
      } else {
        numberOfDaysLabel.text = nil
      }
      delegate?.datePickerDidClick(self, didSelectStartDate: view.startDate, endDate: view.endDate)
    }

    guard let startDate = view.startDate else {
      view.startDate = date
      startDateLabel.text = dateFormatter.string(from: date).capitalized
      endDateLabel.text = nil
      return
    }

    if date > startDate && view.endDate == nil {
      guard Calendar.current.dateComponents([.day], from: startDate, to: date).day! <= 30 else {
        MWMAlertViewController.activeAlert().presentDefaultAlert(withTitle: L("thirty_days_limit_dialog"),
                                                                 message: nil, rightButtonTitle: L("ok"),
                                                                 leftButtonTitle: nil,
                                                                 rightButtonAction: nil)
        return
      }
      view.endDate = date
      endDateLabel.text = dateFormatter.string(from: date).capitalized
      return
    }

    view.startDate = date
    startDateLabel.text = dateFormatter.string(from: date).capitalized
    endDateLabel.text = nil
  }
}