109 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			109 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			Go
		
	
	
	
// Copyright 2013 Unknown
 | 
						|
//
 | 
						|
// 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 cae implements PHP-like Compression and Archive Extensions.
 | 
						|
package cae
 | 
						|
 | 
						|
import (
 | 
						|
	"io"
 | 
						|
	"os"
 | 
						|
	"strings"
 | 
						|
)
 | 
						|
 | 
						|
// A Streamer describes an streamable archive object.
 | 
						|
type Streamer interface {
 | 
						|
	StreamFile(string, os.FileInfo, []byte) error
 | 
						|
	StreamReader(string, os.FileInfo, io.Reader) error
 | 
						|
	Close() error
 | 
						|
}
 | 
						|
 | 
						|
// A HookFunc represents a middleware for packing and extracting archive.
 | 
						|
type HookFunc func(string, os.FileInfo) error
 | 
						|
 | 
						|
// HasPrefix returns true if name has any string in given slice as prefix.
 | 
						|
func HasPrefix(name string, prefixes []string) bool {
 | 
						|
	for _, prefix := range prefixes {
 | 
						|
		if strings.HasPrefix(name, prefix) {
 | 
						|
			return true
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return false
 | 
						|
}
 | 
						|
 | 
						|
// IsEntry returns true if name equals to any string in given slice.
 | 
						|
func IsEntry(name string, entries []string) bool {
 | 
						|
	for _, e := range entries {
 | 
						|
		if e == name {
 | 
						|
			return true
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return false
 | 
						|
}
 | 
						|
 | 
						|
// IsFilter returns true if given name matches any of global filter rule.
 | 
						|
func IsFilter(name string) bool {
 | 
						|
	if strings.Contains(name, ".DS_Store") {
 | 
						|
		return true
 | 
						|
	}
 | 
						|
	return false
 | 
						|
}
 | 
						|
 | 
						|
// IsExist returns true if given path is a file or directory.
 | 
						|
func IsExist(path string) bool {
 | 
						|
	_, err := os.Stat(path)
 | 
						|
	return err == nil || os.IsExist(err)
 | 
						|
}
 | 
						|
 | 
						|
// Copy copies file from source to target path.
 | 
						|
func Copy(dest, src string) error {
 | 
						|
	// Gather file information to set back later.
 | 
						|
	si, err := os.Lstat(src)
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
 | 
						|
	// Handle symbolic link.
 | 
						|
	if si.Mode()&os.ModeSymlink != 0 {
 | 
						|
		target, err := os.Readlink(src)
 | 
						|
		if err != nil {
 | 
						|
			return err
 | 
						|
		}
 | 
						|
		// NOTE: os.Chmod and os.Chtimes don't recoganize symbolic link,
 | 
						|
		// which will lead "no such file or directory" error.
 | 
						|
		return os.Symlink(target, dest)
 | 
						|
	}
 | 
						|
 | 
						|
	sr, err := os.Open(src)
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	defer sr.Close()
 | 
						|
 | 
						|
	dw, err := os.Create(dest)
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	defer dw.Close()
 | 
						|
 | 
						|
	if _, err = io.Copy(dw, sr); err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
 | 
						|
	// Set back file information.
 | 
						|
	if err = os.Chtimes(dest, si.ModTime(), si.ModTime()); err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	return os.Chmod(dest, si.Mode())
 | 
						|
}
 |