package monkey

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

	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().Unix()) + ".ipa"
	pkg_path := "/home/tmp/pkg/" + 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)
	}
}