node/nvm-windows
1
0
Fork 0
mirror of https://github.com/coreybutler/nvm-windows.git synced 2026-01-14 07:03:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add cleanup when download is interrupted by user.

Browse Source
This commit is contained in:
ccccly
2017-03-05 16:44:37 +08:00
parent c75f08806b
commit 8889f643de
1 changed files with 25 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
src/nvm/web/web.go
View File

@@ -5,10 +5,12 @@ import(
"net/http"
"net/url"
"os"
"os/signal"
"io"
"io/ioutil"
"strings"
"syscall"
"crypto/tls"
"strings"
"strconv"
"../arch"
"../file"
@@ -50,7 +52,7 @@ func GetFullNpmUrl(path string) string{
return npmBaseAddress + path;
}
func Download(url string, target string) bool {
func Download(url string, target string, version string) bool {
output, err := os.Create(target)
if err != nil {
@@ -63,12 +65,28 @@ func Download(url string, target string) bool {
fmt.Println("Error while downloading", url, "-", err)
}
defer response.Body.Close()
c := make(chan os.Signal, 2)
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
go func() {
<-c
fmt.Println("Download interrupted.Rolling back...")
output.Close()
response.Body.Close()
var err error
if strings.Contains(target, "node") {
err = os.RemoveAll(os.Getenv("NVM_HOME") + "\\v" + version)
} else {
err = os.Remove(target)
}
if err != nil {
fmt.Println("Error while rolling back", err)
}
os.Exit(1)
}()
_, err = io.Copy(output, response.Body)
if err != nil {
fmt.Println("Error while downloading", url, "-", err)
}
if response.Status[0:3] != "200" {
fmt.Println("Download failed. Rolling Back.")
err := os.Remove(target)
@@ -111,9 +129,9 @@ func GetNodeJS(root string, v string, a string) bool {
} else {
fileName := root+"\\v"+v+"\\node"+a+".exe"
fmt.Printf("Downloading node.js version "+v+" ("+a+"-bit)... ")
fmt.Println("Downloading node.js version "+v+" ("+a+"-bit)... ")
if Download(url,fileName) {
if Download(url,fileName,v) {
fmt.Printf("Complete\n")
return true
} else {
@@ -142,7 +160,7 @@ func GetNpm(root string, v string) bool {
fileName := tempDir+"\\"+"npm-v"+v+".zip"
fmt.Printf("Downloading npm version "+v+"... ")
if Download(url,fileName) {
if Download(url,fileName,v) {
fmt.Printf("Complete\n")
return true
} else {