diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/helsinki.at/rhimport/fetcher.go | 38 |
1 files changed, 35 insertions, 3 deletions
diff --git a/src/helsinki.at/rhimport/fetcher.go b/src/helsinki.at/rhimport/fetcher.go index 97fe7ff..26077b8 100644 --- a/src/helsinki.at/rhimport/fetcher.go +++ b/src/helsinki.at/rhimport/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 } |