gabor-boros/minutes

package utils

import (
    "bufio"
    "fmt"
    "os"
    "strings"
    "time"

    "github.com/spf13/cobra"
)

// Truncate chops the text at length and replaces the remaining with "...".
func Truncate(text string, length int) string {
    if length >= len(text) || length <= 0 {
        return text
    }

    truncated := ""
    maxLength := length - 3

    for i, char := range text {
        if i >= maxLength {
            break
        }

        truncated += string(char)
    }

    return truncated + "..."
}

// IsSliceContains checks if a string slice contains the given element or not.
func IsSliceContains(entry string, slice []string) bool {
    for _, s := range slice {
        if s == entry {
            return true
        }
    }

    return false
}

// Prompt shows the user a message and asks for input, then returns that.
func Prompt(message string) string {
    fmt.Print(message)

    reader := bufio.NewReader(os.Stdin)
    input, err := reader.ReadString('\n')
    cobra.CheckErr(err)

    return strings.TrimSpace(input)
}

// GetTime parses a string based on the given format and returns the time.
// If the rawDate was an empty string, the today's midnight will return.
func GetTime(rawDate string, dateFormat string) (time.Time, error) {
    if rawDate == "" {
        year, month, day := time.Now().Date()
        return time.Date(year, month, day, 0, 0, 0, 0, time.Local), nil
    }

    return time.ParseInLocation(dateFormat, rawDate, time.Local)
}