RYD-Proxy/main.go

105 lines
2.1 KiB
Go

package main
import (
"compress/gzip"
"io"
"log"
"net"
"net/http"
"net/url"
"os"
"regexp"
"time"
"github.com/andybalholm/brotli"
"github.com/gofiber/fiber/v2"
)
var client *http.Client
func main() {
proxy := os.Getenv("PROXY")
if proxy == "" {
proxy = "socks5://tor-proxy:5566"
}
proxyUrl, _ := url.Parse(proxy)
client = &http.Client{
Transport: &http.Transport{
Dial: (&net.Dialer{
Timeout: 30 * time.Second,
KeepAlive: 30 * time.Second,
}).Dial,
TLSHandshakeTimeout: 10 * time.Second,
ResponseHeaderTimeout: 20 * time.Second,
ExpectContinueTimeout: 1 * time.Second,
IdleConnTimeout: 30 * time.Second,
ForceAttemptHTTP2: true,
MaxConnsPerHost: 0,
MaxIdleConnsPerHost: 10,
MaxIdleConns: 0,
Proxy: http.ProxyURL(proxyUrl),
},
}
app := fiber.New(
fiber.Config{
Prefork: true,
},
)
app.Get("/votes/:videoId", handler)
log.Fatal(app.Listen(":3000"))
}
func handler(c *fiber.Ctx) error {
videoId := c.Params("videoId")
match, _ := regexp.Match("^([a-zA-Z0-9_-]{11})", []byte(videoId))
if !match {
return c.Status(400).SendString("Invalid video id")
}
for true {
req, _ := http.NewRequest("GET", "https://returnyoutubedislikeapi.com/Votes?videoId="+videoId, nil)
req.Header.Add("User-Agent", "Mozilla/5.0 (Windows NT 10.0; rv:91.0) Gecko/20100101 Firefox/91.0")
req.Header.Add("Accept", "application/json")
req.Header.Add("Accept-Language", "en-US,en;q=0.5")
req.Header.Add("Accept-Encoding", "gzip, deflate, br")
req.Header.Add("Connection", "keep-alive")
req.Header.Add("Sec-Fetch-Dest", "empty")
req.Header.Add("Sec-Fetch-Mode", "cors")
req.Header.Add("Sec-Fetch-Site", "same-origin")
resp, err := client.Do(req)
if err != nil || resp.StatusCode == 429 {
continue
}
ce := resp.Header.Get("Content-Encoding")
var stream io.Reader
if ce == "gzip" {
stream, _ = gzip.NewReader(resp.Body)
} else if ce == "br" {
stream = brotli.NewReader(resp.Body)
} else {
stream = resp.Body
}
return c.Status(resp.StatusCode).SendStream(stream)
}
// Should never be reached
return nil
}