106 lines
3.8 KiB
Swift
106 lines
3.8 KiB
Swift
//Made by Lumaa
|
|
|
|
import SwiftUI
|
|
import SwiftData
|
|
import WidgetKit
|
|
import AppIntents
|
|
|
|
// MARK: - Account Intents
|
|
|
|
/// Widgets that require to select only an account will use this `ConfigurationIntent`
|
|
struct AccountAppIntent: WidgetConfigurationIntent {
|
|
static var title: LocalizedStringResource = "widget.follow-count"
|
|
static var description = IntentDescription("widget.follow-count.description")
|
|
|
|
@Parameter(title: "widget.select-account")
|
|
var account: AccountEntity?
|
|
}
|
|
|
|
struct AccountGoalAppIntent: WidgetConfigurationIntent {
|
|
static var title: LocalizedStringResource = "widget.follow-goal"
|
|
static var description = IntentDescription("widget.follow-goal.description")
|
|
|
|
@Parameter(title: "widget.select-account")
|
|
var account: AccountEntity?
|
|
|
|
@Parameter(title: "widget.set-goal", default: 1_000)
|
|
var goal: Int
|
|
}
|
|
|
|
struct AccountEntity: AppEntity {
|
|
let client: Client
|
|
let id: String
|
|
let username: String
|
|
/// Bearer token
|
|
let token: OauthToken
|
|
|
|
static var typeDisplayRepresentation: TypeDisplayRepresentation = "account"
|
|
static var defaultQuery = AccountQuery()
|
|
|
|
var displayRepresentation: DisplayRepresentation {
|
|
DisplayRepresentation(stringLiteral: "@\(username)")
|
|
}
|
|
|
|
init(acct: String, username: String, token: OauthToken) {
|
|
self.client = Client(server: String(acct.split(separator: "@")[1]), version: .v2, oauthToken: token)
|
|
self.id = acct
|
|
self.username = username
|
|
self.token = token
|
|
}
|
|
|
|
init(loggedAccount: LoggedAccount) {
|
|
self.client = Client(server: String(loggedAccount.acct.split(separator: "@")[1]), version: .v2, oauthToken: loggedAccount.token)
|
|
self.id = loggedAccount.acct
|
|
self.username = String(loggedAccount.acct.split(separator: "@")[0])
|
|
self.token = loggedAccount.token
|
|
}
|
|
|
|
func toUIImage() -> URL? {
|
|
Task { [self] in
|
|
if let account: String = try? await client.getString(endpoint: Accounts.verifyCredentials) {
|
|
do {
|
|
if let serialized: [String : Any] = try JSONSerialization.jsonObject(with: account.data(using: String.Encoding.utf8) ?? Data()) as? [String : Any] {
|
|
let avatar: String = serialized["avatar"] as! String
|
|
return URL(string: avatar) ?? URL.placeholder
|
|
}
|
|
} catch {
|
|
print("Error fetching image data: \(error)")
|
|
}
|
|
}
|
|
return URL.placeholder
|
|
}
|
|
return URL.placeholder
|
|
}
|
|
}
|
|
|
|
|
|
struct AccountQuery: EntityQuery {
|
|
private static func getAccountsQuery() -> [LoggedAccount] {
|
|
let query = Query<LoggedAccount, [LoggedAccount]>()
|
|
let loggedAccounts: [LoggedAccount] = query.wrappedValue
|
|
return loggedAccounts
|
|
}
|
|
|
|
private static func getAccounts() -> [LoggedAccount] {
|
|
guard let modelContainer: ModelContainer = try? ModelContainer(for: LoggedAccount.self, configurations: ModelConfiguration(isStoredInMemoryOnly: false)) else { return [] }
|
|
let modelContext = ModelContext(modelContainer)
|
|
let loggedAccounts = try? modelContext.fetch(FetchDescriptor<LoggedAccount>())
|
|
|
|
return loggedAccounts ?? []
|
|
}
|
|
|
|
func entities(for identifiers: [AccountEntity.ID]) async throws -> [AccountEntity] {
|
|
let accountEntities: [AccountEntity] = Self.getAccounts().map({ return AccountEntity(loggedAccount: $0) })
|
|
return accountEntities.filter({ identifiers.contains($0.id) })
|
|
}
|
|
|
|
func suggestedEntities() async throws -> [AccountEntity] {
|
|
let accountEntities: [AccountEntity] = Self.getAccounts().map({ return AccountEntity(loggedAccount: $0) })
|
|
return accountEntities
|
|
}
|
|
|
|
func defaultResult() async -> AccountEntity? {
|
|
try? await suggestedEntities().first
|
|
}
|
|
}
|