Add Yahoo Finance stock data integration
- Add StockDataSource to fetch quotes from Yahoo Finance API - Add StockSettingsStore for persisting user's stock symbols - Add StockSettingsView with UI to manage symbols (max 5) - Add STOCK feed item type and ranker weight (0.3) - Integrate stock fetch into ContextOrchestrator pipeline - Stock cards appear in FYI bucket and sync to Glass via BLE 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
186
IrisCompanion/iris/DataSources/StockDataSource.swift
Normal file
186
IrisCompanion/iris/DataSources/StockDataSource.swift
Normal file
@@ -0,0 +1,186 @@
|
||||
//
|
||||
// StockDataSource.swift
|
||||
// iris
|
||||
//
|
||||
|
||||
import Foundation
|
||||
|
||||
struct StockDataSourceConfig: Sendable {
|
||||
var maxSymbols: Int = 5
|
||||
var cacheValiditySec: Int = 300
|
||||
var ttlSec: Int = 600
|
||||
|
||||
init() {}
|
||||
}
|
||||
|
||||
final class StockDataSource {
|
||||
struct StockQuote: Sendable, Equatable {
|
||||
let symbol: String
|
||||
let shortName: String
|
||||
let price: Double
|
||||
let change: Double
|
||||
let changePercent: Double
|
||||
let marketState: String
|
||||
}
|
||||
|
||||
struct StockData: Sendable, Equatable {
|
||||
let quotes: [StockQuote]
|
||||
}
|
||||
|
||||
struct Snapshot: Sendable {
|
||||
let data: StockData
|
||||
let diagnostics: [String: String]
|
||||
}
|
||||
|
||||
enum StockError: Error, LocalizedError, Sendable {
|
||||
case noSymbolsConfigured
|
||||
case networkFailed(message: String, diagnostics: [String: String])
|
||||
case rateLimited(diagnostics: [String: String])
|
||||
case invalidResponse(diagnostics: [String: String])
|
||||
|
||||
var errorDescription: String? {
|
||||
switch self {
|
||||
case .noSymbolsConfigured:
|
||||
return "No symbols configured"
|
||||
case .networkFailed(let message, _):
|
||||
return message
|
||||
case .rateLimited:
|
||||
return "Rate limited by Yahoo Finance"
|
||||
case .invalidResponse:
|
||||
return "Invalid response from Yahoo Finance"
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private let config: StockDataSourceConfig
|
||||
private var cache: (timestamp: Int, data: StockData)?
|
||||
|
||||
init(config: StockDataSourceConfig = .init()) {
|
||||
self.config = config
|
||||
}
|
||||
|
||||
func dataWithDiagnostics(symbols: [String], now: Int) async throws -> Snapshot {
|
||||
var diagnostics: [String: String] = [
|
||||
"now": String(now),
|
||||
"symbols_requested": symbols.joined(separator: ","),
|
||||
"max_symbols": String(config.maxSymbols),
|
||||
]
|
||||
|
||||
guard !symbols.isEmpty else {
|
||||
diagnostics["result"] = "no_symbols"
|
||||
return Snapshot(data: StockData(quotes: []), diagnostics: diagnostics)
|
||||
}
|
||||
|
||||
let limitedSymbols = Array(symbols.prefix(config.maxSymbols))
|
||||
diagnostics["symbols_queried"] = limitedSymbols.joined(separator: ",")
|
||||
|
||||
if let cache = cache, now - cache.timestamp < config.cacheValiditySec {
|
||||
diagnostics["source"] = "cache"
|
||||
diagnostics["cache_age_sec"] = String(now - cache.timestamp)
|
||||
return Snapshot(data: cache.data, diagnostics: diagnostics)
|
||||
}
|
||||
|
||||
let symbolsParam = limitedSymbols.joined(separator: ",")
|
||||
guard let encodedSymbols = symbolsParam.addingPercentEncoding(withAllowedCharacters: .urlQueryAllowed),
|
||||
let url = URL(string: "https://query1.finance.yahoo.com/v7/finance/quote?symbols=\(encodedSymbols)") else {
|
||||
diagnostics["error"] = "invalid_url"
|
||||
throw StockError.networkFailed(message: "Invalid URL", diagnostics: diagnostics)
|
||||
}
|
||||
|
||||
var request = URLRequest(url: url)
|
||||
request.setValue("Mozilla/5.0 (iPhone; CPU iPhone OS 17_0 like Mac OS X)", forHTTPHeaderField: "User-Agent")
|
||||
request.setValue("application/json", forHTTPHeaderField: "Accept")
|
||||
|
||||
let data: Data
|
||||
let response: URLResponse
|
||||
do {
|
||||
(data, response) = try await URLSession.shared.data(for: request)
|
||||
} catch {
|
||||
diagnostics["error"] = "network_error"
|
||||
diagnostics["error_message"] = String(describing: error)
|
||||
if let cached = cache {
|
||||
diagnostics["source"] = "cache_fallback"
|
||||
return Snapshot(data: cached.data, diagnostics: diagnostics)
|
||||
}
|
||||
throw StockError.networkFailed(message: String(describing: error), diagnostics: diagnostics)
|
||||
}
|
||||
|
||||
if let httpResponse = response as? HTTPURLResponse {
|
||||
diagnostics["http_status"] = String(httpResponse.statusCode)
|
||||
if httpResponse.statusCode == 403 {
|
||||
diagnostics["error"] = "rate_limited"
|
||||
if let cached = cache {
|
||||
diagnostics["source"] = "cache_fallback"
|
||||
return Snapshot(data: cached.data, diagnostics: diagnostics)
|
||||
}
|
||||
throw StockError.rateLimited(diagnostics: diagnostics)
|
||||
}
|
||||
guard (200..<300).contains(httpResponse.statusCode) else {
|
||||
diagnostics["error"] = "http_error"
|
||||
if let cached = cache {
|
||||
diagnostics["source"] = "cache_fallback"
|
||||
return Snapshot(data: cached.data, diagnostics: diagnostics)
|
||||
}
|
||||
throw StockError.networkFailed(message: "HTTP \(httpResponse.statusCode)", diagnostics: diagnostics)
|
||||
}
|
||||
}
|
||||
|
||||
let yahooResponse: YahooQuoteResponse
|
||||
do {
|
||||
yahooResponse = try JSONDecoder().decode(YahooQuoteResponse.self, from: data)
|
||||
} catch {
|
||||
diagnostics["error"] = "json_decode_error"
|
||||
diagnostics["error_message"] = String(describing: error)
|
||||
if let cached = cache {
|
||||
diagnostics["source"] = "cache_fallback"
|
||||
return Snapshot(data: cached.data, diagnostics: diagnostics)
|
||||
}
|
||||
throw StockError.invalidResponse(diagnostics: diagnostics)
|
||||
}
|
||||
|
||||
let quotes: [StockQuote] = (yahooResponse.quoteResponse.result ?? []).compactMap { quote in
|
||||
guard let price = quote.regularMarketPrice else { return nil }
|
||||
return StockQuote(
|
||||
symbol: quote.symbol,
|
||||
shortName: quote.shortName ?? quote.symbol,
|
||||
price: price,
|
||||
change: quote.regularMarketChange ?? 0,
|
||||
changePercent: quote.regularMarketChangePercent ?? 0,
|
||||
marketState: quote.marketState ?? "CLOSED"
|
||||
)
|
||||
}
|
||||
|
||||
diagnostics["source"] = "network"
|
||||
diagnostics["quotes_returned"] = String(quotes.count)
|
||||
|
||||
let stockData = StockData(quotes: quotes)
|
||||
cache = (timestamp: now, data: stockData)
|
||||
|
||||
return Snapshot(data: stockData, diagnostics: diagnostics)
|
||||
}
|
||||
}
|
||||
|
||||
// MARK: - Yahoo Finance API Response Models
|
||||
|
||||
private struct YahooQuoteResponse: Codable {
|
||||
let quoteResponse: QuoteResponse
|
||||
|
||||
struct QuoteResponse: Codable {
|
||||
let result: [Quote]?
|
||||
let error: QuoteError?
|
||||
}
|
||||
|
||||
struct Quote: Codable {
|
||||
let symbol: String
|
||||
let shortName: String?
|
||||
let regularMarketPrice: Double?
|
||||
let regularMarketChange: Double?
|
||||
let regularMarketChangePercent: Double?
|
||||
let marketState: String?
|
||||
}
|
||||
|
||||
struct QuoteError: Codable {
|
||||
let code: String?
|
||||
let description: String?
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user