config.go

// 设置相关
package main

import (
	"context"
	"encoding/json"
	"io"
	"io/ioutil"
	"os"
	"path/filepath"
	"reflect"
	"runtime"
	"sort"
	"strings"
	"sync"
	"syscall"
	"time"

	"github.com/fsnotify/fsnotify"
	"github.com/google/go-cmp/cmp"
)

const (
	liveFile   = "live.json"   // 主播设置文件名字
	configFile = "config.json" // 设置文件名字
)

var (
	liveFileLocation   string // 主播设置文件位置
	configFileLocation string // 设置文件位置
)

// 主播的设置数据
type streamer struct {
	UID         int     `json:"uid"`         // 主播uid
	Name        string  `json:"name"`        // 主播名字
	Notify      notify  `json:"notify"`      // 开播提醒相关
	Record      bool    `json:"record"`      // 是否自动下载直播视频
	Danmu       bool    `json:"danmu"`       // 是否自动下载直播弹幕
	KeepOnline  bool    `json:"keepOnline"`  // 是否在该主播的直播间挂机，目前主要用于挂粉丝牌等级
	Bitrate     int     `json:"bitrate"`     // 下载直播视频的最高码率
	Directory   string  `json:"directory"`   // 直播视频和弹幕下载结束后会被移动到该文件夹，会覆盖config.json里的设置
	SendQQ      []int64 `json:"sendQQ"`      // 给这些QQ号发送消息，会覆盖config.json里的设置
	SendQQGroup []int64 `json:"sendQQGroup"` // 给这些QQ群发送消息，会覆盖config.json里的设置
}

// 存放主播的设置数据
var streamers struct {
	sync.Mutex                  // crt的锁
	crt        map[int]streamer // 现在的主播的设置数据
	old        map[int]streamer // 旧的主播的设置数据
}

// 设置数据
type configData struct {
	Source         string    `json:"source"`         // 直播源，有hls和flv两种
	Output         string    `json:"output"`         // 直播下载视频格式的后缀名
	WebPort        int       `json:"webPort"`        // web API的本地端口
	Directory      string    `json:"directory"`      // 直播视频和弹幕下载结束后会被移动到该文件夹，会被live.json里的设置覆盖
	Acfun          acfunUser `json:"acfun"`          // AcFun帐号相关
	AutoKeepOnline bool      `json:"autoKeepOnline"` // 是否自动在有守护徽章的直播间挂机
	Mirai          miraiData `json:"mirai"`          // Mirai相关设置
}

// 默认设置
var config = configData{
	Source:    "flv",
	Output:    "mp4",
	WebPort:   51880,
	Directory: "",
	Acfun: acfunUser{
		Account:  "",
		Password: "",
	},
	AutoKeepOnline: false,
	Mirai: miraiData{
		AdminQQ:       0,
		BotQQ:         0,
		BotQQPassword: "",
		SendQQ:        []int64{},
		SendQQGroup:   []int64{},
	},
}

// AcFun用户帐号数据
type acfunUser struct {
	Account  string `json:"account"`  // AcFun帐号邮箱或手机号
	Password string `json:"password"` // AcFun帐号密码
}

// 从streamers.crt获取streamer
func getStreamer(uid int) (streamer, bool) {
	streamers.Lock()
	defer streamers.Unlock()
	s, ok := streamers.crt[uid]
	return s, ok
}

// 将s放进streamers.crt里
func setStreamer(s streamer) {
	streamers.Lock()
	defer streamers.Unlock()
	streamers.crt[s.UID] = s
}

// 去掉slice里重复的元素
func removeDup(s []int64) []int64 {
	seen := make(map[int64]struct{}, len(s))
	i := 0
	for _, v := range s {
		if _, ok := seen[v]; ok {
			continue
		}
		seen[v] = struct{}{}
		s[i] = v
		i++
	}
	return s[:i]
}

// 将map[int]streamer转换为[]streamer，按照uid大小排序
func getStreamers() []streamer {
	streamers.Lock()
	ss := make([]streamer, 0, len(streamers.crt))
	for _, s := range streamers.crt {
		if s.SendQQ == nil {
			s.SendQQ = []int64{}
		}
		if s.SendQQGroup == nil {
			s.SendQQGroup = []int64{}
		}
		ss = append(ss, s)
	}
	streamers.Unlock()
	// 按uid大小排序
	sort.Slice(ss, func(i, j int) bool {
		return ss[i].UID < ss[j].UID
	})
	return ss
}

