utils.go

package main

import (
	"fmt"
	"io/ioutil"
	"os"
	"testing"

	"github.com/Jeffail/gabs/v2"
)

func readInput(f string) (*gabs.Container, error) {
	var bytes []byte
	if f != "" {
		debug.Printf("Attempting to read file %s", f)
		b, err := ioutil.ReadFile(f)
		if err != nil {
			return nil, err
		}
		bytes = b
	} else {
		stat, _ := os.Stdin.Stat()
		if stat.Mode()&os.ModeNamedPipe != 0 {
			debug.Printf("Attempting to read from stdin")
			b, err := ioutil.ReadAll(os.Stdin)
			if err != nil {
				return nil, err
			}
			bytes = b
		}
	}

	if len(bytes) == 0 {
		e := fmt.Errorf("No input received.")
		return nil, e
	}

	jsonParsed, err := gabs.ParseJSON(bytes)
	if err != nil {
		return nil, err
	}

	return jsonParsed, nil
}

func getDelimiter(d string) string {
	if d != "" {
		return d
	} else {
		return "."
	}
}

func errAndExit(e error) {
	fmt.Printf("Error: %s\n", e)
	os.Exit(1)
}

func readTestFile(t *testing.T, file string) *gabs.Container {
	j, err := readInput("test.json")
	if err != nil {
		t.Fatal(err)
	}

	return j
}