From 446f9eb1007abcf5ccfd5980514d4817ab8cac3e Mon Sep 17 00:00:00 2001
From: Everton Correia <1169768+evertonstz@users.noreply.github.com>
Date: Fri, 3 May 2024 14:59:29 -0300
Subject: [PATCH] Implement haptics message (#51)

Fixes #50
---
 src/components/MainContent/SelectedDevice.tsx |  22 ++++
 src/pax/containers/api/get.ts                 |   3 +
 src/pax/core/messages/HapticsMessage.ts       | 109 ++++++++++++++++++
 src/pax/core/messages/index.ts                |   1 +
 src/state/paxState/actions.ts                 |  10 ++
 src/state/paxState/reducer.ts                 |   3 +
 src/state/paxState/types.ts                   |   1 +
 7 files changed, 149 insertions(+)
 create mode 100644 src/pax/core/messages/HapticsMessage.ts

diff --git a/src/components/MainContent/SelectedDevice.tsx b/src/components/MainContent/SelectedDevice.tsx
index e8647c3..258403c 100644
--- a/src/components/MainContent/SelectedDevice.tsx
+++ b/src/components/MainContent/SelectedDevice.tsx
@@ -6,6 +6,7 @@ import {
   BatteryPercentageMessage,
   BrightnessMessage,
   ColorThemeMessage,
+  HapticsMessage,
   RequestStatusMessage,
 } from '@/pax/core/messages';
 import { Messages } from '@/pax/shared/enums';
@@ -60,6 +61,9 @@ export const SelectedDevice = ({ currentDevice }: SelectedDeviceProps) => {
         if (message instanceof BrightnessMessage) {
           actions.setBrightness(message.brightness);
         }
+        if (message instanceof HapticsMessage) {
+          actions.setHaptics(message.percentage);
+        }
       })
       .catch(e => {
         if (e instanceof BaseBluetoothException) {
@@ -119,6 +123,23 @@ export const SelectedDevice = ({ currentDevice }: SelectedDeviceProps) => {
           void bluetoothState.writeToMainService(toPost.packet);
         }}
       />
+      <Slider
+        disabled={!bluetoothState.connected && !state.haptics}
+        max={1}
+        step={0.1}
+        value={state.haptics ? [state.haptics] : [0]}
+        onValueChange={value => {
+          actions.setHaptics(value[0]);
+        }}
+        onValueCommit={value => {
+          const toPost = post(
+            HapticsMessage.createWithHaptics(value[0]),
+            currentDevice,
+          );
+          void bluetoothState.writeToMainService(toPost.packet);
+        }}
+      />
+
       <HeaterStatus heaterStatus={state.heatingSate} />
       <h1>Device: {!currentDevice ? '' : currentDevice.serial}</h1>
       <Button
@@ -137,6 +158,7 @@ export const SelectedDevice = ({ currentDevice }: SelectedDeviceProps) => {
             RequestStatusMessage.createWithMessage([
               Messages.ATTRIBUTE_COLOR_THEME,
               Messages.ATTRIBUTE_BRIGHTNESS,
+              Messages.ATTRIBUTE_HAPTIC_MODE,
             ]),
             currentDevice,
           );
diff --git a/src/pax/containers/api/get.ts b/src/pax/containers/api/get.ts
index ab923f1..3c0cd29 100644
--- a/src/pax/containers/api/get.ts
+++ b/src/pax/containers/api/get.ts
@@ -4,6 +4,7 @@ import {
   ActualTemperatureMessage,
   BatteryPercentageMessage,
   BrightnessMessage,
+  HapticsMessage,
   HeaterSetPointMessage,
   HeatingStateMessage,
   MessageAbs,
@@ -39,6 +40,8 @@ export const decodeDecryptedPacket = (
       return new BatteryPercentageMessage(packet);
     case Messages.ATTRIBUTE_BRIGHTNESS:
       return BrightnessMessage.createWithPacket(packet);
+    case Messages.ATTRIBUTE_HAPTIC_MODE:
+      return HapticsMessage.createWithPacket(packet);
     default:
       return new UnknownMessage(messageType, packet);
   }
diff --git a/src/pax/core/messages/HapticsMessage.ts b/src/pax/core/messages/HapticsMessage.ts
new file mode 100644
index 0000000..2aa60a7
--- /dev/null
+++ b/src/pax/core/messages/HapticsMessage.ts
@@ -0,0 +1,109 @@
+import { PaxDecryptedPacket } from '@/pax/containers/lib';
+import { Messages } from '@/pax/shared/enums';
+
+import { MessageAbs } from './MessageAbs';
+import { ReadAndWriteMessageAbs } from './ReadAndWriteMessageAbs';
+
+export class HapticsMessage
+  extends ReadAndWriteMessageAbs
+  implements MessageAbs
+{
+  readonly percentage: number;
+  readonly messageType: Messages;
+  readonly packet: PaxDecryptedPacket;
+
+  readonly PaxHapticsAmpMin = 0;
+  readonly PaxHapticsAmpMax = 128;
+
+  constructor(
+    builder:
+      | HapticsMessageBuilderFromPacket<HapticsMessage>
+      | HapticsMessageBuilderFromValue<HapticsMessage>,
+  ) {
+    super();
+    this.messageType = Messages.ATTRIBUTE_HAPTIC_MODE;
+    if (builder instanceof HapticsMessageBuilderFromPacket) {
+      this.packet = builder.getPacket();
+      const hapticsAmplitude = this.packet.getUint8(1);
+      this.percentage = this.hapticsToPercentage(hapticsAmplitude);
+    } else if (builder instanceof HapticsMessageBuilderFromValue) {
+      this.percentage = builder.getHaptics();
+      const buffer = new ArrayBuffer(16);
+      const view = new PaxDecryptedPacket(buffer);
+      view.setUint8(0, this.messageType);
+      view.setUint8(1, this.percentageToHaptics(this.percentage));
+      this.packet = view;
+    } else {
+      throw new Error('Invalid builder');
+    }
+  }
+
+  private hapticsToPercentage(haptics: number): number {
+    return (
+      (haptics - this.PaxHapticsAmpMin) /
+      (this.PaxHapticsAmpMax - this.PaxHapticsAmpMin)
+    );
+  }
+
+  private percentageToHaptics(percentage: number): number {
+    return Math.round(percentage * this.PaxHapticsAmpMax);
+  }
+
+  static createWithPacket(packet: PaxDecryptedPacket): HapticsMessage {
+    const builder = new HapticsMessageBuilderFromPacket<HapticsMessage>();
+    builder.setPacket(packet);
+    return new HapticsMessage(builder);
+  }
+
+  static createWithHaptics(percentage: number): HapticsMessage {
+    const builder = new HapticsMessageBuilderFromValue<HapticsMessage>();
+    builder.setBrightness(percentage);
+    return new HapticsMessage(builder);
+  }
+}
+
+export class HapticsMessageBuilderFromPacket<T extends HapticsMessage> {
+  private packet?: PaxDecryptedPacket;
+
+  setPacket(packet: PaxDecryptedPacket): HapticsMessageBuilderFromPacket<T> {
+    this.packet = packet;
+    return this;
+  }
+
+  getPacket(): PaxDecryptedPacket {
+    if (!this.packet) {
+      throw new Error('Packet is not set');
+    }
+    return this.packet;
+  }
+
+  build(ctor: new (builder: HapticsMessageBuilderFromPacket<T>) => T): T {
+    if (!this.packet) {
+      throw new Error('Packet is not set');
+    }
+    return new ctor(this);
+  }
+}
+
+export class HapticsMessageBuilderFromValue<T extends HapticsMessage> {
+  private percentage?: number;
+
+  setBrightness(percentage: number): HapticsMessageBuilderFromValue<T> {
+    this.percentage = percentage;
+    return this;
+  }
+
+  getHaptics(): number {
+    if (this.percentage === undefined) {
+      throw new Error('Percentage is not set');
+    }
+    return this.percentage;
+  }
+
+  build(ctor: new (builder: HapticsMessageBuilderFromValue<T>) => T): T {
+    if (this.percentage === undefined) {
+      throw new Error('Percentage is not set');
+    }
+    return new ctor(this);
+  }
+}
diff --git a/src/pax/core/messages/index.ts b/src/pax/core/messages/index.ts
index 80c4b2a..5d133d2 100644
--- a/src/pax/core/messages/index.ts
+++ b/src/pax/core/messages/index.ts
@@ -10,3 +10,4 @@ export * from './ColorThemeMessage';
 export * from './BatteryPercentageMessage';
 export * from './BrightnessMessage';
 export * from './RequestStatusMessage';
+export * from './HapticsMessage';
diff --git a/src/state/paxState/actions.ts b/src/state/paxState/actions.ts
index f586f4b..76503df 100644
--- a/src/state/paxState/actions.ts
+++ b/src/state/paxState/actions.ts
@@ -7,6 +7,7 @@ export type PaxActions =
   | { type: 'SET_COLOR_THEME'; payload: Pax.lib.ColorTheme }
   | { type: 'SET_BATTERY_PERCENTAGE'; payload: number }
   | { type: 'SET_BRIGHTNESS'; payload: number }
+  | { type: 'SET_HAPTICS'; payload: number }
   | { type: 'RESET_PAX_STATE' };
 
 export interface BuiltPaxActions {
@@ -16,6 +17,7 @@ export interface BuiltPaxActions {
   setColorTheme: (theme: Pax.lib.ColorTheme) => void;
   setBatteryPercentage: (percentage: number) => void;
   setBrightness: (brightness: number) => void;
+  setHaptics: (haptics: number) => void;
   resetPaxState: () => void;
 }
 
@@ -65,6 +67,13 @@ const setBrightness = (
   dispatch({ type: 'SET_BRIGHTNESS', payload: brightness });
 };
 
+export const setHaptics = (
+  dispatch: React.Dispatch<PaxActions>,
+  haptics: number,
+) => {
+  dispatch({ type: 'SET_HAPTICS', payload: haptics });
+};
+
 export const buildActions = (
   dispatch: React.Dispatch<PaxActions>,
 ): BuiltPaxActions => {
@@ -81,5 +90,6 @@ export const buildActions = (
       setBatteryPercentage(dispatch, percentage),
     resetPaxState: () => resetPaxState(dispatch),
     setBrightness: (brightness: number) => setBrightness(dispatch, brightness),
+    setHaptics: (haptics: number) => setHaptics(dispatch, haptics),
   };
 };
diff --git a/src/state/paxState/reducer.ts b/src/state/paxState/reducer.ts
index 6c240c3..e5d8647 100644
--- a/src/state/paxState/reducer.ts
+++ b/src/state/paxState/reducer.ts
@@ -8,6 +8,7 @@ export const initialPaxState: PaxState = {
   colorTheme: undefined,
   batteryPercentage: undefined,
   brightness: undefined,
+  haptics: undefined,
 };
 
 const reducer = (state: PaxState, action: PaxActions): PaxState => {
@@ -24,6 +25,8 @@ const reducer = (state: PaxState, action: PaxActions): PaxState => {
       return { ...state, batteryPercentage: action.payload };
     case 'SET_BRIGHTNESS':
       return { ...state, brightness: action.payload };
+    case 'SET_HAPTICS':
+      return { ...state, haptics: action.payload };
     case 'RESET_PAX_STATE':
       return initialPaxState;
     default:
diff --git a/src/state/paxState/types.ts b/src/state/paxState/types.ts
index 8b3d3fd..2f2fc11 100644
--- a/src/state/paxState/types.ts
+++ b/src/state/paxState/types.ts
@@ -7,4 +7,5 @@ export interface PaxState {
   colorTheme?: Pax.lib.ColorTheme;
   batteryPercentage?: number;
   brightness?: number;
+  haptics?: number;
 }