70 lines
2.7 KiB
Swift
70 lines
2.7 KiB
Swift
import Foundation
|
|
|
|
/// Builder for the MCMS list-endpoint query parameters.
|
|
///
|
|
/// Supported by most list endpoints: `query`, `projection`, `sort`, `limit`
|
|
/// (server default 1000), `skip`, `next` (cursor = a document `_id`), `distinct`.
|
|
/// Stats/log endpoints additionally require `start-time` and accept `end-time`
|
|
/// (UTC `yyyy-MM-dd HH:mm:ss`).
|
|
struct APIQuery {
|
|
/// Custom Mongo query as `key=value` pairs (joined with commas by the API).
|
|
var query: [String: String] = [:]
|
|
/// Field inclusion/exclusion: `1` include, `0` exclude.
|
|
var projection: [String: Int] = [:]
|
|
/// Sort field → direction (`1` asc, `-1` desc).
|
|
var sort: [String: Int] = [:]
|
|
var limit: Int?
|
|
var skip: Int?
|
|
/// Pagination cursor: excludes all items at/before this `_id`.
|
|
var next: String?
|
|
/// Attribute to fetch unique values for.
|
|
var distinct: String?
|
|
var startTime: Date?
|
|
var endTime: Date?
|
|
|
|
static let empty = APIQuery()
|
|
|
|
private static let timeFormatter: DateFormatter = {
|
|
let f = DateFormatter()
|
|
f.locale = Locale(identifier: "en_US_POSIX")
|
|
f.timeZone = TimeZone(identifier: "UTC")
|
|
f.dateFormat = "yyyy-MM-dd HH:mm:ss"
|
|
return f
|
|
}()
|
|
|
|
func queryItems() -> [URLQueryItem] {
|
|
var items: [URLQueryItem] = []
|
|
|
|
if !query.isEmpty {
|
|
let joined = query.map { "\($0)=\($1)" }.joined(separator: ",")
|
|
items.append(.init(name: "query", value: joined))
|
|
}
|
|
if !projection.isEmpty {
|
|
let joined = projection.map { "\($0)=\($1)" }.joined(separator: ",")
|
|
items.append(.init(name: "projection", value: joined))
|
|
}
|
|
if !sort.isEmpty {
|
|
let joined = sort.map { "\($0)=\($1)" }.joined(separator: ",")
|
|
items.append(.init(name: "sort", value: joined))
|
|
}
|
|
if let limit { items.append(.init(name: "limit", value: "\(limit)")) }
|
|
if let skip { items.append(.init(name: "skip", value: "\(skip)")) }
|
|
if let next { items.append(.init(name: "next", value: next)) }
|
|
if let distinct { items.append(.init(name: "distinct", value: distinct)) }
|
|
if let startTime {
|
|
items.append(.init(name: "start-time", value: Self.timeFormatter.string(from: startTime)))
|
|
}
|
|
if let endTime {
|
|
items.append(.init(name: "end-time", value: Self.timeFormatter.string(from: endTime)))
|
|
}
|
|
return items
|
|
}
|
|
|
|
/// Convenience for a recent time window on stats/log endpoints.
|
|
static func window(lastHours hours: Int, limit: Int? = nil) -> APIQuery {
|
|
var q = APIQuery()
|
|
q.startTime = Calendar.current.date(byAdding: .hour, value: -hours, to: Date())
|
|
q.limit = limit
|
|
return q
|
|
}
|
|
}
|