comaps/iphone/Maps/UI/PlacePage/Components/OpeningHoursViewController.swift

import UIKit

class OpeningHoursTodayViewController: UIViewController {
  @IBOutlet var todayLabel: UILabel!
  @IBOutlet var scheduleLabel: UILabel!
  @IBOutlet var breaksLabel: UILabel!
  @IBOutlet var closedLabel: UILabel!
  @IBOutlet var arrowImageView: UIImageView!

  var workingDay: WorkingDay!
  var closedNow = false
  var onExpand: MWMVoidBlock?

  override func viewDidLoad() {
    super.viewDidLoad()

    todayLabel.text = workingDay.workingDays
    scheduleLabel.text = workingDay.workingTimes
    breaksLabel.text = workingDay.breaks
    closedLabel.isHidden = !closedNow
  }

  @IBAction func onTap(_ sender: UITapGestureRecognizer) {
    onExpand?()
  }
}

class OpeningHoursDayViewController: UIViewController {
  @IBOutlet var todayLabel: UILabel!
  @IBOutlet var scheduleLabel: UILabel!
  @IBOutlet var breaksLabel: UILabel!

  var workingDay: WorkingDay!

  override func viewDidLoad() {
    super.viewDidLoad()

    todayLabel.text = workingDay.workingDays
    scheduleLabel.text = workingDay.workingTimes
    breaksLabel.text = workingDay.breaks
  }
}

class OpeningHoursViewController: UIViewController {
  @IBOutlet var stackView: UIStackView!

  private var otherDaysViews: [OpeningHoursDayViewController] = []

  private lazy var todayView: OpeningHoursTodayViewController = {
    let vc = storyboard!.instantiateViewController(ofType: OpeningHoursTodayViewController.self)
    vc.workingDay = openingHours.days[0]
    vc.closedNow = openingHours.isClosedNow
    return vc
  }()

  private var expanded = false

  var openingHours: OpeningHours!

  override func viewDidLoad() {
    super.viewDidLoad()

    addToStack(todayView)

    if openingHours.days.count == 1 {
      todayView.arrowImageView.isHidden = true
      return
    }
    
    openingHours.days.suffix(from: 1).forEach {
      let vc = createDayItem($0)
      otherDaysViews.append(vc)
      addToStack(vc)
    }

    todayView.onExpand = { [unowned self] in
      self.expanded = !self.expanded
      UIView.animate(withDuration: kDefaultAnimationDuration) {
        self.otherDaysViews.forEach { vc in
          vc.view.isHidden = !self.expanded
        }
        self.todayView.arrowImageView.transform = self.expanded ? CGAffineTransform(rotationAngle: -CGFloat.pi + 0.01)
                                                                : CGAffineTransform.identity
        self.view.layoutIfNeeded()
      }
    }
  }

  private func createDayItem(_ workingDay: WorkingDay) -> OpeningHoursDayViewController {
    let vc = storyboard!.instantiateViewController(ofType: OpeningHoursDayViewController.self)
    vc.workingDay = workingDay
    vc.view.isHidden = true
    return vc
  }
    
  private func addToStack(_ viewController: UIViewController) {
    addChild(viewController)
    stackView.addArrangedSubview(viewController.view)
    viewController.didMove(toParent: self)
  }
}