Collect.swift

/*
 * Copyright 2025, 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 ArgumentParser
import GRPCCore
import GRPCNIOTransportHTTP2

struct Collect: AsyncParsableCommand {
  static let configuration = CommandConfiguration(
    abstract: "Makes a client streaming RPC to the echo-metadata server."
  )

  @OptionGroup
  var arguments: ClientArguments

  func run() async throws {
    try await withGRPCClient(
      transport: .http2NIOPosix(
        target: self.arguments.target,
        transportSecurity: .plaintext
      )
    ) { client in
      let echo = Echo_Echo.Client(wrapping: client)
      let requestMetadata: Metadata = ["echo-message": "\(arguments.message)"]

      print("collect → metadata: \(requestMetadata)")
      try await echo.collect(metadata: requestMetadata) { writer in
        for part in self.arguments.message.split(separator: " ") {
          print("collect → \(part)")
          try await writer.write(.with { $0.text = String(part) })
        }
      } onResponse: { response in
        let initialMetadata = Metadata(response.metadata.filter({ $0.key.starts(with: "echo-") }))
        print("collect ← initial metadata: \(initialMetadata)")

        print("collect ← message: \(try response.message.text)")

        let trailingMetadata = Metadata(
          response.trailingMetadata.filter({ $0.key.starts(with: "echo-") })
        )
        print("collect ← trailing metadata: \(trailingMetadata)")
      }
    }
  }
}