Some PyPIs do not include schema and domain name

[gocheese.git] / gocheese.go

diff --git a/gocheese.go b/gocheese.go
index 2ba14d6616f6a328a98127dd8cf223806ff908ba..fcc353ebeb94e0ed3bd9b89a9d8bb03161743645 100644 (file)
--- a/gocheese.go
+++ b/gocheese.go
@@ -151,12 +151,21 @@ func refreshDir(
                        http.Error(w, err.Error(), http.StatusInternalServerError)
                        return false
                }
+               if !strings.HasPrefix(pkgURL.Fragment, SHA256Prefix) {
+                       log.Println(r.RemoteAddr, "pypi", filename, "no SHA256 digest provided")
+                       http.Error(w, "no SHA256 digest provided", http.StatusBadGateway)
+                       return false
+               }
                digest, err = hex.DecodeString(strings.TrimPrefix(pkgURL.Fragment, SHA256Prefix))
                if err != nil {
                        http.Error(w, err.Error(), http.StatusBadGateway)
                        return false
                }
                pkgURL.Fragment = ""
+               uri = pkgURL.String()
+               if pkgURL.Host == "" {
+                       uri = *pypiURL + strings.TrimPrefix(uri, "/")
+               }
                path = filepath.Join(dirPath, filename)
                if filename == filenameGet {
                        if killed {
@@ -165,7 +174,7 @@ func refreshDir(
                                return false
                        }
                        log.Println(r.RemoteAddr, "pypi download", filename)
-                       resp, err = http.Get(pkgURL.String())
+                       resp, err = http.Get(uri)
                        if err != nil {
                                http.Error(w, err.Error(), http.StatusBadGateway)
                                return false
@@ -211,7 +220,7 @@ func refreshDir(
                        if _, err = os.Stat(path); err != nil {
                                goto GPGSigSkip
                        }
-                       resp, err := http.Get(pkgURL.String() + GPGSigExt)
+                       resp, err := http.Get(uri + GPGSigExt)
                        if err != nil {
                                goto GPGSigSkip
                        }