Initial commit.

1 files changed, 34 insertions, 0 deletions

diff --git a/imagestore/namepathmapper.go b/imagestore/namepathmapper.go
new file mode 100644
index 0000000..919007b
--- /dev/null
+++ b/imagestore/namepathmapper.go
@@ -0,0 +1,34 @@
+package imagestore
+
+import (
+	"regexp"
+	"strings"
+)
+
+type NamePathMapper struct {
+	regex   *regexp.Regexp
+	replace string
+}
+
+func NewNamePathMapper(expr string, mapping string) *NamePathMapper {
+	var r *regexp.Regexp
+	if len(expr) > 0 {
+		r = regexp.MustCompile(expr)
+	}
+
+	return &NamePathMapper{
+		r,
+		mapping,
+	}
+}
+
+func (this *NamePathMapper) mapToPath(obj *StoreObject) string {
+	repl := strings.Replace(this.replace, "${ImageName}", obj.Name, -1)
+	repl = strings.Replace(repl, "${ImageSize}", obj.Type, -1)
+
+	if this.regex != nil {
+		return this.regex.ReplaceAllString(obj.Name, repl)
+	}
+
+	return repl
+}