1 files changed, 49 insertions, 1 deletions
diff --git a/shared/src/commonMain/kotlin/mx/trackermap/TrackerMap/client/models/UnitInformation.kt b/shared/src/commonMain/kotlin/mx/trackermap/TrackerMap/client/models/UnitInformation.kt
index edebff0..a276fda 100644
--- a/shared/src/commonMain/kotlin/mx/trackermap/TrackerMap/client/models/UnitInformation.kt
+++ b/shared/src/commonMain/kotlin/mx/trackermap/TrackerMap/client/models/UnitInformation.kt
@@ -1,6 +1,54 @@
+/**
+ * TrackerMap
+ * Copyright (C) 2021-2022  Iván Ávalos <avalos@disroot.org>, Henoch Ojeda <imhenoch@protonmail.com>
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
 package mx.trackermap.TrackerMap.client.models
 
 data class UnitInformation(
     val device: Device,
     val position: Position?
-)
-\ No newline at end of file
+) {
+    enum class Status {
+        ONLINE, OFFLINE, UNKNOWN
+    }
+
+    enum class EngineStop {
+        ON, OFF, UNKNOWN
+    }
+
+    /**
+     * Status is calculated from speed, because status based on
+     * the actual connection to the server is useless for our clients.
+     */
+    fun getStatus() = position?.speed?.let {
+        if (it >= 2) {
+            Status.ONLINE
+        } else {
+            Status.OFFLINE
+        }
+    } ?: Status.UNKNOWN
+
+    /* Many GPS devices reserve pin 1 for engine stop */
+    fun getEngineStop() = if (position?.attributes?.containsKey("out1") == true) {
+        position.attributes["out1"]?.toString()?.let {
+            when (it) {
+                "true" -> EngineStop.ON
+                "false" -> EngineStop.OFF
+                else -> EngineStop.UNKNOWN
+            }
+        } ?: EngineStop.UNKNOWN
+    } else EngineStop.UNKNOWN
+}
+\ No newline at end of file