fetcher parses SourceUri and selects right fetcher function

1 files changed, 35 insertions, 3 deletions

diff --git a/fetcher.go b/fetcher.go
index 97fe7ff..26077b8 100644
--- a/fetcher.go
+++ b/fetcher.go
@@ -25,14 +25,46 @@
 package rhimport
 
 import (
+	"fmt"
+	"net/url"
 	"path"
 )
 
+func FetchFileHttp(ctx *ImportContext, uri *url.URL) (err error) {
+	rhl.Printf("HTTP/HTTPS fetcher called for '%s'", ctx.SourceUri)
+	ctx.SourceFile = ctx.Config.TempDir + "/" + path.Base(uri.Path)
+	ctx.DeleteSourceFile = true
+	return
+}
+
+func FetchFileLocal(ctx *ImportContext, uri *url.URL) (err error) {
+	rhl.Printf("Local fetcher called for '%s'", ctx.SourceUri)
+	ctx.SourceFile = uri.Path
+	ctx.DeleteSourceFile = false
+	return
+}
+
+type FetchFunc func(*ImportContext, *url.URL) (err error)
+
+var (
+	fetchers = map[string]FetchFunc{
+		"http":  FetchFileHttp,
+		"https": FetchFileHttp,
+		"local": FetchFileLocal,
+	}
+)
+
 func FetchFile(ctx *ImportContext) (err error) {
 
-	// TODO: fetch file from ctx.SourceUri and put it into ctx.Config.TempDir
+	var uri *url.URL
+	if uri, err = url.Parse(ctx.SourceUri); err != nil {
+		return
+	}
 
-	ctx.SourceFile = ctx.Config.TempDir + "/" + path.Base(ctx.SourceUri)
-	ctx.DeleteSourceFile = true
+	if fetcher, ok := fetchers[uri.Scheme]; ok {
+		err = fetcher(ctx, uri)
+	} else {
+		err = fmt.Errorf("No fetcher for uri scheme '%s' found.", uri.Scheme)
+	}
 	return
 }