From 12ea70174e51fe3ec6879cdef0eac3dfb6caa2c6 Mon Sep 17 00:00:00 2001
From: Kevin Hester <kevinh@geeksville.com>
Date: Tue, 23 Mar 2021 13:40:03 +0800
Subject: [PATCH] fix autobug when changing channels

---
 .../geeksville/mesh/service/MeshService.kt    | 21 +++++++++++++------
 1 file changed, 15 insertions(+), 6 deletions(-)

diff --git a/app/src/main/java/com/geeksville/mesh/service/MeshService.kt b/app/src/main/java/com/geeksville/mesh/service/MeshService.kt
index 8071413b8..152f78d2e 100644
--- a/app/src/main/java/com/geeksville/mesh/service/MeshService.kt
+++ b/app/src/main/java/com/geeksville/mesh/service/MeshService.kt
@@ -549,7 +549,7 @@ class MeshService : Service(), Logging {
                 setChannel(it)
             }
 
-            channels = asChannels.toTypedArray()
+            channels = fixupChannelList(asChannels).toTypedArray()
         }
 
     /// Generate a new mesh packet builder with our node as the sender, and the specified node num
@@ -1305,11 +1305,20 @@ class MeshService : Service(), Logging {
         radioConfig = null
 
         // prefill the channel array with null channels
-        channels = Array(myInfo.maxChannels) {
-            val b = ChannelProtos.Channel.newBuilder()
-            b.index = it
-            b.build()
-        }
+        channels = fixupChannelList(listOf<ChannelProtos.Channel>()).toTypedArray()
+    }
+
+    /// scan the channel list and make sure it has one PRIMARY channel and is maxChannels long
+    private fun fixupChannelList(lIn: List<ChannelProtos.Channel>): List<ChannelProtos.Channel> {
+        val mi = myNodeInfo
+        var l = lIn
+        if (mi != null)
+            while (l.size < mi.maxChannels) {
+                val b = ChannelProtos.Channel.newBuilder()
+                b.index = l.size
+                l += b.build()
+            }
+        return l
     }