From cb4a3b0f857bf0c9c50387cb5dd0764a1082092d Mon Sep 17 00:00:00 2001
From: Christian Pointner <equinox@helsinki.at>
Date: Fri, 11 Mar 2016 18:07:19 +0100
Subject: make FilePolicy more precise


diff --git a/rhimport/core.go b/rhimport/core.go
index efb6e58..e6c6624 100644
--- a/rhimport/core.go
+++ b/rhimport/core.go
@@ -70,6 +70,7 @@ const (
 	Auto FilePolicy = iota
 	Keep
 	Delete
+	DeleteWithDir
 )
 
 type Context struct {
diff --git a/rhimport/fetcher.go b/rhimport/fetcher.go
index 1f1e84f..dd4b6df 100644
--- a/rhimport/fetcher.go
+++ b/rhimport/fetcher.go
@@ -415,5 +415,17 @@ func FetchFile(ctx *Context) (res *Result, err error) {
 	} else {
 		err = fmt.Errorf("No fetcher for uri scheme '%s' found.", uri.Scheme)
 	}
+
+	switch ctx.SourceFilePolicy {
+	case Keep:
+		ctx.DeleteSourceFile = false
+		ctx.DeleteSourceDir = false
+	case DeleteWithDir:
+		ctx.DeleteSourceDir = true
+		fallthrough
+	case Delete:
+		ctx.DeleteSourceFile = true
+	}
+
 	return
 }
-- 
cgit v0.10.2