feat: add GetUserHomeDir and WriteTmpFile (#78)

file/file.go

@@ -54,3 +54,21 @@ func ExpandFileList(fileList []string) ([]string, error) {
 
  return result, nil
 }
+
+// WriteTmpFile creates a temporary file with the given name and content, and returns the path to the created file.
+func WriteTmpFile(name, content string) (string, error) {
+ tmpfile, err := os.CreateTemp("", name)
+ if err != nil {
+ return "", err
+ }
+
+ if _, err := tmpfile.Write([]byte(content)); err != nil {
+ return "", err
+ }
+
+ if err := tmpfile.Close(); err != nil {
+ return "", err
+ }
+
+ return tmpfile.Name(), nil
+}

file/file_test.go

@@ -0,0 +1,64 @@
+package file
+
+import (
+ "os"
+ "path/filepath"
+ "testing"
+
+ "github.com/stretchr/testify/assert"
+)
+
+const helloWorld = "Hello, World!"
+
+func TestWriteTmpFile(t *testing.T) {
+ tests := []struct {
+ name string
+ fileName string
+ content string
+ wantErr bool
+ }{
+ {
+ name: "write to temp file",
+ fileName: "test.txt",
+ content: helloWorld,
+ wantErr: false,
+ },
+ {
+ name: "empty file name",
+ fileName: "",
+ content: helloWorld,
+ wantErr: false,
+ },
+ {
+ name: "empty file content",
+ fileName: "test.txt",
+ content: "",
+ wantErr: false,
+ },
+ {
+ name: "create temp file error",
+ fileName: filepath.Join(os.TempDir(), "non-existent", "test.txt"),
+ content: helloWorld,
+ wantErr: true,
+ },
+ }
+
+ for _, tt := range tests {
+ t.Run(tt.name, func(t *testing.T) {
+ tmpFile, err := WriteTmpFile(tt.fileName, tt.content)
+ if tt.wantErr {
+ assert.Error(t, err)
+
+ return
+ }
+
+ assert.NoError(t, err)
+
+ defer os.Remove(tmpFile)
+
+ data, err := os.ReadFile(tmpFile)
+ assert.NoError(t, err)
+ assert.Equal(t, tt.content, string(data))
+ })
+ }
+}

util/user.go

@@ -0,0 +1,15 @@
+package util
+
+import "os/user"
+
+// GetUserHomeDir returns the home directory path for the current user.
+// If the current user cannot be determined, it returns the default "/root" path.
+func GetUserHomeDir() string {
+ home := "/root"
+
+ if currentUser, err := user.Current(); err == nil {
+ home = currentUser.HomeDir
+ }
+
+ return home
+}