mirror of
https://github.com/putyy/res-downloader.git
synced 2026-01-13 06:54:56 +08:00
Compare commits
25 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
046cbb2b83 | ||
|
|
b562f76c69 | ||
|
|
8aaf95fd36 | ||
|
|
983d72d65a | ||
|
|
86378b9fba | ||
|
|
6b18e7fba1 | ||
|
|
ec11132240 | ||
|
|
dc877bd634 | ||
|
|
00b4bf4068 | ||
|
|
51c43564b6 | ||
|
|
820a2671cf | ||
|
|
3b4443110e | ||
|
|
ffd5b29030 | ||
|
|
779f56dd91 | ||
|
|
2beecdade2 | ||
|
|
bca2e110de | ||
|
|
8d55a86c06 | ||
|
|
f61199bed6 | ||
|
|
2d75bbb5c3 | ||
|
|
55d3f06cb6 | ||
|
|
1809847b8a | ||
|
|
da8e8d9641 | ||
|
|
ead622d95e | ||
|
|
c47fcba36b | ||
|
|
54c0da081c |
@@ -24,7 +24,7 @@ wails build -platform "linux/amd64" -s -skipbindings
|
||||
|
||||
# 打包debian
|
||||
cp build/bin/res-downloader build/linux/Debian/usr/local/bin/
|
||||
echo "$(cat build/linux/Debian/DEBIAN/.control | sed -e "s/{{Version}}/$(jq -r '.info.productVersion' wails.json)/g")" > build/linux/Debian/DEBIAN/control
|
||||
echo "$(cat build/linux/Debian/DEBIAN/.control | sed -e "s/{{Version}}/$(jq -r '.info.productVersion' wails.json)/g" -e "s/{{Architecture}}/amd64/g")" > build/linux/Debian/DEBIAN/control
|
||||
dpkg-deb --build ./build/linux/Debian build/bin/res-downloader_$(jq -r '.info.productVersion' wails.json)_linux_amd64.deb
|
||||
|
||||
# 打包AppImage
|
||||
@@ -64,7 +64,7 @@ wails build -platform "linux/arm64" -s -skipbindings
|
||||
|
||||
# 打包debian
|
||||
cp build/bin/res-downloader build/linux/Debian/usr/local/bin/
|
||||
echo "$(cat build/linux/Debian/DEBIAN/.control | sed -e "s/{{Version}}/$(jq -r '.info.productVersion' wails.json)/g")" > build/linux/Debian/DEBIAN/control
|
||||
echo "$(cat build/linux/Debian/DEBIAN/.control | sed -e "s/{{Version}}/$(jq -r '.info.productVersion' wails.json)/g" -e "s/{{Architecture}}/arm64/g")" > build/linux/Debian/DEBIAN/control
|
||||
dpkg-deb --build ./build/linux/Debian build/bin/res-downloader_$(jq -r '.info.productVersion' wails.json)_linux_arm64.deb
|
||||
|
||||
mv -f build/bin/res-downloader build/bin/res-downloader_$(jq -r '.info.productVersion' wails.json)_linux_arm64
|
||||
|
||||
@@ -2,7 +2,7 @@ Package: res-downloader
|
||||
Version: {{Version}}
|
||||
Section: utils
|
||||
Priority: optional
|
||||
Architecture: amd64
|
||||
Architecture: {{Architecture}}
|
||||
Depends: libwebkit2gtk-4.0-37
|
||||
Maintainer: putyy@qq.com
|
||||
Homepage: https://github.com/putyy/res-downloader
|
||||
|
||||
@@ -14,7 +14,7 @@
|
||||
!define INFO_PRODUCTNAME "res-downloader"
|
||||
!endif
|
||||
!ifndef INFO_PRODUCTVERSION
|
||||
!define INFO_PRODUCTVERSION "3.1.0"
|
||||
!define INFO_PRODUCTVERSION "3.1.3"
|
||||
!endif
|
||||
!ifndef INFO_COPYRIGHT
|
||||
!define INFO_COPYRIGHT "Copyright © 2023"
|
||||
|
||||
30
core/app.go
30
core/app.go
@@ -6,6 +6,7 @@ import (
|
||||
"fmt"
|
||||
"github.com/vrischmann/userdir"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
"regexp"
|
||||
"res-downloader/core/shared"
|
||||
@@ -25,6 +26,7 @@ type App struct {
|
||||
PublicCrt []byte `json:"-"`
|
||||
PrivateKey []byte `json:"-"`
|
||||
IsProxy bool `json:"IsProxy"`
|
||||
IsReset bool `json:"-"`
|
||||
}
|
||||
|
||||
var (
|
||||
@@ -35,6 +37,7 @@ var (
|
||||
systemOnce *SystemSetup
|
||||
proxyOnce *Proxy
|
||||
httpServerOnce *HttpServer
|
||||
ruleOnce *RuleSet
|
||||
)
|
||||
|
||||
func GetApp(assets embed.FS, wjs string) *App {
|
||||
@@ -51,6 +54,7 @@ func GetApp(assets embed.FS, wjs string) *App {
|
||||
Version: version,
|
||||
Description: "res-downloader是一款集网络资源嗅探 + 高速下载功能于一体的软件,高颜值、高性能和多样化,提供个人用户下载自己上传到各大平台的网络资源功能!",
|
||||
Copyright: "Copyright © 2023~" + strconv.Itoa(time.Now().Year()),
|
||||
IsReset: false,
|
||||
PublicCrt: []byte(`-----BEGIN CERTIFICATE-----
|
||||
MIIDwzCCAqugAwIBAgIUFAnC6268dp/z1DR9E1UepiWgWzkwDQYJKoZIhvcNAQEL
|
||||
BQAwcDELMAkGA1UEBhMCQ04xEjAQBgNVBAgMCUNob25ncWluZzESMBAGA1UEBwwJ
|
||||
@@ -117,6 +121,7 @@ ILKEQKmPPzKs7kp/7Nz+2cT3
|
||||
initResource()
|
||||
initHttpServer()
|
||||
initSystem()
|
||||
initRule()
|
||||
}
|
||||
return appOnce
|
||||
}
|
||||
@@ -129,6 +134,10 @@ func (a *App) Startup(ctx context.Context) {
|
||||
func (a *App) OnExit() {
|
||||
a.UnsetSystemProxy()
|
||||
globalLogger.Close()
|
||||
if appOnce.IsReset {
|
||||
err := a.ResetApp()
|
||||
fmt.Println("err:", err)
|
||||
}
|
||||
}
|
||||
|
||||
func (a *App) installCert() (string, error) {
|
||||
@@ -179,3 +188,24 @@ func (a *App) lock() error {
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (a *App) ResetApp() error {
|
||||
exePath, err := os.Executable()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
exePath, err = filepath.Abs(exePath)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_ = os.Remove(filepath.Join(appOnce.UserDir, "install.lock"))
|
||||
_ = os.Remove(filepath.Join(appOnce.UserDir, "pass.cache"))
|
||||
_ = os.Remove(filepath.Join(appOnce.UserDir, "config.json"))
|
||||
_ = os.Remove(filepath.Join(appOnce.UserDir, "cert.crt"))
|
||||
|
||||
cmd := exec.Command(exePath)
|
||||
cmd.Start()
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -1,5 +1,9 @@
|
||||
package core
|
||||
|
||||
import (
|
||||
"github.com/wailsapp/wails/v2/pkg/runtime"
|
||||
)
|
||||
|
||||
type Bind struct {
|
||||
}
|
||||
|
||||
@@ -14,3 +18,8 @@ func (b *Bind) Config() *ResponseData {
|
||||
func (b *Bind) AppInfo() *ResponseData {
|
||||
return httpServerOnce.buildResp(1, "ok", appOnce)
|
||||
}
|
||||
|
||||
func (b *Bind) ResetApp() {
|
||||
appOnce.IsReset = true
|
||||
runtime.Quit(appOnce.ctx)
|
||||
}
|
||||
|
||||
@@ -37,6 +37,7 @@ type Config struct {
|
||||
UseHeaders string `json:"UseHeaders"`
|
||||
InsertTail bool `json:"InsertTail"`
|
||||
MimeMap map[string]MimeInfo `json:"MimeMap"`
|
||||
Rule string `json:"Rule"`
|
||||
}
|
||||
|
||||
var (
|
||||
@@ -65,9 +66,10 @@ func initConfig() *Config {
|
||||
TaskNumber: runtime.NumCPU() * 2,
|
||||
DownNumber: 3,
|
||||
UserAgent: "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36",
|
||||
UseHeaders: "User-Agent,Referer,Authorization,Cookie",
|
||||
UseHeaders: "default",
|
||||
InsertTail: true,
|
||||
MimeMap: getDefaultMimeMap(),
|
||||
Rule: "*",
|
||||
}
|
||||
|
||||
rawDefaults, err := json.Marshal(defaultConfig)
|
||||
@@ -161,6 +163,7 @@ func getDefaultMimeMap() map[string]MimeInfo {
|
||||
"application/vnd.apple.mpegurl": {Type: "m3u8", Suffix: ".m3u8"},
|
||||
"application/x-mpegurl": {Type: "m3u8", Suffix: ".m3u8"},
|
||||
"application/x-mpeg": {Type: "m3u8", Suffix: ".m3u8"},
|
||||
"audio/x-mpegurl": {Type: "m3u8", Suffix: ".m3u8"},
|
||||
"application/pdf": {Type: "pdf", Suffix: ".pdf"},
|
||||
"application/vnd.ms-powerpoint": {Type: "ppt", Suffix: ".ppt"},
|
||||
"application/vnd.openxmlformats-officedocument.presentationml.presentation": {Type: "ppt", Suffix: ".pptx"},
|
||||
@@ -172,7 +175,8 @@ func getDefaultMimeMap() map[string]MimeInfo {
|
||||
"text/rtf": {Type: "doc", Suffix: ".rtf"},
|
||||
"application/vnd.oasis.opendocument.text": {Type: "doc", Suffix: ".odt"},
|
||||
"application/vnd.openxmlformats-officedocument.wordprocessingml.document": {Type: "doc", Suffix: ".docx"},
|
||||
"font/woff": {Type: "font", Suffix: ".woff"},
|
||||
"font/woff": {Type: "font", Suffix: ".woff"},
|
||||
"application/octet-stream": {Type: "stream", Suffix: "default"},
|
||||
}
|
||||
}
|
||||
|
||||
@@ -205,6 +209,7 @@ func getDefaultDownloadDir() string {
|
||||
func (c *Config) setConfig(config Config) {
|
||||
oldProxy := c.UpstreamProxy
|
||||
openProxy := c.OpenProxy
|
||||
oldRule := c.Rule
|
||||
c.Host = config.Host
|
||||
c.Port = config.Port
|
||||
c.Theme = config.Theme
|
||||
@@ -223,10 +228,18 @@ func (c *Config) setConfig(config Config) {
|
||||
c.WxAction = config.WxAction
|
||||
c.UseHeaders = config.UseHeaders
|
||||
c.InsertTail = config.InsertTail
|
||||
c.Rule = config.Rule
|
||||
if oldProxy != c.UpstreamProxy || openProxy != c.OpenProxy {
|
||||
proxyOnce.setTransport()
|
||||
}
|
||||
|
||||
if oldRule != c.Rule {
|
||||
err := ruleOnce.Load(c.Rule)
|
||||
if err != nil {
|
||||
globalLogger.Esg(err, "set rule failed")
|
||||
}
|
||||
}
|
||||
|
||||
mimeMux.Lock()
|
||||
c.MimeMap = config.MimeMap
|
||||
mimeMux.Unlock()
|
||||
@@ -279,6 +292,8 @@ func (c *Config) getConfig(key string) interface{} {
|
||||
mimeMux.RLock()
|
||||
defer mimeMux.RUnlock()
|
||||
return c.MimeMap
|
||||
case "Rule":
|
||||
return c.Rule
|
||||
default:
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -1,12 +1,14 @@
|
||||
package core
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"res-downloader/core/shared"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
@@ -47,9 +49,12 @@ type FileDownloader struct {
|
||||
Headers map[string]string
|
||||
DownloadTaskList []*DownloadTask
|
||||
progressCallback ProgressCallback
|
||||
ctx context.Context
|
||||
cancelFunc context.CancelFunc
|
||||
}
|
||||
|
||||
func NewFileDownloader(url, filename string, totalTasks int, headers map[string]string) *FileDownloader {
|
||||
ctx, cancelFunc := context.WithCancel(context.Background())
|
||||
return &FileDownloader{
|
||||
Url: url,
|
||||
FileName: filename,
|
||||
@@ -59,6 +64,8 @@ func NewFileDownloader(url, filename string, totalTasks int, headers map[string]
|
||||
TotalSize: 0,
|
||||
Headers: headers,
|
||||
DownloadTaskList: make([]*DownloadTask, 0),
|
||||
ctx: ctx,
|
||||
cancelFunc: cancelFunc,
|
||||
}
|
||||
}
|
||||
|
||||
@@ -72,12 +79,44 @@ func (fd *FileDownloader) buildClient() *http.Client {
|
||||
}
|
||||
return &http.Client{
|
||||
Transport: transport,
|
||||
Timeout: 60 * time.Second,
|
||||
}
|
||||
}
|
||||
|
||||
var forbiddenDownloadHeaders = map[string]struct{}{
|
||||
"accept-encoding": {},
|
||||
"content-length": {},
|
||||
"host": {},
|
||||
"connection": {},
|
||||
"keep-alive": {},
|
||||
"proxy-connection": {},
|
||||
"transfer-encoding": {},
|
||||
|
||||
"sec-fetch-site": {},
|
||||
"sec-fetch-mode": {},
|
||||
"sec-fetch-dest": {},
|
||||
"sec-fetch-user": {},
|
||||
"sec-ch-ua": {},
|
||||
"sec-ch-ua-mobile": {},
|
||||
"sec-ch-ua-platform": {},
|
||||
|
||||
"if-none-match": {},
|
||||
"if-modified-since": {},
|
||||
|
||||
"x-forwarded-for": {},
|
||||
"x-real-ip": {},
|
||||
}
|
||||
|
||||
func (fd *FileDownloader) setHeaders(request *http.Request) {
|
||||
for key, value := range fd.Headers {
|
||||
if globalConfig.UseHeaders == "default" {
|
||||
lk := strings.ToLower(key)
|
||||
if _, forbidden := forbiddenDownloadHeaders[lk]; forbidden {
|
||||
continue
|
||||
}
|
||||
request.Header.Set(key, value)
|
||||
continue
|
||||
}
|
||||
|
||||
if strings.Contains(globalConfig.UseHeaders, key) {
|
||||
request.Header.Set(key, value)
|
||||
}
|
||||
@@ -143,6 +182,9 @@ func (fd *FileDownloader) init() error {
|
||||
if err := os.MkdirAll(dir, os.ModePerm); err != nil {
|
||||
return fmt.Errorf("create directory failed: %w", err)
|
||||
}
|
||||
|
||||
fd.FileName = shared.GetUniqueFileName(fd.FileName)
|
||||
|
||||
fd.File, err = os.OpenFile(fd.FileName, os.O_RDWR|os.O_CREATE, 0644)
|
||||
if err != nil {
|
||||
return fmt.Errorf("file open failed: %w", err)
|
||||
@@ -268,11 +310,21 @@ func (fd *FileDownloader) startDownloadTask(wg *sync.WaitGroup, progressChan cha
|
||||
return
|
||||
}
|
||||
|
||||
if strings.Contains(err.Error(), "cancelled") {
|
||||
errorChan <- err
|
||||
return
|
||||
}
|
||||
|
||||
task.err = err
|
||||
globalLogger.Warn().Msgf("Task %d failed (attempt %d/%d): %v", task.taskID, retries+1, MaxRetries, err)
|
||||
|
||||
if retries < MaxRetries-1 {
|
||||
time.Sleep(RetryDelay)
|
||||
select {
|
||||
case <-fd.ctx.Done():
|
||||
errorChan <- fmt.Errorf("task %d cancelled during retry", task.taskID)
|
||||
return
|
||||
case <-time.After(RetryDelay):
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -280,7 +332,13 @@ func (fd *FileDownloader) startDownloadTask(wg *sync.WaitGroup, progressChan cha
|
||||
}
|
||||
|
||||
func (fd *FileDownloader) doDownloadTask(progressChan chan ProgressChan, task *DownloadTask) error {
|
||||
request, err := http.NewRequest("GET", fd.Url, nil)
|
||||
select {
|
||||
case <-fd.ctx.Done():
|
||||
return fmt.Errorf("download cancelled")
|
||||
default:
|
||||
}
|
||||
|
||||
request, err := http.NewRequestWithContext(fd.ctx, "GET", fd.Url, nil)
|
||||
if err != nil {
|
||||
return fmt.Errorf("create request failed: %w", err)
|
||||
}
|
||||
@@ -307,6 +365,12 @@ func (fd *FileDownloader) doDownloadTask(progressChan chan ProgressChan, task *D
|
||||
|
||||
buf := make([]byte, 32*1024)
|
||||
for {
|
||||
select {
|
||||
case <-fd.ctx.Done():
|
||||
return fmt.Errorf("download cancelled")
|
||||
default:
|
||||
}
|
||||
|
||||
n, err := resp.Body.Read(buf)
|
||||
if n > 0 {
|
||||
writeSize := int64(n)
|
||||
@@ -364,3 +428,17 @@ func (fd *FileDownloader) Start() error {
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func (fd *FileDownloader) Cancel() {
|
||||
if fd.cancelFunc != nil {
|
||||
fd.cancelFunc()
|
||||
}
|
||||
|
||||
if fd.File != nil {
|
||||
fd.File.Close()
|
||||
}
|
||||
|
||||
if fd.FileName != "" {
|
||||
_ = os.Remove(fd.FileName)
|
||||
}
|
||||
}
|
||||
|
||||
45
core/http.go
45
core/http.go
@@ -84,8 +84,6 @@ func (h *HttpServer) preview(w http.ResponseWriter, r *http.Request) {
|
||||
request.Header.Set("Range", rangeHeader)
|
||||
}
|
||||
|
||||
//request.Header.Set("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36")
|
||||
//request.Header.Set("Referer", parsedURL.Scheme+"://"+parsedURL.Host+"/")
|
||||
resp, err := http.DefaultClient.Do(request)
|
||||
if err != nil {
|
||||
http.Error(w, "Failed to fetch the resource", http.StatusInternalServerError)
|
||||
@@ -93,12 +91,15 @@ func (h *HttpServer) preview(w http.ResponseWriter, r *http.Request) {
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
w.Header().Set("Content-Type", resp.Header.Get("Content-Type"))
|
||||
w.WriteHeader(resp.StatusCode)
|
||||
|
||||
if contentRange := resp.Header.Get("Content-Range"); contentRange != "" {
|
||||
w.Header().Set("Content-Range", contentRange)
|
||||
for k, v := range resp.Header {
|
||||
if strings.ToLower(k) == "access-control-allow-origin" {
|
||||
continue
|
||||
}
|
||||
for _, vv := range v {
|
||||
w.Header().Add(k, vv)
|
||||
}
|
||||
}
|
||||
w.WriteHeader(resp.StatusCode)
|
||||
|
||||
_, err = io.Copy(w, resp.Body)
|
||||
if err != nil {
|
||||
@@ -323,18 +324,20 @@ func (h *HttpServer) clear(w http.ResponseWriter, r *http.Request) {
|
||||
|
||||
func (h *HttpServer) delete(w http.ResponseWriter, r *http.Request) {
|
||||
var data struct {
|
||||
Sign string `json:"sign"`
|
||||
Sign []string `json:"sign"`
|
||||
}
|
||||
err := json.NewDecoder(r.Body).Decode(&data)
|
||||
if err == nil && data.Sign != "" {
|
||||
resourceOnce.delete(data.Sign)
|
||||
if err == nil && len(data.Sign) > 0 {
|
||||
for _, v := range data.Sign {
|
||||
resourceOnce.delete(v)
|
||||
}
|
||||
}
|
||||
h.success(w)
|
||||
}
|
||||
|
||||
func (h *HttpServer) download(w http.ResponseWriter, r *http.Request) {
|
||||
var data struct {
|
||||
MediaInfo
|
||||
shared.MediaInfo
|
||||
DecodeStr string `json:"decodeStr"`
|
||||
}
|
||||
if err := json.NewDecoder(r.Body).Decode(&data); err != nil {
|
||||
@@ -345,9 +348,27 @@ func (h *HttpServer) download(w http.ResponseWriter, r *http.Request) {
|
||||
h.success(w)
|
||||
}
|
||||
|
||||
func (h *HttpServer) cancel(w http.ResponseWriter, r *http.Request) {
|
||||
var data struct {
|
||||
shared.MediaInfo
|
||||
}
|
||||
|
||||
if err := json.NewDecoder(r.Body).Decode(&data); err != nil {
|
||||
h.error(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
err := resourceOnce.cancel(data.Id)
|
||||
if err != nil {
|
||||
h.error(w, err.Error())
|
||||
return
|
||||
}
|
||||
h.success(w)
|
||||
}
|
||||
|
||||
func (h *HttpServer) wxFileDecode(w http.ResponseWriter, r *http.Request) {
|
||||
var data struct {
|
||||
MediaInfo
|
||||
shared.MediaInfo
|
||||
Filename string `json:"filename"`
|
||||
DecodeStr string `json:"decodeStr"`
|
||||
}
|
||||
|
||||
@@ -17,8 +17,10 @@ func Middleware(next http.Handler) http.Handler {
|
||||
func HandleApi(w http.ResponseWriter, r *http.Request) bool {
|
||||
if strings.HasPrefix(r.URL.Path, "/api") {
|
||||
w.Header().Set("Access-Control-Allow-Origin", "*")
|
||||
w.Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS")
|
||||
w.Header().Set("Access-Control-Allow-Headers", "Content-Type")
|
||||
if r.URL.Path != "/api/preview" {
|
||||
w.Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS")
|
||||
w.Header().Set("Access-Control-Allow-Headers", "Content-Type")
|
||||
}
|
||||
if r.Method == http.MethodOptions {
|
||||
w.WriteHeader(http.StatusNoContent)
|
||||
return true
|
||||
@@ -56,6 +58,8 @@ func HandleApi(w http.ResponseWriter, r *http.Request) bool {
|
||||
httpServerOnce.delete(w, r)
|
||||
case "/api/download":
|
||||
httpServerOnce.download(w, r)
|
||||
case "/api/cancel":
|
||||
httpServerOnce.cancel(w, r)
|
||||
case "/api/wx-file-decode":
|
||||
httpServerOnce.wxFileDecode(w, r)
|
||||
case "/api/batch-export":
|
||||
|
||||
@@ -5,8 +5,10 @@ import (
|
||||
"github.com/elazarl/goproxy"
|
||||
gonanoid "github.com/matoous/go-nanoid/v2"
|
||||
"net/http"
|
||||
"path/filepath"
|
||||
"res-downloader/core/shared"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type DefaultPlugin struct {
|
||||
@@ -26,7 +28,7 @@ func (p *DefaultPlugin) OnRequest(r *http.Request, ctx *goproxy.ProxyCtx) (*http
|
||||
}
|
||||
|
||||
func (p *DefaultPlugin) OnResponse(resp *http.Response, ctx *goproxy.ProxyCtx) *http.Response {
|
||||
if resp == nil || resp.Request == nil || (resp.StatusCode != 200 && resp.StatusCode != 206) {
|
||||
if resp == nil || resp.Request == nil || (resp.StatusCode != 200 && resp.StatusCode != 206 && resp.StatusCode != 304) {
|
||||
return resp
|
||||
}
|
||||
|
||||
@@ -39,6 +41,13 @@ func (p *DefaultPlugin) OnResponse(resp *http.Response, ctx *goproxy.ProxyCtx) *
|
||||
isAll, _ := p.bridge.GetResType("all")
|
||||
isClassify, _ := p.bridge.GetResType(classify)
|
||||
|
||||
if suffix == "default" {
|
||||
ext := filepath.Ext(filepath.Base(strings.Split(strings.Split(rawUrl, "?")[0], "#")[0]))
|
||||
if ext != "" {
|
||||
suffix = ext
|
||||
}
|
||||
}
|
||||
|
||||
urlSign := shared.Md5(rawUrl)
|
||||
if ok := p.bridge.MediaIsMarked(urlSign); !ok && (isAll || isClassify) {
|
||||
value, _ := strconv.ParseFloat(resp.Header.Get("content-length"), 64)
|
||||
@@ -51,7 +60,7 @@ func (p *DefaultPlugin) OnResponse(resp *http.Response, ctx *goproxy.ProxyCtx) *
|
||||
Url: rawUrl,
|
||||
UrlSign: urlSign,
|
||||
CoverUrl: "",
|
||||
Size: shared.FormatSize(value),
|
||||
Size: value,
|
||||
Domain: shared.GetTopLevelDomain(rawUrl),
|
||||
Classify: classify,
|
||||
Suffix: suffix,
|
||||
|
||||
@@ -53,6 +53,9 @@ func (p *QqPlugin) OnResponse(resp *http.Response, ctx *goproxy.ProxyCtx) *http.
|
||||
|
||||
classify, _ := p.bridge.TypeSuffix(resp.Header.Get("Content-Type"))
|
||||
if classify == "video" && strings.HasSuffix(host, "finder.video.qq.com") {
|
||||
if strings.Contains(resp.Request.Header.Get("Origin"), "mp.weixin.qq.com") {
|
||||
return nil
|
||||
}
|
||||
return resp
|
||||
}
|
||||
|
||||
@@ -163,7 +166,7 @@ func (p *QqPlugin) handleMedia(body []byte) {
|
||||
Url: rawUrl,
|
||||
UrlSign: urlSign,
|
||||
CoverUrl: "",
|
||||
Size: "0",
|
||||
Size: 0,
|
||||
Domain: shared.GetTopLevelDomain(rawUrl),
|
||||
Classify: "video",
|
||||
Suffix: ".mp4",
|
||||
@@ -198,10 +201,10 @@ func (p *QqPlugin) handleMedia(body []byte) {
|
||||
|
||||
switch size := firstMedia["fileSize"].(type) {
|
||||
case float64:
|
||||
res.Size = shared.FormatSize(size)
|
||||
res.Size = size
|
||||
case string:
|
||||
if value, err := strconv.ParseFloat(size, 64); err == nil {
|
||||
res.Size = shared.FormatSize(value)
|
||||
res.Size = value
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -21,23 +21,6 @@ type Proxy struct {
|
||||
Is bool
|
||||
}
|
||||
|
||||
type MediaInfo struct {
|
||||
Id string
|
||||
Url string
|
||||
UrlSign string
|
||||
CoverUrl string
|
||||
Size string
|
||||
Domain string
|
||||
Classify string
|
||||
Suffix string
|
||||
SavePath string
|
||||
Status string
|
||||
DecodeKey string
|
||||
Description string
|
||||
ContentType string
|
||||
OtherData map[string]string
|
||||
}
|
||||
|
||||
var pluginRegistry = make(map[string]shared.Plugin)
|
||||
|
||||
func init() {
|
||||
@@ -97,7 +80,14 @@ func (p *Proxy) Startup() {
|
||||
//p.Proxy.KeepDestinationHeaders = true
|
||||
//p.Proxy.Verbose = false
|
||||
p.setTransport()
|
||||
p.Proxy.OnRequest().HandleConnect(goproxy.AlwaysMitm)
|
||||
//p.Proxy.OnRequest().HandleConnect(goproxy.AlwaysMitm)
|
||||
p.Proxy.OnRequest().HandleConnectFunc(func(host string, ctx *goproxy.ProxyCtx) (*goproxy.ConnectAction, string) {
|
||||
if ruleOnce.shouldMitm(host) {
|
||||
return goproxy.MitmConnect, host
|
||||
}
|
||||
return goproxy.OkConnect, host
|
||||
})
|
||||
|
||||
p.Proxy.OnRequest().DoFunc(p.httpRequestEvent)
|
||||
p.Proxy.OnResponse().DoFunc(p.httpResponseEvent)
|
||||
}
|
||||
@@ -130,10 +120,14 @@ func (p *Proxy) setTransport() {
|
||||
IdleConnTimeout: 30 * time.Second,
|
||||
}
|
||||
|
||||
p.Proxy.ConnectDial = nil
|
||||
p.Proxy.ConnectDialWithReq = nil
|
||||
|
||||
if globalConfig.UpstreamProxy != "" && globalConfig.OpenProxy && !strings.Contains(globalConfig.UpstreamProxy, globalConfig.Port) {
|
||||
proxyURL, err := url.Parse(globalConfig.UpstreamProxy)
|
||||
if err == nil {
|
||||
transport.Proxy = http.ProxyURL(proxyURL)
|
||||
p.Proxy.ConnectDial = p.Proxy.NewConnectDialToProxy(globalConfig.UpstreamProxy)
|
||||
}
|
||||
}
|
||||
p.Proxy.Tr = transport
|
||||
|
||||
@@ -3,6 +3,7 @@ package core
|
||||
import (
|
||||
"encoding/base64"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/url"
|
||||
@@ -22,6 +23,7 @@ type WxFileDecodeResult struct {
|
||||
|
||||
type Resource struct {
|
||||
mediaMark sync.Map
|
||||
tasks sync.Map
|
||||
resType map[string]bool
|
||||
resTypeMux sync.RWMutex
|
||||
}
|
||||
@@ -86,13 +88,27 @@ func (r *Resource) delete(sign string) {
|
||||
r.mediaMark.Delete(sign)
|
||||
}
|
||||
|
||||
func (r *Resource) download(mediaInfo MediaInfo, decodeStr string) {
|
||||
func (r *Resource) cancel(id string) error {
|
||||
if d, ok := r.tasks.Load(id); ok {
|
||||
d.(*FileDownloader).Cancel()
|
||||
r.tasks.Delete(id) // 可选:取消后清理
|
||||
return nil
|
||||
}
|
||||
return errors.New("task not found")
|
||||
}
|
||||
|
||||
func (r *Resource) download(mediaInfo shared.MediaInfo, decodeStr string) {
|
||||
if globalConfig.SaveDirectory == "" {
|
||||
return
|
||||
}
|
||||
go func(mediaInfo MediaInfo) {
|
||||
go func(mediaInfo shared.MediaInfo) {
|
||||
rawUrl := mediaInfo.Url
|
||||
fileName := shared.Md5(rawUrl)
|
||||
|
||||
if v := shared.GetFileNameFromURL(rawUrl); v != "" {
|
||||
fileName = v
|
||||
}
|
||||
|
||||
if mediaInfo.Description != "" {
|
||||
fileName = regexp.MustCompile(`[^\w\p{Han}]`).ReplaceAllString(mediaInfo.Description, "")
|
||||
fileLen := globalConfig.FilenameLen
|
||||
@@ -107,9 +123,13 @@ func (r *Resource) download(mediaInfo MediaInfo, decodeStr string) {
|
||||
}
|
||||
|
||||
if globalConfig.FilenameTime {
|
||||
mediaInfo.SavePath = filepath.Join(globalConfig.SaveDirectory, fileName+"_"+shared.GetCurrentDateTimeFormatted()+mediaInfo.Suffix)
|
||||
mediaInfo.SavePath = filepath.Join(globalConfig.SaveDirectory, fileName+"_"+shared.GetCurrentDateTimeFormatted())
|
||||
} else {
|
||||
mediaInfo.SavePath = filepath.Join(globalConfig.SaveDirectory, fileName+mediaInfo.Suffix)
|
||||
mediaInfo.SavePath = filepath.Join(globalConfig.SaveDirectory, fileName)
|
||||
}
|
||||
|
||||
if !strings.HasSuffix(mediaInfo.SavePath, mediaInfo.Suffix) {
|
||||
mediaInfo.SavePath = mediaInfo.SavePath + mediaInfo.Suffix
|
||||
}
|
||||
|
||||
if strings.Contains(rawUrl, "qq.com") {
|
||||
@@ -140,9 +160,13 @@ func (r *Resource) download(mediaInfo MediaInfo, decodeStr string) {
|
||||
downloader.progressCallback = func(totalDownloaded, totalSize float64, taskID int, taskProgress float64) {
|
||||
r.progressEventsEmit(mediaInfo, strconv.Itoa(int(totalDownloaded*100/totalSize))+"%", shared.DownloadStatusRunning)
|
||||
}
|
||||
r.tasks.Store(mediaInfo.Id, downloader)
|
||||
err := downloader.Start()
|
||||
mediaInfo.SavePath = downloader.FileName
|
||||
if err != nil {
|
||||
r.progressEventsEmit(mediaInfo, err.Error())
|
||||
if !strings.Contains(err.Error(), "cancelled") {
|
||||
r.progressEventsEmit(mediaInfo, err.Error())
|
||||
}
|
||||
return
|
||||
}
|
||||
if decodeStr != "" {
|
||||
@@ -156,7 +180,7 @@ func (r *Resource) download(mediaInfo MediaInfo, decodeStr string) {
|
||||
}(mediaInfo)
|
||||
}
|
||||
|
||||
func (r *Resource) parseHeaders(mediaInfo MediaInfo) (map[string]string, error) {
|
||||
func (r *Resource) parseHeaders(mediaInfo shared.MediaInfo) (map[string]string, error) {
|
||||
headers := make(map[string]string)
|
||||
|
||||
if hh, ok := mediaInfo.OtherData["headers"]; ok {
|
||||
@@ -175,7 +199,7 @@ func (r *Resource) parseHeaders(mediaInfo MediaInfo) (map[string]string, error)
|
||||
return headers, nil
|
||||
}
|
||||
|
||||
func (r *Resource) wxFileDecode(mediaInfo MediaInfo, fileName, decodeStr string) (string, error) {
|
||||
func (r *Resource) wxFileDecode(mediaInfo shared.MediaInfo, fileName, decodeStr string) (string, error) {
|
||||
sourceFile, err := os.Open(fileName)
|
||||
if err != nil {
|
||||
return "", err
|
||||
@@ -200,7 +224,7 @@ func (r *Resource) wxFileDecode(mediaInfo MediaInfo, fileName, decodeStr string)
|
||||
return mediaInfo.SavePath, nil
|
||||
}
|
||||
|
||||
func (r *Resource) progressEventsEmit(mediaInfo MediaInfo, args ...string) {
|
||||
func (r *Resource) progressEventsEmit(mediaInfo shared.MediaInfo, args ...string) {
|
||||
Status := shared.DownloadStatusError
|
||||
Message := "ok"
|
||||
|
||||
|
||||
126
core/rule.go
Normal file
126
core/rule.go
Normal file
@@ -0,0 +1,126 @@
|
||||
package core
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"net"
|
||||
"strings"
|
||||
"sync"
|
||||
)
|
||||
|
||||
type Rule struct {
|
||||
raw string
|
||||
isNeg bool // 是否否定规则(以 ! 开头)
|
||||
isWildcard bool // 是否为 *.domain 形式
|
||||
isAll bool
|
||||
domain string // 域名部分,不含 "*."
|
||||
}
|
||||
|
||||
type RuleSet struct {
|
||||
mu sync.RWMutex
|
||||
rules []Rule
|
||||
}
|
||||
|
||||
func initRule() *RuleSet {
|
||||
if ruleOnce == nil {
|
||||
ruleOnce = &RuleSet{}
|
||||
err := ruleOnce.Load(globalConfig.Rule)
|
||||
if err != nil {
|
||||
globalLogger.Esg(err, "init rule failed")
|
||||
return nil
|
||||
}
|
||||
}
|
||||
return ruleOnce
|
||||
}
|
||||
|
||||
func (r *RuleSet) Load(rs string) error {
|
||||
reader := strings.NewReader(rs)
|
||||
scanner := bufio.NewScanner(reader)
|
||||
|
||||
var rules []Rule
|
||||
for scanner.Scan() {
|
||||
line := strings.TrimSpace(scanner.Text())
|
||||
if line == "" || strings.HasPrefix(line, "#") {
|
||||
continue
|
||||
}
|
||||
|
||||
isNeg := false
|
||||
if strings.HasPrefix(line, "!") {
|
||||
isNeg = true
|
||||
line = strings.TrimSpace(line[1:])
|
||||
if line == "" {
|
||||
continue
|
||||
}
|
||||
}
|
||||
|
||||
if line == "*" {
|
||||
rules = append(rules, Rule{
|
||||
raw: "*",
|
||||
isAll: true,
|
||||
isNeg: isNeg,
|
||||
})
|
||||
continue
|
||||
}
|
||||
|
||||
isWildcard := false
|
||||
domain := line
|
||||
if strings.HasPrefix(line, "*.") {
|
||||
isWildcard = true
|
||||
domain = line[2:]
|
||||
}
|
||||
|
||||
rules = append(rules, Rule{
|
||||
raw: line,
|
||||
isNeg: isNeg,
|
||||
isWildcard: isWildcard,
|
||||
domain: strings.ToLower(domain),
|
||||
})
|
||||
}
|
||||
|
||||
if err := scanner.Err(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
r.mu.Lock()
|
||||
r.rules = rules
|
||||
r.mu.Unlock()
|
||||
return nil
|
||||
}
|
||||
|
||||
// shouldMitm: 根据当前规则集判断是否对 host 做 MITM
|
||||
// host 可能带端口(example.com:443),函数会只匹配 hostname 部分
|
||||
// 返回 true => MITM(解密),false => 透传
|
||||
func (r *RuleSet) shouldMitm(host string) bool {
|
||||
h := host
|
||||
if strings.HasPrefix(h, "[") {
|
||||
if hostSplitIdx := strings.LastIndex(h, "]"); hostSplitIdx != -1 {
|
||||
h = h[:hostSplitIdx+1]
|
||||
}
|
||||
}
|
||||
if hp, _, err := net.SplitHostPort(host); err == nil {
|
||||
h = hp
|
||||
}
|
||||
h = strings.ToLower(strings.Trim(h, "[]"))
|
||||
|
||||
r.mu.RLock()
|
||||
defer r.mu.RUnlock()
|
||||
|
||||
action := false
|
||||
for _, rule := range r.rules {
|
||||
if rule.isAll {
|
||||
action = !rule.isNeg
|
||||
continue
|
||||
}
|
||||
|
||||
if rule.isWildcard {
|
||||
if h == rule.domain || strings.HasSuffix(h, "."+rule.domain) {
|
||||
action = !rule.isNeg
|
||||
}
|
||||
continue
|
||||
}
|
||||
|
||||
if h == rule.domain {
|
||||
action = !rule.isNeg
|
||||
}
|
||||
}
|
||||
return action
|
||||
}
|
||||
@@ -5,7 +5,7 @@ type MediaInfo struct {
|
||||
Url string
|
||||
UrlSign string
|
||||
CoverUrl string
|
||||
Size string
|
||||
Size float64
|
||||
Domain string
|
||||
Classify string
|
||||
Suffix string
|
||||
|
||||
@@ -9,7 +9,11 @@ import (
|
||||
"net/url"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path"
|
||||
"path/filepath"
|
||||
"regexp"
|
||||
sysRuntime "runtime"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
@@ -61,6 +65,45 @@ func IsDevelopment() bool {
|
||||
return os.Getenv("APP_ENV") == "development"
|
||||
}
|
||||
|
||||
func GetFileNameFromURL(rawUrl string) string {
|
||||
parsedURL, err := url.Parse(rawUrl)
|
||||
if err != nil {
|
||||
return ""
|
||||
}
|
||||
|
||||
fileName := path.Base(parsedURL.Path)
|
||||
if fileName == "" || fileName == "/" {
|
||||
return ""
|
||||
}
|
||||
|
||||
if decoded, err := url.QueryUnescape(fileName); err == nil {
|
||||
fileName = decoded
|
||||
}
|
||||
|
||||
re := regexp.MustCompile(`[<>:"/\\|?*]`)
|
||||
fileName = re.ReplaceAllString(fileName, "_")
|
||||
|
||||
fileName = strings.TrimRightFunc(fileName, func(r rune) bool {
|
||||
return r == '.' || r == ' '
|
||||
})
|
||||
|
||||
const maxFileNameLen = 255
|
||||
runes := []rune(fileName)
|
||||
if len(runes) > maxFileNameLen {
|
||||
ext := path.Ext(fileName)
|
||||
name := strings.TrimSuffix(fileName, ext)
|
||||
|
||||
runes = []rune(name)
|
||||
if len(runes) > maxFileNameLen-len(ext) {
|
||||
runes = runes[:maxFileNameLen-len(ext)]
|
||||
}
|
||||
name = string(runes)
|
||||
fileName = name + ext
|
||||
}
|
||||
|
||||
return fileName
|
||||
}
|
||||
|
||||
func GetCurrentDateTimeFormatted() string {
|
||||
now := time.Now()
|
||||
return fmt.Sprintf("%04d%02d%02d%02d%02d%02d",
|
||||
@@ -72,6 +115,24 @@ func GetCurrentDateTimeFormatted() string {
|
||||
now.Second())
|
||||
}
|
||||
|
||||
func GetUniqueFileName(filePath string) string {
|
||||
if !FileExist(filePath) {
|
||||
return filePath
|
||||
}
|
||||
|
||||
ext := filepath.Ext(filePath)
|
||||
baseName := strings.TrimSuffix(filePath, ext)
|
||||
count := 1
|
||||
|
||||
for {
|
||||
newFileName := fmt.Sprintf("%s(%d)%s", baseName, count, ext)
|
||||
if !FileExist(newFileName) {
|
||||
return newFileName
|
||||
}
|
||||
count++
|
||||
}
|
||||
}
|
||||
|
||||
func OpenFolder(filePath string) error {
|
||||
var cmd *exec.Cmd
|
||||
|
||||
|
||||
2
frontend/components.d.ts
vendored
2
frontend/components.d.ts
vendored
@@ -15,6 +15,7 @@ declare module 'vue' {
|
||||
NaiveProvider: typeof import('./src/components/NaiveProvider.vue')['default']
|
||||
NButton: typeof import('naive-ui')['NButton']
|
||||
NButtonGroup: typeof import('naive-ui')['NButtonGroup']
|
||||
NCheckbox: typeof import('naive-ui')['NCheckbox']
|
||||
NConfigProvider: typeof import('naive-ui')['NConfigProvider']
|
||||
NDataTable: typeof import('naive-ui')['NDataTable']
|
||||
NDialogProvider: typeof import('naive-ui')['NDialogProvider']
|
||||
@@ -48,6 +49,7 @@ declare module 'vue' {
|
||||
RouterView: typeof import('vue-router')['RouterView']
|
||||
Screen: typeof import('./src/components/Screen.vue')['default']
|
||||
ShowLoading: typeof import('./src/components/ShowLoading.vue')['default']
|
||||
ShowOrEdit: typeof import('./src/components/ShowOrEdit.vue')['default']
|
||||
Sider: typeof import('./src/components/layout/Sider.vue')['default']
|
||||
}
|
||||
}
|
||||
|
||||
@@ -92,6 +92,13 @@ export default {
|
||||
data: data
|
||||
})
|
||||
},
|
||||
cancel(data: object) {
|
||||
return request({
|
||||
url: 'api/cancel',
|
||||
method: 'post',
|
||||
data: data
|
||||
})
|
||||
},
|
||||
download(data: object) {
|
||||
return request({
|
||||
url: 'api/download',
|
||||
|
||||
@@ -7,4 +7,12 @@
|
||||
#app {
|
||||
width: 100vw;
|
||||
height: 100vh;
|
||||
}
|
||||
|
||||
.ellipsis-2 {
|
||||
display: -webkit-box;
|
||||
overflow: hidden;
|
||||
text-overflow: ellipsis;
|
||||
-webkit-line-clamp: 2;
|
||||
-webkit-box-orient: vertical;
|
||||
}
|
||||
@@ -23,6 +23,16 @@
|
||||
</NIcon>
|
||||
</template>
|
||||
<div class="flex flex-col">
|
||||
<div class="flex items-center justify-start p-1.5 cursor-pointer" v-if="row.Status === 'running' || row.Status === 'pending'" @click="action('cancel')">
|
||||
<n-icon
|
||||
size="28"
|
||||
class="text-red-500 dark:text-red-300 bg-red-500/20 dark:bg-red-500/30 rounded-full flex items-center justify-center p-1.5 cursor-pointer hover:bg-red-500/40 transition-colors"
|
||||
>
|
||||
<CloseOutline/>
|
||||
</n-icon>
|
||||
<span class="ml-1">{{ t("index.cancel_down") }}</span>
|
||||
</div>
|
||||
|
||||
<div class="flex items-center justify-start p-1.5 cursor-pointer" @click="action('copy')">
|
||||
<n-icon
|
||||
size="28"
|
||||
@@ -76,6 +86,7 @@ import {
|
||||
LockOpenSharp,
|
||||
LinkOutline,
|
||||
GridSharp,
|
||||
CloseOutline,
|
||||
TrashOutline
|
||||
} from "@vicons/ionicons5"
|
||||
|
||||
|
||||
@@ -18,6 +18,14 @@
|
||||
<span class="ml-1">{{ t("index.direct_download") }}</span>
|
||||
</div>
|
||||
|
||||
<div class="flex items-center justify-start p-1.5">
|
||||
<n-icon size="28"
|
||||
class="text-red-500 dark:text-red-300 bg-red-500/20 dark:bg-red-500/30 rounded-full flex items-center justify-center p-1.5 cursor-pointer hover:bg-red-500/40 transition-colors">
|
||||
<CloseOutline/>
|
||||
</n-icon>
|
||||
<span class="ml-1">{{ t("index.cancel_down") }}</span>
|
||||
</div>
|
||||
|
||||
<div class="flex items-center justify-start p-1.5">
|
||||
<n-icon
|
||||
size="28"
|
||||
@@ -91,6 +99,7 @@ import {
|
||||
LinkOutline,
|
||||
LockOpenSharp,
|
||||
GridSharp,
|
||||
CloseOutline,
|
||||
TrashOutline
|
||||
} from "@vicons/ionicons5"
|
||||
|
||||
|
||||
@@ -84,7 +84,7 @@ const playFlvStream = () => {
|
||||
try {
|
||||
if (!flvjs.isSupported() || !videoPlayer.value) return
|
||||
|
||||
flvPlayer = flvjs.createPlayer({ type: "flv", url: props.previewRow.Url })
|
||||
flvPlayer = flvjs.createPlayer({ type: "flv", url: window?.$baseUrl + "/api/preview?url=" + encodeURIComponent(props.previewRow.Url) })
|
||||
flvPlayer.attachMediaElement(videoPlayer.value)
|
||||
flvPlayer.load()
|
||||
flvPlayer.play()
|
||||
@@ -105,7 +105,7 @@ const setupVideoJsPlayer = () => {
|
||||
}
|
||||
|
||||
player.src({
|
||||
src: props.previewRow.Url,
|
||||
src: window?.$baseUrl + "/api/preview?url=" + encodeURIComponent(props.previewRow.Url),
|
||||
type: props.previewRow.ContentType,
|
||||
withCredentials: true,
|
||||
})
|
||||
@@ -113,7 +113,7 @@ const setupVideoJsPlayer = () => {
|
||||
}
|
||||
|
||||
const playVideoWithoutTotalLength = () => {
|
||||
rowUrl = buildUrlWithParams(props.previewRow.Url)
|
||||
rowUrl = window?.$baseUrl + "/api/preview?url=" + encodeURIComponent(buildUrlWithParams(props.previewRow.Url))
|
||||
mediaSource = new MediaSource()
|
||||
videoPlayer.value.src = URL.createObjectURL(mediaSource)
|
||||
videoPlayer.value.play()
|
||||
|
||||
59
frontend/src/components/ShowOrEdit.vue
Normal file
59
frontend/src/components/ShowOrEdit.vue
Normal file
@@ -0,0 +1,59 @@
|
||||
<template>
|
||||
<div
|
||||
class="min-h-6"
|
||||
@click="handleOnClick"
|
||||
>
|
||||
<n-input
|
||||
v-if="isEdit"
|
||||
ref="inputRef"
|
||||
:value="inputValue"
|
||||
@update:value="v => inputValue = v"
|
||||
@change="handleChange"
|
||||
@blur="handleChange"
|
||||
/>
|
||||
|
||||
<n-tooltip
|
||||
v-else
|
||||
trigger="hover"
|
||||
placement="top"
|
||||
>
|
||||
<template #trigger>
|
||||
<div class="ellipsis-2">{{ inputValue }}</div>
|
||||
</template>
|
||||
<div class="ellipsis-2">{{ inputValue }}</div>
|
||||
</n-tooltip>
|
||||
</div>
|
||||
</template>
|
||||
|
||||
<script setup lang="ts">
|
||||
import { ref, nextTick, watch } from 'vue'
|
||||
import type { InputInst } from 'naive-ui'
|
||||
|
||||
interface OnUpdateValue {
|
||||
(value: string): void
|
||||
}
|
||||
|
||||
const props = defineProps<{
|
||||
value: string | number
|
||||
onUpdateValue?: OnUpdateValue
|
||||
}>()
|
||||
|
||||
const isEdit = ref(false)
|
||||
const inputRef = ref<InputInst | null>(null)
|
||||
const inputValue = ref(String(props.value))
|
||||
|
||||
watch(
|
||||
() => props.value,
|
||||
v => inputValue.value = String(v)
|
||||
)
|
||||
|
||||
function handleOnClick() {
|
||||
isEdit.value = true
|
||||
nextTick(() => inputRef.value?.focus())
|
||||
}
|
||||
|
||||
function handleChange() {
|
||||
props.onUpdateValue?.(String(inputValue.value))
|
||||
isEdit.value = false
|
||||
}
|
||||
</script>
|
||||
@@ -26,4 +26,15 @@ export const isValidHost = (host: string) => {
|
||||
export const isValidPort = (port: number) => {
|
||||
const portNumber = Number(port)
|
||||
return Number.isInteger(portNumber) && portNumber > 1024 && portNumber < 65535
|
||||
}
|
||||
|
||||
export const formatSize = (size: number | string) => {
|
||||
if (typeof size === "string") return size
|
||||
if (size > 1048576) {
|
||||
return (size / 1048576).toFixed(2) + 'MB';
|
||||
}
|
||||
if (size > 1024) {
|
||||
return (size / 1024).toFixed(2) + 'KB';
|
||||
}
|
||||
return Math.floor(size) + 'b';
|
||||
}
|
||||
@@ -34,24 +34,29 @@
|
||||
"grab_type": "Grab Type",
|
||||
"clear_list": "Clear List",
|
||||
"clear_list_tip": "Clear all records?",
|
||||
"remember_clear_choice": "Remember this selection and clear it next time",
|
||||
"batch_download": "Batch Download",
|
||||
"batch_export": "Batch Export",
|
||||
"batch_import": "Batch Import",
|
||||
"export_url": "Export Url",
|
||||
"import_success": "Export Success",
|
||||
"total_resources": "total of {count} resources",
|
||||
"all": "All",
|
||||
"image": "Image",
|
||||
"audio": "Audio",
|
||||
"video": "Video",
|
||||
"m3u8": "M3U8",
|
||||
"live": "Live Stream",
|
||||
"stream": "Data Stream",
|
||||
"xls": "Spreadsheet",
|
||||
"doc": "Document",
|
||||
"pdf": "PDF",
|
||||
"font": "Font",
|
||||
"domain": "Domain",
|
||||
"choice": "choice",
|
||||
"type": "Type",
|
||||
"preview": "Preview",
|
||||
"preview_tip": "Preview not supported",
|
||||
"preview_tip": "Cannot preview",
|
||||
"status": "Status",
|
||||
"description": "Description",
|
||||
"resource_size": "Resource Size",
|
||||
@@ -59,6 +64,7 @@
|
||||
"save_path_empty": "Please set save location",
|
||||
"operation": "Operation",
|
||||
"ready": "Ready",
|
||||
"pending": "Pending",
|
||||
"running": "Running",
|
||||
"error": "Error",
|
||||
"done": "Done",
|
||||
@@ -71,6 +77,8 @@
|
||||
"open_link": "Open Link",
|
||||
"open_file": "Open File",
|
||||
"delete_row": "Delete Row",
|
||||
"delete_tip": "Running tasks cannot be deleted",
|
||||
"cancel_down": "Cancel Download",
|
||||
"more_operation": "More Operations",
|
||||
"video_decode": "WxDecrypt",
|
||||
"video_decode_loading": "Decrypting",
|
||||
@@ -80,9 +88,14 @@
|
||||
"import_placeholder": "When adding multiple items, ensure each line contains only one (each link on a new line)",
|
||||
"import_empty": "Please enter data to import",
|
||||
"win_install_tip": "For the first time using this software, please right-click and select 'Run as administrator'",
|
||||
"download_queued": "Download has been added to the queue, current queue length:{count}",
|
||||
"download_queued": "has been added to the queue, current queue length:{count}",
|
||||
"search": "Search",
|
||||
"search_description": "Keyword Search..."
|
||||
"search_description": "Keyword Search...",
|
||||
"start_err_tip": "Error Message",
|
||||
"start_err_content": "The current startup process has encountered an issue. Do you want to reset the application?",
|
||||
"start_err_positiveText": "Clear cache and restart",
|
||||
"start_err_negativeText": "Close the software",
|
||||
"reset_app_tip": "This operation will delete intercepted data and data related to this application. Please proceed with caution!"
|
||||
},
|
||||
"setting": {
|
||||
"restart_tip": "Keep default if unsure, please restart software after modification",
|
||||
@@ -96,7 +109,7 @@
|
||||
"quality_tip": "Effective for video accounts",
|
||||
"full_intercept": "Full Intercept",
|
||||
"full_intercept_tip": "Whether to fully intercept WeChat video accounts, No: only intercept video details",
|
||||
"insert_tail": "insert tail",
|
||||
"insert_tail": "Insert tail",
|
||||
"insert_tail_tip": "Intercept whether new data is added to the end of the list",
|
||||
"upstream_proxy": "Upstream Proxy",
|
||||
"upstream_proxy_tip": "For combining with other proxy tools, format: http://username:password@your.proxy.server:port",
|
||||
@@ -107,9 +120,11 @@
|
||||
"connections_tip": "Keep default if unsure, usually CPU cores * 2, for faster downloads",
|
||||
"down_number": "Download Number",
|
||||
"down_number_tip": "Number of downloads executed simultaneously",
|
||||
"use_headers_tip": "Define headers for downloads, comma separated",
|
||||
"use_headers_tip": "Default system filtering, Define headers for downloads, comma separated",
|
||||
"mime_map": "Intercept Rules",
|
||||
"mime_map_tip": "JSON format, keep default if unsure, please restart software after modification",
|
||||
"domain_rule": "Domain Rule",
|
||||
"domain_rule_tip": "Default * matches all domains, One line for each rule,supports the following: \n*.qq.com\nvideo.qq.com\nexample.com\n\n# Exclude\n!static.qq.com",
|
||||
"port_format_error": "port format error",
|
||||
"host_format_error": "host format error",
|
||||
"basic_setting": "Basic Setting",
|
||||
|
||||
@@ -34,24 +34,29 @@
|
||||
"grab_type": "抓取类型",
|
||||
"clear_list": "清空列表",
|
||||
"clear_list_tip": "清空所有记录?",
|
||||
"remember_clear_choice": "记住此选择,下次直接清除",
|
||||
"batch_download": "批量下载",
|
||||
"batch_export": "批量导出",
|
||||
"batch_import": "批量导入",
|
||||
"export_url": "导出链接",
|
||||
"import_success": "导出成功",
|
||||
"total_resources": "共{count}个资源",
|
||||
"all": "全部",
|
||||
"image": "图片",
|
||||
"audio": "音频",
|
||||
"video": "视频",
|
||||
"m3u8": "m3u8",
|
||||
"live": "直播流",
|
||||
"stream": "流数据",
|
||||
"xls": "表格",
|
||||
"doc": "文档",
|
||||
"pdf": "pdf",
|
||||
"font": "字体",
|
||||
"domain": "域",
|
||||
"choice": "已选",
|
||||
"type": "类型",
|
||||
"preview": "预览",
|
||||
"preview_tip": "暂不支持预览",
|
||||
"preview_tip": "无法预览",
|
||||
"status": "状态",
|
||||
"description": "描述",
|
||||
"resource_size": "资源大小",
|
||||
@@ -59,6 +64,7 @@
|
||||
"save_path_empty": "请设置保存位置",
|
||||
"operation": "操作",
|
||||
"ready": "就绪",
|
||||
"pending": "待处理",
|
||||
"running": "运行中",
|
||||
"error": "错误",
|
||||
"done": "完成",
|
||||
@@ -71,6 +77,8 @@
|
||||
"open_link": "打开链接",
|
||||
"open_file": "打开文件",
|
||||
"delete_row": "删除记录",
|
||||
"delete_tip": "运行中任务无法删除",
|
||||
"cancel_down": "取消下载",
|
||||
"more_operation": "更多操作",
|
||||
"video_decode": "视频解密",
|
||||
"video_decode_loading": "解密中",
|
||||
@@ -80,9 +88,14 @@
|
||||
"import_placeholder": "添加多个时,请确保每行只有一个(每个链接回车换行)",
|
||||
"import_empty": "请输入需要导入的数据",
|
||||
"win_install_tip": "首次启用本软件,请使用鼠标右键选择以管理员身份运行",
|
||||
"download_queued": "下载已加入队列,当前队列长度:{count}",
|
||||
"download_queued": "已加入队列,当前队列长度:{count}",
|
||||
"search": "搜索",
|
||||
"search_description": "关键字搜索..."
|
||||
"search_description": "关键字搜索...",
|
||||
"start_err_tip": "错误提示",
|
||||
"start_err_content": "当前启动过程遇到了问题,是否重置应用?",
|
||||
"start_err_positiveText": "清理缓存并重启",
|
||||
"start_err_negativeText": "关闭软件",
|
||||
"reset_app_tip": "此操作会删除已拦截数据以及本应用相关数据,请谨慎操作!"
|
||||
},
|
||||
"setting": {
|
||||
"restart_tip": "如果不清楚保持默认就行,修改后请重启软件",
|
||||
@@ -107,9 +120,11 @@
|
||||
"connections_tip": "如不清楚请保持默认,通常CPU核心数*2,用于加速下载",
|
||||
"down_number": "下载数",
|
||||
"down_number_tip": "同时进行的下载数量",
|
||||
"use_headers_tip": "定义下载时可使用的header参数,逗号分割",
|
||||
"use_headers_tip": "默认系统过滤,定义下载时可使用的header参数,逗号分割",
|
||||
"mime_map": "拦截规则",
|
||||
"mime_map_tip": "json格式,如果不清楚保持默认就行,修改后请重启软件",
|
||||
"domain_rule": "域名规则",
|
||||
"domain_rule_tip": "默认*匹配所有域,每个规则一行,支持如下: \n*.qq.com\nvideo.qq.com\nexample.com\n\n# 排除\n!static.qq.com",
|
||||
"port_format_error": "port 格式错误",
|
||||
"host_format_error": "host 格式错误",
|
||||
"basic_setting": "基础设置",
|
||||
|
||||
@@ -33,7 +33,8 @@ export const useIndexStore = defineStore("index-store", () => {
|
||||
UserAgent: "",
|
||||
UseHeaders: "",
|
||||
InsertTail: true,
|
||||
MimeMap: {}
|
||||
MimeMap: {},
|
||||
Rule: "*"
|
||||
})
|
||||
|
||||
const envInfo = ref({
|
||||
|
||||
3
frontend/src/types/app.d.ts
vendored
3
frontend/src/types/app.d.ts
vendored
@@ -31,6 +31,7 @@ export namespace appType {
|
||||
UseHeaders: string
|
||||
InsertTail: boolean
|
||||
MimeMap: { [key: string]: MimeMap }
|
||||
Rule: string
|
||||
}
|
||||
|
||||
interface MediaInfo {
|
||||
@@ -38,7 +39,7 @@ export namespace appType {
|
||||
Url: string
|
||||
UrlSign: string
|
||||
CoverUrl: string
|
||||
Size: string
|
||||
Size: number
|
||||
Domain: string
|
||||
Classify: string
|
||||
Suffix: string
|
||||
|
||||
@@ -3,29 +3,54 @@
|
||||
<div class="pb-2 z-40" id="header">
|
||||
<NSpace>
|
||||
<NButton v-if="isProxy" secondary type="primary" @click.stop="close" style="--wails-draggable:no-drag">
|
||||
{{ t("index.close_grab") }}
|
||||
<span class="inline-block w-1.5 h-1.5 bg-red-600 rounded-full mr-1 animate-pulse"></span>
|
||||
{{ t("index.close_grab") }}{{ data.length > 0 ? ` ${t('index.total_resources', {count: data.length})}` : '' }}
|
||||
</NButton>
|
||||
<NButton v-else tertiary type="tertiary" @click.stop="open" style="--wails-draggable:no-drag">
|
||||
{{ t("index.open_grab") }}
|
||||
{{ t("index.open_grab") }}{{ data.length > 0 ? ` ${t('index.total_resources', {count: data.length})}` : '' }}
|
||||
</NButton>
|
||||
<NSelect style="min-width: 100px;--wails-draggable:no-drag" :placeholder="t('index.grab_type')" v-model:value="resourcesType" multiple clearable
|
||||
:max-tag-count="3" :options="classify"></NSelect>
|
||||
<n-popconfirm
|
||||
@positive-click="clear"
|
||||
>
|
||||
<template #trigger>
|
||||
<NButton tertiary type="error" style="--wails-draggable:no-drag">
|
||||
<template #icon>
|
||||
<NButtonGroup style="--wails-draggable:no-drag">
|
||||
|
||||
<NButton v-if="rememberChoice" tertiary type="error" @click.stop="clear" style="--wails-draggable:no-drag">
|
||||
<template #icon>
|
||||
<n-icon>
|
||||
<TrashOutline/>
|
||||
</n-icon>
|
||||
</template>
|
||||
{{ t("index.clear_list") }}
|
||||
</NButton>
|
||||
<n-popconfirm
|
||||
v-else
|
||||
@positive-click="()=>{rememberChoice=rememberChoiceTmp;clear()}"
|
||||
:show-icon="false"
|
||||
>
|
||||
<template #trigger>
|
||||
<NButton tertiary type="error" style="--wails-draggable:no-drag">
|
||||
<template #icon>
|
||||
<n-icon>
|
||||
<TrashOutline/>
|
||||
</n-icon>
|
||||
</template>
|
||||
{{ t("index.clear_list") }}
|
||||
</NButton>
|
||||
</template>
|
||||
<div>
|
||||
<div class="flex flex-row items-center text-red-700 my-2 text-base">
|
||||
<n-icon>
|
||||
<TrashOutline/>
|
||||
</n-icon>
|
||||
</template>
|
||||
{{ t("index.clear_list") }}
|
||||
</NButton>
|
||||
</template>
|
||||
{{ t("index.clear_list_tip") }}
|
||||
</n-popconfirm>
|
||||
<NButtonGroup style="--wails-draggable:no-drag">
|
||||
<p class="ml-1">{{ t("index.clear_list_tip") }}</p>
|
||||
</div>
|
||||
<NCheckbox
|
||||
v-model:checked="rememberChoiceTmp"
|
||||
>
|
||||
<span class="text-gray-400">{{ t('index.remember_clear_choice') }}</span>
|
||||
</NCheckbox>
|
||||
</div>
|
||||
</n-popconfirm>
|
||||
|
||||
<NButton tertiary type="primary" @click.stop="batchDown">
|
||||
<template #icon>
|
||||
<n-icon>
|
||||
@@ -34,21 +59,48 @@
|
||||
</template>
|
||||
{{ t('index.batch_download') }}
|
||||
</NButton>
|
||||
<NButton tertiary type="warning" @click.stop="batchExport">
|
||||
<template #icon>
|
||||
<n-icon>
|
||||
<ArrowRedoCircleOutline/>
|
||||
</n-icon>
|
||||
</template>
|
||||
{{ t('index.batch_export') }}
|
||||
</NButton>
|
||||
<NButton tertiary type="info" @click.stop="showImport=true">
|
||||
<template #icon>
|
||||
<n-icon>
|
||||
<ServerOutline/>
|
||||
</n-icon>
|
||||
</template>
|
||||
{{ t('index.batch_import') }}
|
||||
<NButton tertiary type="info">
|
||||
<NPopover placement="bottom" trigger="hover">
|
||||
<template #trigger>
|
||||
<NIcon size="18" class="">
|
||||
<Apps/>
|
||||
</NIcon>
|
||||
</template>
|
||||
<div class="flex flex-col">
|
||||
<NButton tertiary type="error" @click.stop="batchCancel" class="my-1">
|
||||
<template #icon>
|
||||
<n-icon>
|
||||
<CloseOutline/>
|
||||
</n-icon>
|
||||
</template>
|
||||
{{ t('index.cancel_down') }}
|
||||
</NButton>
|
||||
<NButton tertiary type="warning" @click.stop="batchExport()" class="my-1">
|
||||
<template #icon>
|
||||
<n-icon>
|
||||
<ArrowRedoCircleOutline/>
|
||||
</n-icon>
|
||||
</template>
|
||||
{{ t('index.batch_export') }}
|
||||
</NButton>
|
||||
<NButton tertiary type="info" @click.stop="showImport=true" class="my-1">
|
||||
<template #icon>
|
||||
<n-icon>
|
||||
<ServerOutline/>
|
||||
</n-icon>
|
||||
</template>
|
||||
{{ t('index.batch_import') }}
|
||||
</NButton>
|
||||
<NButton tertiary type="primary" @click.stop="batchExport('url')" class="my-1">
|
||||
<template #icon>
|
||||
<n-icon>
|
||||
<ArrowRedoCircleOutline/>
|
||||
</n-icon>
|
||||
</template>
|
||||
{{ t('index.export_url') }}
|
||||
</NButton>
|
||||
</div>
|
||||
</NPopover>
|
||||
</NButton>
|
||||
</NButtonGroup>
|
||||
</NSpace>
|
||||
@@ -65,6 +117,7 @@
|
||||
:height-for-row="()=> 48"
|
||||
:checked-row-keys="checkedRowKeysValue"
|
||||
@update:checked-row-keys="handleCheck"
|
||||
@update:filters="updateFilters"
|
||||
style="--wails-draggable:no-drag"
|
||||
/>
|
||||
</div>
|
||||
@@ -82,10 +135,10 @@
|
||||
</template>
|
||||
|
||||
<script lang="ts" setup>
|
||||
import {NButton, NIcon, NImage, NInput, NSpace, NTooltip, NPopover} from "naive-ui"
|
||||
import {NButton, NIcon, NImage, NInput, NSpace, NTooltip, NPopover, NGradientText} from "naive-ui"
|
||||
import {computed, h, onMounted, ref, watch} from "vue"
|
||||
import type {appType} from "@/types/app"
|
||||
import type {DataTableRowKey, ImageRenderToolbarProps} from "naive-ui"
|
||||
import type {DataTableRowKey, ImageRenderToolbarProps, DataTableFilterState, DataTableBaseColumn} from "naive-ui"
|
||||
import Preview from "@/components/Preview.vue"
|
||||
import ShowLoading from "@/components/ShowLoading.vue"
|
||||
// @ts-ignore
|
||||
@@ -98,17 +151,25 @@ import ImportJson from "@/components/ImportJson.vue"
|
||||
import {useEventStore} from "@/stores/event"
|
||||
import {BrowserOpenURL, ClipboardSetText} from "../../wailsjs/runtime"
|
||||
import Password from "@/components/Password.vue"
|
||||
import ShowOrEdit from "@/components/ShowOrEdit.vue"
|
||||
import {useI18n} from 'vue-i18n'
|
||||
import {
|
||||
DownloadOutline,
|
||||
ArrowRedoCircleOutline,
|
||||
ServerOutline,
|
||||
SearchOutline,
|
||||
TrashOutline
|
||||
Apps,
|
||||
TrashOutline, CloseOutline
|
||||
} from "@vicons/ionicons5"
|
||||
import {useDialog} from 'naive-ui'
|
||||
import * as bind from "../../wailsjs/go/core/Bind"
|
||||
import {Quit} from "../../wailsjs/runtime"
|
||||
import {DialogOptions} from "naive-ui/es/dialog/src/DialogProvider"
|
||||
import {formatSize} from "@/func"
|
||||
|
||||
const {t} = useI18n()
|
||||
const eventStore = useEventStore()
|
||||
const dialog = useDialog()
|
||||
const isProxy = computed(() => {
|
||||
return store.isProxy
|
||||
})
|
||||
@@ -116,18 +177,22 @@ const certUrl = computed(() => {
|
||||
return store.baseUrl + "/api/cert"
|
||||
})
|
||||
const data = ref<any[]>([])
|
||||
let filterClassify: string[] = []
|
||||
const filterClassify = ref<string[]>([])
|
||||
const filteredData = computed(() => {
|
||||
let result = data.value
|
||||
|
||||
if (filterClassify.length > 0) {
|
||||
result = result.filter(item => filterClassify.includes(item.Classify))
|
||||
if (filterClassify.value.length > 0) {
|
||||
result = result.filter(item => filterClassify.value.includes(item.Classify))
|
||||
}
|
||||
|
||||
if (descriptionSearchValue.value) {
|
||||
result = result.filter(item => item.Description?.toLowerCase().includes(descriptionSearchValue.value.toLowerCase()))
|
||||
}
|
||||
|
||||
if (urlSearchValue.value) {
|
||||
result = result.filter(item => item.Url?.toLowerCase().includes(urlSearchValue.value.toLowerCase()))
|
||||
}
|
||||
|
||||
return result
|
||||
})
|
||||
|
||||
@@ -144,12 +209,14 @@ const classifyAlias: { [key: string]: any } = {
|
||||
xls: computed(() => t("index.xls")),
|
||||
doc: computed(() => t("index.doc")),
|
||||
pdf: computed(() => t("index.pdf")),
|
||||
stream: computed(() => t("index.stream")),
|
||||
font: computed(() => t("index.font"))
|
||||
}
|
||||
|
||||
const dwStatus = computed<any>(() => {
|
||||
return {
|
||||
ready: t("index.ready"),
|
||||
pending: t("index.pending"),
|
||||
running: t("index.running"),
|
||||
error: t("index.error"),
|
||||
done: t("index.done"),
|
||||
@@ -169,15 +236,58 @@ const classify = ref([
|
||||
])
|
||||
|
||||
const descriptionSearchValue = ref("")
|
||||
const urlSearchValue = ref("")
|
||||
const rememberChoice = ref(false)
|
||||
const rememberChoiceTmp = ref(false)
|
||||
|
||||
const columns = ref<any[]>([
|
||||
{
|
||||
type: "selection",
|
||||
},
|
||||
{
|
||||
title: computed(() => t("index.domain")),
|
||||
title: () => {
|
||||
if (checkedRowKeysValue.value.length > 0) {
|
||||
return h(NGradientText, {type: "success"}, t("index.choice") + `(${checkedRowKeysValue.value.length})`)
|
||||
}
|
||||
return h('div', {class: 'flex items-center'}, [
|
||||
t('index.domain'),
|
||||
h(NPopover, {
|
||||
style: "--wails-draggable:no-drag",
|
||||
trigger: 'click',
|
||||
placement: 'bottom',
|
||||
showArrow: true,
|
||||
}, {
|
||||
trigger: () => h(NIcon, {
|
||||
size: "18",
|
||||
class: `ml-1 cursor-pointer ${urlSearchValue.value ? "text-green-600": "text-gray-500"}`,
|
||||
onClick: (e: MouseEvent) => e.stopPropagation()
|
||||
}, h(SearchOutline)),
|
||||
default: () => h('div', {class: 'p-2 w-64'}, [
|
||||
h(NInput, {
|
||||
value: urlSearchValue.value,
|
||||
'onUpdate:value': (val: string) => urlSearchValue.value = val,
|
||||
placeholder: t('index.search_description'),
|
||||
clearable: true
|
||||
}, {
|
||||
prefix: () => h(NIcon, {component: SearchOutline})
|
||||
})
|
||||
])
|
||||
})
|
||||
])
|
||||
},
|
||||
key: "Domain",
|
||||
width: 90,
|
||||
render: (row: appType.MediaInfo) => {
|
||||
return h(NTooltip, {
|
||||
trigger: 'hover',
|
||||
placement: 'top'
|
||||
}, {
|
||||
trigger: () => h('span', {
|
||||
class: 'cursor-default'
|
||||
}, row.Domain),
|
||||
default: () => row.Url
|
||||
})
|
||||
}
|
||||
},
|
||||
{
|
||||
title: computed(() => t("index.type")),
|
||||
@@ -186,7 +296,6 @@ const columns = ref<any[]>([
|
||||
filterOptions: computed(() => Array.from(classify.value).slice(1)),
|
||||
filterMultiple: true,
|
||||
filter: (value: string, row: appType.MediaInfo) => {
|
||||
if (!filterClassify.includes(value)) filterClassify.push(value)
|
||||
return !!~row.Classify.indexOf(String(value))
|
||||
},
|
||||
render: (row: appType.MediaInfo) => {
|
||||
@@ -244,11 +353,18 @@ const columns = ref<any[]>([
|
||||
key: "Status",
|
||||
width: 80,
|
||||
render: (row: appType.MediaInfo, index: number) => {
|
||||
let status = "info"
|
||||
if (row.Status === "done" || row.Status === "running") {
|
||||
status = "success"
|
||||
} else if (row.Status === "pending") {
|
||||
status = "warning"
|
||||
}
|
||||
|
||||
return h(
|
||||
NButton,
|
||||
{
|
||||
tertiary: true,
|
||||
type: row.Status === "done" ? "success" : "info",
|
||||
type: status as any,
|
||||
size: "small",
|
||||
style: {
|
||||
margin: "2px"
|
||||
@@ -273,13 +389,14 @@ const columns = ref<any[]>([
|
||||
title: () => h('div', {class: 'flex items-center'}, [
|
||||
t('index.description'),
|
||||
h(NPopover, {
|
||||
style: "--wails-draggable:no-drag",
|
||||
trigger: 'click',
|
||||
placement: 'bottom',
|
||||
showArrow: true,
|
||||
}, {
|
||||
trigger: () => h(NIcon, {
|
||||
size: "18",
|
||||
class: "ml-1 text-gray-500 cursor-pointer",
|
||||
class: `ml-1 cursor-pointer ${descriptionSearchValue.value ? "text-green-600": "text-gray-500"}`,
|
||||
onClick: (e: MouseEvent) => e.stopPropagation()
|
||||
}, h(SearchOutline)),
|
||||
default: () => h('div', {class: 'p-2 w-64'}, [
|
||||
@@ -297,10 +414,12 @@ const columns = ref<any[]>([
|
||||
key: "Description",
|
||||
width: 150,
|
||||
render: (row: appType.MediaInfo, index: number) => {
|
||||
const d = h("div", {class: "ellipsis-2",}, row.Description)
|
||||
return h(NTooltip, {trigger: 'hover', placement: 'top'}, {
|
||||
trigger: () => d,
|
||||
default: () => d
|
||||
return h(ShowOrEdit, {
|
||||
value: row.Description,
|
||||
onUpdateValue(v: string) {
|
||||
data.value[index].Description = v
|
||||
cacheData()
|
||||
}
|
||||
})
|
||||
}
|
||||
},
|
||||
@@ -308,6 +427,10 @@ const columns = ref<any[]>([
|
||||
title: computed(() => t("index.resource_size")),
|
||||
key: "Size",
|
||||
width: 120,
|
||||
sorter: (row1: appType.MediaInfo, row2: appType.MediaInfo) => row1.Size - row2.Size,
|
||||
render(row: appType.MediaInfo, index: number) {
|
||||
return formatSize(row.Size)
|
||||
}
|
||||
},
|
||||
{
|
||||
title: computed(() => t("index.save_path")),
|
||||
@@ -352,6 +475,7 @@ const showPassword = ref(false)
|
||||
const downloadQueue = ref<appType.MediaInfo[]>([])
|
||||
let activeDownloads = 0
|
||||
let isOpenProxy = false
|
||||
let isInstall = false
|
||||
|
||||
onMounted(() => {
|
||||
try {
|
||||
@@ -360,8 +484,16 @@ onMounted(() => {
|
||||
})
|
||||
loading.value = true
|
||||
handleInstall().then((is: boolean) => {
|
||||
isInstall = true
|
||||
loading.value = false
|
||||
})
|
||||
|
||||
checkLoading()
|
||||
watch(showPassword, () => {
|
||||
if (!showPassword.value) {
|
||||
checkLoading()
|
||||
}
|
||||
})
|
||||
} catch (e) {
|
||||
window.$message?.error(JSON.stringify(e), {duration: 5000})
|
||||
}
|
||||
@@ -379,7 +511,22 @@ onMounted(() => {
|
||||
if (cache) {
|
||||
data.value = JSON.parse(cache)
|
||||
}
|
||||
|
||||
const choiceCache = localStorage.getItem("remember-clear-choice")
|
||||
if (choiceCache === "1") {
|
||||
rememberChoice.value = true
|
||||
}
|
||||
|
||||
watch(rememberChoice, (n, o) => {
|
||||
if (rememberChoice.value) {
|
||||
localStorage.setItem("remember-clear-choice", "1")
|
||||
} else {
|
||||
localStorage.removeItem("remember-clear-choice")
|
||||
}
|
||||
})
|
||||
|
||||
resetTableHeight()
|
||||
|
||||
eventStore.addHandle({
|
||||
type: "newResources",
|
||||
event: (res: appType.MediaInfo) => {
|
||||
@@ -407,6 +554,9 @@ onMounted(() => {
|
||||
item.SavePath = res.SavePath
|
||||
item.Status = 'done'
|
||||
})
|
||||
if (activeDownloads > 0) {
|
||||
activeDownloads--
|
||||
}
|
||||
cacheData()
|
||||
checkQueue()
|
||||
break
|
||||
@@ -415,6 +565,9 @@ onMounted(() => {
|
||||
item.SavePath = res.Message
|
||||
item.Status = 'error'
|
||||
})
|
||||
if (activeDownloads > 0) {
|
||||
activeDownloads--
|
||||
}
|
||||
cacheData()
|
||||
checkQueue()
|
||||
break
|
||||
@@ -434,7 +587,7 @@ watch(resourcesType, (n, o) => {
|
||||
appApi.setType(resourcesType.value)
|
||||
})
|
||||
|
||||
const updateItem = (id: string, updater: (item: any) => void)=>{
|
||||
const updateItem = (id: string, updater: (item: any) => void) => {
|
||||
const item = data.value.find(i => i.Id === id)
|
||||
if (item) updater(item)
|
||||
}
|
||||
@@ -479,6 +632,35 @@ const dataAction = (row: appType.MediaInfo, index: number, type: string) => {
|
||||
case "down":
|
||||
download(row, index)
|
||||
break
|
||||
case "cancel":
|
||||
if (row.Status === "pending") {
|
||||
const queueIndex = downloadQueue.value.findIndex(item => item.Id === row.Id)
|
||||
if (queueIndex !== -1) {
|
||||
downloadQueue.value.splice(queueIndex, 1)
|
||||
}
|
||||
updateItem(row.Id, item => {
|
||||
item.Status = 'ready'
|
||||
item.SavePath = ''
|
||||
})
|
||||
cacheData()
|
||||
} else if (row.Status === "running") {
|
||||
appApi.cancel({id: row.Id}).then((res) => {
|
||||
updateItem(row.Id, item => {
|
||||
item.Status = 'ready'
|
||||
item.SavePath = ''
|
||||
})
|
||||
if (activeDownloads > 0) {
|
||||
activeDownloads--
|
||||
}
|
||||
cacheData()
|
||||
checkQueue()
|
||||
if (res.code === 0) {
|
||||
window?.$message?.error(res.message)
|
||||
return
|
||||
}
|
||||
})
|
||||
}
|
||||
break
|
||||
case "copy":
|
||||
ClipboardSetText(row.Url).then((is: boolean) => {
|
||||
if (is) {
|
||||
@@ -504,7 +686,11 @@ const dataAction = (row: appType.MediaInfo, index: number, type: string) => {
|
||||
decodeWxFile(row, index)
|
||||
break
|
||||
case "delete":
|
||||
appApi.delete({sign: row.UrlSign}).then(() => {
|
||||
if (row.Status === "pending" || row.Status === "running") {
|
||||
window?.$message?.error(t("index.delete_tip"))
|
||||
return
|
||||
}
|
||||
appApi.delete({sign: [row.UrlSign]}).then(() => {
|
||||
data.value.splice(index, 1)
|
||||
cacheData()
|
||||
})
|
||||
@@ -531,6 +717,10 @@ const handleCheck = (rowKeys: DataTableRowKey[]) => {
|
||||
checkedRowKeysValue.value = rowKeys
|
||||
}
|
||||
|
||||
const updateFilters = (filters: DataTableFilterState, initiatorColumn: DataTableBaseColumn) => {
|
||||
filterClassify.value = filters.Classify as string[]
|
||||
}
|
||||
|
||||
const batchDown = async () => {
|
||||
if (checkedRowKeysValue.value.length <= 0) {
|
||||
window?.$message?.error(t("index.use_data"))
|
||||
@@ -551,7 +741,46 @@ const batchDown = async () => {
|
||||
checkedRowKeysValue.value = []
|
||||
}
|
||||
|
||||
const batchExport = () => {
|
||||
const batchCancel = async () => {
|
||||
if (checkedRowKeysValue.value.length <= 0) {
|
||||
window?.$message?.error(t("index.use_data"))
|
||||
return
|
||||
}
|
||||
loading.value = true
|
||||
const cancelTasks: Promise<any>[] = []
|
||||
data.value.forEach((item, index) => {
|
||||
if (!checkedRowKeysValue.value.includes(item.Id)) {
|
||||
return
|
||||
}
|
||||
|
||||
if (item.Status === "pending") {
|
||||
const queueIndex = downloadQueue.value.findIndex(qItem => qItem.Id === item.Id)
|
||||
if (queueIndex !== -1) {
|
||||
downloadQueue.value.splice(queueIndex, 1)
|
||||
}
|
||||
item.Status = 'ready'
|
||||
item.SavePath = ''
|
||||
return
|
||||
}
|
||||
|
||||
if (item.Status === "running") {
|
||||
if (activeDownloads > 0) {
|
||||
activeDownloads--
|
||||
}
|
||||
cancelTasks.push(appApi.cancel({id: item.Id}).then(() => {
|
||||
item.Status = 'ready'
|
||||
item.SavePath = ''
|
||||
checkQueue()
|
||||
}))
|
||||
}
|
||||
})
|
||||
await Promise.allSettled(cancelTasks)
|
||||
loading.value = false
|
||||
checkedRowKeysValue.value = []
|
||||
cacheData()
|
||||
}
|
||||
|
||||
const batchExport = (type?: string) => {
|
||||
if (checkedRowKeysValue.value.length <= 0) {
|
||||
window?.$message?.error(t("index.use_data"))
|
||||
return
|
||||
@@ -565,9 +794,13 @@ const batchExport = () => {
|
||||
loadingText.value = t("common.loading")
|
||||
loading.value = true
|
||||
|
||||
const jsonData = data.value
|
||||
.filter(item => checkedRowKeysValue.value.includes(item.Id))
|
||||
.map(item => encodeURIComponent(JSON.stringify(item)))
|
||||
let jsonData = data.value.filter(item => checkedRowKeysValue.value.includes(item.Id))
|
||||
|
||||
if (type === "url") {
|
||||
jsonData = jsonData.map(item => item.Url)
|
||||
} else {
|
||||
jsonData = jsonData.map(item => encodeURIComponent(JSON.stringify(item)))
|
||||
}
|
||||
|
||||
appApi.batchExport({content: jsonData.join("\n")}).then((res: appType.Res) => {
|
||||
loading.value = false
|
||||
@@ -601,9 +834,9 @@ const download = (row: appType.MediaInfo, index: number) => {
|
||||
}
|
||||
|
||||
if (activeDownloads >= maxConcurrentDownloads.value) {
|
||||
row.Status = "pending"
|
||||
downloadQueue.value.push(row)
|
||||
window?.$message?.info((row.Description ? `「${row.Description}」` : "")
|
||||
+ t("index.download_queued", {count: downloadQueue.value.length}))
|
||||
window?.$message?.info(t("index.download_queued", {count: downloadQueue.value.length}))
|
||||
return
|
||||
}
|
||||
|
||||
@@ -621,9 +854,6 @@ const startDownload = (row: appType.MediaInfo, index: number) => {
|
||||
if (res.code === 0) {
|
||||
window?.$message?.error(res.message)
|
||||
}
|
||||
}).finally(() => {
|
||||
activeDownloads--
|
||||
checkQueue()
|
||||
})
|
||||
}
|
||||
|
||||
@@ -658,25 +888,30 @@ const close = () => {
|
||||
store.unsetProxy()
|
||||
}
|
||||
|
||||
const clear = () => {
|
||||
const clear = async () => {
|
||||
const newData = [] as any[]
|
||||
const signs: string[] = []
|
||||
if (checkedRowKeysValue.value.length > 0) {
|
||||
let newData = [] as any[]
|
||||
data.value.forEach((item, index) => {
|
||||
if (checkedRowKeysValue.value.includes(item.Id)) {
|
||||
appApi.delete({sign: item.UrlSign})
|
||||
if (checkedRowKeysValue.value.includes(item.Id) && item.Status !== "pending" && item.Status !== "running") {
|
||||
signs.push(item.UrlSign)
|
||||
} else {
|
||||
newData.push(item)
|
||||
}
|
||||
})
|
||||
data.value = newData
|
||||
checkedRowKeysValue.value = []
|
||||
cacheData()
|
||||
return
|
||||
} else {
|
||||
data.value.forEach((item, index) => {
|
||||
if (item.Status === "pending" || item.Status === "running") {
|
||||
newData.push(item)
|
||||
} else {
|
||||
signs.push(item.UrlSign)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
data.value = []
|
||||
localStorage.setItem("resources-data", "")
|
||||
appApi.clear()
|
||||
await appApi.delete({sign: signs})
|
||||
data.value = newData
|
||||
cacheData()
|
||||
}
|
||||
|
||||
const decodeWxFile = (row: appType.MediaInfo, index: number) => {
|
||||
@@ -732,8 +967,8 @@ const handleImport = (content: string) => {
|
||||
})
|
||||
if (newItems.length > 0) {
|
||||
data.value = [...newItems, ...data.value]
|
||||
cacheData()
|
||||
}
|
||||
cacheData()
|
||||
showImport.value = false
|
||||
}
|
||||
|
||||
@@ -774,13 +1009,27 @@ const handleInstall = async () => {
|
||||
}
|
||||
return false
|
||||
}
|
||||
</script>
|
||||
<style>
|
||||
.ellipsis-2 {
|
||||
display: -webkit-box;
|
||||
overflow: hidden;
|
||||
text-overflow: ellipsis;
|
||||
-webkit-line-clamp: 2;
|
||||
-webkit-box-orient: vertical;
|
||||
|
||||
const checkLoading = () => {
|
||||
setTimeout(() => {
|
||||
if (loading.value && !isInstall && !showPassword.value) {
|
||||
dialog.warning({
|
||||
title: t("index.start_err_tip"),
|
||||
content: t("index.start_err_content"),
|
||||
positiveText: t("index.start_err_positiveText"),
|
||||
negativeText: t("index.start_err_negativeText"),
|
||||
draggable: false,
|
||||
closeOnEsc: false,
|
||||
closable: false,
|
||||
maskClosable: false,
|
||||
onPositiveClick: () => {
|
||||
bind.ResetApp()
|
||||
},
|
||||
onNegativeClick: () => {
|
||||
Quit()
|
||||
}
|
||||
} as DialogOptions)
|
||||
}
|
||||
}, 6000)
|
||||
}
|
||||
</style>
|
||||
</script>
|
||||
@@ -76,6 +76,17 @@
|
||||
{{ t("setting.insert_tail_tip") }}
|
||||
</NTooltip>
|
||||
</NFormItem>
|
||||
|
||||
<NFormItem >
|
||||
<n-popconfirm @positive-click="resetHandle">
|
||||
<template #trigger>
|
||||
<NButton tertiary type="error" style="--wails-draggable:no-drag">
|
||||
{{ t("index.start_err_positiveText") }}
|
||||
</NButton>
|
||||
</template>
|
||||
{{t("index.reset_app_tip")}}
|
||||
</n-popconfirm>
|
||||
</NFormItem>
|
||||
</NForm>
|
||||
</NTabPane>
|
||||
|
||||
@@ -186,6 +197,22 @@
|
||||
</NTooltip>
|
||||
</NFormItem>
|
||||
|
||||
<NFormItem :label="t('setting.domain_rule')" path="DomainRule">
|
||||
<NInput
|
||||
v-model:value="formValue.Rule"
|
||||
type="textarea"
|
||||
rows="5"
|
||||
:placeholder="t('setting.domain_rule_tip')"
|
||||
/>
|
||||
<NTooltip trigger="hover">
|
||||
<template #trigger>
|
||||
<NIcon size="18" class="ml-1 text-gray-500">
|
||||
<HelpCircleOutline/>
|
||||
</NIcon>
|
||||
</template>
|
||||
{{ t("setting.domain_rule_tip") }}
|
||||
</NTooltip>
|
||||
</NFormItem>
|
||||
<NFormItem :label="t('setting.mime_map')" path="MimeMap">
|
||||
<NInput
|
||||
v-model:value="MimeMap"
|
||||
@@ -217,6 +244,8 @@ import appApi from "@/api/app"
|
||||
import {computed} from "vue"
|
||||
import {useI18n} from 'vue-i18n'
|
||||
import {isValidHost, isValidPort} from '@/func'
|
||||
import {NButton, NIcon} from "naive-ui"
|
||||
import * as bind from "../../wailsjs/go/core/Bind"
|
||||
|
||||
const {t} = useI18n()
|
||||
const store = useIndexStore()
|
||||
@@ -281,6 +310,11 @@ const selectDir = () => {
|
||||
window?.$message?.error(err)
|
||||
})
|
||||
}
|
||||
|
||||
const resetHandle = ()=>{
|
||||
localStorage.clear()
|
||||
bind.ResetApp()
|
||||
}
|
||||
</script>
|
||||
<style lang="scss">
|
||||
.n-tabs-nav--top{
|
||||
|
||||
2
frontend/wailsjs/go/core/Bind.d.ts
vendored
2
frontend/wailsjs/go/core/Bind.d.ts
vendored
@@ -5,3 +5,5 @@ import {core} from '../models';
|
||||
export function AppInfo():Promise<core.ResponseData>;
|
||||
|
||||
export function Config():Promise<core.ResponseData>;
|
||||
|
||||
export function ResetApp():Promise<void>;
|
||||
|
||||
@@ -9,3 +9,7 @@ export function AppInfo() {
|
||||
export function Config() {
|
||||
return window['go']['core']['Bind']['Config']();
|
||||
}
|
||||
|
||||
export function ResetApp() {
|
||||
return window['go']['core']['Bind']['ResetApp']();
|
||||
}
|
||||
|
||||
2
go.mod
2
go.mod
@@ -5,7 +5,7 @@ go 1.22.0
|
||||
toolchain go1.23.2
|
||||
|
||||
require (
|
||||
github.com/elazarl/goproxy v0.0.0-20241223171911-d5978cb8c956
|
||||
github.com/elazarl/goproxy v1.7.2
|
||||
github.com/matoous/go-nanoid/v2 v2.1.0
|
||||
github.com/rs/zerolog v1.33.0
|
||||
github.com/vrischmann/userdir v0.0.0-20151206171402-20f291cebd68
|
||||
|
||||
5
go.sum
5
go.sum
@@ -3,8 +3,8 @@ github.com/bep/debounce v1.2.1/go.mod h1:H8yggRPQKLUhUoqrJC1bO2xNya7vanpDl7xR3IS
|
||||
github.com/coreos/go-systemd/v22 v22.5.0/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc=
|
||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/elazarl/goproxy v0.0.0-20241223171911-d5978cb8c956 h1:HyPt0ZkHkpke+HFl/4dDMz55A/AjFn7ZnLSm8GfdnwU=
|
||||
github.com/elazarl/goproxy v0.0.0-20241223171911-d5978cb8c956/go.mod h1:YfEbZtqP4AetfO6d40vWchF3znWX7C7Vd6ZMfdL8z64=
|
||||
github.com/elazarl/goproxy v1.7.2 h1:Y2o6urb7Eule09PjlhQRGNsqRfPmYI3KKQLFpCAV3+o=
|
||||
github.com/elazarl/goproxy v1.7.2/go.mod h1:82vkLNir0ALaW14Rc399OTTjyNREgmdL2cVoIbS6XaE=
|
||||
github.com/go-ole/go-ole v1.3.0 h1:Dt6ye7+vXGIKZ7Xtk4s6/xVdGDQynvom7xCFEdWr6uE=
|
||||
github.com/go-ole/go-ole v1.3.0/go.mod h1:5LS6F96DhAwUc7C+1HLexzMXY1xGRSryjyPPKW6zv78=
|
||||
github.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA=
|
||||
@@ -90,4 +90,3 @@ golang.org/x/text v0.22.0/go.mod h1:YRoo4H8PVmsu+E3Ou7cqLVH8oXWIHVoX0jqUWALQhfY=
|
||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
|
||||
|
||||
@@ -13,7 +13,7 @@
|
||||
"info": {
|
||||
"companyName": "res-downloader",
|
||||
"productName": "res-downloader",
|
||||
"productVersion": "3.1.0",
|
||||
"productVersion": "3.1.3",
|
||||
"copyright": "Copyright © 2023",
|
||||
"comments": "This is a high-value high-performance and diverse resource downloader called res-downloader."
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user