136 lines
3.4 KiB
Go
136 lines
3.4 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/gin-gonic/gin"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
type CreateDeviceVM struct {
|
|
IsEdit bool
|
|
DescriptorTree *DescriptorRoot
|
|
|
|
// Assets
|
|
Qr *int
|
|
Type string
|
|
|
|
Values map[string]string
|
|
}
|
|
|
|
func (vm *CreateDeviceVM) SetValue(assetType string, field string, value string) {
|
|
if vm.Values == nil {
|
|
vm.Values = make(map[string]string)
|
|
}
|
|
vm.Values[assetType+"-"+field] = value
|
|
}
|
|
|
|
func (vm *CreateDeviceVM) GetValue(assetType string, field string) string {
|
|
if vm.Values == nil {
|
|
return ""
|
|
}
|
|
return vm.Values[assetType+"-"+field]
|
|
}
|
|
|
|
func (vm *CreateDeviceVM) NiceType() string {
|
|
return getAssetTypeById(vm.Type).Name
|
|
}
|
|
|
|
func (vm *CreateDeviceVM) IsPartOf(assertType string) bool {
|
|
if assertType == "asset" {
|
|
return true
|
|
}
|
|
return assertType == vm.Type
|
|
}
|
|
|
|
func (a *App) getCreateDevice(c *gin.Context) {
|
|
var err error
|
|
vm := &CreateDeviceVM{}
|
|
vm.DescriptorTree = DescriptorTree
|
|
vm.Type = c.Query("type")
|
|
|
|
qr := c.Query("id")
|
|
if qr != "" {
|
|
qrInt, err := strconv.Atoi(qr)
|
|
if err != nil {
|
|
c.AbortWithError(http.StatusInternalServerError, fmt.Errorf("invalid qr: %v", err))
|
|
return
|
|
}
|
|
vm.Qr = &qrInt
|
|
}
|
|
|
|
err = a.GetAllGroups(vm)
|
|
if err != nil {
|
|
c.AbortWithError(http.StatusInternalServerError, err)
|
|
return
|
|
}
|
|
|
|
if c.Query("edit") != "" && vm.Qr != nil {
|
|
vm.IsEdit = true
|
|
err := a.pullDataFromDb(*vm.Qr, "asset", vm)
|
|
if err != nil {
|
|
c.AbortWithError(http.StatusInternalServerError, fmt.Errorf("failed to load asset data: %v", err))
|
|
return
|
|
}
|
|
//// Load the asset data from the database
|
|
//err = a.db.Query("SELECT * FROM assets WHERE qr = ?").
|
|
// Bind(*vm.Qr).
|
|
// ScanColumns(&columns).
|
|
// ScanSingle(&vm.Qr, &vm.Type, &vm.Values)
|
|
//if err != nil {
|
|
// c.AbortWithError(http.StatusInternalServerError, fmt.Errorf("failed to load asset: %v", err))
|
|
// return
|
|
//
|
|
//}
|
|
|
|
//err = a.db.Query("SELECT name, type, brand, description FROM assets WHERE qr = ?").
|
|
// Bind(*vm.Qr).
|
|
// ScanSingle(&vm.AssetName, &vm.Type, &vm.AssetBrand, &vm.AssetDescription)
|
|
//if err != nil {
|
|
// c.AbortWithError(http.StatusInternalServerError, err)
|
|
// return
|
|
//}
|
|
//if vm.Type == "ram" {
|
|
// err = a.db.Query("SELECT type, capacity FROM info_ram WHERE asset = ?").
|
|
// Bind(*vm.Qr).
|
|
// ScanSingle(&vm.RamType, &vm.RamCapacity)
|
|
// if err != nil {
|
|
// c.AbortWithError(http.StatusInternalServerError, err)
|
|
// return
|
|
// }
|
|
//} else if vm.Type == "hdd" {
|
|
// err = a.db.Query("SELECT capacity, type, form_factor, connection, rpm FROM info_hdd WHERE asset = ?").
|
|
// Bind(*vm.Qr).
|
|
// ScanSingle(&vm.HddCapacity, &vm.HddType, &vm.HddFormFactor, &vm.HddConnection, &vm.HddRpm)
|
|
// if err != nil {
|
|
// c.AbortWithError(http.StatusInternalServerError, err)
|
|
// return
|
|
// }
|
|
//}
|
|
}
|
|
|
|
c.HTML(http.StatusOK, "create_device", vm)
|
|
}
|
|
|
|
func (a *App) pullDataFromDb(qr int, assetType string, vm *CreateDeviceVM) error {
|
|
var fields []string
|
|
assetDescriptor := getAssetTypeById(assetType)
|
|
for _, field := range assetDescriptor.Fields {
|
|
fields = append(fields, field.Id)
|
|
}
|
|
values := make([]any, len(fields))
|
|
|
|
query := fmt.Sprintf("SELECT %s FROM %s WHERE qr = ?", strings.Join(fields, ", "), assetDescriptor.Table)
|
|
err := a.db.Query(query).Bind(qr).ScanSingle(values...)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to load asset data: %v", err)
|
|
}
|
|
|
|
for i, field := range fields {
|
|
vm.SetValue(assetType, field, fmt.Sprintf("%v", values[i]))
|
|
}
|
|
|
|
return err
|
|
}
|