Use match for mirror

pacur · Sep 6, 2015 · 9ac9895 · 9ac9895
1 parent 1e84114
commit 9ac9895
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 2 deletions.
diff --git a/mirror/errors.go b/mirror/errors.go
@@ -4,6 +4,10 @@ import (
 	"github.com/dropbox/godropbox/errors"
 )
 
+type BuildError struct {
+	errors.DropboxError
+}
+
 type UnknownType struct {
 	errors.DropboxError
 }
diff --git a/mirror/mirror.go b/mirror/mirror.go
@@ -88,7 +88,16 @@ func (m *Mirror) createDebian() (err error) {
 		return
 	}
 
-	debs, err := utils.FindExt(m.Root, ".deb")
+	match, ok := constants.ReleasesMatch[m.Distro + "-" + m.Release]
+	if !ok {
+		err = &BuildError{
+			errors.Newf("mirror: Failed to find match for '%s'",
+				m.Distro + "-" + m.Release),
+		}
+		return
+	}
+
+	debs, err := utils.FindMatch(m.Root, match)
 	if err != nil {
 		return
 	}
@@ -112,7 +121,16 @@ func (m *Mirror) createRedhat() (err error) {
 		return
 	}
 
-	err = utils.RsyncExt(m.Root, outDir, ".rpm")
+	match, ok := constants.ReleasesMatch[m.Distro + "-" + m.Release]
+	if !ok {
+		err = &BuildError{
+			errors.Newf("mirror: Failed to find match for '%s'",
+				m.Distro + "-" + m.Release),
+		}
+		return
+	}
+
+	err = utils.RsyncMatch(m.Root, outDir, match)
 	if err != nil {
 		return
 	}