mirror of
https://github.com/zoriya/Kyoo.git
synced 2025-05-24 02:02:36 -04:00
68 lines
1.6 KiB
Go
68 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/labstack/echo/v4"
|
|
)
|
|
|
|
var client = &http.Client{Timeout: 10 * time.Second}
|
|
|
|
type Item struct {
|
|
Path string `json:"path"`
|
|
}
|
|
|
|
func GetPath(c echo.Context) (string, error) {
|
|
key := c.Request().Header.Get("X-Path")
|
|
if key == "" {
|
|
return "", echo.NewHTTPError(http.StatusBadRequest, "Missing resouce path. You need to set the X-Path header.")
|
|
}
|
|
return key, nil
|
|
}
|
|
|
|
func GetRoute(c echo.Context) string {
|
|
return c.Request().Header.Get("X-Route")
|
|
}
|
|
|
|
func SanitizePath(path string) error {
|
|
if strings.Contains(path, "/") || strings.Contains(path, "..") {
|
|
return echo.NewHTTPError(http.StatusBadRequest, "Invalid parameter. Can't contains path delimiters or ..")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func GetClientId(c echo.Context) (string, error) {
|
|
key := c.Request().Header.Get("X-CLIENT-ID")
|
|
if key == "" {
|
|
return "", echo.NewHTTPError(http.StatusBadRequest, "missing client id. Please specify the X-CLIENT-ID header to a guid constant for the lifetime of the player (but unique per instance)")
|
|
}
|
|
return key, nil
|
|
}
|
|
|
|
func ParseSegment(segment string) (int32, error) {
|
|
var ret int32
|
|
_, err := fmt.Sscanf(segment, "segment-%d.ts", &ret)
|
|
if err != nil {
|
|
return 0, echo.NewHTTPError(http.StatusBadRequest, "Could not parse segment.")
|
|
}
|
|
return ret, nil
|
|
}
|
|
|
|
func ErrorHandler(err error, c echo.Context) {
|
|
code := http.StatusInternalServerError
|
|
var message string
|
|
if he, ok := err.(*echo.HTTPError); ok {
|
|
code = he.Code
|
|
message = fmt.Sprint(he.Message)
|
|
} else {
|
|
c.Logger().Error(err)
|
|
message = "Internal server error"
|
|
}
|
|
c.JSON(code, struct {
|
|
Errors []string `json:"errors"`
|
|
}{Errors: []string{message}})
|
|
}
|