-
Notifications
You must be signed in to change notification settings - Fork 801
/
ioext.go
123 lines (104 loc) · 2.43 KB
/
ioext.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License. See License.txt in the project root for license information.
package ioext
import (
"errors"
"fmt"
"io"
"io/ioutil"
"os"
"path/filepath"
)
// CopyDir recursively copies the specified source directory tree to the
// specified destination. The destination directory must not exist. Any
// symlinks under src are ignored.
func CopyDir(src, dst string) error {
src = filepath.Clean(src)
dst = filepath.Clean(dst)
// verify that src is a directory
srcInfo, err := os.Stat(src)
if err != nil {
return err
}
if !srcInfo.IsDir() {
return fmt.Errorf("source is not a directory")
}
// now verify that dst doesn't exist
_, err = os.Stat(dst)
if err != nil && !os.IsNotExist(err) {
return err
}
if err == nil {
return fmt.Errorf("destination directory already exists")
}
err = os.MkdirAll(dst, srcInfo.Mode())
if err != nil {
return err
}
// get the collection of directory entries under src.
// for each entry build its corresponding path under dst.
entries, err := ioutil.ReadDir(src)
if err != nil {
return err
}
for _, entry := range entries {
// skip symlinks
if entry.Mode()&os.ModeSymlink != 0 {
continue
}
srcPath := filepath.Join(src, entry.Name())
dstPath := filepath.Join(dst, entry.Name())
if entry.IsDir() {
err = CopyDir(srcPath, dstPath)
if err != nil {
return err
}
} else {
err = CopyFile(srcPath, dstPath, true)
if err != nil {
return err
}
}
}
return nil
}
// CopyFile copies the specified source file to the specified destination file.
// Specify true for overwrite to overwrite the destination file if it already exits.
func CopyFile(src, dst string, overwrite bool) error {
srcFile, err := os.Open(src)
if err != nil {
return err
}
defer srcFile.Close()
if !overwrite {
// check if the file exists, if it does then return an error
_, err := os.Stat(dst)
if err != nil && !os.IsNotExist(err) {
return errors.New("won't overwrite destination file")
}
}
dstFile, err := os.Create(dst)
if err != nil {
return err
}
defer dstFile.Close()
_, err = io.Copy(dstFile, srcFile)
if err != nil {
return err
}
// flush the buffer
err = dstFile.Sync()
if err != nil {
return err
}
// copy file permissions
srcInfo, err := os.Stat(src)
if err != nil {
return err
}
err = os.Chmod(dst, srcInfo.Mode())
if err != nil {
return err
}
return nil
}