From 88429412b9d7b2bac70690282da64a9bcca7a202 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Thea=20Sch=C3=B6bl?= <dev@theaninova.de>
Date: Thu, 2 Nov 2023 22:13:37 +0100
Subject: [PATCH] fix: lite breaks layout viewer

---
 src/lib/serial/connection.ts | 3 +--
 src/lib/serial/device.ts     | 2 ++
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/lib/serial/connection.ts b/src/lib/serial/connection.ts
index 70872e7d..bbb670ba 100644
--- a/src/lib/serial/connection.ts
+++ b/src/lib/serial/connection.ts
@@ -61,8 +61,7 @@ export async function initSerial(manual = false) {
 
   const parsedLayout: CharaLayout = [[], [], []]
   for (let layer = 1; layer <= 3; layer++) {
-    // TODO: this will fail for LITE!
-    for (let i = 0; i < 90; i++) {
+    for (let i = 0; i < device.keyCount; i++) {
       parsedLayout[layer - 1][i] = await device.getLayoutKey(layer, i)
     }
   }
diff --git a/src/lib/serial/device.ts b/src/lib/serial/device.ts
index 0ed32abc..ee43aa65 100644
--- a/src/lib/serial/device.ts
+++ b/src/lib/serial/device.ts
@@ -33,6 +33,7 @@ export class CharaDevice {
   company!: "CHARACHORDER"
   device!: "ONE" | "LITE"
   chipset!: "M0" | "S2"
+  keyCount!: 90 | 67
 
   constructor(private readonly baudRate = 115200) {}
 
@@ -62,6 +63,7 @@ export class CharaDevice {
     this.company = company as "CHARACHORDER"
     this.device = device as "ONE" | "LITE"
     this.chipset = chipset as "M0" | "S2"
+    this.keyCount = this.device === "ONE" ? 90 : 67
   }
 
   private async suspend() {