mirror of
https://github.com/jesseduffield/lazygit.git
synced 2024-12-04 10:34:55 +02:00
130 lines
2.4 KiB
Go
130 lines
2.4 KiB
Go
package utils
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"regexp"
|
|
"runtime"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/jesseduffield/gocui"
|
|
)
|
|
|
|
// GetCurrentRepoName gets the repo's base name
|
|
func GetCurrentRepoName() string {
|
|
pwd, err := os.Getwd()
|
|
if err != nil {
|
|
log.Fatalln(err.Error())
|
|
}
|
|
return filepath.Base(pwd)
|
|
}
|
|
|
|
// GetProjectRoot returns the path to the root of the project. Only to be used
|
|
// in testing contexts, as with binaries it's unlikely this path will exist on
|
|
// the machine
|
|
func GetProjectRoot() string {
|
|
dir, err := os.Getwd()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return strings.Split(dir, "lazygit")[0] + "lazygit"
|
|
}
|
|
|
|
// Loader dumps a string to be displayed as a loader
|
|
func Loader() string {
|
|
characters := "|/-\\"
|
|
now := time.Now()
|
|
nanos := now.UnixNano()
|
|
index := nanos / 50000000 % int64(len(characters))
|
|
return characters[index : index+1]
|
|
}
|
|
|
|
// Min returns the minimum of two integers
|
|
func Min(x, y int) int {
|
|
if x < y {
|
|
return x
|
|
}
|
|
return y
|
|
}
|
|
|
|
func AsJson(i interface{}) string {
|
|
bytes, _ := json.MarshalIndent(i, "", " ")
|
|
return string(bytes)
|
|
}
|
|
|
|
// used to keep a number n between 0 and max, allowing for wraparounds
|
|
func ModuloWithWrap(n, max int) int {
|
|
if n >= max {
|
|
return n % max
|
|
} else if n < 0 {
|
|
return max + n
|
|
} else {
|
|
return n
|
|
}
|
|
}
|
|
|
|
// TruncateWithEllipsis returns a string, truncated to a certain length, with an ellipsis
|
|
func TruncateWithEllipsis(str string, limit int) string {
|
|
if limit == 1 && len(str) > 1 {
|
|
return "."
|
|
}
|
|
|
|
if limit == 2 && len(str) > 2 {
|
|
return ".."
|
|
}
|
|
|
|
ellipsis := "..."
|
|
if len(str) <= limit {
|
|
return str
|
|
}
|
|
|
|
remainingLength := limit - len(ellipsis)
|
|
return str[0:remainingLength] + "..."
|
|
}
|
|
|
|
func FindStringSubmatch(str string, regexpStr string) (bool, []string) {
|
|
re := regexp.MustCompile(regexpStr)
|
|
match := re.FindStringSubmatch(str)
|
|
return len(match) > 0, match
|
|
}
|
|
|
|
func MustConvertToInt(s string) int {
|
|
i, err := strconv.Atoi(s)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return i
|
|
}
|
|
|
|
// Safe will close tcell if a panic occurs so that we don't end up in a malformed
|
|
// terminal state
|
|
func Safe(f func()) {
|
|
_ = SafeWithError(func() error { f(); return nil })
|
|
}
|
|
|
|
func SafeWithError(f func() error) error {
|
|
panicking := true
|
|
defer func() {
|
|
if panicking && gocui.Screen != nil {
|
|
gocui.Screen.Fini()
|
|
}
|
|
}()
|
|
|
|
err := f()
|
|
|
|
panicking = false
|
|
|
|
return err
|
|
}
|
|
|
|
func StackTrace() string {
|
|
buf := make([]byte, 10000)
|
|
n := runtime.Stack(buf, false)
|
|
return fmt.Sprintf("%s\n", buf[:n])
|
|
}
|