RYD-Proxy/main.go

119 lines
2.5 KiB
Go
Raw Permalink Normal View History

2022-06-18 01:24:45 +05:30
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")
2022-06-24 02:32:51 +05:30
var httpProxy func(*http.Request) (*url.URL, error)
2022-06-18 01:24:45 +05:30
2022-06-24 02:32:51 +05:30
if proxy != "" {
proxyUrl, _ := url.Parse(proxy)
httpProxy = http.ProxyURL(proxyUrl)
2022-06-18 01:24:45 +05:30
}
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,
2022-06-24 02:32:51 +05:30
Proxy: httpProxy,
2022-06-18 01:24:45 +05:30
},
}
app := fiber.New(
fiber.Config{
Prefork: true,
},
)
// Route for /votes?videoId=:videoId
app.Get("/votes", handleQuery)
// Route for /votes/:videoId
app.Get("/votes/:videoId", handleParam)
2022-06-18 01:24:45 +05:30
log.Fatal(app.Listen(":3000"))
}
func handleQuery(c *fiber.Ctx) error {
videoId := c.Query("videoId")
return getVotes(c, videoId)
}
2022-06-18 01:24:45 +05:30
func handleParam(c *fiber.Ctx) error {
2022-06-18 01:24:45 +05:30
videoId := c.Params("videoId")
return getVotes(c, videoId)
}
2022-06-18 01:24:45 +05:30
func getVotes(c *fiber.Ctx, videoId string) error {
2022-06-18 01:24:45 +05:30
match, _ := regexp.Match("^([a-zA-Z0-9_-]{11})", []byte(videoId))
if !match {
return c.Status(400).SendString("Invalid video id")
}
for {
2022-06-18 01:48:12 +05:30
req, _ := http.NewRequest("GET", "https://returnyoutubedislikeapi.com/Votes?videoId="+videoId+"&likeCount=", nil)
2022-06-18 01:24:45 +05:30
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 {
continue
}
defer resp.Body.Close()
if resp.StatusCode == 429 {
2022-06-18 01:24:45 +05:30
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)
}
}