From 2b1e0dbeab6227f67e2ced780f325938202d51c3 Mon Sep 17 00:00:00 2001
From: Keuin <keuinx@gmail.com>
Date: Mon, 12 Sep 2022 04:42:22 +0800
Subject: Improve config `allowed_network_types`. - Rename values to "ipv4",
 "ipv6" and "any". - Validate them when parsing.

---
 main.go | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

(limited to 'main.go')

diff --git a/main.go b/main.go
index d6a433e..c130301 100644
--- a/main.go
+++ b/main.go
@@ -9,13 +9,16 @@ import (
 	"context"
 	"fmt"
 	"github.com/akamensky/argparse"
+	"github.com/keuin/slbr/bilibili"
 	"github.com/keuin/slbr/common"
 	"github.com/keuin/slbr/logging"
 	"github.com/keuin/slbr/recording"
+	"github.com/mitchellh/mapstructure"
 	"github.com/spf13/viper"
 	"log"
 	"os"
 	"os/signal"
+	"reflect"
 	"sync"
 	"syscall"
 )
@@ -104,7 +107,16 @@ func getTasks() (tasks []recording.TaskConfig) {
 			return
 		}
 		var gc GlobalConfig
-		err = viper.Unmarshal(&gc)
+		netType := reflect.TypeOf(bilibili.IP64)
+		err = viper.Unmarshal(&gc, func(conf *mapstructure.DecoderConfig) {
+			conf.DecodeHook = func(from reflect.Value, to reflect.Value) (interface{}, error) {
+				if to.Type() == netType &&
+					bilibili.IpNetType(from.String()).GetDialNetString() == "" {
+					return nil, fmt.Errorf("invalid IpNetType: %v", from.String())
+				}
+				return from.Interface(), nil
+			}
+		})
 		if err != nil {
 			err = fmt.Errorf("cannot parse config file \"%v\": %w", configFile, err)
 			return
-- 
cgit v1.2.3