1
0
mirror of https://github.com/jesseduffield/lazygit.git synced 2024-12-12 11:15:00 +02:00
lazygit/pkg/commands/oscommands/copy.go

143 lines
3.5 KiB
Go
Raw Normal View History

2020-10-04 13:05:39 +02:00
package oscommands
import (
"fmt"
"io"
"os"
"path/filepath"
)
/* MIT License
*
* Copyright (c) 2017 Roland Singer [roland.singer@desertbit.com]
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
// CopyFile copies the contents of the file named src to the file named
// by dst. The file will be created if it does not already exist. If the
// destination file exists, all it's contents will be replaced by the contents
// of the source file. The file mode will be copied from the source and
// the copied data is synced/flushed to stable storage.
func CopyFile(src, dst string) (err error) {
in, err := os.Open(src)
if err != nil {
2023-06-03 07:49:38 +02:00
return //nolint: nakedret
2020-10-04 13:05:39 +02:00
}
defer in.Close()
out, err := os.Create(dst)
if err != nil {
2023-06-03 07:49:38 +02:00
return //nolint: nakedret
2020-10-04 13:05:39 +02:00
}
defer func() {
if e := out.Close(); e != nil {
err = e
}
}()
_, err = io.Copy(out, in)
if err != nil {
2023-06-03 07:49:38 +02:00
return //nolint: nakedret
2020-10-04 13:05:39 +02:00
}
err = out.Sync()
if err != nil {
2023-06-03 07:49:38 +02:00
return //nolint: nakedret
2020-10-04 13:05:39 +02:00
}
si, err := os.Stat(src)
if err != nil {
2023-06-03 07:49:38 +02:00
return //nolint: nakedret
2020-10-04 13:05:39 +02:00
}
err = os.Chmod(dst, si.Mode())
if err != nil {
2023-06-03 07:49:38 +02:00
return //nolint: nakedret
2020-10-04 13:05:39 +02:00
}
2022-03-25 14:16:24 +02:00
return //nolint: nakedret
2020-10-04 13:05:39 +02:00
}
// CopyDir recursively copies a directory tree, attempting to preserve permissions.
2020-10-06 00:23:09 +02:00
// Source directory must exist. If destination already exists we'll clobber it.
2020-10-04 13:05:39 +02:00
// Symlinks are ignored and skipped.
func CopyDir(src string, dst string) (err error) {
src = filepath.Clean(src)
dst = filepath.Clean(dst)
si, err := os.Stat(src)
if err != nil {
return err
}
if !si.IsDir() {
return fmt.Errorf("source is not a directory")
}
_, err = os.Stat(dst)
if err != nil && !os.IsNotExist(err) {
2023-06-03 07:49:38 +02:00
return //nolint: nakedret
2020-10-04 13:05:39 +02:00
}
if err == nil {
2020-10-06 00:23:09 +02:00
// it exists so let's remove it
if err := os.RemoveAll(dst); err != nil {
return err
}
2020-10-04 13:05:39 +02:00
}
err = os.MkdirAll(dst, si.Mode())
if err != nil {
2023-06-03 07:49:38 +02:00
return //nolint: nakedret
2020-10-04 13:05:39 +02:00
}
2023-08-02 00:21:55 +02:00
entries, err := os.ReadDir(src)
2020-10-04 13:05:39 +02:00
if err != nil {
2023-06-03 07:49:38 +02:00
return //nolint: nakedret
2020-10-04 13:05:39 +02:00
}
for _, entry := range entries {
srcPath := filepath.Join(src, entry.Name())
dstPath := filepath.Join(dst, entry.Name())
if entry.IsDir() {
err = CopyDir(srcPath, dstPath)
if err != nil {
2023-06-03 07:49:38 +02:00
return //nolint: nakedret
2020-10-04 13:05:39 +02:00
}
} else {
2023-08-02 00:21:55 +02:00
var info os.FileInfo
info, err = entry.Info()
if err != nil {
return //nolint: nakedret
}
2020-10-04 13:05:39 +02:00
// Skip symlinks.
2023-08-02 00:21:55 +02:00
if info.Mode()&os.ModeSymlink != 0 {
2020-10-04 13:05:39 +02:00
continue
}
err = CopyFile(srcPath, dstPath)
if err != nil {
2023-06-03 07:49:38 +02:00
return //nolint: nakedret
2020-10-04 13:05:39 +02:00
}
}
}
2022-03-25 14:16:24 +02:00
return //nolint: nakedret
2020-10-04 13:05:39 +02:00
}