Skip to content

Commit

Permalink
Merge pull request #40 from iamtimmy/master
Browse files Browse the repository at this point in the history
  • Loading branch information
@tristanisham
tristanisham authored Jan 28, 2024
2 parents 3a7770e + 53e258d commit ba0ac31
Showing 1 changed file with 120 additions and 72 deletions.
192 changes: 120 additions & 72 deletions cli/install.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -204,8 +204,85 @@ type githubTaggedReleaseResponse struct {
}

type gitHubAsset struct {
Url string // download url for asset
Name string // contains platform information about binary
Url string // url for asset json object
Name string // contains platform information about binary
BrowserDownloadUrl string // download url
}

type zlsCIDownloadIndexResponse struct {
Latest string // most recent ZLS version
LatestTagged string // most recent tagged ZLS version
Versions map[string]zlsCIZLSVersion
}

type zlsCIZLSVersion struct {
ZLSVersion string
Targets []string
}

func getZLSDownloadUrl(version string, archDouble string) (string, error) {
if version == "master" {
resp, err := http.Get("https://zigtools-releases.nyc3.digitaloceanspaces.com/zls/index.json")
if err != nil {
return "", err
}
defer resp.Body.Close()

var releaseBuffer bytes.Buffer
_, err = releaseBuffer.ReadFrom(resp.Body)
if err != nil {
return "", err
}

var ciIndex zlsCIDownloadIndexResponse
if err := json.Unmarshal(releaseBuffer.Bytes(), &ciIndex); err != nil {
return "", err
}

exeName := "zls"
if strings.Contains(archDouble, "windows") {
exeName = "zls.exe"
}

format_url := "https://zigtools-releases.nyc3.digitaloceanspaces.com/zls/%v/%v/%v"
return fmt.Sprintf(format_url, ciIndex.Latest, archDouble, exeName), nil
} else {
url := fmt.Sprintf("https://api.github.com/repos/zigtools/zls/releases/tags/%v", version)

// get release information
resp, err := http.Get(url)
if err != nil {
return "", err
}
defer resp.Body.Close()

var releaseBuffer bytes.Buffer
_, err = releaseBuffer.ReadFrom(resp.Body)
if err != nil {
return "", err
}

// getting list of assets
var taggedReleaseResponse githubTaggedReleaseResponse
if err := json.Unmarshal(releaseBuffer.Bytes(), &taggedReleaseResponse); err != nil {
return "", err
}

// getting platform information
var downloadUrl string
for _, asset := range taggedReleaseResponse.Assets {
if strings.Contains(asset.Name, archDouble) {
downloadUrl = asset.BrowserDownloadUrl
break
}
}

if downloadUrl == "" {
return "", errors.New("invalid release URl")
}

return downloadUrl, nil
}
}

