src/main/scala/com/getbootstrap/no_carrier/http/UserAgentWire.scala


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

package com.getbootstrap.no_carrier.http

import java.util.{Collection=>JavaCollection}
import java.util.Map.{Entry=>MapEntry}
import java.io.InputStream
import scala.collection.JavaConverters._
import com.jcabi.http._

object UserAgentWire {
  private val userAgentHeader = "User-Agent"
}
case class UserAgentWire(private val wire: Wire, userAgent: String) extends Wire {
  @Override
  def send(
    request: Request,
    home: String,
    method: String,
    headers: JavaCollection[MapEntry[String, String]],
    content: InputStream
  ): Response = {
    val header = new ImmutableHeader(UserAgentWire.userAgentHeader, userAgent)
    val newHeaders = header +: headers.asScala.filter{ _.getKey != UserAgentWire.userAgentHeader}.toSeq
    wire.send(request, home, method, newHeaders.asJava, content)
  }
}