Merge branch 'accept-url-as-base64-string' into 'master'

Accept URL as base64 string in scrape request

See merge request saburly/kiviscraplib!2
This commit was merged in pull request #2.
This commit is contained in:
Senad Uka
2020-05-12 11:56:07 +00:00

View File

@@ -1,6 +1,7 @@
package webserver
import (
b64 "encoding/base64"
c "gitlab.com/saburly/kiviscraplib/config"
"gitlab.com/saburly/kiviscraplib/structures"
"gitlab.com/saburly/kiviscraplib/utils"
@@ -41,13 +42,16 @@ func httpHandler(w http.ResponseWriter, r *http.Request) {
return
}
if !utils.IsValidUrl(url[0]) {
decodedUrlBytes, _ := b64.StdEncoding.DecodeString(url[0])
decodedUrl := string(decodedUrlBytes)
if !utils.IsValidUrl(decodedUrl) {
respondWithError(w, 401, "url is malformed")
return
}
request := structures.Request{
Url: url[0],
Url: decodedUrl,
Response: make(chan structures.Response),
}