You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
111 lines
2.4 KiB
111 lines
2.4 KiB
package os
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
)
|
|
|
|
type logger interface {
|
|
Info(msg string, keyvals ...interface{})
|
|
}
|
|
|
|
// TrapSignal catches the SIGTERM/SIGINT and executes cb function. After that it exits
|
|
// with code 0.
|
|
func TrapSignal(logger logger, cb func()) {
|
|
c := make(chan os.Signal, 1)
|
|
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
|
|
go func() {
|
|
for sig := range c {
|
|
logger.Info(fmt.Sprintf("captured %v, exiting...", sig))
|
|
if cb != nil {
|
|
cb()
|
|
}
|
|
os.Exit(0)
|
|
}
|
|
}()
|
|
}
|
|
|
|
// Kill the running process by sending itself SIGTERM.
|
|
func Kill() error {
|
|
p, err := os.FindProcess(os.Getpid())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return p.Signal(syscall.SIGTERM)
|
|
}
|
|
|
|
func Exit(s string) {
|
|
fmt.Printf(s + "\n")
|
|
os.Exit(1)
|
|
}
|
|
|
|
// EnsureDir ensures the given directory exists, creating it if necessary.
|
|
// Errors if the path already exists as a non-directory.
|
|
func EnsureDir(dir string, mode os.FileMode) error {
|
|
err := os.MkdirAll(dir, mode)
|
|
if err != nil {
|
|
return fmt.Errorf("could not create directory %q: %w", dir, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func FileExists(filePath string) bool {
|
|
_, err := os.Stat(filePath)
|
|
return !os.IsNotExist(err)
|
|
}
|
|
|
|
func ReadFile(filePath string) ([]byte, error) {
|
|
return ioutil.ReadFile(filePath)
|
|
}
|
|
|
|
func MustReadFile(filePath string) []byte {
|
|
fileBytes, err := ioutil.ReadFile(filePath)
|
|
if err != nil {
|
|
Exit(fmt.Sprintf("MustReadFile failed: %v", err))
|
|
return nil
|
|
}
|
|
return fileBytes
|
|
}
|
|
|
|
func WriteFile(filePath string, contents []byte, mode os.FileMode) error {
|
|
return ioutil.WriteFile(filePath, contents, mode)
|
|
}
|
|
|
|
func MustWriteFile(filePath string, contents []byte, mode os.FileMode) {
|
|
err := WriteFile(filePath, contents, mode)
|
|
if err != nil {
|
|
Exit(fmt.Sprintf("MustWriteFile failed: %v", err))
|
|
}
|
|
}
|
|
|
|
// CopyFile copies a file. It truncates the destination file if it exists.
|
|
func CopyFile(src, dst string) error {
|
|
srcfile, err := os.Open(src)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer srcfile.Close()
|
|
|
|
info, err := srcfile.Stat()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if info.IsDir() {
|
|
return errors.New("cannot read from directories")
|
|
}
|
|
|
|
// create new file, truncate if exists and apply same permissions as the original one
|
|
dstfile, err := os.OpenFile(dst, os.O_RDWR|os.O_CREATE|os.O_TRUNC, info.Mode().Perm())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer dstfile.Close()
|
|
|
|
_, err = io.Copy(dstfile, srcfile)
|
|
return err
|
|
}
|