mirror of https://github.com/mikefarah/yq.git
86 lines
2.0 KiB
Go
86 lines
2.0 KiB
Go
package yqlib
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
)
|
|
|
|
func tryRenameFile(from string, to string) error {
|
|
if renameError := os.Rename(from, to); renameError != nil {
|
|
log.Debugf("Error renaming from %v to %v, attempting to copy contents", from, to)
|
|
log.Debug(renameError.Error())
|
|
log.Debug("going to try copying instead")
|
|
// can't do this rename when running in docker to a file targeted in a mounted volume,
|
|
// so gracefully degrade to copying the entire contents.
|
|
if copyError := copyFileContents(from, to); copyError != nil {
|
|
return fmt.Errorf("failed copying from %v to %v: %w", from, to, copyError)
|
|
}
|
|
tryRemoveTempFile(from)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func tryRemoveTempFile(filename string) {
|
|
log.Debug("Removing temp file: %v", filename)
|
|
removeErr := os.Remove(filename)
|
|
if removeErr != nil {
|
|
log.Errorf("Failed to remove temp file: %v", filename)
|
|
}
|
|
}
|
|
|
|
// thanks https://stackoverflow.com/questions/21060945/simple-way-to-copy-a-file-in-golang
|
|
func copyFileContents(src, dst string) (err error) {
|
|
// ignore CWE-22 gosec issue - that's more targeted for http based apps that run in a public directory,
|
|
// and ensuring that it's not possible to give a path to a file outside thar directory.
|
|
|
|
in, err := os.Open(src) // #nosec
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer safelyCloseFile(in)
|
|
out, err := os.Create(dst)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer safelyCloseFile(out)
|
|
if _, err = io.Copy(out, in); err != nil {
|
|
return err
|
|
}
|
|
return out.Sync()
|
|
}
|
|
|
|
func SafelyCloseReader(reader io.Reader) {
|
|
switch reader := reader.(type) {
|
|
case *os.File:
|
|
safelyCloseFile(reader)
|
|
}
|
|
}
|
|
|
|
func safelyCloseFile(file *os.File) {
|
|
err := file.Close()
|
|
if err != nil {
|
|
log.Error("Error closing file!")
|
|
log.Error(err.Error())
|
|
}
|
|
}
|
|
|
|
func createTempFile() (*os.File, error) {
|
|
_, err := os.Stat(os.TempDir())
|
|
if os.IsNotExist(err) {
|
|
err = os.Mkdir(os.TempDir(), 0700)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
} else if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
file, err := os.CreateTemp("", "temp")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return file, err
|
|
}
|