nikoksr/proji

package util

import (
    "fmt"
    "io"
    "os"
    "strings"
    "testing"

    "github.com/cavaliercoder/grab"
    "github.com/jedib0t/go-pretty/v6/table"
)

// DoesPathExist checks if a given path exists in the filesystem.
func DoesPathExist(path string) bool {
    _, err := os.Stat(path)
    return !os.IsNotExist(err)
}

// WantTo waits for a valid user input to confirm if he wants to do whatever was asked for.
func WantTo(question string) bool {
    var input string
    for {
        fmt.Print(question + " [y/N] ")
        n, err := fmt.Scanln(&input)
        if err != nil {
            if err.Error() != "unexpected newline" {
                fmt.Printf("Unexpected error: %v", err)
            }
        }
        if n == 1 {
            input = strings.ToLower(input)
            if input == "n" {
                return false
            } else if input == "y" {
                return true
            }
        } else if n == 0 {
            return false
        }
    }
}

// SkipNetworkBasedTests skips network/internet dependent tests when the env variable PROJI_SKIP_NETWORK_TESTS is set to 1.
func SkipNetworkBasedTests(t *testing.T) {
    env := os.Getenv("PROJI_SKIP_NETWORK_TESTS")
    if env == "1" {
        t.Skip("Skipping network based tests")
    }
}

// CreateFolderIfNotExists creates a folder at the given path if it doesn't already exist.
func CreateFolderIfNotExists(path string) error {
    _, err := os.Stat(path)
    if !os.IsNotExist(err) {
        return err
    }
    return os.MkdirAll(path, os.ModePerm)
}

// DownloadFile downloads a file from an url to the local fs.
func DownloadFile(dst, src string) error {
    _, err := grab.Get(dst, src)
    return err
}

// DownloadFileIfNotExists runs downloadFile() if the destination file doesn't already exist.
func DownloadFileIfNotExists(dst, src string) error {
    _, err := os.Stat(dst)
    if os.IsNotExist(err) {
        err = DownloadFile(dst, src)
    }
    return err
}

// NewInfoTable returns a new table.Writer interface bound to the given io.Writer. It sets some sane defaults
// for table styles and behaviour that are used in the cmd package.
func NewInfoTable(out io.Writer) table.Writer {
    infoTable := table.NewWriter()
    infoTable.SetOutputMirror(out)
    infoTable.SuppressEmptyColumns()
    infoTable.SetAutoIndex(true)
    infoTable.SetStyle(table.StyleRounded)
    return infoTable
}