117 lines
2.4 KiB
Go
117 lines
2.4 KiB
Go
// VulcanizeDB
|
|
// Copyright © 2019 Vulcanize
|
|
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package helpers
|
|
|
|
import (
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"syscall"
|
|
|
|
"github.com/mitchellh/go-homedir"
|
|
)
|
|
|
|
func CleanPath(str string) (string, error) {
|
|
path, err := homedir.Expand(filepath.Clean(str))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if strings.Contains(path, "$GOPATH") {
|
|
env := os.Getenv("GOPATH")
|
|
spl := strings.Split(path, "$GOPATH")[1]
|
|
path = filepath.Join(env, spl)
|
|
}
|
|
|
|
return path, nil
|
|
}
|
|
|
|
func ClearFiles(files ...string) error {
|
|
for _, file := range files {
|
|
if _, err := os.Stat(file); err == nil {
|
|
err = os.Remove(file)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
} else if os.IsNotExist(err) {
|
|
continue
|
|
} else {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func CopyFile(src, dst string) error {
|
|
in, err := os.Open(src)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
out, err := os.OpenFile(dst, syscall.O_CREAT|syscall.O_EXCL|os.O_WRONLY, os.FileMode(0666)) // Doesn't overwrite files
|
|
if err != nil {
|
|
in.Close()
|
|
return err
|
|
}
|
|
|
|
_, err = io.Copy(out, in)
|
|
in.Close()
|
|
out.Close()
|
|
return err
|
|
}
|
|
|
|
func CopyDir(src string, dst string, excludeRecursiveDir string) error {
|
|
var err error
|
|
var fds []os.FileInfo
|
|
var srcinfo os.FileInfo
|
|
|
|
srcinfo, err = os.Stat(src)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
mkErr := os.MkdirAll(dst, srcinfo.Mode())
|
|
if mkErr != nil {
|
|
return mkErr
|
|
}
|
|
|
|
fds, readErr := ioutil.ReadDir(src)
|
|
if err != readErr {
|
|
return readErr
|
|
}
|
|
for _, fd := range fds {
|
|
srcfp := filepath.Join(src, fd.Name())
|
|
dstfp := filepath.Join(dst, fd.Name())
|
|
|
|
if fd.IsDir() {
|
|
if fd.Name() != excludeRecursiveDir {
|
|
dirErr := CopyDir(srcfp, dstfp, "")
|
|
if dirErr != nil {
|
|
return dirErr
|
|
}
|
|
}
|
|
} else {
|
|
fileErr := CopyFile(srcfp, dstfp)
|
|
if fileErr != nil {
|
|
return fileErr
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|