wehub-resource-sync
53 lines
1.8 KiB
Swift
53 lines
1.8 KiB
Swift
import Foundation
|
|
|
|
public enum OpenAIAPISettingsReader {
|
|
public static let adminAPIKeyEnvironmentKey = "OPENAI_ADMIN_KEY"
|
|
public static let apiKeyEnvironmentKey = "OPENAI_API_KEY"
|
|
public static let projectIDEnvironmentKey = "OPENAI_PROJECT_ID"
|
|
public static let apiKeyEnvironmentKeys = [
|
|
Self.adminAPIKeyEnvironmentKey,
|
|
Self.apiKeyEnvironmentKey,
|
|
]
|
|
|
|
public static func apiKey(environment: [String: String] = ProcessInfo.processInfo.environment) -> String? {
|
|
for key in self.apiKeyEnvironmentKeys {
|
|
if let token = self.cleaned(environment[key]) { return token }
|
|
}
|
|
return nil
|
|
}
|
|
|
|
public static func adminAPIKey(environment: [String: String] = ProcessInfo.processInfo.environment) -> String? {
|
|
self.cleaned(environment[self.adminAPIKeyEnvironmentKey])
|
|
}
|
|
|
|
public static func projectID(environment: [String: String] = ProcessInfo.processInfo.environment) -> String? {
|
|
self.cleaned(environment[self.projectIDEnvironmentKey])
|
|
}
|
|
|
|
static func cleaned(_ raw: String?) -> String? {
|
|
guard var value = raw?.trimmingCharacters(in: .whitespacesAndNewlines), !value.isEmpty else {
|
|
return nil
|
|
}
|
|
|
|
if (value.hasPrefix("\"") && value.hasSuffix("\"")) ||
|
|
(value.hasPrefix("'") && value.hasSuffix("'"))
|
|
{
|
|
value = String(value.dropFirst().dropLast())
|
|
}
|
|
|
|
value = value.trimmingCharacters(in: .whitespacesAndNewlines)
|
|
return value.isEmpty ? nil : value
|
|
}
|
|
}
|
|
|
|
public enum OpenAIAPISettingsError: LocalizedError, Sendable {
|
|
case missingToken
|
|
|
|
public var errorDescription: String? {
|
|
switch self {
|
|
case .missingToken:
|
|
"OpenAI API key not configured. Set OPENAI_API_KEY or configure an API key in Settings."
|
|
}
|
|
}
|
|
}
|