55 lines
1.1 KiB
Go
55 lines
1.1 KiB
Go
package utils
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
)
|
|
|
|
// LinkVideo 链接视频，解析链接，网络文件会下载到临时目录并返回本地路径
|
|
func LinkVideo(link string) (local string, err error) {
|
|
if link == "" {
|
|
return
|
|
}
|
|
u, err := url.Parse(link)
|
|
if err != nil {
|
|
err = fmt.Errorf("URL 解析错误: %v", err)
|
|
} else {
|
|
if u.Scheme == "file" {
|
|
local, _ = strings.CutPrefix(link, "file://")
|
|
} else if u.Scheme == "http" || u.Scheme == "https" {
|
|
p, _ := url.PathUnescape(u.EscapedPath())
|
|
tmpLocal := path.Join(os.TempDir(), path.Base(p))
|
|
err = Download(link, tmpLocal)
|
|
if err != nil {
|
|
err = fmt.Errorf("链接文件获取失败: %v", err)
|
|
return
|
|
}
|
|
local = tmpLocal
|
|
} else {
|
|
err = fmt.Errorf("不支持的链接协议: %v", u.String())
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
// Download 下载文件
|
|
func Download(link string, local string) (err error) {
|
|
resp, err := http.Get(link)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
f, err := os.OpenFile(local, os.O_CREATE|os.O_WRONLY, 0666)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer f.Close()
|
|
_, err = io.Copy(f, resp.Body)
|
|
return
|
|
}
|