summaryrefslogtreecommitdiff
path: root/main.go
diff options
context:
space:
mode:
authorKeuin <[email protected]>2023-07-11 22:56:07 +0800
committerKeuin <[email protected]>2023-07-11 22:56:07 +0800
commitf5c63cde56eb35c0125a0545f084441cdd4340ab (patch)
treecf24779daaa3067902bef91d21bcdee7a78fd059 /main.go
parent7fd817f6d1ead3de85294c1893a6e1572a1d12e3 (diff)
Refactor: move data structures to a separate package to avoid circular dependency.
Diffstat (limited to 'main.go')
-rw-r--r--main.go8
1 files changed, 4 insertions, 4 deletions
diff --git a/main.go b/main.go
index 7d3fc37..90e0d6a 100644
--- a/main.go
+++ b/main.go
@@ -9,9 +9,9 @@ import (
"context"
"fmt"
"github.com/akamensky/argparse"
- "github.com/keuin/slbr/bilibili"
"github.com/keuin/slbr/logging"
"github.com/keuin/slbr/recording"
+ "github.com/keuin/slbr/types"
"github.com/mitchellh/mapstructure"
"github.com/samber/mo"
"github.com/spf13/viper"
@@ -110,11 +110,11 @@ func getTasks() (tasks []recording.TaskConfig) {
return
}
var gc GlobalConfig
- netType := reflect.TypeOf(bilibili.IP64)
+ netType := reflect.TypeOf(types.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() == "" {
+ types.IpNetType(from.String()).GetDialNetString() == "" {
return nil, fmt.Errorf("invalid IpNetType: %v", from.String())
}
return from.Interface(), nil
@@ -138,7 +138,7 @@ func getTasks() (tasks []recording.TaskConfig) {
}
for i := 0; i < taskCount; i++ {
tasks[i] = recording.TaskConfig{
- RoomId: bilibili.RoomId((*rooms)[i]),
+ RoomId: types.RoomId((*rooms)[i]),
Transport: recording.DefaultTransportConfig(),
Download: recording.DownloadConfig{
DiskWriteBufferBytes: int64(diskBufSize),