custom gestalt tweaks ui

tweaks/custom_gestalt_tweaks.py

@@ -0,0 +1,89 @@
+from enum import Enum
+from json import loads
+from .tweak_classes import MobileGestaltTweak
+
+class ValueType(Enum):
+    Integer = "Integer"
+    Float = "Float"
+    String = "String"
+    Array = "Array"
+    Dictionary = "Dictionary"
+
+ValueTypeStrings: list[ValueType] = [
+    ValueType.Integer.value, ValueType.Float.value,
+    ValueType.String.value,
+    ValueType.Array.value, ValueType.Dictionary.value
+]
+
+class CustomGestaltTweak:
+    def __init__(self, tweak: MobileGestaltTweak, value_type: ValueType):
+        self.tweak = tweak
+        self.value_type = value_type
+
+    # TODO: change everything to not return the dict since it is passed by reference
+    def apply_tweak(self, plist: dict) -> dict:
+        if self.tweak.key == "":
+            # key was not set, don't apply (maybe user added it by accident)
+            return plist
+        self.tweak.enabled = True
+        # set the value to be as the specified value type
+        if self.value_type == ValueType.Integer:
+            self.tweak.value = int(self.tweak.value)
+        elif self.value_type == ValueType.Float:
+            self.tweak.value = float(self.tweak.value)
+        elif self.value_type == ValueType.Array or self.value_type == ValueType.Dictionary:
+            # json convert string to array/dict
+            self.tweak.value = loads(self.tweak.value)
+        
+        # apply the tweak after updating the value
+        plist = self.tweak.apply_tweak(plist)
+        return plist
+            
+
+class CustomGestaltTweaks:
+    custom_tweaks: list[CustomGestaltTweak] = []
+
+    def create_tweak(key: str="", value: str="1", value_type: ValueType = ValueType.Integer) -> int:
+        new_tweak = MobileGestaltTweak("", key, value=value)
+        CustomGestaltTweaks.custom_tweaks.append(CustomGestaltTweak(new_tweak, value_type))
+        # return the tweak id
+        return len(CustomGestaltTweaks.custom_tweaks) - 1
+    
+    def set_tweak_key(id: int, key: str):
+        CustomGestaltTweaks.custom_tweaks[id].tweak.key = key
+            
+    def set_tweak_value(id: int, value: str):
+        CustomGestaltTweaks.custom_tweaks[id].tweak.value = value
+            
+    def set_tweak_value_type(id: int, value_type) -> str:
+        new_value_type = value_type
+        if isinstance(value_type, str):
+            # based on string value
+            new_value_type = ValueType(value_type)
+        elif isinstance(value_type, int):
+            # based on index of the string
+            new_value_type = ValueType(ValueTypeStrings[value_type])
+
+        CustomGestaltTweaks.custom_tweaks[id].value_type = new_value_type
+        # update the value to be of the new type
+        new_value = 1
+        new_str = "1"
+        if new_value_type == ValueType.Float:
+            new_value = 1.0
+            new_str = "1.0"
+        elif new_value_type == ValueType.String:
+            new_value = ""
+            new_str = ""
+        elif new_value_type == ValueType.Array:
+            new_value = []
+            new_str = "[  ]"
+        elif new_value_type == ValueType.Dictionary:
+            new_value = {}
+            new_str = "{  }"
+        CustomGestaltTweaks.custom_tweaks[id].tweak.value = new_value
+        return new_str
+
+    def apply_tweaks(plist: dict):
+        for tweak in CustomGestaltTweaks.custom_tweaks:
+            plist = tweak.apply_tweak(plist)
+        return plist

tweaks/tweaks.py

@@ -42,7 +42,7 @@ tweaks = {
     ## AI Enabler
     "AIEligibility": AITweak(),
     "AIGestalt": MobileGestaltTweak("Enable Apple Intelligence (for Unsupported Devices) (Gestalt)", "A62OafQ85EJAiiqKn4agtg", min_version=Version("18.1")),
-    "SpoofModel": MobileGestaltPickerTweak("Spoofed Device Model", "h9jDsbgj7xIVeIQ8S3/X3Q", ["iPhone16,2", "iPhone17,3", "iPhone17,4", "iPad16,3"], min_version=Version("18.1"), divider_below=True),
+    "SpoofModel": MobileGestaltPickerTweak("Spoofed Device Model", "h9jDsbgj7xIVeIQ8S3/X3Q", values=["iPhone16,2", "iPhone17,3", "iPhone17,4", "iPad16,3"], min_version=Version("18.1"), divider_below=True),
 
     ## Springboard Tweaks
     "LockScreenFootnote": BasicPlistTweak(