generate config file

Author: xhofe

.gitignore

@@ -0,0 +1,23 @@
+.idea/
+.DS_Store
+output/
+dist/
+
+# Binaries for programs and plugins
+*.exe
+*.exe~
+*.dll
+*.so
+*.dylib
+*.db
+*.bin
+
+# Test binary, built with `go test -c`
+*.test
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+
+# Dependency directories (remove the comment below to include it)
+# vendor/
+bin/*

entrypoint.sh

@@ -1,3 +1,4 @@
+mkdir -p /opt/alist/data/
 /main
 
 cd /opt/alist

main.go

@@ -1,26 +1,79 @@
 package main
+
 import (
+	"encoding/json"
 	"fmt"
+	"io/ioutil"
+	"log"
 	"net/url"
 	"os"
+	"strconv"
 )
 
+type Database struct {
+	Type        string `json:"type" env:"DB_TYPE"`
+	Host        string `json:"host" env:"DB_HOST"`
+	Port        int    `json:"port" env:"DB_PORT"`
+	User        string `json:"user" env:"DB_USER"`
+	Password    string `json:"password" env:"DB_PASS"`
+	Name        string `json:"name" env:"DB_NAME"`
+	DBFile      string `json:"db_file" env:"DB_FILE"`
+	TablePrefix string `json:"table_prefix" env:"DB_TABLE_PREFIX"`
+	SslMode     string `json:"ssl_mode" env:"DB_SLL_MODE"`
+}
+
+type Scheme struct {
+	Https    bool   `json:"https" env:"HTTPS"`
+	CertFile string `json:"cert_file" env:"CERT_FILE"`
+	KeyFile  string `json:"key_file" env:"KEY_FILE"`
+}
+
+type CacheConfig struct {
+	Expiration      int64 `json:"expiration" env:"CACHE_EXPIRATION"`
+	CleanupInterval int64 `json:"cleanup_interval" env:"CLEANUP_INTERVAL"`
+}
+
+type Config struct {
+	Force    bool        `json:"force"`
+	Address  string      `json:"address" env:"ADDR"`
+	Port     int         `json:"port" env:"PORT"`
+	Assets   string      `json:"assets" env:"ASSETS"`
+	Database Database    `json:"database"`
+	Scheme   Scheme      `json:"scheme"`
+	Cache    CacheConfig `json:"cache"`
+	TempDir  string      `json:"temp_dir" env:"TEMP_DIR"`
+}
+
 func main() {
-	DatabaseUrl := os.Getenv("DATABASE_URL")
-	fmt.Println("DatabaseUrl",DatabaseUrl)
-	// DATABASE_URL = postgres://user3123:passkja83kd8@ec2-117-21-174-214.compute-1.amazonaws.com:6212/db982398
-	u, err := url.Parse(DatabaseUrl)
+	DATABASE_URL := os.Getenv("DATABASE_URL")
+	fmt.Println("DatabaseUrl", DATABASE_URL)
+	DATABASE_URL = "postgres://user3123:passkja83kd8@ec2-117-21-174-214.compute-1.amazonaws.com:6212/db982398"
+	u, err := url.Parse(DATABASE_URL)
 	if err != nil {
 		fmt.Println(err)
 	}
 	user := u.User.Username()
 	pass, _ := u.User.Password()
 	host := u.Hostname()
-	port := u.Port()
+	port, _ := strconv.Atoi(u.Port())
 	name := u.Path[1:]
-	_ = os.Setenv("DB_HOST", host)
-	_ = os.Setenv("DB_PORT", port)
-	_ = os.Setenv("DB_USER", user)
-	_ = os.Setenv("DB_PASS", pass)
-	_ = os.Setenv("DB_NAME", name)
-}
+	config := Config{
+		Address: "0.0.0.0",
+		TempDir: "data/temp",
+		Database: Database{
+			User:     user,
+			Password: pass,
+			Host:     host,
+			Port:     port,
+			Name:     name,
+		},
+	}
+	confBody, err := json.MarshalIndent(config, "", "  ")
+	if err != nil {
+		log.Fatalf("failed marshal json: %s", err.Error())
+	}
+	err = ioutil.WriteFile("/opt/alist/data/config.json", confBody, 0777)
+	if err != nil {
+		log.Fatalf("failed write json: %s", err.Error())
+	}
+}