package monkey

import (
	"autogo/common"
	"autogo/dbsql"
	"autogo/models"
	"fmt"
	"os/exec"
	"strings"
	"time"

	"github.com/electricbubble/gwda"
	log "github.com/sirupsen/logrus"
	"github.com/spf13/cast"
)

func RuniOSMonkeyByDocker(task models.MonkeyTask, udid string) {
	db, err := dbsql.GetConn(dbsql.DSN)
	if err != nil {
		return
	}
	defer dbsql.Close(db)
	db.Model(models.MonkeyTask{}).Where("id = ?", task.Id).Update("status", "INIT")

	filename := cast.ToString(time.Now().UnixMilli()) + ".ipa"
	pkg_path := "/root/app/autogo/downloads/" + filename
	log.Debug("正在下载ipa：", task.PackageURL)
	err = exec.Command("wget", task.PackageURL, "-O", pkg_path).Run()
	if err != nil {
		log.Error(err)
	}
	log.Debug("下载完毕，安装包路径", pkg_path)

	// python main_adr.py -package music.hising.live.dev -launch_activity sing.lagufun.social.live.ui.activity.SplashActivity
	cmd_content := task.Cmd(udid, filename)

	db.Model(models.MonkeyTask{}).Where("id = ?", task.Id).Last(&task)
	if task.Status != "INIT" {
		common.PushCorntaskLog("任务状态异常，停止发起Monkey任务：" + task.Status)
		return
	}
	log.Debug("开始调用monkey服务，执行命令")
	log.Debug(cmd_content)
	cmd := exec.Command("bash", "-c", cmd_content)
	err = cmd.Run()
	if err != nil {
		log.Error(err)
	}
}

// iOS设备保活，注意需要先启动wda
func KeepiOSDevice(udid string) error {
	// 初始化wda驱动和设备信息
	devices, err := gwda.DeviceList()
	if err != nil {
		fmt.Println()
		return err
	}
	for _, v := range devices {
		if strings.EqualFold(v.SerialNumber(), udid) {
			fmt.Println("匹配到对应设备 -", udid)
			driver, err := gwda.NewUSBDriver(nil, v)
			if err != nil {
				return err
			}
			driver.PressButton(gwda.DeviceButtonHome)
			return nil
		}
	}
	return nil
}