feat: init project

server/profile/profile.go

@@ -0,0 +1,74 @@
+package profile
+
+import (
+	"flag"
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/boojack/corgi/server/version"
+)
+
+// Profile is the configuration to start main server.
+type Profile struct {
+	// Mode can be "prod" or "dev"
+	Mode string `json:"mode"`
+	// Port is the binding port for server
+	Port int `json:"port"`
+	// Data is the data directory
+	Data string `json:"data"`
+	// DSN points to where Corgi stores its own data
+	DSN string `json:"dsn"`
+	// Version is the current version of server
+	Version string `json:"version"`
+}
+
+func checkDSN(dataDir string) (string, error) {
+	// Convert to absolute path if relative path is supplied.
+	if !filepath.IsAbs(dataDir) {
+		absDir, err := filepath.Abs(filepath.Dir(os.Args[0]) + "/" + dataDir)
+		if err != nil {
+			return "", err
+		}
+		dataDir = absDir
+	}
+
+	// Trim trailing / in case user supplies
+	dataDir = strings.TrimRight(dataDir, "/")
+
+	if _, err := os.Stat(dataDir); err != nil {
+		return "", fmt.Errorf("unable to access data folder %s, err %w", dataDir, err)
+	}
+
+	return dataDir, nil
+}
+
+// GetDevProfile will return a profile for dev or prod.
+func GetProfile() (*Profile, error) {
+	profile := Profile{}
+	flag.StringVar(&profile.Mode, "mode", "dev", "mode of server")
+	flag.IntVar(&profile.Port, "port", 8081, "port of server")
+	flag.StringVar(&profile.Data, "data", "", "data directory")
+	flag.Parse()
+
+	if profile.Mode != "dev" && profile.Mode != "prod" {
+		profile.Mode = "dev"
+	}
+
+	if profile.Mode == "prod" && profile.Data == "" {
+		profile.Data = "/var/opt/corgi"
+	}
+
+	dataDir, err := checkDSN(profile.Data)
+	if err != nil {
+		fmt.Printf("Failed to check dsn: %s, err: %+v\n", dataDir, err)
+		return nil, err
+	}
+
+	profile.Data = dataDir
+	profile.DSN = fmt.Sprintf("%s/corgi_%s.db", dataDir, profile.Mode)
+	profile.Version = version.GetCurrentVersion(profile.Mode)
+
+	return &profile, nil
+}