From f97624ac58e80d336f5bb521dac3d5b1dc02756d Mon Sep 17 00:00:00 2001
From: 5ec1cff <ewtqyqyewtqyqy@gmail.com>
Date: Fri, 12 Jul 2024 15:17:00 +0800
Subject: [PATCH] add push service tasks

---
 service/build.gradle.kts | 43 ++++++++++++++++++++++++++++++++++++++--
 1 file changed, 41 insertions(+), 2 deletions(-)

diff --git a/service/build.gradle.kts b/service/build.gradle.kts
index 643ff05..460a76e 100644
--- a/service/build.gradle.kts
+++ b/service/build.gradle.kts
@@ -1,4 +1,4 @@
-import com.android.build.api.dsl.Packaging
+import android.databinding.tool.ext.capitalizeUS
 
 plugins {
     alias(libs.plugins.jetbrains.kotlin.android)
@@ -42,4 +42,43 @@ dependencies {
     compileOnly(project(":stub"))
     compileOnly(libs.annotation)
     implementation(libs.bcpkix.jdk18on)
-}
\ No newline at end of file
+}
+
+androidComponents.onVariants { variant ->
+    afterEvaluate {
+        val variantLowered = variant.name.lowercase()
+        val variantCapped = variant.name.capitalizeUS()
+        val pushTask = task<Task>("pushService$variantCapped") {
+            group = "Service"
+            dependsOn("assemble$variantCapped")
+            doLast {
+                exec {
+                    commandLine(
+                        "adb",
+                        "push",
+                        layout.buildDirectory.file("outputs/apk/$variantLowered/service-$variantLowered.apk")
+                            .get().asFile.absolutePath,
+                        "/data/local/tmp/service.apk"
+                    )
+                }
+                exec {
+                    commandLine(
+                        "adb",
+                        "shell",
+                        "su -c \"mv /data/local/tmp/service.apk /data/adb/modules/tricky_store/\""
+                    )
+                }
+            }
+        }
+
+        task<Task>("pushAndRestartService$variantCapped") {
+            group = "Service"
+            dependsOn(pushTask)
+            doLast {
+                exec {
+                    commandLine("adb", "shell", "su -c \"setprop ctl.restart keystore2\"")
+                }
+            }
+        }
+    }
+}