WIP: Fire tv in Amplitude

src/main/scala/com/mozilla/telemetry/pings/FireTvEventPing.scala

@@ -0,0 +1,69 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+package com.mozilla.telemetry.pings
+
+import com.mozilla.telemetry.heka.Message
+import com.mozilla.telemetry.pings.FireTvEventPing.OptionToType
+import org.json4s.JsonDSL._
+import org.json4s._
+
+case class FireTvEventPing(arch: String,
+                           clientId: String,
+                           created: Long,
+                           device: String,
+                           events: Seq[Event],
+                           locale: String,
+                           os: String,
+                           osversion: String,
+                           seq: Integer,
+                           settings: FireTvSettings,
+                           v: String,
+                           meta: Meta) extends SendsToAmplitude {
+
+  override def getClientId: Option[String] = Some(clientId)
+
+  def sessionStart: Long = created
+
+  def getSessionId: Option[String] = Some((events.map(_.timestamp).max).toString)
+
+  def getOsName: Option[String] = Some(os)
+
+  def getOsVersion: Option[String] = Some(osversion)
+
+  def getCreated: Option[Long] = Some(created)
+
+  override def pingAmplitudeProperties: JObject = {
+    ("device_id" -> getClientId) ~
+    ("device_model" -> device) ~
+    ("arch" -> arch) ~
+    ("locale" -> locale) ~
+    ("user_properties" ->
+      ("tracking_protection_enabled" -> settings.tracking_protection_enabled.asBool) ~
+      ("total_home_tile_count" -> settings.total_home_tile_count) ~
+      ("custom_home_tile_count" -> settings.custom_home_tile_count) ~
+      ("remote_control_name" -> settings.remote_control_name) ~
+      ("app_id" -> settings.app_id))
+  }
+}
+
+case class FireTvSettings(tracking_protection_enabled: Option[String],
+                          total_home_tile_count: Option[String],
+                          custom_home_tile_count: Option[String],
+                          remote_control_name: Option[String],
+                          app_id: Option[String])
+
+object FireTvEventPing {
+  def apply(message: Message): MobileEventPing = {
+    implicit val formats = DefaultFormats
+
+    val ping = Ping.messageToPing(message, List(), eventLocations)
+    ping.extract[MobileEventPing]
+  }
+
+  val eventLocations = List("events" :: Nil)
+
+  implicit class OptionToType(val opt: Option[String]) extends AnyVal {
+    def asBool: Option[Boolean] = opt.map(_ == "true")
+  }
+}

src/main/scala/com/mozilla/telemetry/pings/Ping.scala

@@ -400,11 +400,18 @@ object SendsToAmplitude {
         message.fieldsAsMap.get("appName") match {
           case Some("Focus") => FocusEventPing(message)
           case Some("Zerda") => RocketEventPing(message)
-          case Some(x) => throw new IllegalArgumentException(s"Expect Focus or Zerda for focus-event, but we got $x")
+          case Some(x) => throw new IllegalArgumentException(s"Expect Focus or Zerda for focus-event appName, but we got $x")
+          case _ => throw new IllegalArgumentException(s"No App Name found for focus-event")
+        }
+      }
+      case Some("mobile-event") => {
+        message.fieldsAsMap.get("appName") match {
+          case Some("FirefoxForFireTV") => FireTvEventPing(message)
+          case Some("Fennec") => MobileEventPing(message)
+          case Some(x) => throw new IllegalArgumentException(s"Expect FirefoxForFireTV or Fennec for mobile-event appName, but we got $x")
           case _ => throw new IllegalArgumentException(s"No App Name found for focus-event")
         }
       }
-      case Some("mobile-event") => MobileEventPing(message)
       case Some("main") => MainPing(message)
       case Some("event") => EventPing(message)
       case Some(x) => throw new IllegalArgumentException(s"Unexpected doctype $x")

src/test/scala/com/mozilla/telemetry/streaming/EventsToAmplitudeTest.scala

@@ -395,6 +395,7 @@ class EventsToAmplitudeTest extends FlatSpec with Matchers with BeforeAndAfterAl
   }
 
   "Schema configs" should "validate against meta-schema" in {
+    EventsToAmplitude.readConfigFile("configs/fire_tv_events_schemas.json")
     EventsToAmplitude.readConfigFile("configs/fennec_ios_events_schemas.json")
     EventsToAmplitude.readConfigFile("configs/rocket_android_events_schemas.json")
     EventsToAmplitude.readConfigFile("configs/focus_android_events_schemas.json")