Initial commit: docker compose config

server/utils/filetypes.go

@@ -0,0 +1,99 @@
+package utils
+
+import (
+	"path/filepath"
+	"strings"
+
+	"server/torr/state"
+)
+
+var extVideo = map[string]interface{}{
+	".3g2":   nil,
+	".3gp":   nil,
+	".aaf":   nil,
+	".asf":   nil,
+	".avchd": nil,
+	".avi":   nil,
+	".drc":   nil,
+	".flv":   nil,
+	".m2ts":  nil,
+	".m2v":   nil,
+	".m4p":   nil,
+	".m4v":   nil,
+	".mkv":   nil,
+	".mng":   nil,
+	".mov":   nil,
+	".mp2":   nil,
+	".mp4":   nil,
+	".mpe":   nil,
+	".mpeg":  nil,
+	".mpg":   nil,
+	".mpv":   nil,
+	".mts":   nil,
+	".mxf":   nil,
+	".nsv":   nil,
+	".ogg":   nil,
+	".ogv":   nil,
+	".qt":    nil,
+	".rm":    nil,
+	".rmvb":  nil,
+	".roq":   nil,
+	".svi":   nil,
+	".ts":    nil,
+	".vob":   nil,
+	".webm":  nil,
+	".wmv":   nil,
+	".yuv":   nil,
+}
+
+var extAudio = map[string]interface{}{
+	".aac":  nil,
+	".aiff": nil,
+	".ape":  nil,
+	".au":   nil,
+	".dff":  nil,
+	".dsd":  nil,
+	".dsf":  nil,
+	".flac": nil,
+	".gsm":  nil,
+	".it":   nil,
+	".m3u":  nil,
+	".m4a":  nil,
+	".mid":  nil,
+	".mod":  nil,
+	".mp3":  nil,
+	".mpa":  nil,
+	".mpga": nil,
+	".oga":  nil,
+	".ogg":  nil,
+	".opus": nil,
+	".pls":  nil,
+	".ra":   nil,
+	".s3m":  nil,
+	".sid":  nil,
+	".spx":  nil,
+	".wav":  nil,
+	".wma":  nil,
+	".xm":   nil,
+}
+
+func GetMimeType(filename string) string {
+	ext := strings.ToLower(filepath.Ext(filename))
+	if _, ok := extVideo[ext]; ok {
+		return "video/*"
+	}
+	if _, ok := extAudio[ext]; ok {
+		return "audio/*"
+	}
+	return "*/*"
+}
+
+func GetPlayableFiles(st state.TorrentStatus) []*state.TorrentFileStat {
+	files := make([]*state.TorrentFileStat, 0)
+	for _, f := range st.FileStats {
+		if GetMimeType(f.Path) != "*/*" {
+			files = append(files, f)
+		}
+	}
+	return files
+}

server/utils/location.go

@@ -0,0 +1,22 @@
+package utils
+
+import (
+	"github.com/gin-contrib/location/v2"
+	"github.com/gin-gonic/gin"
+)
+
+func GetScheme(c *gin.Context) string {
+	url := location.Get(c)
+	if url == nil {
+		return "http"
+	}
+	return url.Scheme
+}
+
+func GetHost(c *gin.Context) string {
+	url := location.Get(c)
+	if url == nil {
+		return c.Request.Host
+	}
+	return url.Host
+}

server/utils/prallel.go

@@ -0,0 +1,17 @@
+package utils
+
+import (
+	"sync"
+)
+
+func ParallelFor(begin, end int, fn func(i int)) {
+	var wg sync.WaitGroup
+	wg.Add(end - begin)
+	for i := begin; i < end; i++ {
+		go func(i int) {
+			fn(i)
+			wg.Done()
+		}(i)
+	}
+	wg.Wait()
+}

server/utils/strings.go

@@ -0,0 +1,99 @@
+package utils
+
+import (
+	"fmt"
+	"strconv"
+	"strings"
+	"unicode"
+)
+
+const (
+	_ = 1.0 << (10 * iota) // ignore first value by assigning to blank identifier
+	KB
+	MB
+	GB
+	TB
+	PB
+	EB
+)
+
+func Format(b float64) string {
+	multiple := ""
+	value := b
+
+	switch {
+	case b >= EB:
+		value /= EB
+		multiple = "EB"
+	case b >= PB:
+		value /= PB
+		multiple = "PB"
+	case b >= TB:
+		value /= TB
+		multiple = "TB"
+	case b >= GB:
+		value /= GB
+		multiple = "GB"
+	case b >= MB:
+		value /= MB
+		multiple = "MB"
+	case b >= KB:
+		value /= KB
+		multiple = "KB"
+	case b == 0:
+		return "0"
+	default:
+		return strconv.FormatInt(int64(b), 10) + "B"
+	}
+
+	return fmt.Sprintf("%.2f%s", value, multiple)
+}
+
+func CommonPrefix(first, second string) string {
+	var result strings.Builder
+
+	minLength := len(first)
+	if len(second) < minLength {
+		minLength = len(second)
+	}
+
+	for i := 0; i < minLength; i++ {
+		if first[i] != second[i] {
+			break
+		}
+		result.WriteByte(first[i])
+	}
+
+	return result.String()
+}
+
+func NumberPrefix(str string) (int, error) {
+	var result strings.Builder
+
+	for i := 0; i < len(str); i++ {
+		if !unicode.IsDigit(rune(str[i])) {
+			break
+		}
+		result.WriteByte(str[i])
+	}
+
+	return strconv.Atoi(result.String())
+}
+
+func CompareStrings(first, second string) bool {
+	commonPrefix := CommonPrefix(first, second)
+	resultStr1 := strings.TrimPrefix(first, commonPrefix)
+	resultStr2 := strings.TrimPrefix(second, commonPrefix)
+	num1, err1 := NumberPrefix(resultStr1)
+	num2, err2 := NumberPrefix(resultStr2)
+
+	if err1 == nil && err2 == nil {
+		return num1 < num2
+	}
+	if err1 == nil {
+		return true
+	} else if err2 == nil {
+		return false
+	}
+	return resultStr1 < resultStr2
+}