From d1fbba7908f744694e5462087361b09837c868fd Mon Sep 17 00:00:00 2001 From: doingodswork Date: Sat, 10 Oct 2020 16:31:34 +0200 Subject: [PATCH] Fix AD check in status endpoint --- cmd/deflix-stremio/handlers.go | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/cmd/deflix-stremio/handlers.go b/cmd/deflix-stremio/handlers.go index d9244c3..13f7a80 100644 --- a/cmd/deflix-stremio/handlers.go +++ b/cmd/deflix-stremio/handlers.go @@ -281,9 +281,10 @@ func createStatusHandler(magnetSearchers map[string]imdb2torrent.MagnetSearcher, logger.Debug("statusHandler called", zap.String("request", fmt.Sprintf("%+v", &c.Fasthttp.Request))) imdbID := c.Query("imdbid", "") - apiToken := c.Query("apitoken", "") - if imdbID == "" || apiToken == "" { - logger.Warn("\"/status\" was called without IMDb ID or API token") + rdToken := c.Query("rdtoken", "") + adKey := c.Query("adkey", "") + if imdbID == "" || rdToken == "" || adKey == "" { + logger.Warn("\"/status\" was called without IMDb ID or RD API token or AD API key") c.SendStatus(fiber.StatusBadRequest) } @@ -333,7 +334,7 @@ func createStatusHandler(magnetSearchers map[string]imdb2torrent.MagnetSearcher, res += "\t" + `"RD": {` + "\n" startRD := time.Now() - streamURL, err := rdClient.GetStreamURL(c.Context(), bigBuckBunnyMagnet, apiToken, false) + streamURL, err := rdClient.GetStreamURL(c.Context(), bigBuckBunnyMagnet, rdToken, false) if err != nil { res += "\t\t" + `"err":"` + err.Error() + `",` + "\n" } else { @@ -347,7 +348,7 @@ func createStatusHandler(magnetSearchers map[string]imdb2torrent.MagnetSearcher, res += "\t" + `"AD": {` + "\n" startAD := time.Now() - streamURL, err = adClient.GetStreamURL(c.Context(), bigBuckBunnyMagnet, apiToken) + streamURL, err = adClient.GetStreamURL(c.Context(), bigBuckBunnyMagnet, adKey) if err != nil { res += "\t\t" + `"err":"` + err.Error() + `",` + "\n" } else {