Changeset 96 in code for trunk/morty.go

Timestamp:

Aug 11, 2019, 8:34:33 AM (6 years ago)

Author:

alex

Message:

[mod] follow HTTP redirect (only GET HTTP method)

close #48

File:

• trunk/morty.go (modified) (3 diffs)

trunk/morty.go

@@ -36 +36 @@
 
 const VERSION = "v0.2.0"
+
+const MAX_REDIRECT_COUNT = 5
 
 var CLIENT *fasthttp.Client = &fasthttp.Client{
@@ -288 +290 @@
         }
 
-        parsedURI, err := url.Parse(string(requestURI))
+        p.ProcessUri(ctx, string(requestURI), 0)
+}
+
+func (p *Proxy) ProcessUri(ctx *fasthttp.RequestCtx, requestURI string, redirectCount int) {
+        parsedURI, err := url.Parse(requestURI)
 
         if err != nil {
@@ -339 +345 @@
                         loc := resp.Header.Peek("Location")
                         if loc != nil {
-                                rc := &RequestConfig{Key: p.Key, BaseURL: parsedURI}
-                                url, err := rc.ProxifyURI(loc)
-                                if err == nil {
-                                        ctx.SetStatusCode(resp.StatusCode())
-                                        ctx.Response.Header.Add("Location", url)
-                                        log.Println("redirect to", string(loc))
+                                log.Println("redirect to", string(loc))
+                                if ctx.IsGet() {
+                                        // GET method: Morty follows the redirect
+                                        if redirectCount < MAX_REDIRECT_COUNT {
+                                                p.ProcessUri(ctx, string(loc), redirectCount+1)
+                                        } else {
+                                                p.serveMainPage(ctx, 310, errors.New("Too many redirects"))
+                                        }
                                         return
+                                } else {
+                                        // Other HTTP methods: Morty does NOT follow the redirect
+                                        rc := &RequestConfig{Key: p.Key, BaseURL: parsedURI}
+                                        url, err := rc.ProxifyURI(loc)
+                                        if err == nil {
+                                                ctx.SetStatusCode(resp.StatusCode())
+                                                ctx.Response.Header.Add("Location", url)
+                                                return
+                                        }
                                 }
                         }