feature/add-actions #9

Merged
kjuulh merged 6 commits from feature/add-actions into v0.1 2022-09-14 23:40:59 +02:00
6 changed files with 75 additions and 6 deletions
Showing only changes of commit 896b28188b - Show all commits

View File

@ -3,7 +3,10 @@ package main
import "github.com/bitfield/script" import "github.com/bitfield/script"
func main() { func main() {
script. _, err := script.
Echo("# Readme"). Echo("# Readme").
WriteFile("README.md") WriteFile("README.md")
if err != nil {
panic(err)
}
} }

View File

@ -11,7 +11,7 @@ import (
func CreateKrakenProcessCmd() *cobra.Command { func CreateKrakenProcessCmd() *cobra.Command {
cmd := &cobra.Command{ cmd := &cobra.Command{
Use: "process", Use: "process",
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, _ []string) {
client := http.Client{} client := http.Client{}
var buf bytes.Buffer var buf bytes.Buffer
@ -22,7 +22,7 @@ func CreateKrakenProcessCmd() *cobra.Command {
Path string `json:"path"` Path string `json:"path"`
}{ }{
Repository: "git@git.front.kjuulh.io:kjuulh/kraken.git", Repository: "git@git.front.kjuulh.io:kjuulh/kraken.git",
Branch: "v0.1", Branch: "feature/add-actions",
Path: "_examples/actions/write_a_readme/", Path: "_examples/actions/write_a_readme/",
}) })
if err != nil { if err != nil {

View File

@ -15,6 +15,8 @@ func main() {
} }
_ = logger.Sync() _ = logger.Sync()
zap.ReplaceGlobals(logger)
Execute(logger) Execute(logger)
} }

View File

@ -2,15 +2,33 @@ package actions
import ( import (
"context" "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/schema"
"git.front.kjuulh.io/kjuulh/kraken/internal/services/storage" "git.front.kjuulh.io/kjuulh/kraken/internal/services/storage"
"go.uber.org/zap"
) )
type Action struct { type Action struct {
Schema *schema.KrakenSchema Schema *schema.KrakenSchema
SchemaPath string
} }
func (a *Action) Execute(ctx context.Context, area *storage.Area) error { 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
}
exe(ctx, area.Path)
default:
return errors.New("could not determine action type")
}
}
return nil return nil
} }

View File

@ -70,9 +70,9 @@ func (ac *ActionCreator) Prepare(ctx context.Context, ops *ActionCreatorOps) (*A
} }
ac.logger.Debug("Action creator done") ac.logger.Debug("Action creator done")
return &Action{ return &Action{
Schema: krakenSchema, Schema: krakenSchema,
SchemaPath: executorUrl,
}, nil }, nil
} }

View File

@ -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, fmt.Sprintf("(cd %s; %s/main)", victimPath, modulePath)).Run()
}, nil
}