82 lines
2.4 KiB
Swift
82 lines
2.4 KiB
Swift
//
|
|
// HeuristicRanker.swift
|
|
// iris
|
|
//
|
|
// Created by Codex.
|
|
//
|
|
|
|
import Foundation
|
|
|
|
struct UserContext: Equatable {
|
|
var isMoving: Bool
|
|
var city: String
|
|
|
|
init(isMoving: Bool, city: String = "London") {
|
|
self.isMoving = isMoving
|
|
self.city = city
|
|
}
|
|
}
|
|
|
|
final class HeuristicRanker {
|
|
private let nowProvider: () -> Int
|
|
private let lastShownAtProvider: (String) -> Int?
|
|
|
|
init(now: @escaping () -> Int = { Int(Date().timeIntervalSince1970) },
|
|
lastShownAt: @escaping (String) -> Int? = { _ in nil }) {
|
|
self.nowProvider = now
|
|
self.lastShownAtProvider = lastShownAt
|
|
}
|
|
|
|
func pickWinner(from candidates: [Candidate], now: Int? = nil, context: UserContext) -> Winner {
|
|
let currentTime = now ?? nowProvider()
|
|
|
|
let valid = candidates
|
|
.filter { !$0.isExpired(at: currentTime) }
|
|
.filter { $0.confidence >= 0.0 }
|
|
|
|
guard !valid.isEmpty else {
|
|
return WinnerEnvelope.allQuiet(now: currentTime).winner
|
|
}
|
|
|
|
var best: (candidate: Candidate, score: Double)?
|
|
for candidate in valid {
|
|
let baseWeight = baseWeight(for: candidate.type)
|
|
var score = baseWeight * min(max(candidate.confidence, 0.0), 1.0)
|
|
if let shownAt = lastShownAtProvider(candidate.id),
|
|
currentTime - shownAt <= 2 * 60 * 60 {
|
|
score -= 0.4
|
|
}
|
|
if best == nil || score > best!.score {
|
|
best = (candidate, score)
|
|
}
|
|
}
|
|
|
|
guard let best else {
|
|
return WinnerEnvelope.allQuiet(now: currentTime).winner
|
|
}
|
|
|
|
let priority = min(max(best.score, 0.0), 1.0)
|
|
return Winner(
|
|
id: best.candidate.id,
|
|
type: best.candidate.type,
|
|
title: best.candidate.title.truncated(maxLength: TextConstraints.titleMax),
|
|
subtitle: best.candidate.subtitle.truncated(maxLength: TextConstraints.subtitleMax),
|
|
priority: priority,
|
|
ttlSec: max(1, best.candidate.ttlSec)
|
|
)
|
|
}
|
|
|
|
private func baseWeight(for type: WinnerType) -> Double {
|
|
switch type {
|
|
case .weatherWarning: return 1.0
|
|
case .weatherAlert: return 0.9
|
|
case .transit: return 0.75
|
|
case .poiNearby: return 0.6
|
|
case .info: return 0.4
|
|
case .nowPlaying: return 0.25
|
|
case .currentWeather: return 0.0
|
|
case .allQuiet: return 0.0
|
|
}
|
|
}
|
|
}
|