97 lines
3.7 KiB
Swift
97 lines
3.7 KiB
Swift
//
|
|
// CDVPNProviderServerRepositoryV3.swift
|
|
// Passepartout
|
|
//
|
|
// Created by Davide De Rosa on 10/26/24.
|
|
// Copyright (c) 2024 Davide De Rosa. All rights reserved.
|
|
//
|
|
// https://github.com/passepartoutvpn
|
|
//
|
|
// This file is part of Passepartout.
|
|
//
|
|
// Passepartout is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// Passepartout is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with Passepartout. If not, see <http://www.gnu.org/licenses/>.
|
|
//
|
|
|
|
import AppData
|
|
import CommonUtils
|
|
import CoreData
|
|
import Foundation
|
|
import PassepartoutKit
|
|
|
|
final class CDVPNProviderServerRepositoryV3: VPNProviderServerRepository {
|
|
private let context: NSManagedObjectContext
|
|
|
|
let providerId: ProviderID
|
|
|
|
init(context: NSManagedObjectContext, providerId: ProviderID) {
|
|
self.context = context
|
|
self.providerId = providerId
|
|
}
|
|
|
|
func availableOptions<Configuration>(for configurationType: Configuration.Type) async throws -> VPNFilterOptions where Configuration: ProviderConfigurationIdentifiable {
|
|
try await context.perform {
|
|
let mapper = DomainMapper()
|
|
|
|
let serversRequest = NSFetchRequest<NSDictionary>(entityName: "CDVPNServerV3")
|
|
serversRequest.predicate = self.providerId.predicate
|
|
serversRequest.resultType = .dictionaryResultType
|
|
serversRequest.returnsDistinctResults = true
|
|
serversRequest.propertiesToFetch = [
|
|
"categoryName",
|
|
"countryCode"
|
|
]
|
|
let serversResults = try serversRequest.execute()
|
|
|
|
var countriesByCategoryName: [String: Set<String>] = [:]
|
|
var countryCodes: Set<String> = []
|
|
serversResults.forEach {
|
|
guard let categoryName = $0.object(forKey: "categoryName") as? String,
|
|
let countryCode = $0.object(forKey: "countryCode") as? String else {
|
|
return
|
|
}
|
|
var codes: Set<String> = countriesByCategoryName[categoryName] ?? []
|
|
codes.insert(countryCode)
|
|
countriesByCategoryName[categoryName] = codes
|
|
countryCodes.insert(countryCode)
|
|
}
|
|
|
|
let presetsRequest = CDVPNPresetV3.fetchRequest()
|
|
presetsRequest.predicate = NSPredicate(
|
|
format: "providerId == %@ AND configurationId == %@", self.providerId.rawValue,
|
|
Configuration.providerConfigurationIdentifier
|
|
)
|
|
let presetsResults = try presetsRequest.execute()
|
|
|
|
return VPNFilterOptions(
|
|
countriesByCategoryName: countriesByCategoryName,
|
|
countryCodes: Set(countryCodes),
|
|
presets: Set(try presetsResults.compactMap {
|
|
try mapper.preset(from: $0)
|
|
})
|
|
)
|
|
}
|
|
}
|
|
|
|
func filteredServers(with parameters: VPNServerParameters?) async throws -> [VPNServer] {
|
|
try await context.perform {
|
|
let request = CDVPNServerV3.fetchRequest()
|
|
request.sortDescriptors = parameters?.sorting.map(\.sortDescriptor)
|
|
request.predicate = parameters?.filters.predicate(for: self.providerId)
|
|
let results = try request.execute()
|
|
let mapper = DomainMapper()
|
|
return try results.compactMap(mapper.server(from:))
|
|
}
|
|
}
|
|
}
|