DWS/hadrian
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

handle queries

Browse Source
This commit is contained in:
Gianni Carafa
2023-11-02 22:50:03 +01:00
parent 62e03a384a
commit 63dcaeba3c
4 changed files with 25 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
handlers/api.go
View File

@@ -15,7 +15,8 @@ func Api(c *fiber.Ctx) error {
// Get the url from the URL
urlQuery := c.Params("*")
body, req, resp, err := fetchSite(urlQuery)
queries := c.Queries()
body, req, resp, err := fetchSite(urlQuery, queries)
if err != nil {
log.Println("ERROR:", err)
c.SendStatus(500)

22
handlers/proxy.go
View File

@@ -18,9 +18,10 @@ var ForwardedFor = getenv("X_FORWARDED_FOR", "66.249.66.1")
func ProxySite(c *fiber.Ctx) error {
// Get the url from the URL
urlQuery := c.Params("*")
url := c.Params("*")
body, _, resp, err := fetchSite(urlQuery)
queries := c.Queries()
body, _, resp, err := fetchSite(url, queries)
if err != nil {
log.Println("ERROR:", err)
c.SendStatus(500)
@@ -31,17 +32,26 @@ func ProxySite(c *fiber.Ctx) error {
return c.SendString(body)
}
func fetchSite(urlQuery string) (string, *http.Request, *http.Response, error) {
u, err := url.Parse(urlQuery)
func fetchSite(urlpath string, queries map[string]string) (string, *http.Request, *http.Response, error) {
urlQuery := "?"
if len(queries) > 0 {
for k, v := range queries {
urlQuery += k + "=" + v + "&"
}
}
urlQuery = strings.TrimSuffix(urlQuery, "&")
u, err := url.Parse(urlpath)
if err != nil {
return "", nil, nil, err
}
log.Println(u.String())
log.Println(u.String() + urlQuery)
// Fetch the site
client := &http.Client{}
req, _ := http.NewRequest("GET", u.String(), nil)
req, _ := http.NewRequest("GET", u.String()+urlQuery, nil)
req.Header.Set("User-Agent", UserAgent)
req.Header.Set("X-Forwarded-For", ForwardedFor)
req.Header.Set("Referer", u.String())

3
handlers/raw.go
View File

@@ -10,7 +10,8 @@ func Raw(c *fiber.Ctx) error {
// Get the url from the URL
urlQuery := c.Params("*")
body, _, _, err := fetchSite(urlQuery)
queries := c.Queries()
body, _, _, err := fetchSite(urlQuery, queries)
if err != nil {
log.Println("ERROR:", err)
c.SendStatus(500)