gprimelte: Update releasetools' device detection
diff --git a/releasetools/releasetools.py b/releasetools/releasetools.py
index 8cd98ef..7b82820 100755
--- a/releasetools/releasetools.py
+++ b/releasetools/releasetools.py
@@ -19,12 +19,21 @@
 """Custom OTA commands for gprimelte devices"""
 
 def FullOTA_InstallEnd(info):
-    info.script.Print("Detecting device variant and updating build.prop ...")
-    info.script.AppendExtra('ifelse(is_substring("G530W", getprop("ro.bootloader")), run_program("/sbin/sed", "-i", "s/gprimelte/gprimeltecan/g /system/build.prop"));')
-    info.script.AppendExtra('ifelse(is_substring("G530T1", getprop("ro.bootloader")), run_program("/sbin/sed", "-i", "s/gprimelte/gprimeltemtr/g /system/build.prop"));')
-    info.script.AppendExtra('ifelse(is_substring("G530T", getprop("ro.bootloader")), run_program("/sbin/sed", "-i", "s/gprimelte/gprimeltetmo/g /system/build.prop"));')
-#    info.script.AppendExtra('ifelse(is_substring("P605S", getprop("ro.bootloader")), run_program("/sbin/sh", "-c", "busybox cp /system/lib/gsm/* /system/lib/"));')
-#    info.script.AppendExtra('delete_recursive("/system/lib/gsm/");')
+
+    if ~ is_mounted("/system"):
+        info.script.Print("Mounting /system...")
+        mount("ext4", "MTD", "system", "/system")
+
+    info.script.Print("Detecting device variant ...")
+    if is_substring("G530W", getprop("ro.bootloader")):
+        info.script.Print("Device is SM-G530W. Updating build.prop ...");
+        info.script.AppendExtra('run_program("/sbin/sed", "-i", "s/gprimelte/gprimeltecan/g /system/build.prop");')
+    elif is_substring("G530T1", getprop("ro.bootloader")):
+        info.script.Print("Device is SM-G530T1. Updating build.prop ...");
+        info.script.AppendExtra('run_program("/sbin/sed", "-i", "s/gprimelte/gprimeltemtr/g /system/build.prop");')
+    else:
+        info.script.Print("Device is SM-G530T. Updating build.prop ...");
+        info.script.AppendExtra('run_program("/sbin/sed", "-i", "s/gprimelte/gprimeltetmo/g /system/build.prop");')
 
 def FullOTA_PostValidate(info):
     # run e2fsck