chatbot medium

Author: spenes

src/main/scala/com/snowplowanalytics/refererparser/Medium.scala

@@ -29,6 +29,7 @@ object Medium {
       case SocialMedium.value   => Some(SocialMedium)
       case EmailMedium.value    => Some(EmailMedium)
       case PaidMedium.value     => Some(PaidMedium)
+      case ChatbotMedium.value  => Some(ChatbotMedium)
       case _                    => None
     }
 
@@ -38,6 +39,7 @@ object Medium {
   val Social   = SocialMedium
   val Email    = EmailMedium
   val Paid     = PaidMedium
+  val Chatbot  = ChatbotMedium
 }
 
 case object UnknownMedium extends Medium("unknown")
@@ -46,3 +48,4 @@ case object InternalMedium extends Medium("internal")
 case object SocialMedium extends Medium("social")
 case object EmailMedium extends Medium("email")
 case object PaidMedium extends Medium("paid")
+case object ChatbotMedium extends Medium("chatbot")

src/main/scala/com/snowplowanalytics/refererparser/Parser.scala

@@ -120,6 +120,7 @@ class Parser private[refererparser] (referers: Map[String, RefererLookup]) {
                 case SocialMedium   => SocialReferer(SocialMedium, lookup.source)
                 case EmailMedium    => EmailReferer(EmailMedium, lookup.source)
                 case PaidMedium     => PaidReferer(PaidMedium, lookup.source)
+                case ChatbotMedium  => ChatbotReferer(ChatbotMedium, lookup.source)
               }
             }
             .getOrElse(UnknownReferer(UnknownMedium))

src/main/scala/com/snowplowanalytics/refererparser/Referer.scala

@@ -29,3 +29,4 @@ final case class InternalReferer(medium: Medium) extends Referer
 final case class SocialReferer(medium: Medium, source: String) extends Referer
 final case class EmailReferer(medium: Medium, source: String) extends Referer
 final case class PaidReferer(medium: Medium, source: String) extends Referer
+final case class ChatbotReferer(medium: Medium, source: String) extends Referer

src/test/resources/referer-tests.json

@@ -246,5 +246,37 @@
         "source": null,
         "term": null,
         "known": false
-    }           
+    },
+    {
+        "spec": "ChatGPT",
+        "uri": "https://www.chatgpt.com",
+        "medium": "chatbot",
+        "source": "ChatGPT",
+        "term": null,
+        "known": true
+    },
+    {
+        "spec": "Google Gemini",
+        "uri": "https://gemini.google.com",
+        "medium": "chatbot",
+        "source": "Google Gemini",
+        "term": null,
+        "known": true
+    },
+    {
+        "spec": "Microsoft Copilot",
+        "uri": "https://www.copilot.microsoft.com/account?u=1&id=1",
+        "medium": "chatbot",
+        "source": "Microsoft Copilot",
+        "term": null,
+        "known": true
+    },
+    {
+        "spec": "META.ai",
+        "uri": "https://meta.ai",
+        "medium": "chatbot",
+        "source": "META.ai",
+        "term": null,
+        "known": true
+    }
 ]

src/test/resources/referers.json

@@ -1100,5 +1100,28 @@
                 "odnoklassniki.ru"
             ]
         }
+    },
+    "chatbot": {
+        "ChatGPT": {
+            "domains": [
+                "chatgpt.com",
+                "chat.openai.com"
+            ]
+        },
+        "Google Gemini": {
+            "domains": [
+                "gemini.google.com"
+            ]
+        },
+        "Microsoft Copilot": {
+            "domains": [
+                "copilot.microsoft.com"
+            ]
+        },
+        "META.ai": {
+            "domains": [
+                "meta.ai"
+            ]
+        }
     }
 }

src/test/scala/com/snowplowanalytics/refererparser/JsonParseTest.scala

@@ -72,6 +72,7 @@ class JsonParseTest extends Specification {
           case Some(SocialMedium)   => Some(SocialReferer(SocialMedium, test.source.get))
           case Some(EmailMedium)    => Some(EmailReferer(EmailMedium, test.source.get))
           case Some(PaidMedium)     => Some(PaidReferer(PaidMedium, test.source.get))
+          case Some(ChatbotMedium)  => Some(ChatbotReferer(ChatbotMedium, test.source.get))
           case _                    => throw new Exception(s"Bad medium: ${test.medium}")
         }
         val ioActual   = ioParser.parse(new URI(test.uri), Some(pageHost), internalDomains)

src/test/scala/com/snowplowanalytics/refererparser/ParseTest.scala

@@ -396,6 +396,42 @@ class ParseTest extends Specification with DataTables {
         Some("keyword 1%")
       )
     )
+    test(
+      RefererSpec(
+        "ChatGPT",
+        "https://www.chatgpt.com",
+        Medium.Chatbot,
+        Some("ChatGPT"),
+        None
+      )
+    )
+    test(
+      RefererSpec(
+        "Google Gemini",
+        "https://gemini.google.com",
+        Medium.Chatbot,
+        Some("Google Gemini"),
+        None
+      )
+    )
+    test(
+      RefererSpec(
+        "Microsoft Copilot",
+        "https://www.copilot.microsoft.com/account?u=1&id=1",
+        Medium.Chatbot,
+        Some("Microsoft Copilot"),
+        None
+      )
+    )
+    test(
+      RefererSpec(
+        "META.ai",
+        "https://meta.ai",
+        Medium.Chatbot,
+        Some("META.ai"),
+        None
+      )
+    )
   }
 
   def genExpected(medium: Medium, source: Option[String], term: Option[String]) =
@@ -406,5 +442,6 @@ class ParseTest extends Specification with DataTables {
       case SocialMedium   => SocialReferer(SocialMedium, source.get)
       case EmailMedium    => EmailReferer(EmailMedium, source.get)
       case PaidMedium     => PaidReferer(PaidMedium, source.get)
+      case ChatbotMedium  => ChatbotReferer(ChatbotMedium, source.get)
     }
 }