From 62e85f40923b3485ba60766052de7150acf039c5 Mon Sep 17 00:00:00 2001
From: Flu0r1ne <flur01ne@flu0r1ne.net>
Date: Sun, 5 Sep 2021 04:30:33 -0500
Subject: Add bash adapter

---
 adapters/bash/adapter.go | 96 ++++++++++++++++++++++++++++++++++++++++++++++++
 adapters/bash/config.go  | 64 ++++++++++++++++++++++++++++++++
 cmd/planr/main.go        | 11 +++---
 3 files changed, 166 insertions(+), 5 deletions(-)
 create mode 100644 adapters/bash/adapter.go
 create mode 100644 adapters/bash/config.go

diff --git a/adapters/bash/adapter.go b/adapters/bash/adapter.go
new file mode 100644
index 0000000..b98523e
--- /dev/null
+++ b/adapters/bash/adapter.go
@@ -0,0 +1,96 @@
+package bash
+
+import (
+	"context"
+	"errors"
+	"log"
+	"os"
+	"os/exec"
+	"path"
+	"strings"
+	"time"
+        "fmt"
+
+	"golang.flu0r1ne.net/planr"
+)
+
+type Adapter struct {
+  dirs planr.DirConfig
+}
+
+func (a *Adapter) Config() planr.AdapterConfig {
+  return planr.AdapterConfig {
+    Name: "bash",
+    ParseConfig: ParseConfig,
+    ParseDefaultConfig: ParseDefaultConfig,
+  }
+}
+
+func safeWd() string{
+  wd, err := os.Getwd()
+
+  if err != nil {
+    log.Fatalf("Could not get GtestBuildDir %s %v\n", wd, err)
+  }
+
+  return wd
+}
+
+func (a *Adapter) Init(dirs planr.DirConfig) {
+  a.dirs = dirs
+}
+
+func (adapter Adapter) Build(tcs []planr.TestCase) { }
+
+func executeScriptedTest(testdir string, tc planr.TestCase) planr.TestResult {
+  cfg := tc.AdapterConfig().(*Config)
+
+  timeout := time.Duration(cfg.Timeout) * time.Millisecond
+
+  ctx, cancel := context.WithTimeout(context.Background(), timeout)
+
+  defer cancel()
+
+  path := path.Join(testdir, cfg.Testfile)
+
+  result := planr.TestResult {}
+  result.Tc = tc
+
+  cmd := exec.CommandContext(ctx, "bash", path)
+    
+  if out, err := cmd.CombinedOutput(); err != nil {
+    result.Status = planr.RUNTIME_FAILURE  
+    result.TestOutput = string(out)
+
+    var exiterr *exec.ExitError
+    if !errors.As(err, &exiterr) {
+      log.Fatalf("Test script %s failed with unknown error %v\n", path, err)
+    } else {
+      if strings.Contains(exiterr.String(), "killed") {
+        result.TestOutput += fmt.Sprintf("TEST TERMINATED (Timeout=%d)\n", cfg.Timeout)
+      }
+    }
+
+    return result
+  }
+
+  result.Status = planr.PASSING
+
+
+  return result
+}
+
+func (adapter Adapter) Evaluate(tcs []planr.TestCase) [] planr.TestResult {
+  finalizeConfigs(tcs)
+ 
+  trs := make([]planr.TestResult, len(tcs))
+  for i, tc := range tcs {
+    trs[i] = executeScriptedTest(adapter.dirs.Tests(), tc)
+  }
+
+  return trs
+}
+
+func NewAdapter() *Adapter {
+  return new(Adapter)
+}
diff --git a/adapters/bash/config.go b/adapters/bash/config.go
new file mode 100644
index 0000000..aaa405a
--- /dev/null
+++ b/adapters/bash/config.go
@@ -0,0 +1,64 @@
+package bash
+
+import (
+  "log"
+  "golang.flu0r1ne.net/planr"
+  "github.com/BurntSushi/toml"
+)
+
+const (
+  DEFAULT_TIMEOUT=1000
+)
+
+type Defaults struct {
+  Testfile  string
+  Timeout   uint 
+}
+
+func (child *Defaults) Inherit(p interface{}) {
+  parent := p.(*Defaults)
+
+  if(child.Timeout == 0) { child.Timeout = parent.Timeout }
+}
+
+type Config struct {
+  Defaults
+}
+
+func (c *Config) finalize(path string) {
+  if c.Testfile == "" {
+    log.Fatalf("\"Testfile\" is not defined for unit %s\n", path)
+  }
+
+  if c.Timeout == 0 {
+    c.Timeout = DEFAULT_TIMEOUT;
+  }
+}
+
+func finalizeConfigs(tcs []planr.TestCase) {
+  for i := range tcs {
+    cfg := tcs[i].AdapterConfig().(*Config)
+
+    cfg.finalize(tcs[i].Path)
+  }
+}
+
+func ParseConfig(prim toml.Primitive) (planr.InheritableConfig, error) {
+  config := Config {}
+
+  if err := toml.PrimitiveDecode(prim, &config); err != nil {
+    return nil, err
+  }
+
+  return &config, nil
+}
+
+func ParseDefaultConfig(prim toml.Primitive) (planr.InheritableConfig, error) {
+  config := Defaults{}
+
+  if err := toml.PrimitiveDecode(prim, &config); err != nil {
+    return nil, err
+  }
+
+  return &config, nil
+}
diff --git a/cmd/planr/main.go b/cmd/planr/main.go
index ce20811..d250bbd 100644
--- a/cmd/planr/main.go
+++ b/cmd/planr/main.go
@@ -1,7 +1,6 @@
 package main
 
-import (
-	"flag"
+import ( "flag"
 	"fmt"
 	"io"
 	"log"
@@ -10,6 +9,7 @@ import (
 
 	"golang.flu0r1ne.net/planr"
 	"golang.flu0r1ne.net/planr/adapters/gtest"
+	"golang.flu0r1ne.net/planr/adapters/bash"
 	"golang.flu0r1ne.net/planr/cmd/planr/sub"
 )
 
@@ -28,14 +28,15 @@ func dieUsage() {
 	os.Exit(1)
 }
 
-var src        = flag.String("srcdir", "", "source directory")
-var config     = flag.String("configdir", "", "config directory")
-var build      = flag.String("builddir", "", "build directory")
+var src        = flag.String("srcdir",     "", "source directory")
+var config     = flag.String("configdir",  "", "config directory")
+var build      = flag.String("builddir",   "", "build directory")
 var cpuprofile = flag.String("cpuprofile", "", "write cpu profile to file")
 
 func getConfiguredRunner() planr.Runner {
   r := planr.ConfigureRunner()
   r  = planr.RegisterAdapter(r, gtest.NewAdapter())
+  r  = planr.RegisterAdapter(r, bash.NewAdapter())
 
   if wd, err := os.Getwd(); err == nil {
     r = planr.SetConfigDirFromTree(r, wd) 
-- 
cgit v1.2.3