support lieral

main.go

@@ -9,6 +9,7 @@ import (
 	"log"
 	"math/rand"
 	"net/http"
+	"strings"
 	"time"
 
 	"golang.org/x/time/rate"
@@ -36,22 +37,30 @@ func main() {
 		for len(link) == 0 {
 			link = string(links[rand.Intn(len(links))])
 		}
-		if proxy {
+		if strings.HasPrefix(link, "http") {
-			resp, err := http.Get(link)
+			serveHTTP(w, r, proxy, link)
-			if err != nil {
+		} else if strings.HasPrefix(link, "literal://") {
-				http.Error(w, err.Error(), http.StatusBadGateway)
+			w.Write([]byte(strings.TrimPrefix(link, "literal://") + "\n"))
-			}
-			for k, v := range resp.Header {
-				for _, v2 := range v {
-					w.Header().Add(k, v2)
-				}
-			}
-			w.WriteHeader(resp.StatusCode)
-			io.Copy(w, resp.Body)
-		} else {
-			http.Redirect(w, r, link, http.StatusTemporaryRedirect)
 		}
 	})); err != nil {
 		panic(err)
 	}
 }
+
+func serveHTTP(w http.ResponseWriter, r *http.Request, proxy bool, link string) {
+	if proxy {
+		resp, err := http.Get(link)
+		if err != nil {
+			http.Error(w, err.Error(), http.StatusBadGateway)
+		}
+		for k, v := range resp.Header {
+			for _, v2 := range v {
+				w.Header().Add(k, v2)
+			}
+		}
+		w.WriteHeader(resp.StatusCode)
+		io.Copy(w, resp.Body)
+	} else {
+		http.Redirect(w, r, link, http.StatusTemporaryRedirect)
+	}
+}

testdata/literal

@@ -0,0 +1,2 @@
+literal://hello world
+literal://foo bar