Go code to get the shell of the current user

go.mod

@@ -16,6 +16,7 @@ require (
 	golang.org/x/exp v0.0.0-20230321023759-10a507213a29
 	golang.org/x/image v0.8.0
 	golang.org/x/sys v0.9.0
+	howett.net/plist v1.0.0
 )
 
 require (

go.sum

@@ -23,6 +23,7 @@ github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+
 github.com/hexops/gotextdiff v1.0.3 h1:gitA9+qJrrTCsiCl7+kh75nPqQt1cx4ZkudSTLoUqJM=
 github.com/jamesruan/go-rfc1924 v0.0.0-20170108144916-2767ca7c638f h1:Ko4+g6K16vSyUrtd/pPXuQnWsiHe5BYptEtTxfwYwCc=
 github.com/jamesruan/go-rfc1924 v0.0.0-20170108144916-2767ca7c638f/go.mod h1:eHzfhOKbTGJEGPSdMHzU6jft192tHHt2Bu2vIZArvC0=
+github.com/jessevdk/go-flags v1.4.0/go.mod h1:4FA24M0QyGHXBuZZK/XkWh8h0e1EYbRYJSGM75WSRxI=
 github.com/lufia/plan9stats v0.0.0-20211012122336-39d0f177ccd0/go.mod h1:zJYVVT2jmtg6P3p1VtQj7WsuWi/y4VnjVBn7F8KPB3I=
 github.com/lufia/plan9stats v0.0.0-20230326075908-cb1d2100619a h1:N9zuLhTvBSRt0gWSiJswwQ2HqDmtX/ZCDJURnKUt1Ik=
 github.com/lufia/plan9stats v0.0.0-20230326075908-cb1d2100619a/go.mod h1:JKx41uQRwqlTZabZc+kILPrO/3jlKnQ2Z8b7YiVw5cE=
@@ -98,6 +99,9 @@ golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU=
 golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
 golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v1 v1.0.0-20140924161607-9f9df34309c0/go.mod h1:WDnlLJ4WF5VGsH/HVa3CI79GS0ol3YnhVnKP89i0kNg=
 gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+howett.net/plist v1.0.0 h1:7CrbWYbPPO/PyNy38b2EB/+gYbjCe2DXBxgtOOZbSQM=
+howett.net/plist v1.0.0/go.mod h1:lqaXoTrLY4hg8tnEzNru53gicrbv7rrk+2xJA/7hw9g=

tools/utils/passwd.go

@@ -0,0 +1,171 @@
+// License: GPLv3 Copyright: 2023, Kovid Goyal, <kovid at kovidgoyal.net>
+
+package utils
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+	"os/user"
+	"runtime"
+	"strconv"
+	"strings"
+
+	"howett.net/plist"
+)
+
+var _ = fmt.Print
+
+type PasswdEntry struct {
+	Username, Pass, Uid, Gid, Gecos, Home, Shell string
+}
+
+func ParsePasswdLine(line string) (PasswdEntry, error) {
+	parts := strings.Split(line, ":")
+	if len(parts) == 7 {
+		return PasswdEntry{parts[0], parts[1], parts[2], parts[3], parts[4], parts[5], parts[6]}, nil
+	}
+	return PasswdEntry{}, fmt.Errorf("passwd line has %d colon delimited fields instead of 7", len(parts))
+}
+
+func ParsePasswdDatabase(raw string) (ans map[string]PasswdEntry) {
+	scanner := NewLineScanner(raw)
+	ans = make(map[string]PasswdEntry)
+	for scanner.Scan() {
+		line := scanner.Text()
+		if entry, e := ParsePasswdLine(line); e == nil {
+			ans[entry.Uid] = entry
+		}
+	}
+	return ans
+}
+
+func ParsePasswdFile(path string) (ans map[string]PasswdEntry, err error) {
+	raw, err := os.ReadFile(path)
+	if err != nil {
+		return nil, err
+	}
+	return ParsePasswdDatabase(UnsafeBytesToString(raw)), nil
+}
+
+var passwd_err error
+var passwd_database = Once(func() (ans map[string]PasswdEntry) {
+	ans, passwd_err = ParsePasswdFile("/etc/passwd")
+	return
+})
+
+func PwdEntryForUid(uid string) (ans PasswdEntry, err error) {
+	pwd := passwd_database()
+	if passwd_err != nil {
+		return ans, passwd_err
+	}
+	ans, found := pwd[uid]
+	if !found {
+		return ans, fmt.Errorf("No user matching the UID: %#v found", uid)
+	}
+	return ans, nil
+}
+
+func parse_dscl_data(raw []byte) (ans map[string]PasswdEntry, err error) {
+	var pd []any
+	_, err = plist.Unmarshal(raw, &pd)
+	if err != nil {
+		return
+	}
+	ans = make(map[string]PasswdEntry, 256)
+	for _, entry := range pd {
+		if e, ok := entry.(map[string]any); ok {
+			item := PasswdEntry{}
+			for key, a := range e {
+				array, ok := a.([]any)
+				if !ok || len(array) == 0 || !strings.HasPrefix(key, "dsAttrTypeNative:") {
+					continue
+				}
+				_, key, _ = strings.Cut(key, ":")
+				if val, ok := array[0].(string); ok {
+					switch key {
+					case "uid":
+						item.Uid = val
+					case "gid":
+						item.Gid = val
+					case "home":
+						item.Home = val
+					case "name":
+						item.Username = val
+					case "realname":
+						item.Gecos = val
+					case "shell":
+						item.Shell = val
+					}
+				}
+			}
+			ans[item.Uid] = item
+		}
+	}
+	return
+}
+
+var dscl_error error
+
+var dscl_user_database = Once(func() map[string]PasswdEntry {
+	c := exec.Command("/usr/bin/dscl", "-plist", ".", "-readall", "/Users", "uid", "gid", "name", "realname", "home", "shell")
+	raw, err := c.Output()
+	if err != nil {
+		dscl_error = err
+		return nil
+	}
+	ans, err := parse_dscl_data(raw)
+	if err != nil {
+		dscl_error = err
+		return nil
+	}
+	return ans
+
+})
+
+func LoginShellForUser(u *user.User) (ans string, err error) {
+	var db map[string]PasswdEntry
+	switch runtime.GOOS {
+	case "darwin":
+		db = dscl_user_database()
+		err = dscl_error
+	default:
+		db = passwd_database()
+		err = passwd_err
+	}
+	if err != nil {
+		return
+	}
+	if rec, found := db[u.Uid]; found {
+		return rec.Shell, nil
+	}
+	return ans, fmt.Errorf("No user record available for user with UID: %#v", u.Uid)
+}
+
+func CurrentUser() (ans *user.User, err error) {
+	ans, err = user.Current()
+	if err != nil && runtime.GOOS == "darwin" {
+		uid := strconv.Itoa(os.Geteuid())
+		db := dscl_user_database()
+		if dscl_error != nil {
+			err = dscl_error
+			return
+		}
+		if rec, found := db[uid]; found {
+			u := user.User{Uid: uid, Gid: rec.Gid, Username: rec.Username, Name: rec.Gecos, HomeDir: rec.Home}
+			ans = &u
+			err = nil
+		} else {
+			err = fmt.Errorf("Could not find the current uid: %d in the DSCL user database", os.Geteuid())
+		}
+	}
+	return
+}
+
+func LoginShellForCurrentUser() (ans string, err error) {
+	u, err := CurrentUser()
+	if err != nil {
+		return ans, err
+	}
+	return LoginShellForUser(u)
+}