git media [clean | smudge] vs git-media-clean

technoweenie · technoweenie · commit 24b1939d45cc · 2014-06-02T09:38:39.000-06:00
diff --git a/commands/command_clean.go b/commands/command_clean.go
@@ -0,0 +1,50 @@
+package gitmedia
+
+import (
+	".."
+	"../filters"
+	"os"
+)
+
+type CleanCommand struct {
+	*Command
+}
+
+func (c *CleanCommand) Run() {
+	var filename string
+	if len(c.Args) > 1 {
+		filename = c.Args[1]
+	} else {
+		filename = ""
+	}
+
+	cleaned, err := gitmediafilters.Clean(os.Stdin)
+	if err != nil {
+		gitmedia.Panic(err, "Error cleaning asset")
+	}
+	defer cleaned.Close()
+
+	tmpfile := cleaned.File.Name()
+	mediafile := gitmedia.LocalMediaPath(cleaned.Sha)
+	if stat, _ := os.Stat(mediafile); stat != nil {
+		if stat.Size() != cleaned.Size {
+			gitmedia.Exit("Files don't match:\n%s\n%s", mediafile, tmpfile)
+		}
+		gitmedia.Debug("%s exists", mediafile)
+	} else {
+		if err := os.Rename(tmpfile, mediafile); err != nil {
+			gitmedia.Panic(err, "Unable to move %s to %s\n", tmpfile, mediafile)
+		}
+
+		gitmedia.QueueUpload(cleaned.Sha, filename)
+		gitmedia.Debug("Writing %s", mediafile)
+	}
+
+	gitmedia.Encode(os.Stdout, cleaned.Sha)
+}
+
+func init() {
+	registerCommand("clean", func(c *Command) RunnableCommand {
+		return &CleanCommand{Command: c}
+	})
+}
diff --git a/commands/command_media.go b/commands/command_media.go
diff --git a/commands/command_smudge.go b/commands/command_smudge.go
@@ -0,0 +1,30 @@
+package gitmedia
+
+import (
+	".."
+	"../filters"
+	"os"
+)
+
+type SmudgeCommand struct {
+	*Command
+}
+
+func (c *SmudgeCommand) Run() {
+	sha, err := gitmedia.Decode(os.Stdin)
+	if err != nil {
+		gitmedia.Panic(err, "Error reading git-media meta data from stdin:")
+	}
+
+	err = gitmediafilters.Smudge(os.Stdout, sha)
+	if err != nil {
+		smudgerr := err.(*gitmediafilters.SmudgeError)
+		gitmedia.Panic(err, "Error reading file from local media dir: %s", smudgerr.Filename)
+	}
+}
+
+func init() {
+	registerCommand("smudge", func(c *Command) RunnableCommand {
+		return &SmudgeCommand{Command: c}
+	})
+}
