78 lines
2.6 KiB
Swift
78 lines
2.6 KiB
Swift
//
|
|
// LocationMapper.swift
|
|
// Passepartout
|
|
//
|
|
// Created by Davide De Rosa on 3/14/22.
|
|
// Copyright (c) 2022 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 Foundation
|
|
import CoreData
|
|
import PassepartoutServices
|
|
import PassepartoutUtils
|
|
|
|
struct LocationMapper: DTOMapper, ModelMapper {
|
|
private let context: NSManagedObjectContext
|
|
|
|
init(_ context: NSManagedObjectContext) {
|
|
self.context = context
|
|
}
|
|
|
|
func toDTO(_ ws: WSProviderLocation) -> CDInfrastructureLocation {
|
|
let location = CDInfrastructureLocation(context: context)
|
|
location.countryCode = ws.countryCode
|
|
|
|
let servers = ws.servers.compactMap(ServerMapper(context).toDTO)
|
|
servers.forEach {
|
|
$0.location = location
|
|
}
|
|
location.addToServers(Set(servers) as NSSet)
|
|
|
|
return location
|
|
}
|
|
|
|
static func toModel(_ dto: CDInfrastructureLocation) -> ProviderLocation? {
|
|
guard let infrastructureDTO = dto.category?.infrastructure,
|
|
let providerDTO = infrastructureDTO.provider,
|
|
let providerMetadata = ProviderMapper.toModel(providerDTO),
|
|
let vpnProtocolString = infrastructureDTO.vpnProtocol,
|
|
let vpnProtocol = VPNProtocolType(rawValue: vpnProtocolString),
|
|
let categoryName = dto.category?.name,
|
|
let countryCode = dto.countryCode else {
|
|
|
|
Utils.assertCoreDataDecodingFailed(#file, #function, #line)
|
|
return nil
|
|
}
|
|
|
|
var server: ProviderServer?
|
|
if dto.servers?.count == 1, let serverDTO = dto.servers?.anyObject() as? CDInfrastructureServer {
|
|
server = ServerMapper.toModel(serverDTO)
|
|
}
|
|
|
|
return ProviderLocation(
|
|
providerMetadata: providerMetadata,
|
|
vpnProtocol: vpnProtocol,
|
|
categoryName: categoryName,
|
|
countryCode: countryCode,
|
|
onlyServer: server
|
|
)
|
|
}
|
|
}
|