vaulterm/server/utils/config.go

package utils

import (
	"fmt"
	"os"
	"path/filepath"
)

func GetDataPath(resolveFile string) string {
	dataDir := os.Getenv("DATA_DIR")
	if dataDir != "" {
		return filepath.Join(dataDir, resolveFile)
	}

	// Resolve the app directory
	cwd, err := os.Getwd()
	if err != nil {
		return ""
	}

	return filepath.Join(cwd, resolveFile)
}

func CheckAndCreateEnvFile() error {
	// Skip if ENCRYPTION_KEY is set
	if os.Getenv("ENCRYPTION_KEY") != "" {
		return nil
	}
	// Check if .env file exists
	envFile := GetDataPath(".env")
	if _, err := os.Stat(envFile); !os.IsNotExist(err) {
		return nil
	}

	// File doesn't exist, so create it
	randomKey, err := GenerateRandomKey()
	if err != nil {
		return err
	}

	// Write the random key to the .env file
	envContent := fmt.Sprintf("ENCRYPTION_KEY=%s\n", randomKey)
	err = os.WriteFile(envFile, []byte(envContent), 0644)
	if err != nil {
		return err
	}

	fmt.Println(".env file created with ENCRYPTION_KEY.")
	return nil
}