package handler import ( "context" "database/sql" "encoding/json" "errors" "fmt" "io" "net/http" "vctp/db" queries "vctp/db/queries" models "vctp/server/models" ) // VmImport is used for bulk import of existing VMs func (h *Handler) VmImport(w http.ResponseWriter, r *http.Request) { reqBody, err := io.ReadAll(r.Body) if err != nil { h.Logger.Error("Invalid data received", "error", err) fmt.Fprintf(w, "Invalid data received") w.WriteHeader(http.StatusInternalServerError) return } else { h.Logger.Debug("received input data", "length", len(reqBody)) } // Decode the JSON body into CloudEventReceived struct var inData models.ImportReceived if err := json.Unmarshal(reqBody, &inData); err != nil { h.Logger.Error("unable to decode json", "error", err) http.Error(w, "Invalid JSON body", http.StatusBadRequest) return } else { h.Logger.Debug("successfully decoded JSON") prettyPrint(inData) } ctx := context.Background() // TODO - query Inventory table for this VM before adding it h.Logger.Debug("Checking inventory table for VM record") invParams := queries.GetInventoryVmIdParams{ VmId: sql.NullString{String: inData.VmId, Valid: inData.VmId != ""}, DatacenterName: sql.NullString{String: inData.Datacenter, Valid: inData.Datacenter != ""}, } _, err = h.Database.Queries().GetInventoryVmId(ctx, invParams) if err != nil { if errors.Is(err, sql.ErrNoRows) { // do the insert // Create an instance of CreateInventoryParams var params queries.CreateInventoryParams // Convert vmModel to CreateInventoryParams using the utility function db.ConvertToSQLParams(&inData, ¶ms) prettyPrint(params) // Insert the new inventory record into the database result, err := h.Database.Queries().CreateInventory(ctx, params) if err != nil { h.Logger.Error("unable to perform database insert", "error", err) } else { h.Logger.Debug("created database record", "insert_result", result) } } else { h.Logger.Error("unable to check inventory for vm", "error", err, "vm_id", inData.VmId, "datacenter_name", inData.Datacenter) } } else { h.Logger.Info("not adding vm to inventory table since record alraedy exists", "vm_id", inData.VmId, "datacenter_name", inData.Datacenter) } //h.Logger.Debug("received import request", "body", string(reqBody)) w.WriteHeader(http.StatusOK) fmt.Fprintf(w, "Import Request (%d): %v\n", len(reqBody), string(reqBody)) }