From 6aeb94fa158b8a63ccd2d2b9e1500ea15333576c Mon Sep 17 00:00:00 2001
From: Benoit Marty <benoit@matrix.org>
Date: Mon, 3 Apr 2023 15:18:32 +0200
Subject: [PATCH] Add some shortcuts

---
 .../api/AppNavigationStateExtension.kt        | 62 +++++++++++++++++++
 1 file changed, 62 insertions(+)
 create mode 100644 services/appnavstate/api/src/main/kotlin/io/element/android/services/appnavstate/api/AppNavigationStateExtension.kt

diff --git a/services/appnavstate/api/src/main/kotlin/io/element/android/services/appnavstate/api/AppNavigationStateExtension.kt b/services/appnavstate/api/src/main/kotlin/io/element/android/services/appnavstate/api/AppNavigationStateExtension.kt
new file mode 100644
index 0000000000..00fe638a47
--- /dev/null
+++ b/services/appnavstate/api/src/main/kotlin/io/element/android/services/appnavstate/api/AppNavigationStateExtension.kt
@@ -0,0 +1,62 @@
+/*
+ * Copyright (c) 2023 New Vector Ltd
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.element.android.services.appnavstate.api
+
+import io.element.android.libraries.matrix.api.core.RoomId
+import io.element.android.libraries.matrix.api.core.SessionId
+import io.element.android.libraries.matrix.api.core.SpaceId
+import io.element.android.libraries.matrix.api.core.ThreadId
+
+fun AppNavigationState.currentSessionId(): SessionId? {
+    return when (this) {
+        AppNavigationState.Root -> null
+        is AppNavigationState.Session -> sessionId
+        is AppNavigationState.Space -> parentSession.sessionId
+        is AppNavigationState.Room -> parentSpace.parentSession.sessionId
+        is AppNavigationState.Thread -> parentRoom.parentSpace.parentSession.sessionId
+    }
+}
+
+fun AppNavigationState.currentSpaceId(): SpaceId? {
+    return when (this) {
+        AppNavigationState.Root -> null
+        is AppNavigationState.Session -> null
+        is AppNavigationState.Space -> spaceId
+        is AppNavigationState.Room -> parentSpace.spaceId
+        is AppNavigationState.Thread -> parentRoom.parentSpace.spaceId
+    }
+}
+
+fun AppNavigationState.currentRoomId(): RoomId? {
+    return when (this) {
+        AppNavigationState.Root -> null
+        is AppNavigationState.Session -> null
+        is AppNavigationState.Space -> null
+        is AppNavigationState.Room -> roomId
+        is AppNavigationState.Thread -> parentRoom.roomId
+    }
+}
+
+fun AppNavigationState.currentThreadId(): ThreadId? {
+    return when (this) {
+        AppNavigationState.Root -> null
+        is AppNavigationState.Session -> null
+        is AppNavigationState.Space -> null
+        is AppNavigationState.Room -> null
+        is AppNavigationState.Thread -> threadId
+    }
+}