/*
 * Copyright 2019, gRPC Authors All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
import Commander
import Dispatch
import Foundation
import NIO
import NIOSSL
import GRPC
import GRPCSampleData

// Common flags and options
let sslFlag = Flag("ssl", description: "if true, use SSL for connections")
func addressOption(_ address: String) -> Option<String> {
  return Option("address", default: address, description: "address of server")
}

let portOption = Option("port", default: 8080)
let messageOption = Option("message",
                           default: "Testing 1 2 3",
                           description: "message to send")

func makeClientSSLContext() throws -> NIOSSLContext {
  return try NIOSSLContext(configuration: makeClientTLSConfiguration())
}

func makeServerTLS(enabled: Bool) throws -> GRPCServer.TLSMode {
  guard enabled else {
    return .none
  }
  return .custom(try NIOSSLContext(configuration: makeServerTLSConfiguration()))
}

func makeClientTLSConfiguration() -> TLSConfiguration {
  let caCert = SampleCertificate.ca
  let clientCert = SampleCertificate.client
  precondition(!caCert.isExpired && !clientCert.isExpired,
               "SSL certificates are expired. Please submit an issue at https://github.com/grpc/grpc-swift.")

  return .forClient(certificateVerification: .noHostnameVerification,
                    trustRoots: .certificates([caCert.certificate]),
                    certificateChain: [.certificate(clientCert.certificate)],
                    privateKey: .privateKey(SamplePrivateKey.client),
                    applicationProtocols: GRPCApplicationProtocolIdentifier.allCases.map { $0.rawValue })
}

func makeServerTLSConfiguration() -> TLSConfiguration {
  let caCert = SampleCertificate.ca
  let serverCert = SampleCertificate.server
  precondition(!caCert.isExpired && !serverCert.isExpired,
               "SSL certificates are expired. Please submit an issue at https://github.com/grpc/grpc-swift.")

  return .forServer(certificateChain: [.certificate(serverCert.certificate)],
                    privateKey: .privateKey(SamplePrivateKey.server),
                    trustRoots: .certificates([caCert.certificate]),
                    applicationProtocols: GRPCApplicationProtocolIdentifier.allCases.map { $0.rawValue })
}

/// Create en `EchoClient` and wait for it to initialize. Returns nil if initialisation fails.
func makeEchoClient(address: String, port: Int, ssl: Bool) -> Echo_EchoServiceClient? {
  let eventLoopGroup = MultiThreadedEventLoopGroup(numberOfThreads: 1)

  do {
    let tlsConfiguration: GRPCClientConnection.TLSConfiguration?
    if ssl {
      tlsConfiguration = .init(sslContext: try makeClientSSLContext())
    } else {
      tlsConfiguration = nil
    }

    let configuration = GRPCClientConnection.Configuration(
      target: .hostAndPort(address, port),
      eventLoopGroup: eventLoopGroup,
      tlsConfiguration: tlsConfiguration)

    return try GRPCClientConnection.start(configuration)
      .map { Echo_EchoServiceClient(connection: $0) }
      .wait()
  } catch {
    print("Unable to create an EchoClient: \(error)")
    return nil
  }
}

Group {
  $0.command("serve",
             sslFlag,
             addressOption("localhost"),
             portOption,
             description: "Run an echo server.") { ssl, address, port in
    let sem = DispatchSemaphore(value: 0)
    let eventLoopGroup = MultiThreadedEventLoopGroup(numberOfThreads: 1)

    print(ssl ? "starting secure server" : "starting insecure server")
    _ = try! GRPCServer.start(hostname: address,
                              port: port,
                              eventLoopGroup: eventLoopGroup,
                              serviceProviders: [EchoProvider()],
                              tls: makeServerTLS(enabled: ssl))
      .wait()

    // This blocks to keep the main thread from finishing while the server runs,
    // but the server never exits. Kill the process to stop it.
    _ = sem.wait()
  }

  $0.command(
    "get",
    sslFlag,
    addressOption("localhost"),
    portOption,
    messageOption,
    description: "Perform a unary get()."
  ) { ssl, address, port, message in
    print("calling get")
    guard let echo = makeEchoClient(address: address, port: port, ssl: ssl) else { return }

    var requestMessage = Echo_EchoRequest()
    requestMessage.text = message

    print("get sending: \(requestMessage.text)")
    let get = echo.get(requestMessage)
    get.response.whenSuccess { response in
      print("get received: \(response.text)")
    }

    get.response.whenFailure { error in
      print("get response failed with error: \(error)")
    }

    // wait() on the status to stop the program from exiting.
    do {
      let status = try get.status.wait()
      print("get completed with status: \(status)")
    } catch {
      print("get status failed with error: \(error)")
    }
  }

  $0.command(
    "expand",
    sslFlag,
    addressOption("localhost"),
    portOption,
    messageOption,
    description: "Perform a server-streaming expand()."
  ) { ssl, address, port, message in
    print("calling expand")
    guard let echo = makeEchoClient(address: address, port: port, ssl: ssl) else { return }

    let requestMessage = Echo_EchoRequest.with { $0.text = message }

    print("expand sending: \(requestMessage.text)")
    let expand = echo.expand(requestMessage) { response in
      print("expand received: \(response.text)")
    }

    // wait() on the status to stop the program from exiting.
    do {
      let status = try expand.status.wait()
      print("expand completed with status: \(status)")
    } catch {
      print("expand status failed with error: \(error)")
    }
  }

  $0.command(
    "collect",
    sslFlag,
    addressOption("localhost"),
    portOption,
    messageOption,
    description: "Perform a client-streaming collect()."
  ) { ssl, address, port, message in
    print("calling collect")
    guard let echo = makeEchoClient(address: address, port: port, ssl: ssl) else { return }

    let collect = echo.collect()

    var queue = collect.newMessageQueue()
    for part in message.components(separatedBy: " ") {
      var requestMessage = Echo_EchoRequest()
      requestMessage.text = part
      print("collect sending: \(requestMessage.text)")
      queue = queue.flatMap { collect.sendMessage(requestMessage) }
    }
    queue.whenSuccess { collect.sendEnd(promise: nil) }

    collect.response.whenSuccess { respone in
      print("collect received: \(respone.text)")
    }

    collect.response.whenFailure { error in
      print("collect response failed with error: \(error)")
    }

    // wait() on the status to stop the program from exiting.
    do {
      let status = try collect.status.wait()
      print("collect completed with status: \(status)")
    } catch {
      print("collect status failed with error: \(error)")
    }
  }

  $0.command(
    "update",
    sslFlag,
    addressOption("localhost"),
    portOption,
    messageOption,
    description: "Perform a bidirectional-streaming update()."
  ) { ssl, address, port, message in
    print("calling update")
    guard let echo = makeEchoClient(address: address, port: port, ssl: ssl) else { return }

    let update = echo.update { response in
      print("update received: \(response.text)")
    }

    var queue = update.newMessageQueue()
    for part in message.components(separatedBy: " ") {
      var requestMessage = Echo_EchoRequest()
      requestMessage.text = part
      print("update sending: \(requestMessage.text)")
      queue = queue.flatMap { update.sendMessage(requestMessage) }
    }
    queue.whenSuccess { update.sendEnd(promise: nil) }

    // wait() on the status to stop the program from exiting.
    do {
      let status = try update.status.wait()
      print("update completed with status: \(status)")
    } catch {
      print("update status failed with error: \(error)")
    }
  }
}.run()