fileutils.go

   1 /*
   2 GoCheese -- Python private package repository and caching proxy
   3 Copyright (C) 2019-2023 Sergey Matveev <stargrave@stargrave.org>
   4
   5 This program is free software: you can redistribute it and/or modify
   6 it under the terms of the GNU General Public License as published by
   7 the Free Software Foundation, version 3 of the License.
   8
   9 This program is distributed in the hope that it will be useful,
  10 but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 GNU General Public License for more details.
  13
  14 You should have received a copy of the GNU General Public License
  15 along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16 */
  17
  18 package main
  19
  20 import (
  21         "log"
  22         "net/http"
  23         "os"
  24         "path/filepath"
  25         "time"
  26 )
  27
  28 var (
  29         NoSync   = os.Getenv("GOCHEESE_NO_SYNC") == "1"
  30         UmaskCur int
  31 )
  32
  33 func TempFile(dir string) (*os.File, error) {
  34         tmp, err := os.CreateTemp(dir, "gocheese")
  35         if err != nil {
  36                 return nil, err
  37         }
  38         err = os.Chmod(tmp.Name(), os.FileMode(0666&^UmaskCur))
  39         if err != nil {
  40                 tmp.Close()
  41                 return nil, err
  42         }
  43         return tmp, nil
  44 }
  45
  46 func DirSync(dirPath string) error {
  47         if NoSync {
  48                 return nil
  49         }
  50         fd, err := os.Open(dirPath)
  51         if err != nil {
  52                 return err
  53         }
  54         err = fd.Sync()
  55         if err != nil {
  56                 fd.Close()
  57                 return err
  58         }
  59         return fd.Close()
  60 }
  61
  62 func WriteFileSync(dirPath, filePath string, data []byte, mtime time.Time) error {
  63         dst, err := TempFile(dirPath)
  64         if err != nil {
  65                 return err
  66         }
  67         if _, err = dst.Write(data); err != nil {
  68                 os.Remove(dst.Name())
  69                 dst.Close()
  70                 return err
  71         }
  72         if !NoSync {
  73                 if err = dst.Sync(); err != nil {
  74                         os.Remove(dst.Name())
  75                         dst.Close()
  76                         return err
  77                 }
  78         }
  79         dst.Close()
  80         if err = os.Chtimes(dst.Name(), mtime, mtime); err != nil {
  81                 return err
  82         }
  83         if err = os.Rename(dst.Name(), filePath); err != nil {
  84                 return err
  85         }
  86         return DirSync(dirPath)
  87 }
  88
  89 func mkdirForPkg(w http.ResponseWriter, r *http.Request, pkgName string) bool {
  90         path := filepath.Join(Root, pkgName)
  91         if _, err := os.Stat(path); os.IsNotExist(err) {
  92                 if err = os.Mkdir(path, os.FileMode(0777)); err != nil {
  93                         log.Println("error", r.RemoteAddr, "mkdir", pkgName, err)
  94                         http.Error(w, err.Error(), http.StatusInternalServerError)
  95                         return false
  96                 }
  97                 log.Println(r.RemoteAddr, "mkdir", pkgName)
  98         }
  99         return true
 100 }