More reliable serial calculation

[gocheese.git] / list.go
diff --git a/list.go b/list.go

index 48a9d71498fc7cbaebe315bd76cba7d76b1871d2..7ca9b18f08e8bda641173b42d57c4dac3c97d988 100644 (file)
--- a/list.go
+++ b/list.go
@@ -21,6 +21,7 @@ import (
         "bytes"
         "encoding/hex"
         "errors"
+       "fmt"
         "html/template"
         "io/fs"
         "io/ioutil"
@@ -29,6 +30,7 @@ import (
         "os"
         "path/filepath"
         "sort"
+       "strconv"
         "strings"
         "time"
  )
@@ -39,7 +41,7 @@ var (
  <html>
    <head>
      <meta name="pypi:repository-version" content="1.0">
-    <title>Links for root</title>
+    <title>Simple index</title>
    </head>
    <body>{{$Refresh := .RefreshURLPath}}{{range .Packages}}
      <a href="{{$Refresh}}{{.}}/">{{.}}</a><br/>
@@ -137,7 +139,7 @@ func filenameToVersion(fn string) string {
         return cols[0]
  }
  
-func listDir(pkgName string, doSize bool) (int, []*PkgReleaseInfo, error) {
+func listDir(pkgName string, doSize bool) (int64, []*PkgReleaseInfo, error) {
         dirPath := filepath.Join(Root, pkgName)
         entries, err := os.ReadDir(dirPath)
         if err != nil {
@@ -206,7 +208,15 @@ func listDir(pkgName string, doSize bool) (int, []*PkgReleaseInfo, error) {
                 releases = append(releases, release)
         }
         sort.Sort(PkgReleaseInfoByName(releases))
-       return len(entries), releases, nil
+       fi, err := os.Stat(dirPath)
+       if err != nil {
+               return 0, nil, err
+       }
+       serial := fi.ModTime().Unix()
+       if fi, err = os.Stat(filepath.Join(dirPath, MetadataFile)); err == nil {
+               serial += fi.ModTime().Unix()
+       }
+       return serial, releases, nil
  }
  
  func serveListDir(
@@ -224,7 +234,7 @@ func serveListDir(
                 !refreshDir(w, r, pkgName, "", false) {
                 return
         }
-       _, releases, err := listDir(pkgName, false)
+       serial, releases, err := listDir(pkgName, false)
         if err != nil {
                 log.Println("error", r.RemoteAddr, "list", pkgName, err)
                 http.Error(w, err.Error(), http.StatusInternalServerError)
@@ -258,5 +268,7 @@ func serveListDir(
                 http.Error(w, err.Error(), http.StatusInternalServerError)
                 return
         }
+       w.Header().Set("X-PyPI-Last-Serial", strconv.FormatInt(serial, 10))
         w.Write(buf.Bytes())
+       w.Write([]byte(fmt.Sprintf("<!--SERIAL %d-->\n", serial)))
  }