Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 20 additions & 1 deletion modules/user/user.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,10 +4,29 @@

package user

import "os"
import (
"os"
"os/user"
"runtime"
"strings"
)

// CurrentUsername return current login OS user name
func CurrentUsername() string {
userinfo, err := user.Current()
if err != nil {
return fallbackCurrentUsername()
}
username := userinfo.Username
if runtime.GOOS == "windows" {
parts := strings.Split(username, "\\")
username = parts[len(parts)-1]
}
return username
}

// Old method, used if new method doesn't work on your OS for some reason
func fallbackCurrentUsername() string {
curUserName := os.Getenv("USER")
if len(curUserName) > 0 {
return curUserName
Expand Down
37 changes: 28 additions & 9 deletions modules/user/user_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,21 +2,40 @@ package user

import (
"os"
"os/exec"
"runtime"
"strings"
"testing"
)

func TestCurrentUsername(t *testing.T) {
os.Setenv("USER", "")
os.Setenv("USERNAME", "foobar")
func getWhoamiOutput() (string, error) {
output, err := exec.Command("whoami").Output()
if err != nil {
return "", err
}
return strings.TrimSpace(string(output[:])), nil
}

func TestCurrentUsername(t *testing.T) {
user := CurrentUsername()
if user != "foobar" {
t.Errorf("expected foobar as user, got: %s", user)
if len(user) <= 0 {
t.Errorf("expected non-empty user, got: %s", user)
}

os.Setenv("USER", "gitea")
// Windows whoami is weird, so just skip remaining tests
if runtime.GOOS == "windows" {
t.Skip("skipped test because of weird whoami on Windows")
}
whoami, err := getWhoamiOutput()
if err != nil {
t.Errorf("failed to run whoami to test current user: %f", err)
}
user = CurrentUsername()
if user != whoami {
t.Errorf("expected %s as user, got: %s", whoami, user)
}
os.Setenv("USER", "spoofed")
user = CurrentUsername()
if user != "gitea" {
t.Errorf("expected gitea as user, got: %s", user)
if user != whoami {
t.Errorf("expected %s as user, got: %s", whoami, user)
}
}