Added support checks for WMI methods.

BartoszCichecki · Nov 19, 2021 · 7eb8570 · 7eb8570
1 parent 0cea566
commit 7eb8570
Show file tree

Hide file tree

Showing 5 changed files with 26 additions and 7 deletions.
diff --git a/Lib/Features/AbstractWmiFeature.cs b/Lib/Features/AbstractWmiFeature.cs
@@ -4,18 +4,28 @@
 
 namespace LenovoLegionToolkit.Lib.Features
 {
-    public class AbstractWmiFeature<T> : IFeature<T> where T : struct, IComparable
+    public abstract class AbstractWmiFeature<T> : IFeature<T> where T : struct, IComparable
     {
         private readonly string _methodNameSuffix;
         private readonly int _offset;
+        private readonly string _supportMethodName;
+        private readonly int _supportOffset;
 
-        protected AbstractWmiFeature(string methodNameSuffix, int offset)
+        protected AbstractWmiFeature(string methodNameSuffix, int offset, string supportMethodName = null, int supportOffset = 0)
         {
             _methodNameSuffix = methodNameSuffix;
             _offset = offset;
+            _supportMethodName = supportMethodName;
+            _supportOffset = supportOffset;
         }
 
-        public T GetState() => FromInternal(ExecuteGamezone("Get" + _methodNameSuffix, "Data"));
+        public T GetState()
+        {
+            if (!IsSupported())
+                throw new NotSupportedException($"Feature {_methodNameSuffix} is not supported.");
+
+            return FromInternal(ExecuteGamezone("Get" + _methodNameSuffix, "Data"));
+        }
 
         public void SetState(T state)
         {
@@ -26,6 +36,15 @@ public void SetState(T state)
                 });
         }
 
+        private bool IsSupported()
+        {
+            if (_supportMethodName == null)
+                return true;
+
+            var value = ExecuteGamezone(_supportMethodName, "Data");
+            return value > _supportOffset;
+        }
+
         private int ToInternal(T state) => (int)(object)state + _offset;
 
         private T FromInternal(int state) => (T)(object)(state - _offset);

diff --git a/Lib/Features/HybridModeFeature.cs b/Lib/Features/HybridModeFeature.cs
@@ -8,6 +8,6 @@ public enum HybridModeState
 
     public class HybridModeFeature : AbstractWmiFeature<HybridModeState>
     {
-        public HybridModeFeature() : base("GSyncStatus", 0) { }
+        public HybridModeFeature() : base("GSyncStatus", 0, "IsSupportGSync") { }
     }
 }
diff --git a/Lib/Features/OverDriveFeature.cs b/Lib/Features/OverDriveFeature.cs
@@ -8,6 +8,6 @@ public enum OverDriveState
 
     public class OverDriveFeature : AbstractWmiFeature<OverDriveState>
     {
-        public OverDriveFeature() : base("ODStatus", 0) { }
+        public OverDriveFeature() : base("ODStatus", 0, "IsSupportOD") { }
     }
 }
diff --git a/Lib/Features/PowerModeFeature.cs b/Lib/Features/PowerModeFeature.cs
@@ -25,6 +25,6 @@ public static string GetPowerPlanGuid(this PowerModeState state)
 
     public class PowerModeFeature : AbstractWmiFeature<PowerModeState>
     {
-        public PowerModeFeature() : base("SmartFanMode", 1) { }
+        public PowerModeFeature() : base("SmartFanMode", 1, "IsSupportSmartFan") { }
     }
 }
diff --git a/Lib/Features/TouchpadLockFeature.cs b/Lib/Features/TouchpadLockFeature.cs
@@ -8,6 +8,6 @@ public enum TouchpadLockState
 
     public class TouchpadLockFeature : AbstractWmiFeature<TouchpadLockState>
     {
-        public TouchpadLockFeature() : base("TPStatus", 0) { }
+        public TouchpadLockFeature() : base("TPStatus", 0, "IsSupportDisableTP") { }
     }
 }