From 2df8627eaed145ee3814728e2e388febc85c9de7 Mon Sep 17 00:00:00 2001
From: Nikola Shabanov <60219580+shamanec@users.noreply.github.com>
Date: Sun, 30 Oct 2022 11:51:07 +0200
Subject: [PATCH] add launch and kill endpoints (#191)

---
 restapi/README.md            |   2 +-
 restapi/api/app_endpoints.go | 124 ++++++++++++++++++++++++++++++++++-
 restapi/api/routes.go        |   5 +-
 3 files changed, 127 insertions(+), 4 deletions(-)

diff --git a/restapi/README.md b/restapi/README.md
index 42e3b948..4c764e98 100644
--- a/restapi/README.md
+++ b/restapi/README.md
@@ -2,7 +2,7 @@
 
 ## getting started:
 - go install github.com/swaggo/swag/cmd/swag@latest
-- swag init
+- swag init --parseDependency
 - go run main.go
 
 plug an ios device into your machine and test on localhost:8080
diff --git a/restapi/api/app_endpoints.go b/restapi/api/app_endpoints.go
index 05a9badd..f4044a77 100644
--- a/restapi/api/app_endpoints.go
+++ b/restapi/api/app_endpoints.go
@@ -1,12 +1,22 @@
 package api
 
 import (
+	"net/http"
+
 	"github.com/danielpaulus/go-ios/ios"
 	"github.com/danielpaulus/go-ios/ios/installationproxy"
+	"github.com/danielpaulus/go-ios/ios/instruments"
 	"github.com/gin-gonic/gin"
-	"net/http"
 )
 
+// List apps on a device
+// @Summary      List apps on a device
+// @Description  List the installed apps on a device
+// @Tags         apps
+// @Produce      json
+// @Success      200 {object} []installationproxy.AppInfo
+// @Failure      500 {object} GenericResponse
+// @Router       /device/{udid}/apps [post]
 func ListApps(c *gin.Context) {
 	device := c.MustGet(IOS_KEY).(ios.DeviceEntry)
 	svc, _ := installationproxy.New(device)
@@ -14,7 +24,117 @@ func ListApps(c *gin.Context) {
 	var response []installationproxy.AppInfo
 	response, err = svc.BrowseAllApps()
 	if err != nil {
-		c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"error": err})
+		c.JSON(http.StatusInternalServerError, GenericResponse{Error: err.Error()})
 	}
 	c.IndentedJSON(http.StatusOK, response)
 }
+
+// Launch app on a device
+// @Summary      Launch app on a device
+// @Description  Launch app on a device by provided bundleID
+// @Tags         apps
+// @Produce      json
+// @Param        bundleID query string true "bundle identifier of the targeted app"
+// @Success      200  {object} GenericResponse
+// @Failure      500  {object} GenericResponse
+// @Router       /device/{udid}/apps/launch [post]
+func LaunchApp(c *gin.Context) {
+	device := c.MustGet(IOS_KEY).(ios.DeviceEntry)
+
+	bundleID := c.Query("bundleID")
+	if bundleID == "" {
+		c.JSON(http.StatusUnprocessableEntity, GenericResponse{Error: "bundleID query param is missing"})
+		return
+	}
+
+	pControl, err := instruments.NewProcessControl(device)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, GenericResponse{Error: err.Error()})
+		return
+	}
+
+	_, err = pControl.LaunchApp(bundleID)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, GenericResponse{Error: err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, GenericResponse{Message: bundleID + " launched successfully"})
+}
+
+// Kill running app on a device
+// @Summary      Kill running app on a device
+// @Description  Kill running app on a device by provided bundleID
+// @Tags         apps
+// @Produce      json
+// @Param        bundleID query string true "bundle identifier of the targeted app"
+// @Success      200 {object} GenericResponse
+// @Failure      500 {object} GenericResponse
+// @Router       /device/{udid}/apps/kill [post]
+func KillApp(c *gin.Context) {
+	device := c.MustGet(IOS_KEY).(ios.DeviceEntry)
+	var processName = ""
+
+	bundleID := c.Query("bundleID")
+	if bundleID == "" {
+		c.JSON(http.StatusUnprocessableEntity, GenericResponse{Error: "bundleID query param is missing"})
+		return
+	}
+
+	pControl, err := instruments.NewProcessControl(device)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, GenericResponse{Error: err.Error()})
+		return
+	}
+
+	svc, err := installationproxy.New(device)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, GenericResponse{Error: err.Error()})
+		return
+	}
+
+	response, err := svc.BrowseAllApps()
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, GenericResponse{Error: err.Error()})
+		return
+	}
+
+	for _, app := range response {
+		if app.CFBundleIdentifier == bundleID {
+			processName = app.CFBundleExecutable
+			break
+		}
+	}
+
+	if processName == "" {
+		c.JSON(http.StatusNotFound, GenericResponse{Message: bundleID + " is not installed"})
+		return
+	}
+
+	service, err := instruments.NewDeviceInfoService(device)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, GenericResponse{Error: err.Error()})
+		return
+	}
+	defer service.Close()
+
+	processList, err := service.ProcessList()
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, GenericResponse{Error: err.Error()})
+		return
+	}
+
+	for _, p := range processList {
+		if p.Name == processName {
+			err = pControl.KillProcess(p.Pid)
+			if err != nil {
+				c.JSON(http.StatusInternalServerError, GenericResponse{Error: err.Error()})
+				return
+			}
+			c.JSON(http.StatusOK, GenericResponse{Message: bundleID + " successfully killed"})
+			return
+		}
+	}
+
+	c.JSON(http.StatusOK, GenericResponse{Message: bundleID + " is not running"})
+}
diff --git a/restapi/api/routes.go b/restapi/api/routes.go
index eaeb4e74..d8523276 100644
--- a/restapi/api/routes.go
+++ b/restapi/api/routes.go
@@ -25,8 +25,11 @@ func registerRoutes(router *gin.RouterGroup) {
 	go reservation.CleanReservationsCRON()
 }
 func initAppRoutes(group *gin.RouterGroup) {
-	router := group.Group("/app")
+	router := group.Group("/apps")
 	router.GET("/", ListApps)
+	router.POST("/launch", LaunchApp)
+	router.POST("/kill", KillApp)
+	router.Use(LimitNumClientsUDID())
 }
 
 func initStreamingResponseRoutes(device *gin.RouterGroup, router *gin.RouterGroup) {