import Foundation #if canImport(FoundationNetworking)import FoundationNetworking#endif public struct PerplexityUsageFetcher: Sendable { private static let log = CodexBarLog.logger(LogCategories.perplexityAPI) private static let creditsURL = URL(string: "https://www.perplexity.ai/rest/billing/credits?version=2.18&source=default")! @TaskLocal static var fetchCreditsOverride: (@Sendable (String, String, Date) async throws -> PerplexityUsageSnapshot)? /// Testing hook: parse a raw JSON response without making network calls. public static func _parseResponseForTesting(_ data: Data, now: Date = Date()) throws -> PerplexityUsageSnapshot { do { let decoded = try JSONDecoder().decode(PerplexityCreditsResponse.self, from: data) return PerplexityUsageSnapshot(response: decoded, now: now) } catch { throw PerplexityAPIError.parseFailed(error.localizedDescription) } } public static func fetchCredits( sessionToken: String, cookieName: String = PerplexityCookieHeader.defaultSessionCookieName, now: Date = Date()) async throws -> PerplexityUsageSnapshot { if let override = self.fetchCreditsOverride { return try await override(sessionToken, cookieName, now) } var request = URLRequest(url: self.creditsURL) request.httpMethod = "GET" request.timeoutInterval = 15 request.setValue("application/json", forHTTPHeaderField: "Accept") request.setValue( "\(cookieName)=\(sessionToken)", forHTTPHeaderField: "Cookie") request.setValue("https://www.perplexity.ai", forHTTPHeaderField: "Origin") request.setValue("https://www.perplexity.ai/account/usage", forHTTPHeaderField: "Referer") let userAgent = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) " + "AppleWebKit/537.36 (KHTML, like Gecko) Chrome/143.0.0.0 Safari/537.36" request.setValue(userAgent, forHTTPHeaderField: "User-Agent") let response = try await ProviderHTTPClient.shared.response(for: request) let data = response.data guard response.statusCode == 200 else { let statusCode = response.statusCode let body = String(data: data, encoding: .utf8) ?? "<binary>" let truncated = body.count > 200 ? String(body.prefix(200)) + "…" : body Self.log.error("Perplexity API returned \(statusCode): \(truncated)") if statusCode == 401 || statusCode == 403 { throw PerplexityAPIError.invalidToken } throw PerplexityAPIError.apiError("HTTP \(statusCode)") } do { let decoded = try JSONDecoder().decode(PerplexityCreditsResponse.self, from: data) let snapshot = PerplexityUsageSnapshot(response: decoded, now: now) Self.log.debug( "Perplexity credits parsed balance=\(snapshot.balanceCents) totalUsage=\(snapshot.totalUsageCents)") return snapshot } catch { let preview = String(data: data.prefix(500), encoding: .utf8) ?? "<binary>" Self.log.error("Perplexity parse failed: \(error) — response: \(preview)") throw PerplexityAPIError.parseFailed(error.localizedDescription) } }}
import Foundation
#if canImport(FoundationNetworking)
import FoundationNetworking
#endif
public struct PerplexityUsageFetcher: Sendable {
private static let log = CodexBarLog.logger(LogCategories.perplexityAPI)
private static let creditsURL =
URL(string: "https://www.perplexity.ai/rest/billing/credits?version=2.18&source=default")!
@TaskLocal static var fetchCreditsOverride:
(@Sendable (String, String, Date) async throws -> PerplexityUsageSnapshot)?
/// Testing hook: parse a raw JSON response without making network calls.
public static func _parseResponseForTesting(_ data: Data, now: Date = Date()) throws -> PerplexityUsageSnapshot {
do {
let decoded = try JSONDecoder().decode(PerplexityCreditsResponse.self, from: data)
return PerplexityUsageSnapshot(response: decoded, now: now)
} catch {
throw PerplexityAPIError.parseFailed(error.localizedDescription)
}
}
public static func fetchCredits(
sessionToken: String,
cookieName: String = PerplexityCookieHeader.defaultSessionCookieName,
now: Date = Date()) async throws -> PerplexityUsageSnapshot
{
if let override = self.fetchCreditsOverride {
return try await override(sessionToken, cookieName, now)
}
var request = URLRequest(url: self.creditsURL)
request.httpMethod = "GET"
request.timeoutInterval = 15
request.setValue("application/json", forHTTPHeaderField: "Accept")
request.setValue(
"\(cookieName)=\(sessionToken)",
forHTTPHeaderField: "Cookie")
request.setValue("https://www.perplexity.ai", forHTTPHeaderField: "Origin")
request.setValue("https://www.perplexity.ai/account/usage", forHTTPHeaderField: "Referer")
let userAgent = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) " +
"AppleWebKit/537.36 (KHTML, like Gecko) Chrome/143.0.0.0 Safari/537.36"
request.setValue(userAgent, forHTTPHeaderField: "User-Agent")
let response = try await ProviderHTTPClient.shared.response(for: request)
let data = response.data
guard response.statusCode == 200 else {
let statusCode = response.statusCode
let body = String(data: data, encoding: .utf8) ?? "<binary>"
let truncated = body.count > 200 ? String(body.prefix(200)) + "…" : body
Self.log.error("Perplexity API returned \(statusCode): \(truncated)")
if statusCode == 401 || statusCode == 403 {
throw PerplexityAPIError.invalidToken
}
throw PerplexityAPIError.apiError("HTTP \(statusCode)")
}
do {
let decoded = try JSONDecoder().decode(PerplexityCreditsResponse.self, from: data)
let snapshot = PerplexityUsageSnapshot(response: decoded, now: now)
Self.log.debug(
"Perplexity credits parsed balance=\(snapshot.balanceCents) totalUsage=\(snapshot.totalUsageCents)")
return snapshot
} catch {
let preview = String(data: data.prefix(500), encoding: .utf8) ?? "<binary>"
Self.log.error("Perplexity parse failed: \(error) — response: \(preview)")
throw PerplexityAPIError.parseFailed(error.localizedDescription)
}
}
}