ZeroBot-Plugin/plugin_manager/timer/migrate/main.go

package main

import (
	"fmt"
	io "io"
	"os"
	"time"

	"github.com/FloatTech/ZeroBot-Plugin/utils/file"
)

var timersmap TimersMapOld
var timersmapnew TimersMap

func loadTimers(pbfile string) bool {
	if file.IsExist(pbfile) {
		f, err := os.Open(pbfile)
		if err == nil {
			data, err1 := io.ReadAll(f)
			if err1 == nil {
				if len(data) > 0 {
					err1 = timersmap.Unmarshal(data)
					if err1 == nil {
						return true
					}
				}
			}
		}
	}
	return false
}

// saveTimers 保存当前计时器
func saveTimers(pbfile string) error {
	data, err := timersmapnew.Marshal()
	if err == nil {
		f, err1 := os.OpenFile(pbfile, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0644)
		if err1 != nil {
			return err1
		} else {
			_, err2 := f.Write(data)
			f.Close()
			return err2
		}
	}
	return err
}

func main() {
	if len(os.Args) == 3 {
		if loadTimers(os.Args[1]) {
			timersmapnew.Timers = make(map[string]*Timer)
			for s, t := range timersmap.Timers {
				tm := &Timer{
					Alert: t.Alert,
					Url:   t.Url,
				}
				tm.SetMonth(time.Month(t.Month))
				tm.SetDay(int(t.Day))
				tm.SetHour(int(t.Hour))
				tm.SetMinute(int(t.Minute))
				tm.SetWeek(time.Weekday(t.Week))
				tm.SetEn(t.Enable)
				timersmapnew.Timers[s] = tm
			}
			saveTimers(os.Args[2])
		}
	} else {
		fmt.Println("用法：旧文件路径 新文件路径")
	}
}