Chat search filters improvements

submodules/ChatListUI/Sources/DateSuggestion.swift

@@ -0,0 +1,144 @@
+import Foundation
+import TelegramPresentationData
+
+private let telegramReleaseDate = Date(timeIntervalSince1970: 1376438400.0)
+
+func suggestDates(for string: String, strings: PresentationStrings, dateTimeFormat: PresentationDateTimeFormat) -> [(Date, String?)] {
+    let string = string.folding(options: .diacriticInsensitive, locale: .current).trimmingCharacters(in: .whitespacesAndNewlines).lowercased()
+    if string.count < 3 {
+        return []
+    }
+    
+    let months: [Int: (String, String)] = [
+        1:  (strings.Month_GenJanuary, strings.Month_ShortJanuary),
+        2:  (strings.Month_GenFebruary, strings.Month_ShortFebruary),
+        3:  (strings.Month_GenMarch, strings.Month_ShortMarch),
+        4:  (strings.Month_GenApril, strings.Month_ShortApril),
+        5:  (strings.Month_GenMay, strings.Month_ShortMay),
+        6:  (strings.Month_GenJune, strings.Month_ShortJune),
+        7:  (strings.Month_GenJuly, strings.Month_ShortJuly),
+        8:  (strings.Month_GenAugust, strings.Month_ShortAugust),
+        9:  (strings.Month_GenSeptember, strings.Month_ShortSeptember),
+        10: (strings.Month_GenOctober, strings.Month_ShortOctober),
+        11: (strings.Month_GenNovember, strings.Month_ShortNovember),
+        12: (strings.Month_GenDecember, strings.Month_ShortDecember)
+    ]
+    
+    let weekDays: [Int: (String, String)] = [
+        1:  (strings.Weekday_Monday, strings.Weekday_ShortMonday),
+        2:  (strings.Weekday_Tuesday, strings.Weekday_ShortTuesday),
+        3:  (strings.Weekday_Wednesday, strings.Weekday_ShortWednesday),
+        4:  (strings.Weekday_Thursday, strings.Weekday_ShortThursday),
+        5:  (strings.Weekday_Friday, strings.Weekday_ShortFriday),
+        6:  (strings.Weekday_Saturday, strings.Weekday_ShortSaturday),
+        7:  (strings.Weekday_Sunday, strings.Weekday_ShortSunday.lowercased()),
+    ]
+   
+    let today = strings.Weekday_Today
+    let yesterday = strings.Weekday_Yesterday
+    let dateSeparator = dateTimeFormat.dateSeparator
+    
+    var result: [(Date, String?)] = []
+    
+    let calendar = Calendar.current
+    func getUpperDate(for date: Date) -> Date {
+        let components = calendar.dateComponents(in: .current, from: date)
+        let upperComponents = DateComponents(year: components.year, month: components.month, day: components.day, hour: 23, minute: 59, second: 59)
+        return calendar.date(from: upperComponents)!
+    }
+    
+    let now = Date()
+    let nowComponents = calendar.dateComponents(in: .current, from: now)
+    guard let year = nowComponents.year else {
+        return []
+    }
+    
+    let midnight = calendar.startOfDay(for: now)
+    if today.lowercased().hasPrefix(string) {
+        let todayDate = getUpperDate(for: midnight)
+        result.append((todayDate, today))
+    }
+    if yesterday.lowercased().hasPrefix(string) {
+        let yesterdayMidnight = calendar.date(byAdding: .day, value: -1, to: midnight)!
+        let yesterdayDate = getUpperDate(for: yesterdayMidnight)
+        result.append((yesterdayDate, yesterday))
+    }
+    
+    func getUpperMonthDate(month: Int, year: Int) -> Date {
+        let monthComponents = DateComponents(year: year, month: month)
+        let date = calendar.date(from: monthComponents)!
+        let range = calendar.range(of: .day, in: .month, for: date)!
+        let numDays = range.count
+        let upperComponents = DateComponents(year: year, month: month, day: numDays, hour: 23, minute: 59, second: 59)
+        return calendar.date(from: upperComponents)!
+    }
+        
+    let decimalRange = string.rangeOfCharacter(from: .decimalDigits)
+    if decimalRange != nil {
+        if string.count == 4, let value = Int(string), value <= year {
+            let date = getUpperMonthDate(month: 12, year: value)
+            if date > telegramReleaseDate {
+                result.append((date, "\(value)"))
+            }
+        } else if !dateSeparator.isEmpty && string.contains(dateSeparator) {
+            let stringComponents = string.components(separatedBy: dateSeparator)
+            if stringComponents.count > 1 {
+                let locale = Locale(identifier: strings.baseLanguageCode)
+                do {
+                    let dd = try NSDataDetector(types: NSTextCheckingResult.CheckingType.date.rawValue)
+                    if let match = dd.firstMatch(in: string, options: [], range: NSMakeRange(0, string.utf16.count)), let date = match.date, date > telegramReleaseDate {
+                        var resultDate = date
+                        if resultDate > now {
+                            if let date = calendar.date(byAdding: .year, value: -1, to: resultDate) {
+                                resultDate = date
+                            }
+                        }
+                        
+                        for i in 0..<5 {
+                            if let date = calendar.date(byAdding: .year, value: -i, to: resultDate) {
+                                result.append((date, nil))
+                            }
+                        }
+                    }
+                } catch {
+                
+                }
+            }
+        }
+    } else {
+        for (day, value) in weekDays {
+            let dayName = value.0.lowercased()
+            let shortDayName = value.1.lowercased()
+            if string == shortDayName || (string.count >= shortDayName.count && dayName.hasPrefix(string)) {
+                var nextDateComponent = calendar.dateComponents([.hour, .minute, .second], from: now)
+                nextDateComponent.weekday = day + calendar.firstWeekday
+                if let date = calendar.nextDate(after: now, matching: nextDateComponent, matchingPolicy: .nextTime, direction: .backward) {
+                    let upperDate = getUpperDate(for: date)
+                    for i in 0..<5 {
+                        if let date = calendar.date(byAdding: .hour, value: -24 * 7 * i, to: upperDate) {
+                            if calendar.isDate(date, equalTo: now, toGranularity: .weekOfYear) {
+                                result.append((date, value.0))
+                            } else {
+                                result.append((date, nil))
+                            }
+                        }
+                    }
+                }
+            }
+        }
+        for (month, value) in months {
+            let monthName = value.0.lowercased()
+            let shortMonthName = value.1.lowercased()
+            if string == shortMonthName || (string.count >= shortMonthName.count && monthName.hasPrefix(string)) {
+                for i in (year - 7 ... year).reversed() {
+                    let date = getUpperMonthDate(month: month, year: i)
+                    if date <= now && date > telegramReleaseDate {
+                        result.append((date, nil))
+                    }
+                }
+            }
+        }
+    }
+    
+    return result
+}