From 52844da70db823f32cf8a0c23ffe348500d690af Mon Sep 17 00:00:00 2001
From: Qudix <17361645+Qudix@users.noreply.github.com>
Date: Wed, 23 Sep 2020 19:00:18 -0500
Subject: [PATCH] Added ModDataChecker to STALKER Anomaly

---
 games/game_stalkeranomaly.py | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/games/game_stalkeranomaly.py b/games/game_stalkeranomaly.py
index 6728866..a322d15 100644
--- a/games/game_stalkeranomaly.py
+++ b/games/game_stalkeranomaly.py
@@ -6,10 +6,22 @@ import mobase
 
 from ..basic_game import BasicGame
 
+class StalkerAnomalyModDataChecker(mobase.ModDataChecker):
+    def __init__(self):
+        super().__init__()
+    
+    def dataLooksValid(
+        self, tree: mobase.IFileTree
+    ) -> mobase.ModDataChecker.CheckReturn:
+        if tree.exists("db") or tree.exists("appdata") or tree.exists("gamedata"):
+            return mobase.ModDataChecker.VALID
+        else:
+            return mobase.ModDataChecker.INVALID
+
 class StalkerAnomalyGame(BasicGame):
     Name = "STALKER Anomaly"
     Author = "Qudix"
-    Version = "0.1.0"
+    Version = "0.2.0"
     Description = "Adds support for STALKER Anomaly"
 
     GameName = "STALKER Anomaly"
@@ -22,6 +34,7 @@ class StalkerAnomalyGame(BasicGame):
     
     def init(self, organizer: mobase.IOrganizer):
         super().init(organizer)
+        self._featureMap[mobase.ModDataChecker] = StalkerAnomalyModDataChecker()
         return True
 
     def executables(self):