Raise copyright years

[gocheese.git] / fileutils.go

diff --git a/fileutils.go b/fileutils.go
index b1b9ef95714edea8321e5832341f4d005a19f3fa..8a3df63e1a6f9af7155f2f934e8889bd2e474630 100644 (file)
--- a/fileutils.go
+++ b/fileutils.go
@@ -1,6 +1,6 @@
 /*
 GoCheese -- Python private package repository and caching proxy
-Copyright (C) 2019-2020 Sergey Matveev <stargrave@stargrave.org>
+Copyright (C) 2019-2023 Sergey Matveev <stargrave@stargrave.org>
 
 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
@@ -18,20 +18,35 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>.
 package main
 
 import (
+       "log"
+       "net/http"
        "os"
        "path/filepath"
-       "strconv"
        "time"
 )
 
+var (
+       NoSync   = os.Getenv("GOCHEESE_NO_SYNC") == "1"
+       UmaskCur int
+)
+
 func TempFile(dir string) (*os.File, error) {
-       // Assume that probability of suffix collision is negligible
-       suffix := strconv.FormatInt(time.Now().UnixNano()+int64(os.Getpid()), 16)
-       name := filepath.Join(dir, "nncp"+suffix)
-       return os.OpenFile(name, os.O_RDWR|os.O_CREATE|os.O_EXCL, os.FileMode(0666))
+       tmp, err := os.CreateTemp(dir, "gocheese")
+       if err != nil {
+               return nil, err
+       }
+       err = os.Chmod(tmp.Name(), os.FileMode(0666&^UmaskCur))
+       if err != nil {
+               tmp.Close()
+               return nil, err
+       }
+       return tmp, nil
 }
 
 func DirSync(dirPath string) error {
+       if NoSync {
+               return nil
+       }
        fd, err := os.Open(dirPath)
        if err != nil {
                return err
@@ -44,7 +59,7 @@ func DirSync(dirPath string) error {
        return fd.Close()
 }
 
-func WriteFileSync(dirPath, filePath string, data []byte) error {
+func WriteFileSync(dirPath, filePath string, data []byte, mtime time.Time) error {
        dst, err := TempFile(dirPath)
        if err != nil {
                return err
@@ -54,14 +69,32 @@ func WriteFileSync(dirPath, filePath string, data []byte) error {
                dst.Close()
                return err
        }
-       if err = dst.Sync(); err != nil {
-               os.Remove(dst.Name())
-               dst.Close()
-               return err
+       if !NoSync {
+               if err = dst.Sync(); err != nil {
+                       os.Remove(dst.Name())
+                       dst.Close()
+                       return err
+               }
        }
        dst.Close()
+       if err = os.Chtimes(dst.Name(), mtime, mtime); err != nil {
+               return err
+       }
        if err = os.Rename(dst.Name(), filePath); err != nil {
                return err
        }
        return DirSync(dirPath)
 }
+
+func mkdirForPkg(w http.ResponseWriter, r *http.Request, pkgName string) bool {
+       path := filepath.Join(Root, pkgName)
+       if _, err := os.Stat(path); os.IsNotExist(err) {
+               if err = os.Mkdir(path, os.FileMode(0777)); err != nil {
+                       log.Println("error", r.RemoteAddr, "mkdir", pkgName, err)
+                       http.Error(w, err.Error(), http.StatusInternalServerError)
+                       return false
+               }
+               log.Println(r.RemoteAddr, "mkdir", pkgName)
+       }
+       return true
+}