// 查看设置文件是否存在
func isConfigFileExist(filename string) bool {
	fileLocation := filepath.Join(*configDir, filename)
	info, err := os.Stat(fileLocation)
	if os.IsNotExist(err) {
		return false
	}
	checkErr(err)
	if info.IsDir() {
		lPrintErr(fileLocation + " 不能是目录")
		os.Exit(1)
	}
	return true
}

// 读取live.json
func loadLiveConfig() {
	if isConfigFileExist(liveFile) {
		data, err := ioutil.ReadFile(liveFileLocation)
		checkErr(err)

		if json.Valid(data) {
			var ss []streamer
			err = json.Unmarshal(data, &ss)
			checkErr(err)
			news := make(map[int]streamer)
			for _, s := range ss {
				s.SendQQ = removeDup(s.SendQQ)
				s.SendQQGroup = removeDup(s.SendQQGroup)
				news[s.UID] = s
			}
			streamers.crt = news
		} else {
			lPrintErr("设置文件" + liveFile + "的内容不符合json格式，请检查其内容")
		}
	}
}

// 读取config.json
func loadConfig() {
	if isConfigFileExist(configFile) {
		data, err := ioutil.ReadFile(configFileLocation)
		checkErr(err)

		if json.Valid(data) {
			err = json.Unmarshal(data, &config)
			checkErr(err)
		} else {
			lPrintErr("设置文件" + configFile + "的内容不符合json格式，请检查其内容")
		}
	}
}

// 保存live.json
func saveLiveConfig() {
	data, err := json.MarshalIndent(getStreamers(), "", "    ")
	checkErr(err)

	err = ioutil.WriteFile(liveFileLocation, data, 0644)
	checkErr(err)
}

// 设置里删除指定uid的主播
func deleteStreamer(uid int) bool {
	streamers.Lock()
	if s, ok := streamers.crt[uid]; ok {
		delete(streamers.crt, uid)
		lPrintln("删除" + s.Name + "的设置数据")
	}
	streamers.Unlock()

	saveLiveConfig()
	return true
}

// 监控config.json是否被修改，是的话重新设置
func cycleConfig(ctx context.Context) {
	defer func() {
		if err := recover(); err != nil {
			lPrintErr("Recovering from panic in cycleConfig(), the error is:", err)
			lPrintErr("监控设置文件" + liveFile + "时出错，请重启本程序")
		}
	}()

	lPrintln("开始监控设置文件" + liveFile)

	watcher, err := fsnotify.NewWatcher()
	checkErr(err)
	defer watcher.Close()

	var wg sync.WaitGroup
	wg.Add(1)
	go func() {
		defer wg.Done()
		for {
			select {
			case <-ctx.Done():
				return
			case event, ok := <-watcher.Events:
				if !ok {
					return
				}
				// 很多时候保存文件会分为数段写入，避免读取未完成写入的设置文件
				time.Sleep(100 * time.Millisecond)
			Outer:
				for {
					select {
					case event, ok = <-watcher.Events:
						if !ok {
							return
						}
						time.Sleep(100 * time.Millisecond)
					default:
						if event.Op&fsnotify.Write == fsnotify.Write {
							lPrintln("设置文件" + liveFile + "被修改，重新读取设置")
							loadNewConfig()
						}
						break Outer
					}
				}
			case err, ok := <-watcher.Errors:
				if !ok {
					return
				}
				lPrintErr("监控设置文件"+liveFile+"时出现错误：", err)
			}
		}
	}()

	err = watcher.Add(liveFileLocation)
	checkErr(err)
	wg.Wait()
	lPrintln("停止监控设置文件" + liveFile)
}

