From 37e92a063b6dbea81aaf0d2c5acea0bd4bd9226a Mon Sep 17 00:00:00 2001
From: Amaru8 <52407090+Amaru8@users.noreply.github.com>
Date: Wed, 5 Oct 2022 09:00:31 +0200
Subject: [PATCH] Implement SetRepeat to WSAPI

---
 src/main/base/wsapi.ts | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/main/base/wsapi.ts b/src/main/base/wsapi.ts
index 06cc632e..9ff8c83b 100644
--- a/src/main/base/wsapi.ts
+++ b/src/main/base/wsapi.ts
@@ -163,6 +163,15 @@ export class wsapi {
           case "repeat":
             this._win.webContents.executeJavaScript(`wsapi.toggleRepeat()`);
             break;
+          case "set-repeat":
+            if (data.repeat === 0 || data.repeat === 1 || data.repeat === 2) {
+              this._win.webContents.executeJavaScript(`MusicKit.getInstance().repeatMode = ${data.repeat}`);
+              response.message = "Repeat " + data.repeat;
+            } else {
+              this._win.webContents.executeJavaScript(`wsapi.toggleRepeat()`);
+              response.message = "Invalid Repeat, toggling";
+            }
+            break;
           case "seek":
             this._win.webContents.executeJavaScript(`MusicKit.getInstance().seekToTime(${parseFloat(data.time)})`);
             response.message = "Seek";