Skip to content

Commit

Permalink
wip
Browse files Browse the repository at this point in the history
  • Loading branch information
larsxschneider committed Oct 24, 2016
1 parent 798998b commit 46a8ee5
Show file tree
Hide file tree
Showing 15 changed files with 451 additions and 9 deletions.
16 changes: 15 additions & 1 deletion .travis.yml
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ env:
global:
- GIT_LFS_TEST_DIR="$HOME/git-lfs-tests"
- GIT_SOURCE_REPO="https://github.com/git/git.git"
- GIT_SOURCE_BRANCH="master"
- GIT_SOURCE_BRANCH="next"

matrix:
fast_finish: true
Expand All @@ -26,6 +26,20 @@ matrix:
make --jobs=2;
make install;
cd ..;
- env: git-from-source
os: osx
before_script:
- >
export NO_OPENSSL=YesPlease;
export APPLE_COMMON_CRYPTO=YesPlease;
brew install gettext;
brew link --force gettext;
git clone $GIT_SOURCE_REPO git-source;
cd git-source;
git checkout $GIT_SOURCE_BRANCH;
make --jobs=2;
make install;
cd ..;
- env: git-latest
os: linux
addons:
Expand Down
2 changes: 1 addition & 1 deletion commands/command_env.go
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ func envCommand(cmd *cobra.Command, args []string) {
Print(env)
}

for _, key := range []string{"filter.lfs.smudge", "filter.lfs.clean"} {
for _, key := range []string{"filter.lfs.process", "filter.lfs.smudge", "filter.lfs.clean"} {
value, _ := cfg.Git.Get(key)
Print("git config %s = %q", key, value)
}
Expand Down
165 changes: 165 additions & 0 deletions commands/command_filter.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,165 @@
package commands

import (
"bytes"
"fmt"
"io"
"os"

"github.com/github/git-lfs/config"
"github.com/github/git-lfs/errors"
"github.com/github/git-lfs/git"
"github.com/github/git-lfs/lfs"
"github.com/github/git-lfs/progress"
"github.com/spf13/cobra"
)

var (
filterSmudgeSkip = false
)

func clean(reader io.Reader, fileName string) ([]byte, error) {
var cb progress.CopyCallback
var file *os.File
var fileSize int64
if len(fileName) > 0 {
stat, err := os.Stat(fileName)
if err == nil && stat != nil {
fileSize = stat.Size()

localCb, localFile, err := lfs.CopyCallbackFile("clean", fileName, 1, 1)
if err != nil {
Error(err.Error())
} else {
cb = localCb
file = localFile
}
}
}

cleaned, err := lfs.PointerClean(reader, fileName, fileSize, cb)
if file != nil {
file.Close()
}

if cleaned != nil {
defer cleaned.Teardown()
}

if errors.IsCleanPointerError(err) {
// TODO: report errors differently!
// os.Stdout.Write(errors.GetContext(err, "bytes").([]byte))
return errors.GetContext(err, "bytes").([]byte), nil
}

if err != nil {
Panic(err, "Error cleaning asset.")
}

tmpfile := cleaned.Filename
mediafile, err := lfs.LocalMediaPath(cleaned.Oid)
if err != nil {
Panic(err, "Unable to get local media path.")
}

if stat, _ := os.Stat(mediafile); stat != nil {
if stat.Size() != cleaned.Size && len(cleaned.Pointer.Extensions) == 0 {
Exit("Files don't match:\n%s\n%s", mediafile, tmpfile)
}
Debug("%s exists", mediafile)
} else {
if err := os.Rename(tmpfile, mediafile); err != nil {
Panic(err, "Unable to move %s to %s\n", tmpfile, mediafile)
}

Debug("Writing %s", mediafile)
}

return []byte(cleaned.Pointer.Encoded()), nil
}

func smudge(reader io.Reader, filename string) ([]byte, error) {
ptr, err := lfs.DecodePointer(reader)
if err != nil {
// mr := io.MultiReader(b, reader)
// _, err := io.Copy(os.Stdout, mr)
// if err != nil {
// Panic(err, "Error writing data to stdout:")
// }
var content []byte
reader.Read(content)
return content, nil
}

lfs.LinkOrCopyFromReference(ptr.Oid, ptr.Size)

cb, file, err := lfs.CopyCallbackFile("smudge", filename, 1, 1)
if err != nil {
Error(err.Error())
}

cfg := config.Config
download := lfs.FilenamePassesIncludeExcludeFilter(filename, cfg.FetchIncludePaths(), cfg.FetchExcludePaths())

if filterSmudgeSkip || cfg.Os.Bool("GIT_LFS_SKIP_SMUDGE", false) {
download = false
}

buf := new(bytes.Buffer)
err = ptr.Smudge(buf, filename, download, TransferManifest(), cb)
if file != nil {
file.Close()
}

if err != nil {
// 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() {
// TODO: What to do best here?
os.Exit(2)
}
}

return []byte(ptr.Encoded()), nil
}

return buf.Bytes(), nil
}

func filterCommand(cmd *cobra.Command, args []string) {
requireStdin("This command should be run by the Git filter process")
lfs.InstallHooks(false)

s := git.NewObjectScanner(os.Stdin, os.Stdout)
s.Init()
s.NegotiateCapabilities()
for {
request, data, err := s.ReadRequest()
if err != nil {
break
}

// TODO:
// ReadRequest should return data as Reader instead of []byte ?!
// clean/smudge should also take a Writer instead of returning []byte
var outputData []byte
switch request["command"] {
case "clean":
outputData, _ = clean(bytes.NewReader(data), request["pathname"])
case "smudge":
outputData, _ = smudge(bytes.NewReader(data), request["pathname"])
default:
fmt.Errorf("Unknown command %s", cmd)
break
}

s.WriteResponse(outputData)
}
}

func init() {
RegisterCommand("filter", filterCommand, func(cmd *cobra.Command) {
cmd.Flags().BoolVarP(&filterSmudgeSkip, "skip", "s", false, "")
})
}
1 change: 1 addition & 0 deletions git/git.go
Original file line number Diff line number Diff line change
Expand Up @@ -809,6 +809,7 @@ func CloneWithoutFilters(flags CloneFlags, args []string) error {
// with --skip-smudge is costly across many files in a checkout
cmdargs := []string{
"-c", fmt.Sprintf("filter.lfs.smudge=%v", filterOverride),
"-c", "filter.lfs.process=",
"-c", "filter.lfs.required=false",
"clone"}

Expand Down
Loading

0 comments on commit 46a8ee5

Please sign in to comment.