var _ = Describe("Filter", func() {
Describe("Download Links", func() {
It("returns the download links", func() {
productFiles := pivnet.ProductFiles{[]pivnet.ProductFile{
{ID: 3,
AWSObjectKey: "product_files/banana/file-name-1.zip",
Links: &pivnet.Links{Download: map[string]string{"href": "/products/banana/releases/666/product_files/6/download"}},
},
{ID: 4,
AWSObjectKey: "product_files/banana/file-name-2.zip",
Links: &pivnet.Links{Download: map[string]string{"href": "/products/banana/releases/666/product_files/8/download"}},
},
}}
links := filter.DownloadLinks(productFiles)
Expect(links).To(HaveLen(2))
Expect(links).To(Equal(map[string]string{
"file-name-1.zip": "/products/banana/releases/666/product_files/6/download",
"file-name-2.zip": "/products/banana/releases/666/product_files/8/download",
}))
})
})
Describe("Download Links by Glob", func() {
It("returns the download links that match the glob filters", func() {
downloadLinks := map[string]string{
"android-file.zip": "/products/banana/releases/666/product_files/6/download",
"ios-file.zip": "/products/banana/releases/666/product_files/8/download",
"random-file.zip": "/products/banana/releases/666/product_files/8/download",
}
func main() {
if version == "" {
version = "dev"
}
var input concourse.InRequest
if len(os.Args) < 2 {
log.Fatalln(fmt.Sprintf(
"not enough args - usage: %s <sources directory>", os.Args[0]))
}
downloadDir := os.Args[1]
err := json.NewDecoder(os.Stdin).Decode(&input)
if err != nil {
log.Fatalln(err)
}
logFile, err := ioutil.TempFile("", "pivnet-resource-in.log")
if err != nil {
log.Fatalln(err)
}
fmt.Fprintf(logFile, "PivNet Resource version: %s\n", version)
fmt.Fprintf(os.Stderr, "logging to %s\n", logFile.Name())
sanitized := concourse.SanitizedSource(input.Source)
sanitizer := sanitizer.NewSanitizer(sanitized, logFile)
l := logger.NewLogger(sanitizer)
token := input.Source.APIToken
mustBeNonEmpty(token, "api_token")
l.Debugf("Received input: %+v\n", input)
clientConfig := pivnet.NewClientConfig{
URL: pivnet.URL,
Token: input.Source.APIToken,
UserAgent: fmt.Sprintf("pivnet-resource/%s", version),
}
client := pivnet.NewClient(
clientConfig,
l,
)
productVersion := input.Version.ProductVersion
productSlug := input.Source.ProductSlug
l.Debugf(
"Getting release: {product_slug: %s, product_version: %s}\n",
productSlug,
productVersion,
)
release, err := client.GetRelease(productSlug, productVersion)
if err != nil {
log.Fatalf("Failed to get Release: %s\n", err.Error())
}
l.Debugf(
"Accepting EULA: {product_slug: %s, release_id: %d}\n",
productSlug,
release.ID,
)
err = client.AcceptEULA(productSlug, release.ID)
if err != nil {
log.Fatalf("EULA acceptance failed for the release: %s\n", err.Error())
}
l.Debugf(
"Getting product files: {release_id: %d}\n",
release.ID,
)
productFiles, err := client.GetProductFiles(release)
if err != nil {
log.Fatalf("Failed to get Product Files: %s\n", err.Error())
}
l.Debugf(
"Getting download links: {product_files: %+v}\n",
productFiles,
)
downloadLinks := filter.DownloadLinks(productFiles)
if len(input.Params.Globs) > 0 {
l.Debugf(
"Filtering download links with globs: {globs: %+v}\n",
input.Params.Globs,
)
var err error
downloadLinks, err = filter.DownloadLinksByGlob(downloadLinks, input.Params.Globs)
if err != nil {
log.Fatalf("Failed to filter Product Files: %s\n", err.Error())
}
l.Debugf(
"Downloading files: {download_links: %+v, download_dir: %s}\n",
downloadLinks,
downloadDir,
)
err = downloader.Download(downloadDir, downloadLinks, token)
if err != nil {
log.Fatalf("Failed to Download Files: %s\n", err.Error())
}
}
versionFilepath := filepath.Join(downloadDir, "version")
l.Debugf(
"Writing version to file: {version: %s, version_filepath: %s}\n",
version,
versionFilepath,
)
err = ioutil.WriteFile(versionFilepath, []byte(productVersion), os.ModePerm)
if err != nil {
log.Fatalln(err)
}
out := concourse.InResponse{
Version: concourse.Version{
ProductVersion: productVersion,
},
Metadata: []concourse.Metadata{
{Name: "release_type", Value: release.ReleaseType},
{Name: "release_date", Value: release.ReleaseDate},
{Name: "description", Value: release.Description},
{Name: "eula_slug", Value: release.Eula.Slug},
},
}
l.Debugf("Returning output: %+v\n", out)
err = json.NewEncoder(os.Stdout).Encode(out)
if err != nil {
log.Fatalln(err)
}
}