main.go

package main

import (
	"flag"
	"log"
	"os"
	"setuServer/config"
	"setuServer/server"
	"strings"
)

var (
	setuApiUrl     = flag.String("setu-api-url", "https://api.lolicon.app/setu/v2", "Api Url of setu")
	wechatUrl      = flag.String("wechat-url", "", "Wechat Web Hook Url")
	intervals      = flag.Uint("intervals", 60, "Intervals of post setu.(second) [Minimum is 10s]")
	r18            = flag.Bool("r18", false, "Post R18 picture")
	atAll          = flag.Bool("at-all", false, "@all group member")
	picMsg         = flag.Bool("pic-msg", false, "Download picture & send picture msg")
	newsMsg        = flag.Bool("news-msg", true, "Send picture use news message")
	dlDir          = flag.String("dl-dir", "./", "Dir of download picture")
	picDump        = flag.Bool("pic-dump", false, "Dump setu pictures to local server")
	dumpServer     = flag.String("dump-server", "", "Server info to dump pictures")
	dumpUrl        = flag.String("dump-url", "", "Url for user get pictures")
	setuTransmit   = flag.Bool("setu-transmit", false, "Transmit setu messages to local server")
	transmitServer = flag.String("transmit-server", "", "Server info to transmit setu")
	tags           = flag.String("tags", "", "Tags of pictures")
	picSize        = flag.String("pic-size", "original", "Size list of pictures")
	once           = flag.Bool("once", false, "Run once and exit")
	keep           = flag.Bool("keep", true, "Keep local original image")
)

// tagsContentAnalysis analyze tags according to rules
func tagsContentAnalysis(tagStr string) (tagArr []string) {
	tagArr = append(tagArr, tagStr)
	return
}

// getPicSize get pictures size list
func getPicSize(sizeStr string) (sizeArr []string) {
	// Note: 'original' size must be exists!!!
	sizeArr = strings.Split(sizeStr, "|")
	for _, v := range sizeArr {
		if v == "original" {
			return
		}
	}
	sizeArr = append(sizeArr, "original")
	return
}

// cmdConfigSetToGlobal store command config to global config.
func cmdConfigSetToGlobal(cfg *config.Config) {
	cfg.SetuApiUrl = *setuApiUrl
	cfg.WeChatUrl = *wechatUrl
	cfg.Intervals = *intervals
	cfg.R18 = *r18
	cfg.AtAll = *atAll
	cfg.PicMsg = *picMsg
	cfg.NewsMsg = *newsMsg
	cfg.PicDownloadDir = *dlDir
	cfg.PicDump = *picDump
	cfg.DumpServer = *dumpServer
	cfg.DumpUrl = *dumpUrl
	cfg.SetuTransmit = *setuTransmit
	cfg.TransmitServer = *transmitServer
	cfg.Tags = tagsContentAnalysis(*tags)
	cfg.PicSize = getPicSize(*picSize)
	cfg.Once = *once
	cfg.Keep = *keep
}

func main() {
	help := flag.Bool("help", false, "show the usage")
	version := flag.Bool("version", false, "version of server")
	flag.Parse()
	if *help {
		flag.Usage()
		os.Exit(0)
	}
	if *version {
		// TODO: Print version info.(Version/GitCommit/CompileTime...)
		log.Println("Version: v0.2.0")
		os.Exit(0)
	}
	if *wechatUrl == "" && *transmitServer == "" {
		log.Println("Error! Both of Wechat hook url and transmit server are empty.")
		flag.Usage()
		os.Exit(1)
	}
	config.InitializeConfig(cmdConfigSetToGlobal)
	server.Run()
}