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 }