mirror of
https://github.com/acepanel/panel.git
synced 2026-02-05 00:39:32 +08:00
142 lines
2.6 KiB
Go
142 lines
2.6 KiB
Go
package tools
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/goravel/framework/support"
|
|
)
|
|
|
|
// Write 写入文件
|
|
func Write(path string, data string, permission os.FileMode) error {
|
|
if err := os.MkdirAll(filepath.Dir(path), permission); err != nil {
|
|
return err
|
|
}
|
|
|
|
err := os.WriteFile(path, []byte(data), permission)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Read 读取文件
|
|
func Read(path string) (string, error) {
|
|
data, err := os.ReadFile(path)
|
|
return string(data), err
|
|
}
|
|
|
|
// Remove 删除文件/目录
|
|
func Remove(path string) error {
|
|
return os.RemoveAll(path)
|
|
}
|
|
|
|
// Exec 执行 shell 命令
|
|
func Exec(shell string) (string, error) {
|
|
cmd := exec.Command("bash", "-c", "LC_ALL=C "+shell)
|
|
|
|
var stdoutBuf, stderrBuf bytes.Buffer
|
|
cmd.Stdout = &stdoutBuf
|
|
cmd.Stderr = &stderrBuf
|
|
|
|
err := cmd.Run()
|
|
if err != nil {
|
|
return "", errors.New(strings.TrimSpace(stderrBuf.String()))
|
|
}
|
|
|
|
return strings.TrimSpace(stdoutBuf.String()), err
|
|
}
|
|
|
|
// ExecAsync 异步执行 shell 命令
|
|
func ExecAsync(shell string) error {
|
|
cmd := exec.Command("bash", "-c", shell)
|
|
err := cmd.Start()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
go func() {
|
|
err := cmd.Wait()
|
|
if err != nil {
|
|
if support.Env == support.EnvTest {
|
|
fmt.Println(err.Error())
|
|
panic(err)
|
|
}
|
|
}
|
|
}()
|
|
|
|
return nil
|
|
}
|
|
|
|
// Mkdir 创建目录
|
|
func Mkdir(path string, permission os.FileMode) error {
|
|
return os.MkdirAll(path, permission)
|
|
}
|
|
|
|
// Chmod 修改文件/目录权限
|
|
func Chmod(path string, permission os.FileMode) error {
|
|
return os.Chmod(path, permission)
|
|
}
|
|
|
|
// Chown 修改文件/目录所有者
|
|
func Chown(path, user, group string) error {
|
|
cmd := exec.Command("chown", "-R", user+":"+group, path)
|
|
return cmd.Run()
|
|
}
|
|
|
|
// Exists 判断路径是否存在
|
|
func Exists(path string) bool {
|
|
_, err := os.Stat(path)
|
|
return err == nil
|
|
}
|
|
|
|
// Empty 判断路径是否为空
|
|
func Empty(path string) bool {
|
|
files, err := os.ReadDir(path)
|
|
if err != nil {
|
|
return true
|
|
}
|
|
|
|
return len(files) == 0
|
|
}
|
|
|
|
// Mv 移动文件/目录
|
|
func Mv(src, dst string) error {
|
|
cmd := exec.Command("mv", src, dst)
|
|
|
|
return cmd.Run()
|
|
}
|
|
|
|
// Cp 复制文件/目录
|
|
func Cp(src, dst string) error {
|
|
cmd := exec.Command("cp", "-r", src, dst)
|
|
|
|
return cmd.Run()
|
|
}
|
|
|
|
// Size 获取路径大小
|
|
func Size(path string) (int64, error) {
|
|
var size int64
|
|
|
|
err := filepath.Walk(path, func(filePath string, info os.FileInfo, walkErr error) error {
|
|
if walkErr != nil {
|
|
return walkErr
|
|
}
|
|
size += info.Size()
|
|
return nil
|
|
})
|
|
|
|
return size, err
|
|
}
|
|
|
|
// FileInfo 获取文件大小
|
|
func FileInfo(path string) (os.FileInfo, error) {
|
|
return os.Stat(path)
|
|
}
|