Update msx.go (#400)

server/web/msx/msx.go

@@ -3,7 +3,6 @@ package msx
 import (
 	"encoding/json"
 	"errors"
-	"io/fs"
 	"net/http"
 	"os"
 	"strconv"
@@ -24,86 +23,149 @@ var parameter = param
 
 func SetupRoute(r gin.IRouter) {
 	authorized := r.Group("/", auth.CheckAuth())
-	authorized.GET("/msx/stg", func(c *gin.Context) {
+	authorized.Any("/msx", mng)
-		r := map[string]any{"version": version.Version, "search": settings.BTsets.EnableRutorSearch}
+	authorized.GET("/msx/*pth", func(c *gin.Context) { proxy(c, "https://damiva.github.io/msx"+c.Param("pth")) })
-		if p, o := c.GetQuery("parameter"); o {
+
-			if p == "" {
+	authorized.GET("/files", fls)
-				p = param
+	authorized.StaticFS("/files/", gin.Dir(files, true))
-			}
+
-			parameter = p
+	authorized.GET("/imdb/:id", imdb)
-		}
+}
-		r["parameter"] = parameter
+func proxy(c *gin.Context, u string, h ...string) {
-		if p, o := c.GetQuery(files); o {
+	if u == "" {
-			e := os.Remove(files)
+		c.AbortWithStatus(http.StatusBadRequest)
-			if os.IsNotExist(e) {
+	} else if q, e := http.NewRequest(c.Request.Method, u, c.Request.Body); e != nil {
-				e = nil
+		c.AbortWithError(http.StatusInternalServerError, e)
-			}
-			if e == nil && p != "" {
-				var f fs.FileInfo
-				if f, e = os.Stat(p); e == nil {
-					if !f.IsDir() {
-						e = errors.New(p + " is not a folder")
 	} else {
-						e = os.Symlink(p, files)
+		for _, v := range h {
+			if v := strings.SplitN(v, ":", 2); len(v) == 2 {
+				q.Header.Add(v[0], v[1])
 			}
 		}
-			}
+		if r, e := http.DefaultClient.Do(q); e != nil {
-			if e == nil {
+			c.AbortWithError(http.StatusInternalServerError, e)
-				r[files] = p
 		} else {
-				r = map[string]any{"error": e.Error()}
+			c.DataFromReader(r.StatusCode, r.ContentLength, r.Header.Get("Content-Type"), r.Body, nil)
+			r.Body.Close()
 		}
-		} else if l, e := os.Readlink(files); e == nil {
-			r[files] = l
-		} else if !os.IsNotExist(e) {
-			r["error"] = e.Error()
 	}
-		c.JSON(200, r)
+}
-	})
+func mng(c *gin.Context) {
-	authorized.GET("/msx/trn", func(c *gin.Context) {
+	if p := c.Query("url"); p != "" {
-		if h := c.Query("indb"); h != "" {
+		proxy(c, p, c.QueryArray("header")...)
+	} else if c.Request.Method == "POST" {
+		trn(c)
+	} else if p = c.Query("indb"); p != "" {
 		var r bool
 		for _, t := range settings.ListTorrent() {
-				if r = t.InfoHash.HexString() == h; r {
+			if r = (t != nil && t.InfoHash.HexString() == p); r {
 				break
 			}
 		}
 		c.JSON(200, r)
-		} else if h = c.Query("hash"); h != "" {
+	} else {
-			st, sc := trn(h)
+		if p, o := c.GetQuery("parameter"); o {
+			if p == "" {
+				parameter = param
+			} else {
+				parameter = p
+			}
+		}
+		c.JSON(200, map[string]any{"version": version.Version, "search": settings.BTsets.EnableRutorSearch, "parameter": parameter})
+	}
+}
+func trn(c *gin.Context) {
+	var (
+		h, a string
+		q    struct{ Data any }
+		r    struct {
+			R struct {
+				S int    `json:"status"`
+				T string `json:"text"`
+				M string `json:"message,omitempty"`
+				D any    `json:"data,omitempty"`
+			} `json:"response"`
+		}
+	)
+	if e := json.NewDecoder(c.Request.Body).Decode(&q); e != nil {
+		r.R.M = e.Error()
+	} else if s, o := q.Data.(string); o {
+		a, h = s, a[strings.LastIndexByte(s, ':')+1:]
+	} else if s, o := q.Data.(map[string]any); o {
+		if s, o := s["info"].(map[string]any); o {
+			if s, o := s["content"].(map[string]any); o {
+				h, _ = s["flag"].(string)
+			}
+		}
+	}
+	if h != "" {
+		var st, sc string
+		if h := torr.GetTorrent(h); h != nil {
+			if h := h.Status(); h != nil && h.Stat < 5 {
+				switch h.Stat {
+				case 4:
+					sc = "msx-red"
+				case 3:
+					sc = "msx-green"
+				default:
+					sc = "msx-yellow"
+				}
+				st = "{ico:north} " + strconv.Itoa(h.ActivePeers) + " / " + strconv.Itoa(h.TotalPeers) + " {ico:south} " + strconv.Itoa(h.ConnectedSeeders)
+			}
+		}
+		if a != "" {
+			r.R.D = map[string]any{"action": a, "data": map[string]any{
+				"stamp": st, "stampColor": sc, "live": map[string]any{
+					"type": "airtime", "duration": 1000, "over": map[string]any{
+						"action": "execute:" + utils.GetScheme(c) + c.Request.Host + c.Request.URL.Path, "data": a,
+					},
+				},
+			}}
+		} else {
 			if sc != "" {
 				sc = "{col:" + sc + "}"
 			}
-			msx(c, map[string]string{"action": "player:label:position:{VALUE}{tb}{tb}" + sc + st})
+			r.R.D = map[string]string{"action": "player:label:position:{VALUE}{tb}{tb}" + sc + st}
-		} else {
-			c.AbortWithStatus(http.StatusBadRequest)
 		}
-	})
+	} else if r.R.M == "" {
-	authorized.POST("/msx/trn", func(c *gin.Context) {
+		r.R.M = "wrong data struct"
-		var j struct{ Data string }
-		if e := json.NewDecoder(c.Request.Body).Decode(&j); e != nil {
-			msx(c, e)
-		} else {
-			st, sc := trn(j.Data[strings.LastIndexByte(j.Data, ':')+1:])
-			msx(c, map[string]any{"stamp": st, "stampColor": sc, "live": map[string]any{
-				"type": "airtime", "duration": 1000, "over": map[string]any{
-					"action": "execute:" + utils.GetScheme(c) + "://" + c.Request.Host + c.Request.URL.Path,
-					"data":   j.Data,
-				},
-			}})
 	}
-	})
+	if r.R.D == nil {
-	authorized.Any("/msx/proxy", func(c *gin.Context) {
+		r.R.S = http.StatusBadRequest
-		proxy(c, c.Query("url"), c.QueryArray("header")...)
+	} else {
-	})
+		r.R.S = http.StatusOK
-	authorized.GET("/msx/start.json", func(c *gin.Context) {
+	}
-		c.JSON(200, map[string]any{"name": "TorrServer", "version": version.Version, "parameter": parameter})
+	r.R.T = http.StatusText(r.R.S)
-	})
+	c.JSON(200, &r)
-	authorized.GET("/msx/", func(c *gin.Context) {
+}
-		proxy(c, "https://damiva.github.io"+c.Request.URL.EscapedPath())
+func fls(c *gin.Context) {
-	})
+	var e error
-	authorized.GET("/imdb/:id", func(c *gin.Context) {
+	p, o := c.GetQuery("path")
-		i, l, j := c.Param("id"), "", false
+	if o {
+		if e = os.Remove(files); e != nil && os.IsNotExist(e) {
+			e = nil
+		}
+		if e == nil && p != "" {
+			var f os.FileInfo
+			if f, e = os.Stat(p); e == nil {
+				if f.IsDir() {
+					e = os.Symlink(p, files)
+				} else {
+					e = errors.New(p + " is not a directory")
+				}
+			}
+		}
+	} else if p, e = os.Readlink(files); e != nil && os.IsNotExist(e) {
+		e = nil
+	}
+	if e == nil {
+		c.JSON(200, p)
+	} else {
+		c.AbortWithError(http.StatusInternalServerError, e)
+	}
+}
+func imdb(c *gin.Context) {
+	i, l, j := strings.TrimPrefix(c.Param("id"), "/"), "", false
 	if j = strings.HasSuffix(i, ".json"); !j {
 		i += ".json"
 	}
@@ -125,60 +187,4 @@ func SetupRoute(r gin.IRouter) {
 	} else {
 		c.Redirect(http.StatusMovedPermanently, l)
 	}
-	})
-	authorized.Static("/files", files)
-}
-func proxy(c *gin.Context, u string, h ...string) {
-	if u == "" {
-		c.AbortWithStatus(http.StatusBadRequest)
-	} else if q, e := http.NewRequest(c.Request.Method, u, c.Request.Body); e != nil {
-		c.AbortWithError(http.StatusInternalServerError, e)
-	} else {
-		for _, v := range h {
-			if v := strings.SplitN(v, ":", 2); len(v) == 2 {
-				q.Header.Add(v[0], v[1])
-			}
-		}
-		if r, e := http.DefaultClient.Do(q); e != nil {
-			c.AbortWithError(http.StatusInternalServerError, e)
-		} else {
-			c.DataFromReader(r.StatusCode, r.ContentLength, r.Header.Get("Content-Type"), r.Body, nil)
-			r.Body.Close()
-		}
-	}
-}
-func msx(c *gin.Context, d any) {
-	var r struct {
-		R struct {
-			S int    `json:"status"`
-			T string `json:"text"`
-			M string `json:"message,omitempty"`
-			D any    `json:"data,omitempty"`
-		} `json:"response"`
-	}
-	if e, o := d.(error); o {
-		r.R.S = http.StatusBadRequest
-		r.R.M = e.Error()
-	} else {
-		r.R.S = http.StatusOK
-		r.R.D = d
-	}
-	r.R.T = http.StatusText(r.R.S)
-	c.JSON(200, &r)
-}
-func trn(h string) (t, c string) {
-	if h := torr.GetTorrent(h); h != nil {
-		if h := h.Status(); h != nil && h.Stat < 5 {
-			switch h.Stat {
-			case 4:
-				c = "msx-red"
-			case 3:
-				c = "msx-green"
-			default:
-				c = "msx-yellow"
-			}
-			t = "{ico:north} " + strconv.Itoa(h.ActivePeers) + " / " + strconv.Itoa(h.TotalPeers) + " {ico:south} " + strconv.Itoa(h.ConnectedSeeders)
-		}
-	}
-	return
 }