utils.go

package main

import (
	"fmt"
	"os"
	"os/exec"
	"os/user"
	"path/filepath"
	"strings"
)

// formats cursor and output to STDOUT
func printInputIndicator() {
	user := fmt.Sprintf("%s%s%s", GREEN, getCurrentUser(), Reset)
	hostname := fmt.Sprintf("@%s%s%s", SOFT_RED, getCurrntHostname(), Reset)
	folder := fmt.Sprintf("%s%s$%s", CYAN_BLUE, getCurrentFolder(), Reset)
	git := fmt.Sprintf("%s%s%s", AMBER, getCurrentBranch(), Reset)

	fmt.Fprintf(os.Stdout, "%s%s ~/%s %s > ", user, hostname, folder, git)
}

func getCurrentFolder() string {
	dir, err := os.Getwd()
	if err != nil {
		fmt.Fprintln(os.Stderr, err)
		return ""
	}

	folder := filepath.Base(dir)

	if len(folder) > 20 {
		return fmt.Sprintf("%s..", folder[:20])
	}

	return folder
}

func getCurrntHostname() string {
	hostname, err := os.Hostname()
	if err != nil {
		fmt.Fprintln(os.Stderr, err)
		return ""
	}

	return hostname
}

func getCurrentUser() string {
	currntUser, err := user.Current()
	if err != nil {
		fmt.Fprintln(os.Stderr, err)
		return ""
	}

	return currntUser.Username
}

func getCurrentBranch() string {
	args := []string{"rev-parse", "--abbrev-ref", "HEAD"}

	out, err := exec.Command("git", args...).Output()
	if err != nil {
		return ""
	}

	branch := string(out)

	branch = strings.TrimSuffix(branch, "\n")

	return fmt.Sprintf("(%s)", branch)
}

func isSSHSession() bool {
	if os.Getenv("SSH_CLIENT") != "" || os.Getenv("SSH_CONNECTION") != "" || os.Getenv("SSH_TTY") != "" {
		return true
	}

	return false
}