func (z *ZVM) InstallZls(version string) error {
Expand All @@ -214,6 +291,7 @@ func (z *ZVM) InstallZls(version string) error {
}

fmt.Println("Finding ZLS executable...")

// make sure dir exists
installDir := filepath.Join(z.zvmBaseDir, version)
err := os.MkdirAll(installDir, 0755)
Expand All @@ -222,118 +300,88 @@ func (z *ZVM) InstallZls(version string) error {
}

arch, osType := zigStyleSysInfo()
expectedArchOs := fmt.Sprintf("%v-%v", arch, osType)

filename := "zls"
if osType == "windows" {
filename += ".exe"
}

// master does not need unzipping, zpm just serves full binary
var releaseUrl string
shouldUnzip := true
if version == "master" {
releaseUrl = "https://api.github.com/repos/zigtools/zls/releases/latest"
} else {
// build url for tagged version
releaseUrl = fmt.Sprintf("https://api.github.com/repos/zigtools/zls/releases/tags/%v", version)
}

// get release information
resp, err := http.Get(releaseUrl)
if err != nil {
return err
shouldUnzip = false
}
defer resp.Body.Close()

var releaseBuffer bytes.Buffer
_, err = releaseBuffer.ReadFrom(resp.Body)
downloadUrl, err := getZLSDownloadUrl(version, expectedArchOs)
if err != nil {
return err
}

// some github releases use tar.gz, some tar.xz
expectedArchOs := fmt.Sprintf("%v-%v", arch, osType)
zipName := ""
var taggedReleaseResponse githubTaggedReleaseResponse
// getting list of assets
if err := json.Unmarshal(releaseBuffer.Bytes(), &taggedReleaseResponse); err != nil {
return err
}

// getting platform information
downloadUrl := ""
for _, asset := range taggedReleaseResponse.Assets {
if strings.Contains(asset.Name, expectedArchOs) {
downloadUrl = asset.Url
zipName = asset.Name
break
}
}

// couldn't find the file
if downloadUrl == "" {
return fmt.Errorf("could not find zls-%v", expectedArchOs)
}

if version == "master" {
downloadUrl = fmt.Sprintf("https://zig.pm/zls/downloads/%v/bin/zls", expectedArchOs)
}

client := &http.Client{}

// download tarball to temp
req, err := http.NewRequest("GET", downloadUrl, nil)
if err != nil {
return err
}
req.Header.Set("Accept", "application/octet-stream")
resp, err = client.Do(req)
request, err := http.NewRequest("GET", downloadUrl, nil)
if err != nil {
return err
}
defer resp.Body.Close()

// creating file to place contents
tempName := filename
if version != "master" {
_, pathEnding, _ := strings.Cut(zipName, ".")
tempName = "*." + pathEnding
}
request.Header.Set("User-Agent", "zvm "+meta.VERSION)

tempDir, err := os.CreateTemp(z.zvmBaseDir, tempName)
response, err := http.DefaultClient.Do(request)
if err != nil {
return err
}

defer tempDir.Close()
defer os.RemoveAll(tempDir.Name())
defer response.Body.Close()

// if resp.ContentLength == 0 {
// return fmt.Errorf("invalid ZLS content length (%d bytes)", resp.ContentLength)
// }

pbar := progressbar.DefaultBytes(
int64(resp.ContentLength),
int64(response.ContentLength),
"Downloading ZLS",
)

if _, err := io.Copy(io.MultiWriter(pbar, tempDir), resp.Body); err != nil {
return err
}
// untar to destination
fmt.Println("Extracting ZLS...")
versionPath := filepath.Join(z.zvmBaseDir, version)
if version == "master" {
if err := os.Rename(tempDir.Name(), filepath.Join(versionPath, filename)); err != nil {
binaryLocation := filepath.Join(versionPath, filename)

if shouldUnzip == false {
file, err := os.Create(binaryLocation)
if err != nil {
return err
}
defer file.Close()

if _, err := io.Copy(io.MultiWriter(pbar, file), response.Body); err != nil {
return err
}
} else {
var pathEnding string
if runtime.GOOS == "windows" {
pathEnding = "*.zip"
} else {
pathEnding = "*.tar.xz"
}

tempDir, err := os.CreateTemp(z.zvmBaseDir, pathEnding)
if err != nil {
return err
}

defer tempDir.Close()
defer os.RemoveAll(tempDir.Name())

if _, err := io.Copy(io.MultiWriter(pbar, tempDir), response.Body); err != nil {
return err
}

fmt.Println("Extracting ZLS...")
if err := ExtractBundle(tempDir.Name(), filepath.Join(z.zvmBaseDir, version)); err != nil {
log.Fatal(err)
}
if err := os.Rename(filepath.Join(versionPath, "bin", filename), filepath.Join(versionPath, filename)); err != nil {
return err
}
}

if err := os.Chmod(filepath.Join(versionPath, filename), 0755); err != nil {
return err
}
Expand Down

0 comments on commit ba0ac31

Please # to comment.