mirror of
https://github.com/garraflavatra/rolens.git
synced 2025-01-18 21:17:59 +00:00
83 lines
2.3 KiB
Go
83 lines
2.3 KiB
Go
package app
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"github.com/ncruces/zenity"
|
|
"github.com/wailsapp/wails/v2/pkg/runtime"
|
|
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
mongoOptions "go.mongodb.org/mongo-driver/mongo/options"
|
|
)
|
|
|
|
func (a *App) UpdateItems(hostKey, dbKey, collKey string, formJson string) int64 {
|
|
var form struct {
|
|
Upsert bool `json:"upsert"`
|
|
Many bool `json:"many"`
|
|
Query string `json:"query"`
|
|
Parameters []struct {
|
|
Type string `json:"type"`
|
|
Value string `json:"value"`
|
|
} `json:"parameters"`
|
|
}
|
|
|
|
err := json.Unmarshal([]byte(formJson), &form)
|
|
if err != nil {
|
|
runtime.LogError(a.ctx, "Could not parse update form:")
|
|
runtime.LogError(a.ctx, err.Error())
|
|
zenity.Error(err.Error(), zenity.Title("Could not parse form"), zenity.ErrorIcon)
|
|
return 0
|
|
}
|
|
|
|
client, ctx, close, err := a.connectToHost(hostKey)
|
|
if err != nil {
|
|
return 0
|
|
}
|
|
|
|
defer close()
|
|
var query bson.M
|
|
update := bson.M{}
|
|
|
|
err = bson.UnmarshalExtJSON([]byte(form.Query), true, &query)
|
|
if err != nil {
|
|
runtime.LogWarning(a.ctx, "Invalid update query:")
|
|
runtime.LogWarning(a.ctx, form.Query)
|
|
runtime.LogWarning(a.ctx, err.Error())
|
|
zenity.Error(err.Error(), zenity.Title("Invalid update query"), zenity.ErrorIcon)
|
|
return 0
|
|
}
|
|
|
|
for _, param := range form.Parameters {
|
|
var unmarshalled bson.M
|
|
err = json.Unmarshal([]byte(param.Value), &unmarshalled)
|
|
if err == nil {
|
|
update[param.Type] = unmarshalled
|
|
} else {
|
|
runtime.LogWarning(a.ctx, "Invalid update parameter value:")
|
|
runtime.LogWarning(a.ctx, param.Value)
|
|
runtime.LogWarning(a.ctx, err.Error())
|
|
zenity.Error(err.Error(), zenity.Title("Invalid update query"), zenity.ErrorIcon)
|
|
return 0
|
|
}
|
|
}
|
|
|
|
var result *mongo.UpdateResult
|
|
options := mongoOptions.Update().SetUpsert(form.Upsert)
|
|
|
|
if form.Many {
|
|
result, err = client.Database(dbKey).Collection(collKey).UpdateMany(ctx, query, update, options)
|
|
} else {
|
|
result, err = client.Database(dbKey).Collection(collKey).UpdateOne(ctx, query, update, options)
|
|
}
|
|
|
|
if err != nil {
|
|
runtime.LogWarning(a.ctx, "Encountered an error while performing update:")
|
|
runtime.LogWarning(a.ctx, err.Error())
|
|
zenity.Error(err.Error(), zenity.Title("Error while performing update"), zenity.ErrorIcon)
|
|
return 0
|
|
}
|
|
|
|
return result.ModifiedCount
|
|
}
|