105 lines
2.7 KiB
Go
105 lines
2.7 KiB
Go
package commands
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/git-lfs/git-lfs/errors"
|
|
"github.com/git-lfs/git-lfs/filepathfilter"
|
|
"github.com/git-lfs/git-lfs/lfs"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
// smudgeSkip is a command-line flag belonging to the "git-lfs smudge"
|
|
// command specifying whether to skip the smudge process.
|
|
smudgeSkip = false
|
|
)
|
|
|
|
// smudge smudges the given `*lfs.Pointer`, "ptr", and writes its objects
|
|
// contents to the `io.Writer`, "to".
|
|
//
|
|
// If the smudged object did not "pass" the include and exclude filterset, it
|
|
// will not be downloaded, and the object will remain a pointer on disk, as if
|
|
// the smudge filter had not been applied at all.
|
|
//
|
|
// Any errors encountered along the way will be returned immediately if they
|
|
// were non-fatal, otherwise execution will halt and the process will be
|
|
// terminated by using the `commands.Panic()` func.
|
|
func smudge(to io.Writer, ptr *lfs.Pointer, filename string, skip bool, filter *filepathfilter.Filter) error {
|
|
lfs.LinkOrCopyFromReference(ptr.Oid, ptr.Size)
|
|
cb, file, err := lfs.CopyCallbackFile("smudge", filename, 1, 1)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
download := !skip
|
|
if download {
|
|
download = filter.Allows(filename)
|
|
}
|
|
|
|
err = ptr.Smudge(to, filename, download, getTransferManifest(), cb)
|
|
if file != nil {
|
|
file.Close()
|
|
}
|
|
|
|
if err != nil {
|
|
ptr.Encode(to)
|
|
// Download declined error is ok to skip if we weren't requesting download
|
|
if !(errors.IsDownloadDeclinedError(err) && !download) {
|
|
LoggedError(err, "Error downloading object: %s (%s)", filename, ptr.Oid)
|
|
if !cfg.SkipDownloadErrors() {
|
|
os.Exit(2)
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func smudgeCommand(cmd *cobra.Command, args []string) {
|
|
requireStdin("This command should be run by the Git 'smudge' filter")
|
|
lfs.InstallHooks(false)
|
|
|
|
// keeps the initial buffer from lfs.DecodePointer
|
|
b := &bytes.Buffer{}
|
|
r := io.TeeReader(os.Stdin, b)
|
|
|
|
ptr, perr := lfs.DecodePointer(r)
|
|
if perr != nil {
|
|
mr := io.MultiReader(b, os.Stdin)
|
|
if _, err := io.Copy(os.Stdout, mr); err != nil {
|
|
Panic(err, "Error writing data to stdout:")
|
|
}
|
|
return
|
|
}
|
|
|
|
if !smudgeSkip && cfg.Os.Bool("GIT_LFS_SKIP_SMUDGE", false) {
|
|
smudgeSkip = true
|
|
}
|
|
filter := filepathfilter.New(cfg.FetchIncludePaths(), cfg.FetchExcludePaths())
|
|
if err := smudge(os.Stdout, ptr, smudgeFilename(args, perr), smudgeSkip, filter); err != nil {
|
|
Error(err.Error())
|
|
}
|
|
}
|
|
|
|
func smudgeFilename(args []string, err error) string {
|
|
if len(args) > 0 {
|
|
return args[0]
|
|
}
|
|
|
|
if errors.IsSmudgeError(err) {
|
|
return filepath.Base(errors.GetContext(err, "FileName").(string))
|
|
}
|
|
|
|
return "<unknown file>"
|
|
}
|
|
|
|
func init() {
|
|
RegisterCommand("smudge", smudgeCommand, func(cmd *cobra.Command) {
|
|
cmd.Flags().BoolVarP(&smudgeSkip, "skip", "s", false, "")
|
|
})
|
|
}
|