Mirrors/iptv-proxy
1
0
Fork 0
mirror of https://github.com/pierre-emmanuelJ/iptv-proxy.git synced 2026-03-15 17:09:16 +01:00
Code Issues Packages Projects Releases Wiki Activity

Forward HTTP headers on request

Browse Source 
Signed-off-by: Pierre-Emmanuel Jacquier <15922119+pierre-emmanuelJ@users.noreply.github.com>
This commit is contained in:
Pierre-Emmanuel Jacquier
2021-03-21 23:11:02 +01:00
parent 66d0a5c486
commit c9f2c6398b
4 changed files with 58 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
pkg/xtream-proxy/xtream-proxy.go
View File

@@ -104,36 +104,46 @@ func (c *Client) Action(config *config.ProxyConfig, action string, q url.Values)
case getLiveCategories:
respBody, err = c.GetLiveCategories()
case getLiveStreams:
respBody, err = c.GetLiveStreams("")
categoryID := ""
if len(q["category_id"]) > 0 {
categoryID = q["category_id"][0]
}
respBody, err = c.GetLiveStreams(categoryID)
case getVodCategories:
respBody, err = c.GetVideoOnDemandCategories()
case getVodStreams:
respBody, err = c.GetVideoOnDemandStreams("")
categoryID := ""
if len(q["category_id"]) > 0 {
categoryID = q["category_id"][0]
}
respBody, err = c.GetVideoOnDemandStreams(categoryID)
case getVodInfo:
if len(q["vod_id"]) < 1 {
err = fmt.Errorf(`bad body url query parameters: missing "vod_id"`)
httpcode = http.StatusBadRequest
httpcode, err = validateParams(q, "vod_id")
if err != nil {
return
}
respBody, err = c.GetVideoOnDemandInfo(q["vod_id"][0])
case getSeriesCategories:
respBody, err = c.GetSeriesCategories()
case getSeries:
respBody, err = c.GetSeries("")
categoryID := ""
if len(q["category_id"]) > 0 {
categoryID = q["category_id"][0]
}
respBody, err = c.GetSeries(categoryID)
case getSerieInfo:
if len(q["series_id"]) < 1 {
err = fmt.Errorf(`bad body url query parameters: missing "series_id"`)
httpcode = http.StatusBadRequest
httpcode, err = validateParams(q, "series_id")
if err != nil {
return
}
respBody, err = c.GetSeriesInfo(q["series_id"][0])
case getShortEPG:
if len(q["stream_id"]) < 1 {
err = fmt.Errorf(`bad body url query parameters: missing "stream_id"`)
httpcode = http.StatusBadRequest
limit := 0
httpcode, err = validateParams(q, "stream_id")
if err != nil {
return
}
limit := 0
if len(q["limit"]) > 0 {
limit, err = strconv.Atoi(q["limit"][0])
if err != nil {
@@ -143,9 +153,8 @@ func (c *Client) Action(config *config.ProxyConfig, action string, q url.Values)
}
respBody, err = c.GetShortEPG(q["stream_id"][0], limit)
case getSimpleDataTable:
if len(q["stream_id"]) < 1 {
err = fmt.Errorf(`bad body url query parameters: missing "stream_id"`)
httpcode = http.StatusBadRequest
httpcode, err = validateParams(q, "stream_id")
if err != nil {
return
}
respBody, err = c.GetEPG(q["stream_id"][0])
@@ -155,3 +164,14 @@ func (c *Client) Action(config *config.ProxyConfig, action string, q url.Values)
return
}
func validateParams(u url.Values, params ...string) (int, error) {
for _, p := range params {
if len(u[p]) < 1 {
return http.StatusBadRequest, fmt.Errorf("missing %q", p)
}
}
return 0, nil
}