mirror of
https://github.com/tickstep/aliyunpan.git
synced 2025-01-23 22:42:15 +08:00
131 lines
3.5 KiB
Go
131 lines
3.5 KiB
Go
// Copyright (c) 2020 tickstep.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
package cmdutil
|
|
|
|
import (
|
|
"github.com/kardianos/osext"
|
|
"github.com/tickstep/library-go/logger"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
)
|
|
|
|
func IsPipeInput() bool {
|
|
fileInfo, err := os.Stdin.Stat()
|
|
if err != nil {
|
|
return false
|
|
}
|
|
return (fileInfo.Mode() & os.ModeNamedPipe) == os.ModeNamedPipe
|
|
}
|
|
|
|
// IsIPhoneOS 是否为苹果移动设备
|
|
func IsIPhoneOS() bool {
|
|
if runtime.GOOS == "darwin" && (runtime.GOARCH == "arm" || runtime.GOARCH == "arm64") {
|
|
_, err := os.Stat("Info.plist")
|
|
return err == nil
|
|
}
|
|
return false
|
|
}
|
|
|
|
// ChWorkDir 切换回工作目录
|
|
func ChWorkDir() {
|
|
if !IsIPhoneOS() {
|
|
return
|
|
}
|
|
|
|
dir, err := filepath.Abs("")
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
subPath := filepath.Dir(os.Args[0])
|
|
os.Chdir(strings.TrimSuffix(dir, subPath))
|
|
}
|
|
|
|
// Executable 获取程序所在的真实目录或真实相对路径
|
|
func Executable() string {
|
|
executablePath, err := osext.Executable()
|
|
if err != nil {
|
|
logger.Verbosef("DEBUG: osext.Executable: %s\n", err)
|
|
executablePath, err = filepath.Abs(filepath.Dir(os.Args[0]))
|
|
if err != nil {
|
|
logger.Verbosef("DEBUG: filepath.Abs: %s\n", err)
|
|
executablePath = filepath.Dir(os.Args[0])
|
|
}
|
|
}
|
|
|
|
if IsIPhoneOS() {
|
|
executablePath = filepath.Join(strings.TrimSuffix(executablePath, os.Args[0]), filepath.Base(os.Args[0]))
|
|
}
|
|
|
|
// 读取链接
|
|
linkedExecutablePath, err := filepath.EvalSymlinks(executablePath)
|
|
if err != nil {
|
|
logger.Verbosef("DEBUG: filepath.EvalSymlinks: %s\n", err)
|
|
return executablePath
|
|
}
|
|
return linkedExecutablePath
|
|
}
|
|
|
|
// ExecutablePath 获取程序所在目录
|
|
func ExecutablePath() string {
|
|
return filepath.Dir(Executable())
|
|
}
|
|
|
|
// ExecutablePathJoin 返回程序所在目录的子目录
|
|
func ExecutablePathJoin(subPath string) string {
|
|
return filepath.Join(ExecutablePath(), subPath)
|
|
}
|
|
|
|
// WalkDir 获取指定目录及所有子目录下的所有文件，可以匹配后缀过滤。
|
|
// 支持 Linux/macOS 软链接
|
|
func WalkDir(dirPth, suffix string) (files []string, err error) {
|
|
files = make([]string, 0, 32)
|
|
suffix = strings.ToUpper(suffix) //忽略后缀匹配的大小写
|
|
|
|
var walkFunc filepath.WalkFunc
|
|
walkFunc = func(filename string, fi os.FileInfo, err error) error { //遍历目录
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if fi.IsDir() { // 忽略目录
|
|
return nil
|
|
}
|
|
if fi.Mode()&os.ModeSymlink != 0 { // 读取 symbol link
|
|
err = filepath.Walk(filename+string(os.PathSeparator), walkFunc)
|
|
return err
|
|
}
|
|
|
|
if strings.HasSuffix(strings.ToUpper(fi.Name()), suffix) {
|
|
files = append(files, path.Clean(filename))
|
|
}
|
|
return nil
|
|
}
|
|
|
|
err = filepath.Walk(dirPth, walkFunc)
|
|
return files, err
|
|
}
|
|
|
|
// ConvertToUnixPathSeparator 将 windows 目录分隔符转换为 Unix 的
|
|
func ConvertToUnixPathSeparator(p string) string {
|
|
return strings.Replace(p, "\\", "/", -1)
|
|
}
|
|
|
|
// ConvertToWindowsPathSeparator 将路径中的所有分隔符转换成windows格式
|
|
func ConvertToWindowsPathSeparator(p string) string {
|
|
return strings.Replace(p, "/", "\\", -1)
|
|
}
|