diff --git a/Calendar/CalendarKit/CalendarView.swift b/Calendar/CalendarKit/CalendarView.swift index 8514f0a..216d728 100644 --- a/Calendar/CalendarKit/CalendarView.swift +++ b/Calendar/CalendarKit/CalendarView.swift @@ -27,11 +27,22 @@ class CalendarView: UIView, UICollectionViewDataSource, UICollectionViewDelegate var baseDate: Date? { didSet { collectionData = [CalendarLogic]() - if let date = baseDate { - var dateIter1 = date, dateIter2 = date + + let df = DateFormatter() + df.dateFormat = "yyyy-MM-dd" + + let str = df.string(from: baseDate!) + + let dateNew = df.date(from: str) + + if let date = dateNew { + + var dateIter1 = date + + var dateIter2 = date var set = Set() - set.insert(CalendarLogic(date: baseDate!)) + set.insert(CalendarLogic(date: dateNew!)) // advance one year (0..