112 lines
2.5 KiB
Go
112 lines
2.5 KiB
Go
//go:build windows
|
|
|
|
package info
|
|
|
|
import (
|
|
"encoding/xml"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"github.com/gookit/color"
|
|
|
|
"github.com/axgle/mahonia"
|
|
)
|
|
|
|
type task struct {
|
|
RegistrationInfo struct {
|
|
Description string
|
|
}
|
|
Actions struct {
|
|
Exec struct {
|
|
Command string
|
|
Arguments string
|
|
}
|
|
}
|
|
Triggers struct {
|
|
CalendarTrigger struct {
|
|
StartBoundary string
|
|
}
|
|
}
|
|
Principals struct {
|
|
Principal struct {
|
|
UserId string
|
|
}
|
|
}
|
|
}
|
|
|
|
type CronTab struct {
|
|
Name string `json:"name,omitempty"`
|
|
Command string `json:"command,omitempty"`
|
|
Arg string `json:"arg,omitempty"`
|
|
User string `json:"user,omitempty"`
|
|
Rule string `json:"rule,omitempty"`
|
|
Description string `json:"description,omitempty"`
|
|
}
|
|
|
|
func DisplayPlanTask() {
|
|
crontab := GetCronTab()
|
|
DisplayCronTab(crontab)
|
|
}
|
|
|
|
// GetCronTab 获取计划任务
|
|
func GetCronTab() (resultData []CronTab) {
|
|
var taskPath string
|
|
if runtime.GOARCH == "386" {
|
|
taskPath = `C:\Windows\SysNative\Tasks\`
|
|
} else {
|
|
taskPath = `C:\Windows\System32\Tasks\`
|
|
}
|
|
dir, err := ioutil.ReadDir(taskPath)
|
|
if err != nil {
|
|
return resultData
|
|
}
|
|
for _, f := range dir {
|
|
if f.IsDir() {
|
|
continue
|
|
}
|
|
dat, err := ioutil.ReadFile(taskPath + f.Name())
|
|
if err != nil {
|
|
continue
|
|
}
|
|
v := task{}
|
|
dec := mahonia.NewDecoder("utf-16")
|
|
data := dec.ConvertString(string(dat))
|
|
err = xml.Unmarshal([]byte(strings.Replace(data, "UTF-16", "UTF-8", 1)), &v)
|
|
if err != nil {
|
|
log.Println("Windows crontab info xml Unmarshal error: ", err.Error())
|
|
continue
|
|
}
|
|
m := CronTab{}
|
|
m.Name = f.Name()
|
|
m.Command = v.Actions.Exec.Command
|
|
m.Arg = v.Actions.Exec.Arguments
|
|
m.User = v.Principals.Principal.UserId
|
|
m.Rule = v.Triggers.CalendarTrigger.StartBoundary
|
|
m.Description = v.RegistrationInfo.Description
|
|
resultData = append(resultData, m)
|
|
}
|
|
return resultData
|
|
}
|
|
|
|
func DisplayCronTab(cronTab []CronTab) {
|
|
color.Greenp("==============================================================================================\n")
|
|
for _, item := range cronTab {
|
|
color.Greenp("* ")
|
|
fmt.Println("NAME: ", item.Name)
|
|
color.Greenp("* ")
|
|
fmt.Println("COMMAND: ", item.Command)
|
|
color.Greenp("* ")
|
|
fmt.Println("ARG: ", item.Arg)
|
|
color.Greenp("* ")
|
|
fmt.Println("USER: ", item.User)
|
|
color.Greenp("* ")
|
|
fmt.Println("RULE: ", item.Rule)
|
|
color.Greenp("* ")
|
|
fmt.Println("DESCRIPTION: ", item.Description)
|
|
color.Greenp("==============================================================================================\n")
|
|
}
|
|
}
|