feature/add-actions (#9)

Co-authored-by: kjuulh <contact@kjuulh.io>
Reviewed-on: https://git.front.kjuulh.io/kjuulh/kraken/pulls/9

internal/actions/action.go

@@ -2,15 +2,38 @@ package actions
 
 import (
 	"context"
+	"errors"
 
+	"git.front.kjuulh.io/kjuulh/kraken/internal/actions/builders"
 	"git.front.kjuulh.io/kjuulh/kraken/internal/schema"
 	"git.front.kjuulh.io/kjuulh/kraken/internal/services/storage"
+	"go.uber.org/zap"
 )
 
 type Action struct {
-	Schema *schema.KrakenSchema
+	Schema     *schema.KrakenSchema
+	SchemaPath string
 }
 
 func (a *Action) Execute(ctx context.Context, area *storage.Area) error {
+	for _, action := range a.Schema.Actions {
+		switch action.Type {
+		case "go":
+			exe, err := builders.NewGo(zap.L()).Build(ctx, a.SchemaPath, action.Entry)
+			if err != nil {
+				return err
+			}
+			err = exe(ctx, area.Path)
+			if err != nil {
+				return err
+			}
+
+			zap.L().Debug("Execution done")
+
+		default:
+			return errors.New("could not determine action type")
+		}
+	}
+
 	return nil
 }

internal/actions/action_creator.go

@@ -47,18 +47,13 @@ func (ac *ActionCreator) Prepare(ctx context.Context, ops *ActionCreatorOps) (*A
 		return nil, err
 	}
 
-	cloneCtx, _ := context.WithTimeout(ctx, time.Second*5)
-	repo, err := ac.git.Clone(cloneCtx, area, ops.RepositoryUrl)
+	cloneCtx, _ := context.WithTimeout(ctx, time.Second*10)
+	_, err = ac.git.CloneBranch(cloneCtx, area, ops.RepositoryUrl, ops.Branch)
 	if err != nil {
 		ac.logger.Error("could not clone repo", zap.Error(err))
 		return nil, err
 	}
 
-	err = ac.git.Checkout(ctx, repo, ops.Branch)
-	if err != nil {
-		return nil, err
-	}
-
 	executorUrl := path.Join(area.Path, ops.Path)
 	if _, err = os.Stat(executorUrl); os.IsNotExist(err) {
 		return nil, fmt.Errorf("path is invalid: %s", ops.Path)
@@ -74,8 +69,10 @@ func (ac *ActionCreator) Prepare(ctx context.Context, ops *ActionCreatorOps) (*A
 		return nil, err
 	}
 
+	ac.logger.Debug("Action creator done")
 	return &Action{
-		Schema: krakenSchema,
+		Schema:     krakenSchema,
+		SchemaPath: executorUrl,
 	}, nil
 }
 

internal/actions/builders/go.go

@@ -0,0 +1,46 @@
+package builders
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"os"
+	"os/exec"
+
+	"go.uber.org/zap"
+)
+
+type Go struct {
+	logger *zap.Logger
+}
+
+func NewGo(logger *zap.Logger) *Go {
+	return &Go{logger: logger}
+}
+
+type GoExecutable func(ctx context.Context, victimPath string) error
+
+func (g *Go) Build(ctx context.Context, modulePath, entryPath string) (GoExecutable, error) {
+	g.logger.Debug("Building go binary", zap.String("actiondir", modulePath), zap.String("entry", entryPath))
+
+	if _, err := os.Stat(fmt.Sprintf("%s/%s", modulePath, entryPath)); os.IsNotExist(err) {
+		return nil, errors.New("could not find entry")
+	}
+
+	err := exec.CommandContext(
+		ctx,
+		"/bin/bash",
+		"-c",
+		fmt.Sprintf("(cd %s; go build -o main %s)", modulePath, entryPath),
+	).Run()
+	if err != nil {
+		return nil, err
+	}
+
+	g.logger.Debug("Go binary built!")
+
+	return func(ctx context.Context, victimPath string) error {
+		g.logger.Debug("Executing script", zap.String("victim", victimPath))
+		return exec.CommandContext(ctx, "/bin/bash", "-c", fmt.Sprintf("(cd %s; %s/main)", victimPath, modulePath)).Run()
+	}, nil
+}