short_prompt_pwd similar to fish

short_prompt_pwd.go

package main

import (
        "fmt"
        "log"
        "os"
        "strings"
)

func getdir(str string) (string, string) {
        index := strings.Index(str, "/")

        if index < 0 {
                return str, ""
        }
        if index == len(str) {
                return str, ""
        }
        if index == 0 {
                return "/", str[index+1:]
        }

        return str[0:1] + "/", str[index+1:]
}

func main() {
        dirname, err := os.Getwd()
        home := os.Getenv("HOME")
        if strings.HasPrefix(dirname, home) {
                dirname = strings.Replace(dirname, home, "~", 1)
        }
        if err != nil {
                log.Println(err)
        }
        cur, rest := getdir(dirname)
        output := cur
        for rest != "" {
                cur, rest = getdir(rest)
                output = output + cur
        }
        fmt.Println(output)
}