main.go

package main

import (
	"flag"
	"grass/browser"
	"grass/config"
	"log/slog"
	"math/rand"
	"os"
	"os/signal"
	"sync"
	"syscall"
	"time"
)

var (
	configs  []config.Config
	file     string
	help     bool
	browsers sync.Map
)

func main() {
	flags()
	loadConfigs()
	for _, conf := range configs {
		func(c config.Config) {
			for _, p := range c.Proxies {
				go touchGrass(c.User, p)
			}
		}(conf)
		if len(conf.Proxies) == 0 {
			go touchGrass(conf.User, "")
		}
	}
	mainLoop()
}

func flags() {
	flag.StringVar(&file, "file", "config.yaml", "config file path")
	flag.BoolVar(&help, "help", false, "help")
	flag.Parse()
	if help {
		flag.Usage()
		os.Exit(0)
	}
}
func loadConfigs() {
	var err error
	configs, err = config.GetConfigs(file)
	if err != nil {
		slog.Error("load config err:", err)
		os.Exit(1)
	}
}

//func watchConfigs() error {
//	watcher, err := fsnotify.NewWatcher()
//	if err != nil {
//		return err
//	}
//	go func() {
//		defer watcher.Close()
//		for {
//			select {
//			case event, ok := <-watcher.Events:
//				if !ok {
//					return
//				}
//				log.Println("event:", event)
//				if event.Has(fsnotify.Write) {
//					log.Println("modified file:", event.Name)
//				}
//			case err, ok := <-watcher.Errors:
//				if !ok {
//					return
//				}
//				log.Println("error:", err)
//			}
//		}
//	}()
//}

func touchGrass(user config.User, addr string) {
	grassBrowser := browser.New(user.Id, addr)
	browsers.Store(grassBrowser.ProxyAddr, grassBrowser)
	for {
		select {
		default:
			<-time.After(time.Second * time.Duration(rand.Intn(60))) //随机延迟
			grassBrowser.Run()
		}
	}
}

func mainLoop() {
	signalChan := make(chan os.Signal, 1)
	signal.Notify(signalChan, syscall.SIGINT)
loop:
	for {
		select {
		case s := <-signalChan:
			switch s {
			case syscall.SIGINT:
				break loop
			}
			break
		}
	}
}