MailPocket/shared/utils.go

package shared

import (
	"log"
	"os"
)

func GetArgs() (string, string, string) {
	var PORT string
	var ADMINPORT string
	if len(os.Args) > 2 {
		ADMINPORT = os.Args[2]
		PORT = os.Args[1]
	} else if len(os.Args) > 1 {
		PORT = os.Args[1]
	}

	if PORT == "" {
		PORT = "15521"
	}

	if ADMINPORT == "" {
		ADMINPORT = "15522"
	}

	dbdir := "data"

	isDocker := os.Getenv("container") == "docker" || os.Getenv("DOCKER") == "true" || func() bool { _, err := os.Stat("/.dockerenv"); return err == nil }()
	if isDocker {
		dbdir = "/app/data"
	}

	if _, err := os.Stat(dbdir); os.IsNotExist(err) {
		if err := os.MkdirAll(dbdir, 0755); err != nil {
			log.Fatalf("Failed to create directory: %v", err)
		}
	}

	return ADMINPORT, PORT, dbdir
}