mirror of https://github.com/helm/helm
commit
dde6bb37dd
@ -0,0 +1,51 @@
|
|||||||
|
package repo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"io/ioutil"
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"gopkg.in/yaml.v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
// IndexFile represents the index file in a chart repository
|
||||||
|
type IndexFile struct {
|
||||||
|
Entries map[string]*ChartRef
|
||||||
|
}
|
||||||
|
|
||||||
|
// ChartRef represents a chart entry in the IndexFile
|
||||||
|
type ChartRef struct {
|
||||||
|
Name string `yaml:"name"`
|
||||||
|
URL string `yaml:"url"`
|
||||||
|
Keywords []string `yaml:"keywords"`
|
||||||
|
Removed bool `yaml:"removed,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// DownloadIndexFile uses
|
||||||
|
func DownloadIndexFile(repoName, url, indexFileName string) error {
|
||||||
|
var indexURL string
|
||||||
|
|
||||||
|
indexURL = strings.TrimSuffix(url, "/") + "/index.yaml"
|
||||||
|
resp, err := http.Get(indexURL)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
var r IndexFile
|
||||||
|
|
||||||
|
b, err := ioutil.ReadAll(resp.Body)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := yaml.Unmarshal(b, &r); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := ioutil.WriteFile(indexFileName, b, 0644); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
@ -0,0 +1,56 @@
|
|||||||
|
package repo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"io/ioutil"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"gopkg.in/yaml.v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
testRepo = "test-repo"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestDownloadIndexFile(t *testing.T) {
|
||||||
|
fileBytes, err := ioutil.ReadFile("testdata/local-index.yaml")
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("%#v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
w.Header().Set("Content-Type", "binary/octet-stream")
|
||||||
|
fmt.Fprintln(w, string(fileBytes))
|
||||||
|
}))
|
||||||
|
|
||||||
|
dirName, err := ioutil.TempDir("testdata", "tmp")
|
||||||
|
path := filepath.Join(dirName, testRepo+"-index.yaml")
|
||||||
|
if err := DownloadIndexFile(testRepo, ts.URL, path); err != nil {
|
||||||
|
t.Errorf("%#v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := os.Stat(path); err != nil {
|
||||||
|
t.Errorf("error finding created index file: %#v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
b, err := ioutil.ReadFile(path)
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("error reading index file: %#v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var i IndexFile
|
||||||
|
if err = yaml.Unmarshal(b, &i); err != nil {
|
||||||
|
t.Errorf("error unmarshaling index file: %#v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
numEntries := len(i.Entries)
|
||||||
|
if numEntries != 2 {
|
||||||
|
t.Errorf("Expected 2 entries in index file but got %v", numEntries)
|
||||||
|
}
|
||||||
|
os.Remove(path)
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in new issue