Migs647/amplitude (#8)

* Added Amplitude Session tracking

Author: migs647

samples/kotlin-android-app-destinations/src/main/java/com/segment/analytics/destinations/MainApplication.kt

@@ -2,7 +2,8 @@ package com.segment.analytics.destinations
 
 import android.app.Application
 import com.segment.analytics.*
-import com.segment.analytics.next.plugins.WebhookPlugin
+import com.segment.analytics.destinations.plugins.AmplitudeSession
+import com.segment.analytics.destinations.plugins.WebhookPlugin
 import com.segment.analytics.platform.Plugin
 import com.segment.analytics.utilities.*
 import kotlinx.coroutines.CoroutineScope
@@ -50,5 +51,8 @@ class MainApplication : Application() {
                 Executors.newSingleThreadExecutor()
             )
         )
+
+        // Try out amplitude session
+        analytics.add(AmplitudeSession())
     }
-}
+}

samples/kotlin-android-app-destinations/src/main/java/com/segment/analytics/destinations/plugins/AmplitudeSession.kt

@@ -0,0 +1,91 @@
+package com.segment.analytics.destinations.plugins
+
+import androidx.lifecycle.Lifecycle
+import androidx.lifecycle.LifecycleObserver
+import androidx.lifecycle.OnLifecycleEvent
+import androidx.lifecycle.ProcessLifecycleOwner
+import com.segment.analytics.*
+import com.segment.analytics.platform.DestinationPlugin
+import com.segment.analytics.platform.plugins.LogType
+import com.segment.analytics.platform.plugins.log
+import com.segment.analytics.utilities.putIntegrations
+import java.util.*
+import kotlin.concurrent.schedule
+
+// A Destination plugin that adds session tracking to Amplitude cloud mode.
+class AmplitudeSession() : DestinationPlugin() {
+
+    override val name: String = "AmplitudeSession"
+    var sessionID: Long = -1
+
+    private var timer: TimerTask? = null
+    private val fireTime: Long = 300000
+
+    override fun setup(analytics: Analytics) {
+        super.setup(analytics)
+    }
+
+    // Add the session_id to the Amplitude payload for cloud mode to handle.
+    private inline fun <reified T: BaseEvent?> insertSession(payload: T?): BaseEvent? {
+        var returnPayload = payload
+        payload?.let {
+            analytics.log(message = "Running ${payload.type} payload through $name", event = payload, type = LogType.INFO)
+            returnPayload = payload.putIntegrations("Amplitude", mapOf("session_id" to sessionID)) as T?
+        }
+        return returnPayload
+    }
+
+    override fun track(payload: TrackEvent): BaseEvent? {
+        if (payload.event == "Application Backgrounded") {
+            onBackground()
+        } else if (payload.event == "Application Opened") {
+            onForeground()
+        }
+        insertSession(payload)
+        return payload
+    }
+
+    override fun identify(payload: IdentifyEvent): BaseEvent? {
+        insertSession(payload)
+        return payload
+    }
+
+    override fun screen(payload: ScreenEvent): BaseEvent? {
+        insertSession(payload)
+        return payload
+    }
+
+    override fun group(payload: GroupEvent): BaseEvent? {
+        insertSession(payload)
+        return payload
+    }
+
+    override fun alias(payload: AliasEvent): BaseEvent? {
+        insertSession(payload)
+        return payload
+    }
+
+    fun onBackground() {
+        stopTimer()
+    }
+
+    fun onForeground() {
+        startTimer()
+    }
+
+    fun startTimer() {
+
+        // Set the session id
+        sessionID = Calendar.getInstance().timeInMillis
+
+        timer = Timer().schedule(fireTime) {
+            stopTimer()
+            startTimer()
+        }
+    }
+
+    fun stopTimer() {
+        timer?.cancel()
+        sessionID = -1
+    }
+}

samples/kotlin-android-app-destinations/src/main/java/com/segment/analytics/destinations/plugins/WebhookPlugin.kt

@@ -1,4 +1,4 @@
-package com.segment.analytics.next.plugins
+package com.segment.analytics.destinations.plugins
 
 import com.segment.analytics.*
 import com.segment.analytics.platform.DestinationPlugin
@@ -14,7 +14,7 @@ import java.net.MalformedURLException
 import java.net.URL
 import java.util.concurrent.ExecutorService
 
-// A Destination plugin that doesnt modify the incoming payload, and sends it to the configured webhook
+// A Destination plugin that doesn't modify the incoming payload, and sends it to the configured webhook
 class WebhookPlugin(private val webhookUrl: String, private val networkExecutor: ExecutorService) :
     DestinationPlugin() {
     override val name: String = "WebhookDestination-$webhookUrl"