wide/util/file.go

160 lines
3.3 KiB
Go
Raw Normal View History

// Copyright (c) 2014-2019, b3log.org & hacpai.com
2014-12-13 13:47:41 +03:00
//
2014-11-24 12:24:35 +03:00
// 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
2014-12-13 13:47:41 +03:00
//
2018-03-12 07:28:33 +03:00
// https://www.apache.org/licenses/LICENSE-2.0
2014-12-13 13:47:41 +03:00
//
2014-11-24 12:24:35 +03:00
// 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.
2014-11-16 11:25:27 +03:00
package util
import (
2015-10-12 19:08:37 +03:00
"io"
2014-11-16 11:25:27 +03:00
"os"
2015-10-12 19:08:37 +03:00
"path/filepath"
2014-11-16 11:25:27 +03:00
"strings"
2014-12-13 13:47:41 +03:00
"github.com/b3log/wide/log"
2014-11-16 11:25:27 +03:00
)
2014-12-13 13:47:41 +03:00
// Logger.
var fileLogger = log.NewLogger(os.Stdout)
2014-11-16 11:25:27 +03:00
type myfile struct{}
// File utilities.
var File = myfile{}
// GetFileSize get the length in bytes of file of the specified path.
func (*myfile) GetFileSize(path string) int64 {
2014-11-24 05:48:37 +03:00
fi, err := os.Stat(path)
if nil != err {
2015-12-28 11:00:17 +03:00
fileLogger.Error(err)
return -1
}
return fi.Size()
}
2014-11-16 11:25:27 +03:00
// IsExist determines whether the file spcified by the given path is exists.
func (*myfile) IsExist(path string) bool {
_, err := os.Stat(path)
return err == nil || os.IsExist(err)
}
// IsBinary determines whether the specified content is a binary file content.
func (*myfile) IsBinary(content string) bool {
for _, b := range content {
if 0 == b {
return true
}
}
return false
}
// IsImg determines whether the specified extension is a image.
func (*myfile) IsImg(extension string) bool {
ext := strings.ToLower(extension)
switch ext {
case ".jpg", ".jpeg", ".bmp", ".gif", ".png", ".svg", ".ico":
return true
default:
return false
}
}
// IsDir determines whether the specified path is a directory.
func (*myfile) IsDir(path string) bool {
fio, err := os.Lstat(path)
if nil != err {
2014-12-13 13:47:41 +03:00
fileLogger.Warnf("Determines whether [%s] is a directory failed: [%v]", path, err)
2014-11-16 11:25:27 +03:00
return false
}
return fio.IsDir()
}
2015-10-12 19:08:37 +03:00
// CopyFile copies the source file to the dest file.
func (*myfile) CopyFile(source string, dest string) (err error) {
sourcefile, err := os.Open(source)
if err != nil {
return err
}
defer sourcefile.Close()
destfile, err := os.Create(dest)
if err != nil {
return err
}
defer destfile.Close()
_, err = io.Copy(destfile, sourcefile)
if err == nil {
sourceinfo, err := os.Stat(source)
if err != nil {
err = os.Chmod(dest, sourceinfo.Mode())
}
}
return nil
}
// CopyDir copies the source directory to the dest directory.
func (*myfile) CopyDir(source string, dest string) (err error) {
sourceinfo, err := os.Stat(source)
if err != nil {
return err
}
// create dest dir
err = os.MkdirAll(dest, sourceinfo.Mode())
if err != nil {
return err
}
2015-10-26 09:19:46 +03:00
directory, err := os.Open(source)
if err != nil {
return err
}
defer directory.Close()
2015-10-12 19:08:37 +03:00
objects, err := directory.Readdir(-1)
2015-10-26 09:19:46 +03:00
if err != nil {
return err
}
2015-10-12 19:08:37 +03:00
for _, obj := range objects {
srcFilePath := filepath.Join(source, obj.Name())
destFilePath := filepath.Join(dest, obj.Name())
if obj.IsDir() {
// create sub-directories - recursively
err = File.CopyDir(srcFilePath, destFilePath)
if err != nil {
fileLogger.Error(err)
}
} else {
err = File.CopyFile(srcFilePath, destFilePath)
if err != nil {
fileLogger.Error(err)
}
}
}
return nil
}