feat: added some preference options

Ljzd-PRO · Apr 4, 2024 · 139a76f · 139a76f
1 parent 1df84b3
commit 139a76f
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 11 deletions.
diff --git a/ButtPlugReporter.cs b/ButtPlugReporter.cs
@@ -9,20 +9,30 @@ namespace DeppartPrototypeHentaiPlayMod
 {
     public class ButtPlugReporter : BaseReporter
     {
+        private readonly double _activeVibrateScalar;
         private readonly ButtplugClient _buttplugClient;
         private readonly Mutex _mutexLock = new Mutex();
+        private readonly double _shotVibrateScalar;
         private double _baseVibrateScalar;
 
-        public ButtPlugReporter(HentaiPlayMod melonMod, Uri buttPlugServerUrl) : base(melonMod)
+        public ButtPlugReporter
+        (
+            HentaiPlayMod melonMod,
+            string buttPlugServerUrl,
+            double activeVibrateScalar = 0.5,
+            double shotVibrateScalar = 1
+        ) : base(melonMod)
         {
+            _activeVibrateScalar = activeVibrateScalar;
+            _shotVibrateScalar = shotVibrateScalar;
             _buttplugClient = new ButtplugClient(MelonMod.Info.Name);
             _buttplugClient.DeviceAdded +=
                 (sender, args) => MelonMod.LoggerInstance.Msg($"ButtPlug device added: {args.Device.Name}");
             _buttplugClient.DeviceRemoved +=
                 (sender, args) => MelonMod.LoggerInstance.Msg($"ButtPlug device removed: {args.Device.Name}");
             _buttplugClient.ScanningFinished += (sender, args) =>
                 MelonMod.LoggerInstance.Msg($"ButtPlug scanning finished: {args}");
-            var connector = new ButtplugWebsocketConnector(buttPlugServerUrl);
+            var connector = new ButtplugWebsocketConnector(new Uri(buttPlugServerUrl));
             connector.Disconnected +=
                 (sender, args) => MelonMod.LoggerInstance.Msg($"ButtPlug scanning finished: {args}");
             new Thread(() =>
@@ -80,11 +90,11 @@ public override void ReportActivateEvent(string eventName)
                 { EventEnum.BulbBroken.ToString(), EventEnum.ZombieRun.ToString(), EventEnum.EnterLevel1.ToString() };
             if (tempEvents.Contains(eventName))
             {
-                SendCommand(new[] { 0.5, _baseVibrateScalar }, 5000);
+                SendCommand(new[] { _activeVibrateScalar, _baseVibrateScalar }, 5000);
             }
             else
             {
-                _baseVibrateScalar = 0.5;
+                _baseVibrateScalar = _activeVibrateScalar;
                 SendCommand(new[] { _baseVibrateScalar });
             }
         }
@@ -111,7 +121,7 @@ public override void ReportGameExitEvent()
         public override void ReportShot()
         {
             base.ReportShot();
-            SendCommand(new[] { 1, _baseVibrateScalar });
+            SendCommand(new[] { _shotVibrateScalar, _baseVibrateScalar });
         }
     }
 }
diff --git a/HentaiPlayMod.cs b/HentaiPlayMod.cs
@@ -1,5 +1,4 @@
-using System;
-using System.Collections.Generic;
+using System.Collections.Generic;
 using System.Linq;
 using DeppartPrototypeHentaiPlayMod;
 using MelonLoader;
@@ -23,7 +22,10 @@ public class HentaiPlayMod : MelonMod
             { EventEnum.PlayerDied.ToString(), false }
         };
 
-        private MelonPreferences_Entry<Uri> _buttPlugServerUrlEntry;
+        private MelonPreferences_Entry<double> _buttPlugActiveVibrateScalar;
+
+        private MelonPreferences_Entry<string> _buttPlugServerUrlEntry;
+        private MelonPreferences_Entry<double> _buttPlugShotVibrateScalar;
         private MelonPreferences_Entry<bool> _disableEventLogEntry;
 
         private IEventReporter _eventReporter;
@@ -44,7 +46,8 @@ public override void OnInitializeMelon()
             (
                 "EventReporterType",
                 nameof(ButtPlugReporter),
-                description: "Type of reporter that report events in game"
+                description: "Type of reporter that report events in game " +
+                             $"(Available: {nameof(BaseReporter)}, {nameof(HttpReporter)}, {nameof(ButtPlugReporter)})"
             );
             _httpReporterUrlEntry = _preferencesCategory.CreateEntry
             (
@@ -67,9 +70,21 @@ public override void OnInitializeMelon()
             _buttPlugServerUrlEntry = _preferencesCategory.CreateEntry
             (
                 "ButtPlugServerUrl",
-                new Uri("ws://localhost:12345"),
+                "ws://localhost:12345",
                 description: "Websocket URL of ButtPlug server (Intiface Central)"
             );
+            _buttPlugActiveVibrateScalar = _preferencesCategory.CreateEntry
+            (
+                "ButtPlugActiveVibrateScalar",
+                0.5,
+                description: "Set the ButtPlug vibrate scalar when game events active"
+            );
+            _buttPlugShotVibrateScalar = _preferencesCategory.CreateEntry
+            (
+                "ButtPlugShotVibrateScalar",
+                1.0,
+                description: "Set the ButtPlug vibrate scalar when gun shot"
+            );
         }
 
         public override void OnLateInitializeMelon()
@@ -119,7 +134,13 @@ private void SetupEventReporter()
                     );
                     break;
                 case nameof(ButtPlugReporter):
-                    _eventReporter = new ButtPlugReporter(this, _buttPlugServerUrlEntry.Value);
+                    _eventReporter = new ButtPlugReporter
+                    (
+                        this,
+                        _buttPlugServerUrlEntry.Value,
+                        _buttPlugActiveVibrateScalar.Value,
+                        _buttPlugShotVibrateScalar.Value
+                    );
                     break;
             }