/*
GoCheese -- Python private package repository and caching proxy
-Copyright (C) 2019 Sergey Matveev <stargrave@stargrave.org>
+Copyright (C) 2019-2022 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
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
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
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
+}