Add support for Mottainai API key

Author: Daniele Rondina

README.md

@@ -40,6 +40,7 @@ Available Commands:
   print                   Show configuration params
 
 Flags:
+  -k, --apikey string       Mottainai API Key
   -c, --config string       SimpleStreams Builder configuration file
   -h, --help                help for this command
   -t, --target-dir string   Target dir of operations.

cmd/root.go

@@ -57,9 +57,11 @@ func initCommand(rootCmd *cobra.Command, config *conf.BuilderTreeConfig) {
 
 	pflags.StringP("config", "c", "", "SimpleStreams Builder configuration file")
 	pflags.StringP("target-dir", "t", "", "Target dir of operations.")
+	pflags.StringP("apikey", "k", "", "Mottainai API Key")
 
 	config.Viper.BindPFlag("config", pflags.Lookup("config"))
 	config.Viper.BindPFlag("target-dir", pflags.Lookup("target-dir"))
+	config.Viper.BindPFlag("apikey", pflags.Lookup("apikey"))
 
 	rootCmd.AddCommand(
 		newPrintCommand(config),

pkg/images/file.go

@@ -77,7 +77,7 @@ func BuildImagesFile(config *config.BuilderTreeConfig, sourceDir string) (*lxd_s
 				strings.TrimRight(v.Directory, "/"),
 			)
 			// Fetch and parse remote ssb.json file
-			manifest, err = ReadVersionsManifestJsonFromUrl(ssbPath)
+			manifest, err = ReadVersionsManifestJsonFromUrl(ssbPath, config.Viper.GetString("apikey"))
 
 		} else {
 			// POST: Try to search ssb.json file under local filesystem.

pkg/images/versions.go

@@ -229,8 +229,9 @@ func WriteVersionsManifestJson(manifest *VersionsSSBuilderManifest, out io.Write
 	return enc.Encode(manifest)
 }
 
-func ReadVersionsManifestJsonFromUrl(url string) (*VersionsSSBuilderManifest, error) {
+func ReadVersionsManifestJsonFromUrl(url, apiKey string) (*VersionsSSBuilderManifest, error) {
 	var ans *VersionsSSBuilderManifest = nil
+	var req *http.Request = nil
 	var err error
 
 	transport := &http.Transport{
@@ -251,7 +252,16 @@ func ReadVersionsManifestJsonFromUrl(url string) (*VersionsSSBuilderManifest, er
 		Timeout:   60 * time.Second,
 	}
 
-	resp, err := client.Get(url)
+	req, err = http.NewRequest("GET", url, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	if apiKey != "" {
+		req.Header.Add("Authorization", "token "+apiKey)
+	}
+
+	resp, err := client.Do(req)
 	if err != nil {
 		return nil, err
 	}

pkg/index/index.go

@@ -76,7 +76,8 @@ func BuildIndexStruct(config *config.BuilderTreeConfig, sourceDir string) (*lxd_
 				strings.TrimRight(v.Directory, "/"),
 			)
 			// Fetch and parse remote ssb.json file
-			manifest, err = images.ReadVersionsManifestJsonFromUrl(ssbPath)
+			manifest, err = images.ReadVersionsManifestJsonFromUrl(ssbPath,
+				config.Viper.GetString("apikey"))
 
 		} else {
 			// POST: Try to search ssb.json file under local filesystem.