ghttp.CombineHandlers(
ghttp.VerifyRequest("POST", "/the-first-post", ""),
ghttp.VerifyHeaderKV("Authorization", fmt.Sprintf("Token %s", token)),
ghttp.RespondWith(http.StatusFound, nil, header),
),
ghttp.CombineHandlers(
ghttp.VerifyRequest("GET", "/some-redirect-link"),
ghttp.RespondWith(http.StatusOK, make([]byte, 10, 14)),
),
)
fileNames := map[string]string{
"the-first-post": apiAddress + "/the-first-post",
}
err := downloader.Download(dir, fileNames, token)
Expect(err).NotTo(HaveOccurred())
})
It("Downloads the files into the directory provided", func() {
fileNames := map[string]string{
"file-0": apiAddress + "/post-0",
"file-1": apiAddress + "/post-1",
"file-2": apiAddress + "/post-2",
}
for i := 0; i < len(fileNames); i++ {
url := fmt.Sprintf("/post-%d", i)
server.RouteToHandler("POST", url, ghttp.CombineHandlers(
ghttp.VerifyRequest("POST", url, ""),
ghttp.RespondWith(http.StatusOK, fmt.Sprintf("contents-%d", i)),
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)
}
}
