package dao

import (
	"gorm.io/gorm"
	"log"
	"videoplayer/proto"
)

type Shell struct {
	gorm.Model
	AuthID       uint   `gorm:"column:auth_id"`
	Server       string `gorm:"column:server"`
	ShellName    string `gorm:"column:shell_name"`
	ShellContent string `gorm:"column:shell_content"`
	Status       int    `gorm:"column:status"` // 0 未执行 1 执行中 2 执行完成
	ShellResult  string `gorm:"column:shell_result"`
}

func CreateShell(shellName, shellContent, server string, uid uint) uint {
	shell := Shell{ShellName: shellName, ShellContent: shellContent, Server: server, AuthID: uid}
	var res *gorm.DB
	if proto.Config.SERVER_SQL_LOG {
		res = DB.Debug().Create(&shell)
	} else {
		res = DB.Create(&shell)
	}
	if res.Error != nil {
		return 0
	}
	return shell.ID
}

func FindShellByID(id, uid uint) []Shell {
	var shell Shell
	var result *gorm.DB

	if proto.Config.SERVER_SQL_LOG {
		result = DB.Debug().Where("id = ? and auth_id = ?", id, uid).First(&shell)
	} else {
		result = DB.Where("id = ? and auth_id = ?", id, uid).First(&shell)
	}
	var res []Shell
	res = append(res, shell)
	if result.Error != nil {
		log.Printf("FindShellByID failed: %v", result.Error)
	}
	return res
}

func FindShellByAuthID(auth_id int) []Shell {
	var shells []Shell
	if proto.Config.SERVER_SQL_LOG {
		DB.Debug().Where("auth_id = ?", auth_id).Order("created_at DESC").Limit(100).Find(&shells)
	} else {
		DB.Where("auth_id = ?", auth_id).Order("created_at DESC").Limit(100).Find(&shells)
	}
	return shells
}

func UpdateShellByID(id, authId uint, shellName, shellContent string, status int, shellResult string) bool {
	pd := FindShellByID(id, authId)
	if pd[0].ID == 0 {
		return false
	}
	var result *gorm.DB
	if proto.Config.SERVER_SQL_LOG {
		result = DB.Debug().Model(&Shell{}).Where("id = ? and auth_id = ?", id, authId).Updates(Shell{ShellName: shellName, ShellContent: shellContent, Status: status, ShellResult: shellResult})
	} else {
		result = DB.Model(&Shell{}).Where("id = ? and auth_id = ?", id, authId).Updates(Shell{ShellName: shellName, ShellContent: shellContent, Status: status, ShellResult: shellResult})
	}
	if result.Error != nil {
		return false
	}
	return true
}

func FindShellWillRunByServer(server string, uid uint) []Shell {
	var shells []Shell
	if proto.Config.SERVER_SQL_LOG {
		result := DB.Debug().Where("server = ? AND auth_id = ? AND status = 0", server, uid).
			Order("created_at DESC").
			Limit(100).
			Find(&shells)
		if result.Error != nil {
			// 若查询过程中出现错误，输出错误信息
			log.Printf("Failed to query shells: %v\n", result.Error)
		}
	} else {
		//DB.Exec("select * from shells where server = ? and auth_id = ? and status = 0 order by created_at desc limit 100", server, uid).Scan(&shells)
		result := DB.Where("server = ? AND auth_id = ? AND status = 0", server, uid).
			Order("created_at DESC").
			Limit(100).
			Find(&shells)
		if result.Error != nil {
			// 若查询过程中出现错误，输出错误信息
			log.Printf("Failed to query shells: %v\n", result.Error)
		}
	}
	return shells
}

func DeleteShellByID(id, authId uint) bool {
	var db2 *gorm.DB
	if proto.Config.SERVER_SQL_LOG {
		db2 = DB
	} else {
		db2 = DB.Debug()
	}
	result := db2.Where("id = ? and auth_id = ?", id, authId).Delete(&Shell{})

	if result.Error != nil {
		log.Printf("DeleteShellByID failed: %v", result.Error)
		return false
	}
	return true
}