DataHearth
/
doggo-fetcher
Archived
1
0
Fork 0
Code Issues Pull Requests Releases 3 Activity

WIP

This commit is contained in:
DataHearth 2022-06-18 22:45:29 +02:00
parent 0afd0b88f1
commit 25cafd23e3
No known key found for this signature in database
GPG Key ID: E88FD356ACC5F3C4
6 changed files with 290 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

73
cmd/main.go
View File

@ -1,9 +1,12 @@
package main
import (
"crypto/sha256"
"errors"
"fmt"
"io/fs"
"os"
"path/filepath"
"github.com/datahearth/doggo-fetcher/pkg"
"github.com/urfave/cli/v2"
@ -11,21 +14,23 @@ import (
var app = &cli.App{
Name: "dg",
Usage: "I bring you your latest Golang version with ease and efficiency (like a stick) !",
Description: `Doggo-fetcher is a utility tool that manage for you your Golang version.
You can select a specific go version or even set version for directories.`,
Usage: "I bring you your latest GoLang release with ease and efficiency (like a stick) !",
Description: `Doggo-fetcher is a utility tool that manage for you your installed GoLang releases.
You can select a specific GoLang release or even set a specific one for directories.`,
EnableBashCompletion: true,
Authors: []*cli.Author{{
Name: "Antoine <DataHearth> Langlois",
Email: "antoine.l@antoine-langlois.net",
}},
Authors: []*cli.Author{
{
Name: "Antoine <DataHearth> Langlois",
Email: "antoine.l@antoine-langlois.net",
},
},
Suggest: true,
Version: "0.1.0",
Commands: []*cli.Command{
{
Name: "install",
Usage: "Download a given release",
Description: "Download a given release and set it as first release to be use",
Usage: "Download a release",
Description: "Download a release. If the given release is already stored, nothing will happen.",
Flags: []cli.Flag{
&cli.BoolFlag{
Name: "keep-release",
@ -34,7 +39,7 @@ You can select a specific go version or even set version for directories.`,
&cli.BoolFlag{
Name: "latest",
Aliases: []string{"lts"},
Usage: "Download and install the latest release",
Usage: "Use latest release",
},
&cli.BoolFlag{
Name: "rc",
@ -51,7 +56,7 @@ You can select a specific go version or even set version for directories.`,
if !ctx.Bool("latest") {
return errors.New("a release is required if \"--latest|--lts\" is not passed")
}
release = "lts"
release = pkg.LTS
} else {
release = ctx.Args().First()
}
@ -62,8 +67,43 @@ You can select a specific go version or even set version for directories.`,
return err
}
fmt.Printf("release: %v\n", release)
dlRelease := pkg.NewDownload(release, ctx.Bool("keep-release"))
releasePath, err := dlRelease.DownloadRelease()
if err != nil {
return err
}
extractedRelease, err := pkg.ExtractRelease(releasePath, release)
if err != nil {
return err
}
totalSum := sha256.New()
err = filepath.Walk(extractedRelease, func(path string, info fs.FileInfo, err error) error {
if err != nil {
return err
}
if !info.Mode().IsRegular() || info.IsDir() {
return nil
}
b, err := os.ReadFile(path)
if err != nil {
return err
}
})
// todo: install release
return nil
},
},
{
Name: "use",
Usage: "Set a specific golang version",
Description: `Use a specific golang version as primary golang binary for the user.
If the version is not already downloaded, it'll downloaded and installed automatically.`,
Action: func(ctx *cli.Context) error {
fmt.Println("to be implemented!")
return nil
},
},
@ -74,6 +114,15 @@ You can select a specific go version or even set version for directories.`,
},
}
// todo: use (download if not present, install and set in path)
// todo: uninstall
// todo: ls (list installed-releases)
// todo: ls-remote (list remote releases)
// todo: exec (without setting release=
// todo: alias
// todo: run-bg (check each 10mins)
// todo: auto-use (automatically switch when changing directory)
func main() {
if err := app.Run(os.Args); err != nil {
fmt.Println(err)

5
go.mod
View File

@ -5,10 +5,15 @@ go 1.18
require github.com/urfave/cli/v2 v2.8.1
require (
github.com/briandowns/spinner v1.18.1 // indirect
github.com/cpuguy83/go-md2man/v2 v2.0.1 // indirect
github.com/fatih/color v1.7.0 // indirect
github.com/google/go-github/v45 v45.1.0 // indirect
github.com/google/go-querystring v1.1.0 // indirect
github.com/mattn/go-colorable v0.1.2 // indirect
github.com/mattn/go-isatty v0.0.8 // indirect
github.com/russross/blackfriday/v2 v2.1.0 // indirect
github.com/xrash/smetrics v0.0.0-20201216005158-039620a65673 // indirect
golang.org/x/crypto v0.0.0-20210817164053-32db794688a5 // indirect
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1 // indirect
)

11
go.sum
View File

@ -1,10 +1,18 @@
github.com/briandowns/spinner v1.18.1 h1:yhQmQtM1zsqFsouh09Bk/jCjd50pC3EOGsh28gLVvwY=
github.com/briandowns/spinner v1.18.1/go.mod h1:mQak9GHqbspjC/5iUx3qMlIho8xBS/ppAL/hX5SmPJU=
github.com/cpuguy83/go-md2man/v2 v2.0.1 h1:r/myEWzV9lfsM1tFLgDyu0atFtJ1fXn261LKYj/3DxU=
github.com/cpuguy83/go-md2man/v2 v2.0.1/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=
github.com/fatih/color v1.7.0 h1:DkWD4oS2D8LGGgTQ6IvwJJXSL5Vp2ffcQg58nFV38Ys=
github.com/fatih/color v1.7.0/go.mod h1:Zm6kSWBoL9eyXnKyktHP6abPY2pDugNf5KwzbycvMj4=
github.com/google/go-cmp v0.5.2/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
github.com/google/go-github/v45 v45.1.0 h1:SbUjHMRiCe9cHfu6Me4idWxLQEV8ZW9DLPz69zopyWo=
github.com/google/go-github/v45 v45.1.0/go.mod h1:FObaZJEDSTa/WGCzZ2Z3eoCDXWJKMenWWTrd8jrta28=
github.com/google/go-querystring v1.1.0 h1:AnCroh3fv4ZBgVIf1Iwtovgjaw/GiKJo8M8yD/fhyJ8=
github.com/google/go-querystring v1.1.0/go.mod h1:Kcdr2DB4koayq7X8pmAG4sNG59So17icRSOU623lUBU=
github.com/mattn/go-colorable v0.1.2 h1:/bC9yWikZXAL9uJdulbSfyVNIR3n3trXl+v8+1sx8mU=
github.com/mattn/go-colorable v0.1.2/go.mod h1:U0ppj6V5qS13XJ6of8GYAs25YV2eR4EVcfRqFIhoBtE=
github.com/mattn/go-isatty v0.0.8 h1:HLtExJ+uU2HOZ+wI0Tt5DtUDrx8yhUqDcp7fYERX4CE=
github.com/mattn/go-isatty v0.0.8/go.mod h1:Iq45c/XA43vh69/j3iqttzPXn0bhXyGjM0Hdxcsrc5s=
github.com/russross/blackfriday/v2 v2.1.0 h1:JIOH55/0cWyOuilr9/qlrm0BSXldqnqwMsf35Ld67mk=
github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
github.com/urfave/cli/v2 v2.8.1 h1:CGuYNZF9IKZY/rfBe3lJpccSoIY1ytfvmgQT90cNOl4=
@ -13,4 +21,7 @@ github.com/xrash/smetrics v0.0.0-20201216005158-039620a65673 h1:bAn7/zixMGCfxrRT
github.com/xrash/smetrics v0.0.0-20201216005158-039620a65673/go.mod h1:N3UwUGtsrSj3ccvlPHLoLsHnpR27oXr4ZE984MbSER8=
golang.org/x/crypto v0.0.0-20210817164053-32db794688a5 h1:HWj/xjIHfjYU5nVXpTM0s39J9CbLn7Cc5a7IC5rwsMQ=
golang.org/x/crypto v0.0.0-20210817164053-32db794688a5/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
golang.org/x/sys v0.0.0-20190222072716-a9d3bda3a223/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1 h1:SrN+KX8Art/Sf4HNj6Zcz06G7VEz+7w9tdXTPOZ7+l4=
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=

140
pkg/download.go
View File

@ -1,4 +1,142 @@
package pkg
const GO_DL_SERVER = "https://dl.google.com/go"
import (
"errors"
"fmt"
"io"
"net/http"
"os"
"runtime"
"strconv"
"time"
"github.com/briandowns/spinner"
)
const GO_DL_SERVER = "https://dl.google.com/go"
const DGF_LOCAL_FOLDER = ".local/doggo-fetcher"
var (
ErrNonSuccessStatusCode = errors.New("golang download server return a non success status code")
)
type DownloadAction interface {
DownloadRelease() (string, error)
}
type Download struct {
release string
keep bool
client *http.Client
}
func NewDownload(release string, keep bool) DownloadAction {
return Download{
release: release,
keep: keep,
client: http.DefaultClient,
}
}
func (d Download) DownloadRelease() (string, error) {
var dstDir string
osName := runtime.GOOS
if d.keep {
homeDir, err := os.UserHomeDir()
if err != nil {
return "", err
}
dstDir = fmt.Sprintf("%s/%s", homeDir, DGF_LOCAL_FOLDER)
if err := os.MkdirAll(dstDir, 0755); err != nil {
return "", err
}
} else {
dstDir = os.TempDir()
}
releasePath := fmt.Sprintf("%s/%s.%s-amd64.tar.gz", dstDir, d.release, osName)
if d.keep {
fi, err := os.Stat(releasePath)
if err != nil && !os.IsNotExist(err) {
fmt.Printf("error type: %T\n", err)
return "", err
} else if os.IsNotExist(err) {
return releasePath, d.writeRelease(releasePath)
}
if fi.IsDir() {
if err := os.RemoveAll(releasePath); err != nil {
return "", err
}
return releasePath, d.writeRelease(releasePath)
}
rsp, err := d.client.Head(fmt.Sprintf("%s/%s.%s-amd64.tar.gz", GO_DL_SERVER, d.release, osName))
if err != nil {
return "", err
}
if rsp.StatusCode != 200 {
fmt.Printf("rsp.Request.URL: %v\n", rsp.Request.URL)
fmt.Fprintf(os.Stderr, "%s: %d\n", ErrNonSuccessStatusCode.Error(), rsp.StatusCode)
return "", ErrNonSuccessStatusCode
}
if rsp.Header.Get("content-length") != "" {
contentLength, err := strconv.ParseInt(rsp.Header.Get("content-length"), 10, 64)
if err != nil {
return "", err
}
if contentLength == 0 || contentLength != fi.Size() {
return releasePath, d.writeRelease(releasePath)
}
fmt.Printf("release \"%s\" already downloaded and has a valid size. Skipping...\n", d.release)
return releasePath, nil
}
return releasePath, d.writeRelease(releasePath)
}
return releasePath, d.writeRelease(releasePath)
}
func (d Download) writeRelease(releasePath string) error {
f, err := os.Create(releasePath)
if err != nil {
return err
}
defer f.Close()
return d.downloadFile(f)
}
func (d Download) downloadFile(f *os.File) error {
osName := runtime.GOOS
s := spinner.New(spinner.CharSets[35], 100*time.Millisecond)
s.FinalMSG = "Release downloaded!\n"
s.Suffix = " Downloading release..."
s.Start()
defer s.Stop()
rsp, err := d.client.Get(fmt.Sprintf("%s/%s.%s-amd64.tar.gz", GO_DL_SERVER, d.release, osName))
if err != nil {
return err
}
defer rsp.Body.Close()
if rsp.StatusCode != 200 {
fmt.Printf("rsp.Request.URL: %v\n", rsp.Request.URL)
fmt.Fprintf(os.Stderr, "%s: %d\n", ErrNonSuccessStatusCode.Error(), rsp.StatusCode)
return ErrNonSuccessStatusCode
}
if _, err := io.Copy(f, rsp.Body); err != nil {
return err
}
return nil
}

12
pkg/tags.go
View File

@ -11,15 +11,17 @@ import (
"github.com/google/go-github/v45/github"
)
const LTS = "lts"
var (
ErrBadResponse = errors.New("github API responde with a non success code")
ErrEmptyTags = errors.New("no tags found")
)
type TagsAction interface {
CheckReleaseExists(beta, rc bool) (string, error)
GetRelease(beta, rc bool) (string, error)
getTagsRef() ([]*github.Reference, error)
getLatestRelease(beta, rc bool) (string, error)
getLatestTag(beta, rc bool) (string, error)
}
type Tags struct {
@ -28,7 +30,7 @@ type Tags struct {
ctx context.Context
}
func NewTags(release string, ctx context.Context) Tags {
func NewTags(release string, ctx context.Context) TagsAction {
return Tags{
release: release,
client: *github.NewClient(nil),
@ -100,7 +102,7 @@ func (t Tags) getTagsRef() ([]*github.Reference, error) {
return nil, err
}
if response.StatusCode != http.StatusOK {
fmt.Fprintf(os.Stderr, "non success github status code %d\n", response.StatusCode)
fmt.Fprintf(os.Stderr, "%s: %d\n", ErrBadResponse.Error(), response.StatusCode)
return nil, ErrBadResponse
}
if len(refs) == 0 {
@ -123,9 +125,7 @@ func (t Tags) getLatestTag(beta, rc bool) (string, error) {
return "", err
}
fmt.Printf("len(tags): %v\n", len(refs))
for i := len(refs) - 1; i >= 0; i-- {
fmt.Printf("[%d] tags[i].Name: %v\n", i, *refs[i].Ref)
if (!beta && strings.Contains(*refs[i].Ref, "beta")) || (!rc && strings.Contains(*refs[i].Ref, "rc")) {
continue
}

68
pkg/tar.go Normal file
View File

@ -0,0 +1,68 @@
package pkg
import (
"archive/tar"
"compress/gzip"
"fmt"
"io"
"os"
"path/filepath"
)
func ExtractRelease(releasePath, releaseName string) (string, error) {
f, err := os.Open(releasePath)
if err != nil {
return "", err
}
defer f.Close()
gzipReader, err := gzip.NewReader(f)
if err != nil {
return "", err
}
defer gzipReader.Close()
dst, _ := filepath.Split(releasePath)
tarReader := tar.NewReader(gzipReader)
dstFolder := ""
for {
h, err := tarReader.Next()
if err != nil {
if err == io.EOF {
extractedRelease := fmt.Sprintf("%s/%s", filepath.Dir(dstFolder), releaseName)
if err := os.Rename(dstFolder, extractedRelease); err != nil {
return "", err
}
return extractedRelease, os.Remove(releasePath)
}
return "", err
}
target := filepath.Join(dst, h.Name)
if dstFolder == "" {
dstFolder = target
}
switch h.Typeflag {
case tar.TypeDir:
if err = os.MkdirAll(target, h.FileInfo().Mode()); err != nil {
return "", err
}
case tar.TypeReg:
file, err := os.OpenFile(target, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, h.FileInfo().Mode())
if err != nil {
return "", err
}
defer file.Close()
if _, err = io.Copy(file, tarReader); err != nil {
return "", err
}
}
}
}