lagi di service :

upload, extract, tinggal verify

internal/artifact/repo/artifact.go

@@ -15,5 +15,6 @@ type ArtifactList struct {
 // Repo interface to operate with artifact
 type Repo interface {
 	GetArtifactList(pageNum int64, rows int64) (ArtifactList, error)
-	PutTarballToFile(tarball *string) error
+	PutTarballToFile(tarball *string, taskUUID string) error
+	ExtractSubmittedTarball(taskUUID string) error
 }

internal/artifact/repo/file_impl.go

@@ -1,10 +1,15 @@
 package repo
 
 import (
+	"encoding/base64"
+	"io/ioutil"
+	"os"
 	"path/filepath"
 	"strings"
 
 	model "github.com/blankon/irgsh-go/internal/artifact/model"
+
+	tar "github.com/blankon/irgsh-go/pkg/tar"
 )
 
 // FileRepo interface with file system based artifact information
@@ -38,7 +43,36 @@ func (A *FileRepo) GetArtifactList(pageNum int64, rows int64) (artifactsList Art
 }
 
 // PutTarballToFile not it's just general function to write string of base64 to file
-func (A *FileRepo) PutTarballToFile(tarball *string) (err error) {
+func (A *FileRepo) PutTarballToFile(tarball *string, taskUUID string) (err error) {
+
+	// create artifact directory
+	submissionDir := A.generateSubmissionPath(taskUUID)
+	err = os.MkdirAll(submissionDir, os.ModeDir)
+	if err != nil {
+		return
+	}
+
+	// write the tarball
+	filePath := submissionDir + "/" + taskUUID + ".tar.gz"
+	buff, err := base64.StdEncoding.DecodeString(*tarball)
+	if err != nil {
+		return
+	}
+	err = ioutil.WriteFile(filePath, buff, 07440)
+	if err != nil {
+		return
+	}
+
+	return
+}
+
+// ExtractSubmittedTarball ...
+func (A *FileRepo) ExtractSubmittedTarball(taskUUID string) (err error) {
+	submissionDir := A.generateSubmissionPath(taskUUID)
+	filePath := submissionDir + "/" + taskUUID + ".tar.gz"
+
+	err = tar.ExtractTarball(filePath, submissionDir)
+
 	return
 }
 
@@ -49,3 +83,8 @@ func getArtifactFilename(filePath string) (fileName string) {
 	}
 	return
 }
+
+func (A *FileRepo) generateSubmissionPath(taskUUID string) (path string) {
+	path = A.Workdir + "/submissions/" + taskUUID
+	return
+}

internal/artifact/repo/file_impl_test.go

@@ -17,7 +17,6 @@ func TestMain(m *testing.M) {
 	file002.Close()
 
 	exitVal := m.Run()
-	// time.Sleep(2 * time.Second)
 
 	// clean up test directory
 	os.RemoveAll("./artifacts")
@@ -106,3 +105,39 @@ func TestFileRepo_GetArtifactList(t *testing.T) {
 		})
 	}
 }
+
+func TestFileRepo_generateSubmissionPath(t *testing.T) {
+	type fields struct {
+		Workdir string
+	}
+	type args struct {
+		taskUUID string
+	}
+	tests := []struct {
+		name     string
+		fields   fields
+		args     args
+		wantPath string
+	}{
+		{
+			name:     "empty",
+			wantPath: "/submissions/",
+		},
+		{
+			name:     "complete",
+			fields:   fields{Workdir: "/home/workingdir"},
+			args:     args{taskUUID: "randomnumberhere"},
+			wantPath: "/home/workingdir/submissions/randomnumberhere",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			A := &FileRepo{
+				Workdir: tt.fields.Workdir,
+			}
+			if gotPath := A.generateSubmissionPath(tt.args.taskUUID); gotPath != tt.wantPath {
+				t.Errorf("FileRepo.generateSubmissionPath() = %v, want %v", gotPath, tt.wantPath)
+			}
+		})
+	}
+}

internal/artifact/service/artifact.go

@@ -1,6 +1,7 @@
 package service
 
 import (
+	"errors"
 	"time"
 
 	artifactModel "github.com/blankon/irgsh-go/internal/artifact/model"
@@ -63,7 +64,17 @@ func (A *ArtifactService) SubmitPackage(tarball string) (job SubmissionJob, err
 
 	submittedJob.TaskUUID = generateSubmissionUUID(submittedJob.Timestamp)
 
-	// err = A.repo.(tarball)
+	err = A.repo.PutTarballToFile(&tarball, submittedJob.TaskUUID)
+	if err != nil {
+		return job, errors.New("Can't store tarball " + err.Error())
+	}
+
+	err = A.repo.ExtractSubmittedTarball(submittedJob.TaskUUID)
+	if err != nil {
+		return job, errors.New("Can't extract tarball " + err.Error())
+	}
+
+	// verify the package
 
 	return
 }