Abhishek
5 years ago
3 changed files with 219 additions and 188 deletions
@ -0,0 +1,185 @@
|
||||
// Copyright © 2015 Abhishek Banthia |
||||
|
||||
import Cocoa |
||||
|
||||
enum RowType { |
||||
case timezoneHeader |
||||
case cityHeader |
||||
case city |
||||
case timezone |
||||
} |
||||
|
||||
struct TimezoneMetadata { |
||||
let timezone: NSTimeZone |
||||
let tags: Set<String> |
||||
let formattedName: String |
||||
} |
||||
|
||||
class SearchDataSource: NSObject { |
||||
private var searchField: NSSearchField! |
||||
private var finalArray: [RowType] = [] |
||||
private var dataTask: URLSessionDataTask? = .none |
||||
private var timezoneMetadataDictionary: [String: [String]] = |
||||
["IST": ["india", "indian", "kolkata", "calcutta", "mumbai", "delhi", "hyderabad", "noida"], |
||||
"PST": ["los", "los angeles", "california", "san francisco", "bay area", "pacific standard time"], |
||||
"UTC": ["utc", "universal"], |
||||
"EST": ["florida", "new york"]] |
||||
|
||||
var filteredArray: [Any] = [] |
||||
var timezoneArray: [TimezoneMetadata] = [] |
||||
var timezoneFilteredArray: [TimezoneMetadata] = [] |
||||
|
||||
init(with searchField: NSSearchField) { |
||||
super.init() |
||||
self.searchField = searchField |
||||
setupTimezoneDatasource() |
||||
calculateArray() |
||||
} |
||||
|
||||
func cleanupFilterArray() { |
||||
filteredArray = [] |
||||
} |
||||
|
||||
func setFilteredArrayValue(_ newArray: [Any]) { |
||||
filteredArray = newArray |
||||
} |
||||
|
||||
func placeForRow(_ row: Int) -> RowType { |
||||
return finalArray[row] |
||||
} |
||||
|
||||
private func setupTimezoneDatasource() { |
||||
timezoneArray = [] |
||||
|
||||
let anywhereOnEarth = TimezoneMetadata(timezone: NSTimeZone(abbreviation: "GMT-1200")!, |
||||
tags: ["aoe", "anywhere on earth"], |
||||
formattedName: "Anywhere on Earth") |
||||
timezoneArray.append(anywhereOnEarth) |
||||
|
||||
for (abbreviation, timezone) in TimeZone.abbreviationDictionary { |
||||
var tags: Set<String> = [abbreviation.lowercased(), timezone.lowercased()] |
||||
var extraTags: [String] = [] |
||||
if let tagsPresent = timezoneMetadataDictionary[abbreviation] { |
||||
extraTags = tagsPresent |
||||
} |
||||
|
||||
extraTags.forEach { tag in |
||||
tags.insert(tag) |
||||
} |
||||
|
||||
let timezoneIdentifier = NSTimeZone(name: timezone)! |
||||
let formattedName = timezone + " (\(abbreviation))" |
||||
let timezoneMetadata = TimezoneMetadata(timezone: timezoneIdentifier, |
||||
tags: tags, |
||||
formattedName: formattedName) |
||||
timezoneArray.append(timezoneMetadata) |
||||
} |
||||
|
||||
print(TimeZone.knownTimeZoneIdentifiers.count) |
||||
print(timezoneArray.count) |
||||
} |
||||
|
||||
func calculateArray() { |
||||
finalArray = [] |
||||
|
||||
func addTimezonesIfNeeded(_ data: [TimezoneMetadata]) { |
||||
if !data.isEmpty { |
||||
finalArray.append(.timezoneHeader) |
||||
} |
||||
data.forEach { _ in |
||||
finalArray.append(.timezone) |
||||
} |
||||
} |
||||
|
||||
if searchField.stringValue.isEmpty { |
||||
addTimezonesIfNeeded(timezoneArray) |
||||
} else { |
||||
if !filteredArray.isEmpty { |
||||
finalArray.append(.cityHeader) |
||||
} |
||||
filteredArray.forEach { _ in |
||||
finalArray.append(.city) |
||||
} |
||||
addTimezonesIfNeeded(timezoneFilteredArray) |
||||
} |
||||
} |
||||
} |
||||
|
||||
extension SearchDataSource: NSTableViewDataSource { |
||||
func numberOfRows(in _: NSTableView) -> Int { |
||||
return finalArray.count |
||||
} |
||||
|
||||
func tableView(_ tableView: NSTableView, viewFor _: NSTableColumn?, row: Int) -> NSView? { |
||||
let currentRowType = finalArray[row] |
||||
|
||||
switch currentRowType { |
||||
case .timezoneHeader, .cityHeader: |
||||
return headerCell(tableView, currentRowType) |
||||
case .timezone: |
||||
return timezoneCell(tableView, currentRowType, row) |
||||
case .city: |
||||
return cityCell(tableView, currentRowType, row) |
||||
} |
||||
} |
||||
} |
||||
|
||||
extension SearchDataSource: NSTableViewDelegate { |
||||
func tableView(_: NSTableView, isGroupRow row: Int) -> Bool { |
||||
let currentRowType = finalArray[row] |
||||
return |
||||
currentRowType == .timezoneHeader || |
||||
currentRowType == .cityHeader |
||||
} |
||||
|
||||
func tableView(_: NSTableView, shouldSelectRow row: Int) -> Bool { |
||||
let currentRowType = finalArray[row] |
||||
return !(currentRowType == .timezoneHeader || currentRowType == .cityHeader) |
||||
} |
||||
} |
||||
|
||||
extension SearchDataSource { |
||||
private func timezoneCell(_ tableView: NSTableView, _: RowType, _ row: Int) -> NSView? { |
||||
if let message = tableView.makeView(withIdentifier: NSUserInterfaceItemIdentifier(rawValue: "resultCell"), owner: self) as? SearchResultTableViewCell { |
||||
let datasource = searchField.stringValue.isEmpty ? timezoneArray : timezoneFilteredArray |
||||
guard !datasource.isEmpty else { |
||||
return nil |
||||
} |
||||
let index = searchField.stringValue.isEmpty ? row - 1 : row |
||||
message.sourceName.stringValue = datasource[index % datasource.count].formattedName |
||||
return message |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
private func cityCell(_ tableView: NSTableView, _: RowType, _ row: Int) -> NSView? { |
||||
if let cityCell = tableView.makeView(withIdentifier: NSUserInterfaceItemIdentifier(rawValue: "resultCell"), owner: self) as? SearchResultTableViewCell { |
||||
guard let timezoneData = filteredArray[row % filteredArray.count] as? TimezoneData else { |
||||
assertionFailure() |
||||
return nil |
||||
} |
||||
|
||||
cityCell.sourceName.stringValue = timezoneData.formattedAddress ?? "Error" |
||||
return cityCell |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
private func headerCell(_ tableView: NSTableView, _ headerType: RowType) -> NSView? { |
||||
if let message = tableView.makeView(withIdentifier: NSUserInterfaceItemIdentifier(rawValue: "headerCell"), owner: self) as? HeaderTableViewCell { |
||||
message.headerField.stringValue = headerType == .timezoneHeader ? "Timezones" : "Places" |
||||
return message |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
} |
||||
|
||||
class SearchResultTableViewCell: NSTableCellView { |
||||
@IBOutlet var sourceName: NSTextField! |
||||
} |
||||
|
||||
class HeaderTableViewCell: NSTableCellView { |
||||
@IBOutlet var headerField: NSTextField! |
||||
} |
||||
Loading…
Reference in new issue