Browse Source
remove server
No known key found for this signature in database
GPG Key ID: A2ECAB73CE12147F
1 changed files with 0 additions and 178 deletions
Unified View
Diff Options
-
+0 -178server.go
+ 0
- 178
server.go
View File
| @ -1,178 +0,0 @@ | |||||
| package pypihub | |||||
| import ( | |||||
| "encoding/base64" | |||||
| "fmt" | |||||
| "io" | |||||
| "net/http" | |||||
| "strconv" | |||||
| "strings" | |||||
| "time" | |||||
| ) | |||||
| type Server struct { | |||||
| client *Client | |||||
| config Config | |||||
| assets []Asset | |||||
| timer *time.Timer | |||||
| } | |||||
| func NewServer(cfg Config) *Server { | |||||
| return &Server{ | |||||
| client: NewClient(cfg), | |||||
| config: cfg, | |||||
| assets: make([]Asset, 0), | |||||
| } | |||||
| } | |||||
| func (s *Server) findAsset(owner string, repo string, id int) *Asset { | |||||
| for _, a := range s.assets { | |||||
| if a.Owner == owner && a.Repo == repo && a.ID == id { | |||||
| return &a | |||||
| } | |||||
| } | |||||
| return nil | |||||
| } | |||||
| func (s *Server) refetchAssets() { | |||||
| var err error | |||||
| s.assets, err = s.client.GetAllAssets() | |||||
| if err != nil { | |||||
| fmt.Println(err) | |||||
| } | |||||
| } | |||||
| func (s *Server) startTimer() { | |||||
| if s.timer != nil { | |||||
| s.timer.Stop() | |||||
| } | |||||
| s.timer = time.AfterFunc(time.Duration(10*time.Minute), func() { | |||||
| go s.refetchAssets() | |||||
| }) | |||||
| } | |||||
| func (s *Server) listAssets(w http.ResponseWriter, r *http.Request) { | |||||
| var repo = strings.Trim(r.URL.Path, "/") | |||||
| w.Header().Add("Content-Type", "text/html") | |||||
| fmt.Fprintf(w, "<html><title>Links for %s</title><meta name=\"api-version\" content=\"2\" /><body>", repo) | |||||
| fmt.Fprintf(w, "<h1>Links for %s</h1>", repo) | |||||
| for _, a := range s.assets { | |||||
| if a.Repo == repo { | |||||
| fmt.Fprintf(w, "<a href=\"%s\">%s</a> ", a.URL(), a.Name) | |||||
| } | |||||
| } | |||||
| fmt.Fprintf(w, "</body></html>") | |||||
| } | |||||
| func (s *Server) listRepoAssets(w http.ResponseWriter, r *http.Request) { | |||||
| var parts = strings.SplitN(strings.Trim(r.URL.Path, "/"), "/", 2) | |||||
| var owner = parts[0] | |||||
| var repo = parts[1] | |||||
| w.Header().Add("Content-Type", "text/html") | |||||
| fmt.Fprintf(w, "<html><title>Links for %s</title><meta name=\"api-version\" content=\"2\" /><body>", repo) | |||||
| fmt.Fprintf(w, "<h1>Links for %s</h1>", repo) | |||||
| for _, a := range s.assets { | |||||
| if a.Owner == owner && a.Repo == repo { | |||||
| fmt.Fprintf(w, "<a href=\"%s\">%s</a> ", a.URL(), a.Name) | |||||
| } | |||||
| } | |||||
| fmt.Fprintf(w, "</body></html>") | |||||
| } | |||||
| func (s *Server) listAllAssets(w http.ResponseWriter, r *http.Request) { | |||||
| w.Header().Add("Content-Type", "text/html") | |||||
| fmt.Fprintf(w, "<html><title>All asset links</title><meta name=\"api-version\" content=\"2\" /><body>") | |||||
| fmt.Fprintf(w, "<h1>All asset links</h1>") | |||||
| for _, a := range s.assets { | |||||
| fmt.Fprintf(w, "<a href=\"%s\">%s</a> ", a.URL(), a.Name) | |||||
| } | |||||
| fmt.Fprintf(w, "</body></html>") | |||||
| } | |||||
| func (s *Server) listRepos(w http.ResponseWriter, r *http.Request) { | |||||
| w.Header().Add("Content-Type", "text/html") | |||||
| fmt.Fprintf(w, "<html><title>Simple index</title><meta name=\"api-version\" content=\"2\" /><body>") | |||||
| for _, r := range s.config.RepoNames { | |||||
| var parts = strings.SplitN(r, "/", 2) | |||||
| fmt.Fprintf(w, "<a href=\"/%s\">%s</a> ", parts[1], parts[1]) | |||||
| } | |||||
| fmt.Fprintf(w, "</body></html>") | |||||
| } | |||||
| func (s *Server) fetchAsset(w http.ResponseWriter, r *http.Request) { | |||||
| var url = strings.Trim(r.URL.Path, "/") | |||||
| var parts = strings.SplitN(url, "/", 4) | |||||
| if len(parts) != 4 { | |||||
| w.WriteHeader(http.StatusNotFound) | |||||
| return | |||||
| } | |||||
| var asset *Asset | |||||
| var id int64 | |||||
| var err error | |||||
| id, err = strconv.ParseInt(parts[2], 10, 32) | |||||
| if err != nil { | |||||
| w.WriteHeader(http.StatusNotFound) | |||||
| return | |||||
| } | |||||
| asset = s.findAsset(parts[0], parts[1], int(id)) | |||||
| if asset == nil { | |||||
| w.WriteHeader(http.StatusNotFound) | |||||
| return | |||||
| } | |||||
| var rc io.ReadCloser | |||||
| rc, err = s.client.DownloadAsset(*asset) | |||||
| if err != nil || rc == nil { | |||||
| w.WriteHeader(http.StatusNotFound) | |||||
| return | |||||
| } | |||||
| defer rc.Close() | |||||
| io.Copy(w, rc) | |||||
| } | |||||
| func (s *Server) serveFavicon(w http.ResponseWriter, r *http.Request) { | |||||
| var decoded []byte | |||||
| var err error | |||||
| decoded, err = base64.StdEncoding.DecodeString("AAABAAEAEBAAAAEACABoBQAAFgAAACgAAAAQAAAAIAAAAAEACAAAAAAAAAEAAAAAAAAAAAAAAAEAAAAAAAAAAAAAPs7/AJ9vNwBO3f8APMz/AEzb/wCdbjYAStn/AJZqNgBI1/8ARtX/AETT/wBA0f8AqnY3AD7P/wCjcjcApHI3ADzN/wBQ3v8AOsv/AE7c/wA4yf8Amm02ALF6OACYazYARtb/AETU/wCveTcAQtL/AKh1NwCtdzcAQND/AKFxNwCmczcAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAcDA4RExUAAAAAAAAAAAAZGhwfDhEAFQAAAAAAAAAACRkaHB8OABMAAAAAAAAAAAcJGRocHw4RAAAAAAAgAgYFBwkZGhwfARETFQAhDyACAwUHCQoaHB8BERMVHSEPIBIUBQcJChocHwEREw0dIQ8AEhQFBwkKGhwfAQQeDR0hDyACBhYYCAALHB8BGx4NHSEPIAIGFhgICgscHxcbHg0dIRAgAgYWGAkKCxwAFxseDR0hECACBhYHCQoAAAAAAB4NHSEQIAIGAAAAAAAAAAAbAA0dIRAgAgAAAAAAAAAAFwAeDR0hECAAAAAAAAAAAAAXGx4NHSEAAAAAAPgfAADwLwAA8C8AAPAPAACAAQAAAAAAAAAAAAAIAAAAABAAAAAAAAAAAAAAgAEAAPAPAAD0DwAA9A8AAPgfAAA=") | |||||
| if err != nil { | |||||
| w.WriteHeader(http.StatusNotFound) | |||||
| return | |||||
| } | |||||
| fmt.Fprintf(w, "%s", decoded) | |||||
| } | |||||
| func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) { | |||||
| var parts = strings.Split(strings.Trim(r.URL.Path, "/"), "/") | |||||
| parts = removeEmpty(parts) | |||||
| fmt.Println(r.URL.Path) | |||||
| switch len(parts) { | |||||
| case 0: | |||||
| s.listAllAssets(w, r) | |||||
| case 1: | |||||
| if parts[0] == "favicon.ico" { | |||||
| s.serveFavicon(w, r) | |||||
| } else { | |||||
| s.listAssets(w, r) | |||||
| } | |||||
| case 2: | |||||
| s.listRepoAssets(w, r) | |||||
| default: | |||||
| s.fetchAsset(w, r) | |||||
| } | |||||
| } | |||||
| func (s *Server) ListenAndServe() error { | |||||
| s.refetchAssets() | |||||
| s.startTimer() | |||||
| http.Handle("/", s) | |||||
| fmt.Println("Server listening at", s.config.Bind) | |||||
| return http.ListenAndServe(s.config.Bind, nil) | |||||
| } | |||||