From 9fed62e4a7aa80a34846fdd289056bc1e9288041 Mon Sep 17 00:00:00 2001
From: Benoit Marty <benoitm@matrix.org>
Date: Thu, 14 May 2020 01:31:15 +0200
Subject: [PATCH] Improve template

---
 .../RiotXFeature/root/src/app_package/ViewModel.kt.ftl      | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/tools/templates/RiotXFeature/root/src/app_package/ViewModel.kt.ftl b/tools/templates/RiotXFeature/root/src/app_package/ViewModel.kt.ftl
index f4090b40e6..1d2ec0a069 100644
--- a/tools/templates/RiotXFeature/root/src/app_package/ViewModel.kt.ftl
+++ b/tools/templates/RiotXFeature/root/src/app_package/ViewModel.kt.ftl
@@ -6,6 +6,7 @@ import com.airbnb.mvrx.MvRxViewModelFactory
 import com.airbnb.mvrx.ViewModelContext
 import com.squareup.inject.assisted.Assisted
 import com.squareup.inject.assisted.AssistedInject
+import im.vector.riotx.core.extensions.exhaustive
 import im.vector.riotx.core.platform.VectorViewModel
 
 <#if createViewEvents>
@@ -38,7 +39,8 @@ class ${viewModelClass} @AssistedInject constructor(@Assisted initialState: ${vi
     }
 
     override fun handle(action: ${actionClass}) {
-		//TODO
-    }
+        when (action) {
 
+        }.exhaustive
+    }
 }