mirror of
https://github.com/balzack/databag.git
synced 2025-02-12 03:29:16 +00:00
196 lines
6.4 KiB
Go
196 lines
6.4 KiB
Go
package databag
|
|
|
|
import (
|
|
"databag/internal/store"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
"net/http"
|
|
)
|
|
|
|
//SetNodeConfig sets node configuration
|
|
func SetNodeConfig(w http.ResponseWriter, r *http.Request) {
|
|
|
|
// validate login
|
|
if code, err := ParamSessionToken(r); err != nil {
|
|
ErrResponse(w, code, err)
|
|
return
|
|
}
|
|
|
|
// update open access
|
|
updateAccess := r.FormValue("setOpenAccess") == "true"
|
|
|
|
// parse node config
|
|
var config NodeConfig
|
|
if err := ParseRequest(r, w, &config); err != nil {
|
|
ErrResponse(w, http.StatusBadRequest, err)
|
|
return
|
|
}
|
|
|
|
// store credentials
|
|
err := store.DB.Transaction(func(tx *gorm.DB) error {
|
|
|
|
// upsert domain config
|
|
if res := tx.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "config_id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"str_value"}),
|
|
}).Create(&store.Config{ConfigID: CNFDomain, StrValue: config.Domain}).Error; res != nil {
|
|
return res
|
|
}
|
|
|
|
// upsert account storage config
|
|
if res := tx.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "config_id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"num_value"}),
|
|
}).Create(&store.Config{ConfigID: CNFStorage, NumValue: config.AccountStorage}).Error; res != nil {
|
|
return res
|
|
}
|
|
|
|
// upsert enable image processing
|
|
if res := tx.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "config_id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"bool_value"}),
|
|
}).Create(&store.Config{ConfigID: CNFEnableImage, BoolValue: config.EnableImage}).Error; res != nil {
|
|
return res
|
|
}
|
|
|
|
// upsert enable audio processing
|
|
if res := tx.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "config_id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"bool_value"}),
|
|
}).Create(&store.Config{ConfigID: CNFEnableAudio, BoolValue: config.EnableAudio}).Error; res != nil {
|
|
return res
|
|
}
|
|
|
|
// upsert enable video processing
|
|
if res := tx.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "config_id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"bool_value"}),
|
|
}).Create(&store.Config{ConfigID: CNFEnableVideo, BoolValue: config.EnableVideo}).Error; res != nil {
|
|
return res
|
|
}
|
|
|
|
// upsert enable binary attachments
|
|
if res := tx.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "config_id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"bool_value"}),
|
|
}).Create(&store.Config{ConfigID: CNFEnableBinary, BoolValue: config.EnableBinary}).Error; res != nil {
|
|
return res
|
|
}
|
|
|
|
// upsert allow unsealed channels
|
|
if res := tx.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "config_id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"bool_value"}),
|
|
}).Create(&store.Config{ConfigID: CNFAllowUnsealed, BoolValue: config.AllowUnsealed}).Error; res != nil {
|
|
return res
|
|
}
|
|
|
|
// upsert push supported
|
|
if res := tx.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "config_id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"bool_value"}),
|
|
}).Create(&store.Config{ConfigID: CNFPushSupported, BoolValue: config.PushSupported}).Error; res != nil {
|
|
return res
|
|
}
|
|
|
|
// upsert key type
|
|
if res := tx.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "config_id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"str_value"}),
|
|
}).Create(&store.Config{ConfigID: CNFKeyType, StrValue: config.KeyType}).Error; res != nil {
|
|
return res
|
|
}
|
|
|
|
// upsert ice supported
|
|
if res := tx.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "config_id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"bool_value"}),
|
|
}).Create(&store.Config{ConfigID: CNFEnableIce, BoolValue: config.EnableIce}).Error; res != nil {
|
|
return res
|
|
}
|
|
|
|
// upsert ice service name
|
|
if res := tx.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "config_id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"str_value"}),
|
|
}).Create(&store.Config{ConfigID: CNFIceService, StrValue: config.IceService}).Error; res != nil {
|
|
return res
|
|
}
|
|
|
|
// upsert key type
|
|
if res := tx.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "config_id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"str_value"}),
|
|
}).Create(&store.Config{ConfigID: CNFIceUrl, StrValue: config.IceURL}).Error; res != nil {
|
|
return res
|
|
}
|
|
|
|
// upsert key type
|
|
if res := tx.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "config_id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"str_value"}),
|
|
}).Create(&store.Config{ConfigID: CNFIceUsername, StrValue: config.IceUsername}).Error; res != nil {
|
|
return res
|
|
}
|
|
|
|
// upsert key type
|
|
if res := tx.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "config_id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"str_value"}),
|
|
}).Create(&store.Config{ConfigID: CNFIcePassword, StrValue: config.IcePassword}).Error; res != nil {
|
|
return res
|
|
}
|
|
|
|
if updateAccess {
|
|
// upsert enable open access
|
|
if res := tx.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "config_id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"bool_value"}),
|
|
}).Create(&store.Config{ConfigID: CNFEnableOpenAccess, BoolValue: config.EnableOpenAccess}).Error; res != nil {
|
|
return res
|
|
}
|
|
|
|
// upsert open access limit
|
|
if res := tx.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "config_id"}},
|
|
DoUpdates: clause.AssignmentColumns([]string{"num_value"}),
|
|
}).Create(&store.Config{ConfigID: CNFOpenAccessLimit, NumValue: config.OpenAccessLimit}).Error; res != nil {
|
|
return res
|
|
}
|
|
}
|
|
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
ErrResponse(w, http.StatusInternalServerError, err)
|
|
return
|
|
}
|
|
|
|
// increment revision of all account data
|
|
var accounts []*store.Account
|
|
if err := store.DB.Find(&accounts).Error; err != nil {
|
|
ErrMsg(err);
|
|
return
|
|
}
|
|
err = store.DB.Transaction(func(tx *gorm.DB) error {
|
|
for _, account := range accounts {
|
|
if res := tx.Model(account).Update("account_revision", account.AccountRevision+1).Error; res != nil {
|
|
return res
|
|
}
|
|
if res := tx.Model(account).Update("profile_revision", account.ProfileRevision+1).Error; res != nil {
|
|
return res
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
ErrMsg(err);
|
|
return
|
|
}
|
|
for _, account := range accounts {
|
|
SetStatus(account)
|
|
}
|
|
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|