gogs
/
random-redirect
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

norepeat

master^2
bel 2023-01-17 22:14:42 -07:00
parent 90e1dd6c78
commit 762567080d
1 changed files with 33 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
main.go
View File

@ -31,15 +31,23 @@ func main() {
} }
links := bytes.Split(b, []byte{'\n'}) links := bytes.Split(b, []byte{'\n'})
log.Print(port) log.Print(port)
limiter := rate.NewLimiter(1, 1) server := Server{
if err := http.ListenAndServe(fmt.Sprintf(":%d", port), ServeHTTP(limiter, links, proxy)); err != nil { limiter: rate.NewLimiter(1, 1),
links: links,
proxy: proxy,
}
if err := http.ListenAndServe(fmt.Sprintf(":%d", port), server); err != nil {
panic(err) panic(err)
} }
} }
func ServeHTTP(limiter *rate.Limiter, links [][]byte, proxy bool) http.HandlerFunc { type Server struct {
handler := serveHTTP(limiter, links, proxy) limiter *rate.Limiter
return func(w http.ResponseWriter, r *http.Request) { proxy bool
links [][]byte
}
func (server Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
s := r.URL.Query().Get("n") s := r.URL.Query().Get("n")
if s == "" { if s == "" {
s = "1" s = "1"
@ -49,21 +57,19 @@ func ServeHTTP(limiter *rate.Limiter, links [][]byte, proxy bool) http.HandlerFu
http.Error(w, err.Error(), http.StatusBadRequest) http.Error(w, err.Error(), http.StatusBadRequest)
return return
} }
for i := 0; i < n; i++ {
handler(w, r) links := make([]string, n)
} for i := range links {
link := ""
for len(link) == 0 || strings.Contains(fmt.Sprint(links), link) {
link = string(server.links[rand.Intn(len(server.links))])
} }
links[i] = link
} }
func serveHTTP(limiter *rate.Limiter, links [][]byte, proxy bool) http.HandlerFunc { for _, link := range links {
return func(w http.ResponseWriter, r *http.Request) {
limiter.Wait(r.Context())
link := ""
for len(link) == 0 {
link = string(links[rand.Intn(len(links))])
}
if strings.HasPrefix(link, "http") { if strings.HasPrefix(link, "http") {
_serveHTTP(w, r, proxy, link) server.serveHTTP(w, r, link)
} else if strings.HasPrefix(link, "literal://") { } else if strings.HasPrefix(link, "literal://") {
w.Write([]byte(strings.TrimPrefix(link, "literal://"))) w.Write([]byte(strings.TrimPrefix(link, "literal://")))
} }
@ -71,8 +77,9 @@ func serveHTTP(limiter *rate.Limiter, links [][]byte, proxy bool) http.HandlerFu
} }
} }
func _serveHTTP(w http.ResponseWriter, r *http.Request, proxy bool, link string) { func (server Server) serveHTTP(w http.ResponseWriter, r *http.Request, link string) {
if proxy { if server.proxy {
server.limiter.Wait(r.Context())
resp, err := http.Get(link) resp, err := http.Get(link)
if err != nil { if err != nil {
http.Error(w, err.Error(), http.StatusBadGateway) http.Error(w, err.Error(), http.StatusBadGateway)