wust校园网认证登录(web + pppoe拨号)
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.
canti/app/commons/file.go

157 lines
3.3 KiB

package commons
import (
"errors"
"fmt"
"io"
"log"
"os"
"path/filepath"
"strings"
)
// 这段代码大部分直接搬运自nps,经过一定的修改
func GetDefaultInstallPath() (path string) {
if IsWindows() {
return `C:\Programs Files\canti`
} else {
return `/etc`
}
}
func CopyProgram(srcPath, binFile string) string {
binPath, _ := filepath.Abs(os.Args[0])
if IsWindows() {
copyForWindows(srcPath, binFile)
} else {
binPath = copyForUnix(srcPath, binFile)
}
chMod(binPath, 0755)
return binPath
}
func copyForUnix(srcPath, binFile string) (targetPath string) {
_, err := copyFile(filepath.Join(srcPath, binFile), "/usr/bin/"+binFile)
if err != nil {
_, err := copyFile(filepath.Join(srcPath, binFile), "/usr/local/bin/"+binFile)
if err != nil {
log.Fatal("复制文件时发生错误:", err)
} else {
return "/usr/local/bin/" + binFile
}
} else {
return "/usr/bin/" + binFile
}
return ""
}
func copyForWindows(srcPath, binFile string) {
copyFile(filepath.Join(srcPath, binFile+".exe"), filepath.Join(GetAppPath(), binFile+".exe"))
}
func MkidrDirAll(path string, v ...string) {
for _, item := range v {
if err := os.MkdirAll(filepath.Join(path, item), 0755); err != nil {
log.Fatalf("Failed to create directory %s error:%s", path, err.Error())
}
}
}
func CopyDir(srcPath string, destPath string) error {
//检测目录正确性
if srcInfo, err := os.Stat(srcPath); err != nil {
fmt.Println(err.Error())
return err
} else {
if !srcInfo.IsDir() {
e := errors.New("SrcPath is not the right directory!")
return e
}
}
if destInfo, err := os.Stat(destPath); err != nil {
return err
} else {
if !destInfo.IsDir() {
e := errors.New("DestInfo is not the right directory!")
return e
}
}
err := filepath.Walk(srcPath, func(path string, f os.FileInfo, err error) error {
if f == nil {
return err
}
if !f.IsDir() {
destNewPath := strings.Replace(path, srcPath, destPath, -1)
log.Println("copy file ::" + path + " to " + destNewPath)
copyFile(path, destNewPath)
if !IsWindows() {
chMod(destNewPath, 0766)
}
}
return nil
})
return err
}
//生成目录并拷贝文件
func copyFile(src, dest string) (w int64, err error) {
srcFile, err := os.Open(src)
if err != nil {
return
}
defer srcFile.Close()
//分割path目录
destSplitPathDirs := strings.Split(dest, string(filepath.Separator))
//检测时候存在目录
destSplitPath := ""
for index, dir := range destSplitPathDirs {
if index < len(destSplitPathDirs)-1 {
destSplitPath = destSplitPath + dir + string(filepath.Separator)
b, _ := pathExists(destSplitPath)
if b == false {
log.Println("mkdir:" + destSplitPath)
//创建目录
err := os.Mkdir(destSplitPath, os.ModePerm)
if err != nil {
log.Fatalln(err)
}
}
}
}
dstFile, err := os.Create(dest)
if err != nil {
return
}
defer dstFile.Close()
return io.Copy(dstFile, srcFile)
}
//检测文件夹路径时候存在
func pathExists(path string) (bool, error) {
_, err := os.Stat(path)
if err == nil {
return true, nil
}
if os.IsNotExist(err) {
return false, nil
}
return false, err
}
func chMod(name string, mode os.FileMode) {
if !IsWindows() {
os.Chmod(name, mode)
}
}
func GetAppPath() string {
if path, err := filepath.Abs(filepath.Dir(os.Args[0])); err == nil {
return path
}
return os.Args[0]
}