From 8f07f6f4d7e91a9e4b7164a39759907fac5fb8a1 Mon Sep 17 00:00:00 2001 From: Keuin Date: Sat, 1 Jul 2023 22:06:01 +0800 Subject: Refactor: encapsulate custom url manipulation function into object method. --- recording/runner.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'recording/runner.go') diff --git a/recording/runner.go b/recording/runner.go index 8f892fa..fe7aa21 100644 --- a/recording/runner.go +++ b/recording/runner.go @@ -12,6 +12,7 @@ import ( "fmt" "github.com/keuin/slbr/bilibili" "github.com/keuin/slbr/common" + "github.com/keuin/slbr/common/myurl" "github.com/keuin/slbr/logging" "github.com/samber/mo" "io" @@ -277,7 +278,7 @@ func record( var extName string // the real extension name (without renaming) - originalExtName := mo.TupleToResult(common.GetFileExtensionFromUrl(streamSource.URL)).OrElse("flv") + originalExtName := mo.TupleToResult(myurl.Url(streamSource.URL).FileExtension()).OrElse("flv") if task.Download.UseSpecialExtNameBeforeFinishing { extName = kSpecialExtName -- cgit v1.2.3