-
Notifications
You must be signed in to change notification settings - Fork 271
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #692 from kongfei605/feature_install
self upgrade
- v0.4.3
- v0.4.2
- v0.4.1
- v0.3.87
- v0.3.86
- v0.3.85
- v0.3.84
- v0.3.83
- v0.3.82
- v0.3.81
- v0.3.80
- v0.3.79
- v0.3.78
- v0.3.77
- v0.3.76
- v0.3.75
- v0.3.74
- v0.3.73
- v0.3.72
- v0.3.71
- v0.3.70
- v0.3.69
- v0.3.68
- v0.3.67
- v0.3.66
- v0.3.65
- v0.3.64
- v0.3.63
- v0.3.62
- v0.3.61
- v0.3.60
- v0.3.59
- v0.3.58
- v0.3.57
- v0.3.56
- v0.3.55
- v0.3.54
- v0.3.53
- v0.3.52
- v0.3.51
- v0.3.50
- v0.3.49
- v0.3.48
- v0.3.47
- v0.3.46
- v0.3.45
- v0.3.44
- v0.3.44-pre
- v0.3.43
- v0.3.42
- v0.3.41
- v0.3.40
- v0.3.39
- v0.3.38
- v0.3.37
- v0.3.36
Showing
6 changed files
with
233 additions
and
3 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,10 @@ | ||
package update | ||
|
||
import ( | ||
"fmt" | ||
) | ||
|
||
func Update(tar string) error { | ||
// binary | ||
return fmt.Errorf("linux support only") | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,9 @@ | ||
package update | ||
|
||
import ( | ||
"fmt" | ||
) | ||
|
||
func Update(tar string) error { | ||
return fmt.Errorf("linux support only") | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,155 @@ | ||
package update | ||
|
||
import ( | ||
"archive/tar" | ||
"bufio" | ||
"compress/gzip" | ||
"fmt" | ||
"io" | ||
"log" | ||
"net/http" | ||
"os" | ||
"path/filepath" | ||
"strings" | ||
) | ||
|
||
func download(file string) (string, error) { | ||
fname := filepath.Base(file) | ||
log.Println("downloading file:", file, "save to:", fname) | ||
res, err := http.Get(file) | ||
if err != nil { | ||
return fname, fmt.Errorf("cannot download file from %s", file) | ||
} | ||
defer res.Body.Close() | ||
|
||
if res.StatusCode != 200 { | ||
return fname, fmt.Errorf("download %s error response: %s", file, res.Status) | ||
} | ||
f, err := os.OpenFile(fname, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0755) | ||
if err != nil { | ||
return fname, err | ||
} | ||
defer f.Close() | ||
bufWriter := bufio.NewWriter(f) | ||
|
||
_, err = io.Copy(bufWriter, res.Body) | ||
if err != nil { | ||
return fname, err | ||
} | ||
bufWriter.Flush() | ||
return fname, nil | ||
} | ||
|
||
func Update(tar string) error { | ||
// download | ||
fname, err := download(tar) | ||
if err != nil { | ||
return err | ||
} | ||
nv, err := UnTar("./", fname) | ||
if err != nil { | ||
return err | ||
} | ||
|
||
// old version | ||
ov, err := os.Executable() | ||
if err != nil { | ||
return err | ||
} | ||
fm, err := os.Stat(ov) | ||
if err != nil { | ||
return err | ||
} | ||
fi, err := os.Stat(ov) | ||
if err != nil { | ||
return err | ||
} | ||
if fi.Mode().IsDir() { | ||
return fmt.Errorf("%s is directory", ov) | ||
} | ||
log.Printf("I! replace old version:%s with new version:%s", ov, "./"+nv) | ||
|
||
// replace | ||
err = os.Rename(nv, ov) | ||
if err != nil { | ||
return err | ||
} | ||
err = os.RemoveAll("./" + filepath.Dir(nv)) | ||
if err != nil { | ||
log.Println("E! clean dir:", "./"+filepath.Dir(nv), "error:", err) | ||
} else { | ||
log.Println("I! clean dir:", "./"+filepath.Dir(nv), "success") | ||
} | ||
err = os.Remove("./" + fname) | ||
if err != nil { | ||
log.Println("E! clean file:", "./"+fname, "error:", err) | ||
} else { | ||
log.Println("I! clean file:", "./"+fname, "success") | ||
} | ||
return os.Chmod(ov, fm.Mode().Perm()) | ||
} | ||
|
||
func UnTar(dst, src string) (target string, err error) { | ||
fr, err := os.Open(src) | ||
if err != nil { | ||
return | ||
} | ||
defer fr.Close() | ||
|
||
gr, err := gzip.NewReader(fr) | ||
if err != nil { | ||
return | ||
} | ||
defer gr.Close() | ||
|
||
tr := tar.NewReader(gr) | ||
|
||
for { | ||
hdr, err := tr.Next() | ||
|
||
switch { | ||
case err == io.EOF: | ||
return target, nil | ||
case err != nil: | ||
return target, err | ||
case hdr == nil: | ||
continue | ||
} | ||
|
||
dstFileDir := filepath.Join(dst, hdr.Name) | ||
|
||
switch hdr.Typeflag { | ||
case tar.TypeDir: | ||
if b := ExistDir(dstFileDir); !b { | ||
if err := os.MkdirAll(dstFileDir, 0775); err != nil { | ||
return target, err | ||
} | ||
} | ||
case tar.TypeReg: | ||
err := os.MkdirAll(filepath.Dir(dstFileDir), 0755) | ||
if err != nil { | ||
log.Printf("mdkir:%s, error:%s", filepath.Base(dstFileDir), err) | ||
return target, err | ||
} | ||
file, err := os.OpenFile(dstFileDir, os.O_CREATE|os.O_RDWR, os.FileMode(hdr.Mode)) | ||
if err != nil { | ||
return target, err | ||
} | ||
if strings.HasSuffix(dstFileDir, "categraf") { | ||
target = dstFileDir | ||
} | ||
_, err = io.Copy(file, tr) | ||
if err != nil { | ||
return target, err | ||
} | ||
file.Close() | ||
} | ||
} | ||
|
||
return target, nil | ||
} | ||
|
||
func ExistDir(dirname string) bool { | ||
fi, err := os.Stat(dirname) | ||
return (err == nil || os.IsExist(err)) && fi.IsDir() | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,9 @@ | ||
package update | ||
|
||
import ( | ||
"fmt" | ||
) | ||
|
||
func Update(tar string) error { | ||
return fmt.Errorf("linux support only") | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters