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 := ParamAdminToken(r); err != nil { ErrResponse(w, code, err) return } // 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 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 push 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 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 } 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 } } return nil }) if err != nil { ErrMsg(err); return } for _, account := range accounts { SetStatus(account) } w.WriteHeader(http.StatusOK) }