init: 1.0.0

pkg/fs/files+polyfill.go

@@ -0,0 +1,8 @@
+//go:build windows
+
+package fs
+
+import "os"
+
+func CloseOnExec(*os.File) {
+}

pkg/fs/files.go

@@ -0,0 +1,15 @@
+//go:build linux || darwin
+
+package fs
+
+import (
+	"os"
+	"syscall"
+)
+
+// CloseOnExec makes sure closing the file on process forking.
+func CloseOnExec(file *os.File) {
+	if file != nil {
+		syscall.CloseOnExec(int(file.Fd()))
+	}
+}

pkg/fs/files_test.go

@@ -0,0 +1,15 @@
+package fs
+
+import (
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCloseOnExec(t *testing.T) {
+	file := os.NewFile(0, os.DevNull)
+	assert.NotPanics(t, func() {
+		CloseOnExec(file)
+	})
+}

pkg/fs/temps.go

@@ -0,0 +1,41 @@
+package fs
+
+import (
+	"os"
+
+	"github.com/perfect-panel/ppanel-server/pkg/hash"
+)
+
+// TempFileWithText creates the temporary file with the given content,
+// and returns the opened *os.File instance.
+// The file is kept as open, the caller should close the file handle,
+// and remove the file by name.
+func TempFileWithText(text string) (*os.File, error) {
+	tmpFile, err := os.CreateTemp(os.TempDir(), hash.Md5Hex([]byte(text)))
+	if err != nil {
+		return nil, err
+	}
+
+	if err := os.WriteFile(tmpFile.Name(), []byte(text), os.ModeTemporary); err != nil {
+		return nil, err
+	}
+
+	return tmpFile, nil
+}
+
+// TempFilenameWithText creates the file with the given content,
+// and returns the filename (full path).
+// The caller should remove the file after use.
+func TempFilenameWithText(text string) (string, error) {
+	tmpFile, err := TempFileWithText(text)
+	if err != nil {
+		return "", err
+	}
+
+	filename := tmpFile.Name()
+	if err = tmpFile.Close(); err != nil {
+		return "", err
+	}
+
+	return filename, nil
+}

pkg/fs/temps_test.go

@@ -0,0 +1,49 @@
+package fs
+
+import (
+	"io"
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestTempFileWithText(t *testing.T) {
+	f, err := TempFileWithText("test")
+	if err != nil {
+		t.Error(err)
+	}
+	if f == nil {
+		t.Error("TempFileWithText returned nil")
+	}
+	if f.Name() == "" {
+		t.Error("TempFileWithText returned empty file name")
+	}
+	defer os.Remove(f.Name())
+
+	bs, err := io.ReadAll(f)
+	assert.Nil(t, err)
+	if len(bs) != 4 {
+		t.Error("TempFileWithText returned wrong file size")
+	}
+	if f.Close() != nil {
+		t.Error("TempFileWithText returned error on close")
+	}
+}
+
+func TestTempFilenameWithText(t *testing.T) {
+	f, err := TempFilenameWithText("test")
+	if err != nil {
+		t.Error(err)
+	}
+	if f == "" {
+		t.Error("TempFilenameWithText returned empty file name")
+	}
+	defer os.Remove(f)
+
+	bs, err := os.ReadFile(f)
+	assert.Nil(t, err)
+	if len(bs) != 4 {
+		t.Error("TempFilenameWithText returned wrong file size")
+	}
+}