feat: torrent file support on web (#171)

pkg/download/downloader.go

@@ -169,9 +169,6 @@ func (d *Downloader) Setup() error {
 func (d *Downloader) parseFb(url string) (fetcher.FetcherBuilder, error) {
 	schema := util.ParseSchema(url)
 	fetchBuilder, ok := d.fetchBuilders[schema]
-	if !ok {
-		fetchBuilder, ok = d.fetchBuilders[util.FileSchema]
-	}
 	if ok {
 		return fetchBuilder, nil
 	}

pkg/util/path.go

@@ -88,3 +88,13 @@ func CreateDirIfNotExist(dir string) error {
 	}
 	return nil
 }
+
+// IsExistsFile check file exists and is a file
+func IsExistsFile(path string) bool {
+	info, err := os.Stat(path)
+	// 判断路径是否存在，并且是一个文件
+	if err == nil && !info.IsDir() {
+		return true
+	}
+	return false
+}

pkg/util/path_test.go

@@ -199,3 +199,43 @@ func doCheckDuplicateAndRename(t *testing.T, exitsPaths []string, path string, e
 		t.Errorf("CheckDuplicateAndRename() = %v, want %v", got, except)
 	}
 }
+
+func TestIsExistsFile(t *testing.T) {
+	type args struct {
+		path string
+	}
+	tests := []struct {
+		name string
+		args args
+		want bool
+	}{
+		{
+			name: "exist",
+			args: args{
+				path: "./path.go",
+			},
+			want: true,
+		},
+		{
+			name: "not exist",
+			args: args{
+				path: "./path_not_exist.go",
+			},
+			want: false,
+		},
+		{
+			name: "is dir",
+			args: args{
+				path: "../util",
+			},
+			want: false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := IsExistsFile(tt.args.path); got != tt.want {
+				t.Errorf("IsExistsFile() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}

pkg/util/url.go

@@ -1,13 +1,42 @@
 package util
 
-import "strings"
+import (
+	"encoding/base64"
+	"regexp"
+	"strings"
+)
 
 const FileSchema = "FILE"
 
 func ParseSchema(url string) string {
 	index := strings.Index(url, ":")
 	if index == -1 {
-		return FileSchema
+		// check is file path
+		if IsExistsFile(url) {
+			return FileSchema
+		}
+		return ""
 	}
-	return strings.ToUpper(url[:index])
+	schema := url[:index]
+	if schema == "data" {
+		schema, _ = ParseDataUri(url)
+	}
+	return strings.ToUpper(schema)
+}
+
+// ParseDataUri parses a data URI and returns the MIME type and decode data.
+func ParseDataUri(uri string) (string, []byte) {
+	re := regexp.MustCompile(`^data:(.*);base64,(.*)$`)
+	matches := re.FindStringSubmatch(uri)
+	if len(matches) != 3 {
+		return "", nil
+	}
+	mime := matches[1]
+	base64Data := matches[2]
+	// 解码Base64数据
+	data, err := base64.StdEncoding.DecodeString(base64Data)
+	if err != nil {
+		return "", nil
+	}
+	return mime, data
 }

pkg/util/url_test.go

@@ -1,6 +1,10 @@
 package util
 
-import "testing"
+import (
+	"encoding/base64"
+	"reflect"
+	"testing"
+)
 
 func TestParseSchema(t *testing.T) {
 	type args struct {
@@ -32,13 +36,27 @@ func TestParseSchema(t *testing.T) {
 			},
 			want: "FILE",
 		},
+		{
+			name: "file-not-exists",
+			args: args{
+				url: "not.txt",
+			},
+			want: "",
+		},
 		{
 			name: "file-no-scheme",
 			args: args{
-				url: "/home/bt.torrent",
+				url: "./url.go",
 			},
 			want: "FILE",
 		},
+		{
+			name: "data-uri",
+			args: args{
+				url: "data:application/x-bittorrent;base64,test",
+			},
+			want: "APPLICATION/X-BITTORRENT",
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
@@ -48,3 +66,65 @@ func TestParseSchema(t *testing.T) {
 		})
 	}
 }
+
+func TestParseDataUri(t *testing.T) {
+	type args struct {
+		uri string
+	}
+	type result struct {
+		mime string
+		data []byte
+	}
+
+	testData := []byte("test")
+	testData64 := base64.StdEncoding.EncodeToString(testData)
+
+	tests := []struct {
+		name string
+		args args
+		want result
+	}{
+		{
+			name: "success",
+			args: args{
+				uri: "data:application/x-bittorrent;base64," + testData64,
+			},
+			want: result{
+				mime: "application/x-bittorrent",
+				data: testData,
+			},
+		},
+		{
+			name: "fail-dirty-data",
+			args: args{
+				uri: "data::application/x-bittorrent;base64,!@$",
+			},
+			want: result{
+				mime: "",
+				data: nil,
+			},
+		},
+		{
+			name: "fail-miss-data",
+			args: args{
+				uri: ":application/x-bittorrent;base64," + testData64,
+			},
+			want: result{
+				mime: "",
+				data: nil,
+			},
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			mime, data := ParseDataUri(tt.args.uri)
+			got := result{
+				mime: mime,
+				data: data,
+			}
+			if !reflect.DeepEqual(got, tt.want) {
+				t.Errorf("ParseDataUri() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}