ari/arimelody.me
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

dynamic data directory

Browse source
This commit is contained in:
ari melody 2024-11-10 00:37:01 +00:00
parent d0b392f6a0
commit 5284b8a7cc
Signed by: ari
GPG key ID: CF99829C92678188
3 changed files with 24 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

22
global/data.go
View file

@ -3,6 +3,7 @@ package global
import (
"fmt"
"os"
"path/filepath"
"strings"
"github.com/jmoiron/sqlx"
@ -34,7 +35,7 @@ var Args = func() map[string]string {
return args
}()
var HTTP_DOMAIN = func() string {
var HTTP_DOMAIN = func() string {
domain := os.Getenv("ARIMELODY_HTTP_DOMAIN")
if domain == "" {
return "https://arimelody.me"
@ -42,4 +43,23 @@ var HTTP_DOMAIN = func() string {
return domain
}()
var DATA_DIR = func() string {
dir, err := filepath.Abs(os.Getenv("ARIMELODY_DATA_DIR"))
if err != nil {
fmt.Printf("FATAL: Failed to get working directory: %s\n", err.Error())
os.Exit(1)
}
if dir != "" {
os.MkdirAll(dir, os.ModePerm)
} else {
var err error
dir, err = os.Getwd()
if err != nil {
fmt.Printf("FATAL: Failed to get working directory: %s\n", err.Error())
os.Exit(1)
}
}
return dir
}()
var DB *sqlx.DB