mirror of https://github.com/containers/podman.git
125 lines
2.7 KiB
Go
125 lines
2.7 KiB
Go
//go:build amd64 || arm64
|
|
// +build amd64 arm64
|
|
|
|
package machine
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"path/filepath"
|
|
"regexp"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
const (
|
|
githubURL = "http://github.com/fedora-cloud/docker-brew-fedora/"
|
|
)
|
|
|
|
var fedoraxzRegex = regexp.MustCompile(`fedora[^\"]+xz`)
|
|
|
|
type FedoraDownload struct {
|
|
Download
|
|
}
|
|
|
|
func NewFedoraDownloader(vmType, vmName, releaseStream string) (DistributionDownload, error) {
|
|
imageName, downloadURL, size, err := getFedoraDownload(releaseStream)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
dataDir, err := GetDataDir(vmType)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
f := FedoraDownload{
|
|
Download: Download{
|
|
Arch: getFcosArch(),
|
|
Artifact: artifact,
|
|
Format: Format,
|
|
ImageName: imageName,
|
|
LocalPath: filepath.Join(dataDir, imageName),
|
|
URL: downloadURL,
|
|
VMName: vmName,
|
|
Size: size,
|
|
},
|
|
}
|
|
f.Download.LocalUncompressedFile = f.getLocalUncompressedName()
|
|
return f, nil
|
|
}
|
|
|
|
func (f FedoraDownload) Get() *Download {
|
|
return &f.Download
|
|
}
|
|
|
|
func (f FedoraDownload) HasUsableCache() (bool, error) {
|
|
info, err := os.Stat(f.LocalPath)
|
|
if err != nil {
|
|
if errors.Is(err, os.ErrNotExist) {
|
|
return false, nil
|
|
}
|
|
return false, err
|
|
}
|
|
return info.Size() == f.Size, nil
|
|
}
|
|
|
|
func truncRead(url string) ([]byte, error) {
|
|
resp, err := http.Get(url)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
defer func() {
|
|
if err := resp.Body.Close(); err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
}()
|
|
|
|
body, err := ioutil.ReadAll(io.LimitReader(resp.Body, 10*1024*1024))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, _ = io.Copy(io.Discard, resp.Body)
|
|
|
|
return body, nil
|
|
}
|
|
|
|
func getFedoraDownload(releaseStream string) (string, *url.URL, int64, error) {
|
|
dirURL := githubURL + "tree/" + releaseStream + "/" + getFcosArch() + "/"
|
|
body, err := truncRead(dirURL)
|
|
if err != nil {
|
|
return "", nil, -1, err
|
|
}
|
|
|
|
file := fedoraxzRegex.FindString(string(body))
|
|
if len(file) == 0 {
|
|
return "", nil, -1, fmt.Errorf("could not locate Fedora download at %s", dirURL)
|
|
}
|
|
|
|
rawURL := githubURL + "raw/" + releaseStream + "/" + getFcosArch() + "/"
|
|
newLocation := rawURL + file
|
|
downloadURL, err := url.Parse(newLocation)
|
|
if err != nil {
|
|
return "", nil, -1, errors.Wrapf(err, "invalid URL generated from discovered Fedora file: %s", newLocation)
|
|
}
|
|
|
|
resp, err := http.Head(newLocation)
|
|
if err != nil {
|
|
return "", nil, -1, errors.Wrapf(err, "head request failed: %s", newLocation)
|
|
}
|
|
_ = resp.Body.Close()
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
return "", nil, -1, fmt.Errorf("head request failed [%d] on download: %s", resp.StatusCode, newLocation)
|
|
}
|
|
|
|
return file, downloadURL, resp.ContentLength, nil
|
|
}
|