// 读取修改后的config.json
func loadNewConfig() {
	streamers.Lock()

	loadLiveConfig()

	for uid, s := range streamers.crt {
		if olds, ok := streamers.old[uid]; ok {
			if !cmp.Equal(s, olds) {
				// olds的设置被修改
				lPrintln(s.longID() + "的设置被修改，重新设置")
				restart := controlMsg{s: s, c: startCycle}
				sInfoMap.Lock()
				if m, ok := sInfoMap.info[s.UID]; ok {
					m.modify = true
					m.ch <- restart
				} else {
					lPrintErr("sInfoMap没有%s的key", s.longID())
				}
				sInfoMap.Unlock()
			}
		} else {
			// s为新增的主播
			lPrintln("新增" + s.longID() + "的设置")
			start := controlMsg{s: s, c: startCycle}
			sInfoMap.Lock()
			if m, ok := sInfoMap.info[s.UID]; ok {
				lPrintErr("sInfoMap不应该有%s的key", s.longID())
				m.modify = true
			} else {
				sInfoMap.info[s.UID] = &streamerInfo{modify: true}
			}
			sInfoMap.Unlock()
			mainCh <- start
		}
	}

	for uid, olds := range streamers.old {
		if _, ok := streamers.crt[uid]; !ok {
			// olds为被删除的主播
			lPrintln(olds.longID() + "的设置被删除")
			stop := controlMsg{s: olds, c: stopCycle}
			sInfoMap.Lock()
			if m, ok := sInfoMap.info[olds.UID]; ok {
				m.modify = true
				m.ch <- stop
			} else {
				lPrintErr("sInfoMap没有%s的key", olds.longID())
			}
			sInfoMap.Unlock()
		}
	}

	oldstreamers := make(map[int]streamer)
	for uid, s := range streamers.crt {
		oldstreamers[uid] = s
	}
	streamers.old = oldstreamers

	streamers.Unlock()
}

// 移动文件
func (s *streamer) moveFile(oldFile string) {
	if oldFile == "" {
		return
	}
	directory := config.Directory
	if s.Directory != "" {
		directory = s.Directory
	}

	if directory != "" {
		info, err := os.Stat(directory)
		checkErr(err)
		if !info.IsDir() {
			lPrintErrf("%s 或 %s 里的directory必须是存在的文件夹：%s", configFile, liveFile, directory)
			return
		}

		_, err = os.Stat(oldFile)
		if os.IsNotExist(err) {
			lPrintErrf("文件 %s 不存在", oldFile)
			return
		}
		checkErr(err)

		filename := filepath.Base(oldFile)
		newFile := filepath.Join(directory, filename)

		// https://github.com/cloudfoundry/bosh-utils/blob/master/fileutil/mover.go
		err = os.Rename(oldFile, newFile)
		if err != nil {
			le, ok := err.(*os.LinkError)
			if !ok {
				lPrintErrf("将文件 %s 移动到 %s 失败：%v", oldFile, newFile, err)
				return
			}

			if le.Err == syscall.EXDEV || (runtime.GOOS == "windows" && le.Err == syscall.Errno(0x11)) {
				inputFile, err := os.Open(oldFile)
				checkErr(err)
				defer inputFile.Close()
				outputFile, err := os.Create(newFile)
				checkErr(err)
				defer outputFile.Close()
				_, err = io.Copy(outputFile, inputFile)
				checkErr(err)
				_ = inputFile.Close()
				err = os.Remove(oldFile)
				checkErr(err)
			} else {
				lPrintErrf("将文件 %s 移动到 %s 失败：%+v", oldFile, newFile, le)
				return
			}
		}

		lPrintf("成功将文件 %s 移动到 %s", oldFile, newFile)
	}
}

// 设置live.json里类型为bool的值
func (s streamer) setBoolConfig(tag string, value bool) bool {
	if v, ok := seekField(&s, tag); ok {
		if v.Kind() != reflect.Bool {
			lPrintErrf("%s不是bool类型，设置失败", tag)
			return false
		}
		if v.Bool() == value {
			lPrintWarnf("%s的%s已经被设置成%v", s.longID(), tag, value)
			return true
		}
		v.SetBool(value)
		setStreamer(s)
		lPrintf("成功设置%s的%s为%v", s.longID(), tag, value)
		saveLiveConfig()
		return true
	}

	lPrintErrf("设置里没有%s，设置失败", tag)
	return false
}

// 递归寻找tag指定的value
func seekField(iface interface{}, tag string) (reflect.Value, bool) {
	ifv := reflect.Indirect(reflect.ValueOf(iface))
	for i := 0; i < ifv.NumField(); i++ {
		value := ifv.Field(i)
		if value.Kind() == reflect.Struct {
			if v, ok := seekField(value.Addr().Interface(), tag); ok {
				return v, true
			}
		}
		if t := ifv.Type().Field(i).Tag.Get("json"); strings.ToLower(t) == tag {
			return value, true
		}
	}

	return reflect.Value{}, false
}