Add http client for installer #3

pkg/installer/client/http.go

@@ -0,0 +1,97 @@
+// Copyright © 2019 Ettore Di Giacinto <mudler@gentoo.org>
+//
+// This program is free software; you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation; either version 2 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License along
+// with this program; if not, see <http://www.gnu.org/licenses/>.
+
+package client
+
+import (
+	"io/ioutil"
+	"net/url"
+	"os"
+	"path"
+	"path/filepath"
+
+	. "github.com/mudler/luet/pkg/logger"
+
+	"github.com/mudler/luet/pkg/compiler"
+	"github.com/mudler/luet/pkg/helpers"
+
+	"github.com/cavaliercoder/grab"
+)
+
+type HttpClient struct {
+	RepoData RepoData
+}
+
+func NewHttpClient(r RepoData) *HttpClient {
+	return &HttpClient{RepoData: r}
+}
+
+func (c *HttpClient) DownloadArtifact(artifact compiler.Artifact) (compiler.Artifact, error) {
+	artifactName := path.Base(artifact.GetPath())
+	Info("Downloading artifact", artifactName, "from", c.RepoData.Uri)
+
+	temp, err := ioutil.TempDir(os.TempDir(), "tree")
+	if err != nil {
+		return nil, err
+	}
+
+	file, err := ioutil.TempFile(temp, "HttpClient")
+	if err != nil {
+		return nil, err
+	}
+
+	u, err := url.Parse(c.RepoData.Uri)
+	if err != nil {
+		return nil, err
+	}
+	u.Path = path.Join(u.Path, artifactName)
+
+	_, err = grab.Get(temp, u.String())
+	if err != nil {
+		return nil, err
+	}
+
+	err = helpers.CopyFile(filepath.Join(temp, artifactName), file.Name())
+
+	return compiler.NewPackageArtifact(file.Name()), nil
+}
+
+func (c *HttpClient) DownloadFile(name string) (string, error) {
+	temp, err := ioutil.TempDir(os.TempDir(), "tree")
+	if err != nil {
+		return "", err
+	}
+	file, err := ioutil.TempFile(os.TempDir(), "HttpClient")
+	if err != nil {
+		return "", err
+	}
+	//defer os.Remove(file.Name())
+	u, err := url.Parse(c.RepoData.Uri)
+	if err != nil {
+		return "", err
+	}
+	u.Path = path.Join(u.Path, name)
+
+	Info("Downloading", u.String())
+
+	_, err = grab.Get(temp, u.String())
+	if err != nil {
+		return "", err
+	}
+
+	err = helpers.CopyFile(filepath.Join(temp, name), file.Name())
+
+	return file.Name(), err
+}

pkg/installer/repository.go

@@ -192,6 +192,8 @@ func (r *LuetRepository) Client() Client {
 	switch r.GetType() {
 	case "local":
 		return client.NewLocalClient(client.RepoData{Uri: r.GetUri()})
+	case "http":
+		return client.NewHttpClient(client.RepoData{Uri: r.GetUri()})
 	}
 
 	return nil