feature/add-actions #9
@ -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)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
@ -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 {
|
||||||
|
@ -15,6 +15,8 @@ func main() {
|
|||||||
}
|
}
|
||||||
_ = logger.Sync()
|
_ = logger.Sync()
|
||||||
|
|
||||||
|
zap.ReplaceGlobals(logger)
|
||||||
|
|
||||||
Execute(logger)
|
Execute(logger)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -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
|
||||||
}
|
}
|
||||||
|
@ -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
|
||||||
}
|
}
|
||||||
|
|
||||||
|
46
internal/actions/builders/go.go
Normal file
46
internal/actions/builders/go.go
Normal 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
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user