diff --git a/app/routes/account.go b/app/routes/account.go index 20f16c9..e7e7d7a 100644 --- a/app/routes/account.go +++ b/app/routes/account.go @@ -2,6 +2,7 @@ package routes import ( "fmt" + "math" "net/http" "proxmoxaas-dashboard/app/common" @@ -17,6 +18,68 @@ type Account struct { Min int Max int } + Resources map[string]any +} + +type Constraint struct { + Max int64 + Used int64 + Avail int64 +} + +type Match struct { + Name string + Match string + Max int64 + Used int64 + Avail int64 +} + +type NumericResource struct { + Type string + Name string + Multiplier int64 + Base int64 + Compact bool + Unit string + Display bool + Global Constraint + Nodes map[string]Constraint + Total Constraint +} + +type StorageResource struct { + Type string + Name string + Multiplier int64 + Base int64 + Compact bool + Unit string + Display bool + Disks []string + Global Constraint + Nodes map[string]Constraint + Total Constraint +} + +type ListResource struct { + Type string + Whitelist bool + Display bool + Global []Match + Nodes map[string][]Match + Total []Match +} + +type ResourceChart struct { + Type string + Display bool + Name string + Used int64 + Max int64 + Avail float64 + Prefix string + Unit string } func HandleGETAccount(c *gin.Context) { @@ -28,6 +91,58 @@ func HandleGETAccount(c *gin.Context) { return } + for k, v := range account.Resources { + switch t := v.(type) { + case NumericResource: + avail, prefix := FormatNumber(t.Total.Avail*t.Multiplier, t.Base) + account.Resources[k] = ResourceChart{ + Type: t.Type, + Display: t.Display, + Name: t.Name, + Used: t.Total.Used, + Max: t.Total.Max, + Avail: avail, + Prefix: prefix, + Unit: t.Unit, + } + case StorageResource: + avail, prefix := FormatNumber(t.Total.Avail*t.Multiplier, t.Base) + account.Resources[k] = ResourceChart{ + Type: t.Type, + Display: t.Display, + Name: t.Name, + Used: t.Total.Used, + Max: t.Total.Max, + Avail: avail, + Prefix: prefix, + Unit: t.Unit, + } + case ListResource: + l := struct { + Type string + Display bool + Resources []ResourceChart + }{ + Type: t.Type, + Display: t.Display, + Resources: []ResourceChart{}, + } + + for _, r := range t.Total { + l.Resources = append(l.Resources, ResourceChart{ + Type: t.Type, + Display: t.Display, + Name: r.Name, + Used: r.Used, + Max: r.Max, + Avail: float64(r.Avail), // usually an int + Unit: "", + }) + } + account.Resources[k] = l + } + } + c.HTML(http.StatusOK, "html/account.html", gin.H{ "global": common.Global, "page": "account", @@ -39,7 +154,9 @@ func HandleGETAccount(c *gin.Context) { } func GetUserAccount(username string, token string, csrf string) (Account, error) { - account := Account{} + account := Account{ + Resources: map[string]any{}, + } ctx := common.RequestContext{ Cookies: map[string]string{ @@ -49,6 +166,8 @@ func GetUserAccount(username string, token string, csrf string) (Account, error) }, Body: map[string]any{}, } + + // get user account basic data res, code, err := common.RequestGetAPI("/user/config/cluster", ctx) if err != nil { return account, err @@ -56,12 +175,89 @@ func GetUserAccount(username string, token string, csrf string) (Account, error) if code != 200 { return account, fmt.Errorf("request to /user/config/cluster resulted in %+v", res) } - err = mapstructure.Decode(ctx.Body, &account) if err != nil { return account, err } else { account.Username = username - return account, nil + } + + ctx.Body = map[string]any{} + // get user resources + res, code, err = common.RequestGetAPI("/user/dynamic/resources", ctx) + if err != nil { + return account, err + } + if code != 200 { + return account, fmt.Errorf("request to /user/dynamic/resources resulted in %+v", res) + } + resources := ctx.Body + + ctx.Body = map[string]any{} + // get resource meta data + res, code, err = common.RequestGetAPI("/global/config/resources", ctx) + if err != nil { + return account, err + } + if code != 200 { + return account, fmt.Errorf("request to /global/config/resources resulted in %+v", res) + } + meta := ctx.Body["resources"].(map[string]any) + + // build each resource by its meta type + for k, v := range meta { + m := v.(map[string]any) + t := m["type"].(string) + r := resources[k].(map[string]any) + if t == "numeric" { + n := NumericResource{} + n.Type = t + err_m := mapstructure.Decode(m, &n) + err_r := mapstructure.Decode(r, &n) + if err_m != nil || err_r != nil { + return account, fmt.Errorf("%s\n%s", err_m.Error(), err_r.Error()) + } + account.Resources[k] = n + } else if t == "storage" { + n := StorageResource{} + n.Type = t + err_m := mapstructure.Decode(m, &n) + err_r := mapstructure.Decode(r, &n) + if err_m != nil || err_r != nil { + return account, fmt.Errorf("%s\n%s", err_m.Error(), err_r.Error()) + } + account.Resources[k] = n + } else if t == "list" { + n := ListResource{} + n.Type = t + err_m := mapstructure.Decode(m, &n) + err_r := mapstructure.Decode(r, &n) + if err_m != nil || err_r != nil { + return account, fmt.Errorf("%s\n%s", err_m.Error(), err_r.Error()) + } + account.Resources[k] = n + } + } + + return account, nil +} + +func FormatNumber(val int64, base int64) (float64, string) { + valf := float64(val) + basef := float64(base) + steps := 0 + for math.Abs(valf) > basef && steps < 4 { + valf /= basef + steps++ + } + + if base == 1000 { + prefixes := []string{"", "K", "M", "G", "T"} + return valf, prefixes[steps] + } else if base == 1024 { + prefixes := []string{"", "Ki", "Mi", "Gi", "Ti"} + return valf, prefixes[steps] + } else { + return 0, "" } } diff --git a/web/html/account.html b/web/html/account.html index e91ece7..eeb74d4 100644 --- a/web/html/account.html +++ b/web/html/account.html @@ -3,8 +3,6 @@
{{template "head" .}} - - - -