passepartout-apple/Passepartout/Sources/Model/Profiles/ProviderConnectionProfile.swift

//
//  ProviderConnectionProfile.swift
//  Passepartout
//
//  Created by Davide De Rosa on 9/2/18.
//  Copyright (c) 2018 Davide De Rosa. All rights reserved.
//
//  https://github.com/keeshux
//
//  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 TunnelKit

class ProviderConnectionProfile: ConnectionProfile, Codable, Equatable {
    let name: Infrastructure.Name

    var infrastructure: Infrastructure {
        return InfrastructureFactory.shared.get(name)
    }

    var poolId: String {
        didSet {
            validateEndpoint()
        }
    }

    var pool: Pool? {
        return infrastructure.pool(for: poolId)
    }

    var presetId: String {
        didSet {
            validateEndpoint()
        }
    }
    
    var preset: InfrastructurePreset? {
        return infrastructure.preset(for: presetId)
    }
    
    var manualAddress: String?

    var manualProtocol: TunnelKitProvider.EndpointProtocol?
    
    var usesProviderEndpoint: Bool {
        return (manualAddress != nil) || (manualProtocol != nil)
    }
    
    init(name: Infrastructure.Name) {
        self.name = name
        poolId = ""
        presetId = ""

        id = "provider.\(name.rawValue)"
        username = nil

        poolId = infrastructure.defaults.pool
        presetId = infrastructure.defaults.preset
    }
    
    func sortedPools() -> [Pool] {
        return infrastructure.pools.sorted { $0.name < $1.name }
    }
    
    private func validateEndpoint() {
        guard let pool = pool, let preset = preset else {
            manualAddress = nil
            manualProtocol = nil
            return
        }
        if let address = manualAddress, !pool.hasAddress(address) {
            manualAddress = nil
        }
        if let proto = manualProtocol, !preset.hasProtocol(proto) {
            manualProtocol = nil
        }
    }
    
    // MARK: ConnectionProfile
    
    let id: String
    
    var title: String {
        return name.rawValue
    }
    
    var username: String?
    
    var isConfigured: Bool {
        return (pool != nil) && (preset != nil)
    }

    func generate(from configuration: TunnelKitProvider.Configuration, preferences: Preferences) throws -> TunnelKitProvider.Configuration {
        guard let pool = pool else {
            throw ApplicationError.providerPool
        }
        guard let preset = preset else {
            throw ApplicationError.providerPreset
        }

//        assert(!pool.numericAddresses.isEmpty)

        // XXX: copy paste, error prone
        var builder = preset.configuration.builder()
        builder.mtu = configuration.mtu
        builder.shouldDebug = configuration.shouldDebug
        builder.debugLogFormat = configuration.debugLogFormat
        builder.debugLogKey = configuration.debugLogKey

        if let address = manualAddress {
            builder.prefersResolvedAddresses = true
            builder.resolvedAddresses = [address]
        } else {
            builder.prefersResolvedAddresses = !preferences.resolvesHostname
            builder.resolvedAddresses = pool.addresses(sorted: false)
        }
        
        if let proto = manualProtocol {
            builder.endpointProtocols = [proto]
        } else {
            builder.endpointProtocols = preset.configuration.endpointProtocols
//            builder.endpointProtocols = [
//                TunnelKitProvider.EndpointProtocol(.udp, 8080),
//                TunnelKitProvider.EndpointProtocol(.tcp, 443)
//            ]
        }
        return builder.build()
    }
}

extension ProviderConnectionProfile {
    static func ==(lhs: ProviderConnectionProfile, rhs: ProviderConnectionProfile) -> Bool {
        return lhs.id == rhs.id
    }
}

extension ProviderConnectionProfile {
    var mainAddress: String {
        assert(pool != nil, "Getting provider main address but no pool set")
        return pool?.hostname ?? ""
    }
    
    var addresses: [String] {
        return pool?.addresses(sorted: true) ?? []
    }
    
    var protocols: [TunnelKitProvider.EndpointProtocol] {
        return preset?.configuration.endpointProtocols ?? []
    }
    
    var canCustomizeEndpoint: Bool {
        return true
    }
    
    var customAddress: String? {
        return manualAddress
    }

    var customProtocol: TunnelKitProvider.EndpointProtocol? {
        return manualProtocol
    }
}