/*
 * 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 Logging
import NIOCore

/// Keys for `Logger` metadata.
enum MetadataKey {
  static let requestID = "grpc_request_id"
  static let connectionID = "grpc_connection_id"

  static let eventLoop = "event_loop"

  static let h2StreamID = "h2_stream_id"
  static let h2ActiveStreams = "h2_active_streams"
  static let h2EndStream = "h2_end_stream"
  static let h2Payload = "h2_payload"
  static let h2Headers = "h2_headers"
  static let h2DataBytes = "h2_data_bytes"
  static let h2GoAwayError = "h2_goaway_error"
  static let h2GoAwayLastStreamID = "h2_goaway_last_stream_id"
  static let h2PingAck = "h2_ping_ack"

  static let delayMs = "delay_ms"
  static let intervalMs = "interval_ms"

  static let error = "error"
}

extension Logger {
  internal mutating func addIPAddressMetadata(local: SocketAddress?, remote: SocketAddress?) {
    if let local = local?.ipAddress {
      self[metadataKey: "grpc.conn.addr_local"] = "\(local)"
    }
    if let remote = remote?.ipAddress {
      self[metadataKey: "grpc.conn.addr_remote"] = "\(remote)"
    }
  }
}