add human readable time field

Author: skudasov

framework/input.go framework/config.goframework/input.go renamed to framework/config.go

@@ -3,6 +3,7 @@ package framework
 import (
 	"bytes"
 	"context"
+	"encoding/json"
 	"errors"
 	"fmt"
 	"github.com/davecgh/go-spew/spew"
@@ -18,6 +19,7 @@ import (
 	"strings"
 	"testing"
 	"text/template"
+	"time"
 )
 
 const (
@@ -297,3 +299,59 @@ func applyEnvConfig(prefix string, input interface{}) error {
 	}
 	return nil
 }
+
+func UseCache() bool {
+	return os.Getenv("CTF_USE_CACHED_OUTPUTS") == "true"
+}
+
+func getBaseConfigPath() (string, error) {
+	configs := os.Getenv("CTF_CONFIGS")
+	if configs == "" {
+		return "", fmt.Errorf("no %s env var is provided, you should provide at least one test promtailConfig in TOML", EnvVarTestConfigs)
+	}
+	return strings.Split(configs, ",")[0], nil
+}
+
+func Store[T any](cfg *T) error {
+	if UseCache() {
+		return nil
+	}
+	baseConfigPath, err := getBaseConfigPath()
+	if err != nil {
+		return err
+	}
+	cachedOutName := fmt.Sprintf("%s-cache.toml", strings.Replace(baseConfigPath, ".toml", "", -1))
+	L.Info().Str("OutputFile", cachedOutName).Msg("Storing configuration output")
+	d, err := toml.Marshal(cfg)
+	if err != nil {
+		return err
+	}
+	return os.WriteFile(filepath.Join(DefaultConfigDir, cachedOutName), d, os.ModePerm)
+}
+
+// JSONStrDuration is JSON friendly duration that can be parsed from "1h2m0s" Go format
+type JSONStrDuration struct {
+	time.Duration
+}
+
+func (d *JSONStrDuration) MarshalJSON() ([]byte, error) {
+	return json.Marshal(d.String())
+}
+
+func (d *JSONStrDuration) UnmarshalJSON(b []byte) error {
+	var v interface{}
+	if err := json.Unmarshal(b, &v); err != nil {
+		return err
+	}
+	switch value := v.(type) {
+	case string:
+		var err error
+		d.Duration, err = time.ParseDuration(value)
+		if err != nil {
+			return err
+		}
+		return nil
+	default:
+		return errors.New("invalid duration")
+	}
+}