feat: initialize

internal/jar/jar.go

@@ -0,0 +1,70 @@
+package jar
+
+import (
+	"errors"
+	"io"
+	"net/http"
+	"os"
+	"strings"
+
+	"gitea.com/logicamp/lc"
+	api "github.com/logi-camp/go-flink-client"
+	gonanoid "github.com/matoous/go-nanoid/v2"
+	"go.uber.org/zap"
+)
+
+type JarFile struct {
+	uri      string
+	filePath string
+}
+
+func NewJarFile(URI string) (*JarFile, error) {
+	jarFile := &JarFile{
+		uri: URI,
+	}
+	err := jarFile.Download()
+	if err != nil {
+		return nil, err
+	}
+	return jarFile, nil
+}
+
+func (JarFile JarFile) Upload(flinkClient *api.Client) (fileName string, err error) {
+
+	resp, err := flinkClient.UploadJar(JarFile.filePath)
+	if err != nil {
+		lc.Logger.Error("[main] error uploading jar", zap.Error(err))
+	}
+	filePathParts := strings.Split(resp.FileName, "/")
+	fileName = filePathParts[len(filePathParts)-1]
+	if resp.Status != "success" {
+		err = errors.New("jar upload was not success")
+	}
+	return
+}
+
+func (jarFile *JarFile) Download() error {
+	fileName, _ := gonanoid.New()
+	jarFile.filePath = fileName + ".jar"
+	out, err := os.Create(jarFile.filePath)
+	if err != nil {
+		return err
+	}
+
+	defer out.Close()
+	resp, err := http.Get(jarFile.uri)
+	if err != nil {
+		return err
+	}
+
+	defer resp.Body.Close()
+	_, err = io.Copy(out, resp.Body)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (jarFile JarFile) Delete() error {
+	return os.Remove(jarFile.filePath)
+}