Mirror/filebrowser
1
0
Fork 0
mirror of https://github.com/filebrowser/filebrowser.git synced 2024-06-07 23:00:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add replace opt

Browse Source
This commit is contained in:
Henrique Dias 2019-01-08 12:02:49 +00:00
parent fc06f642f2
commit 76bf72902d
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
cmd/users_import.go
View File

@ -12,6 +12,7 @@ import (
func init() { func init() {
usersCmd.AddCommand(usersImportCmd) usersCmd.AddCommand(usersImportCmd)
usersImportCmd.Flags().Bool("overwrite", false, "overwrite users with the same id/username combo") usersImportCmd.Flags().Bool("overwrite", false, "overwrite users with the same id/username combo")
usersImportCmd.Flags().Bool("replace", false, "replace the entire user base")
} }
var usersImportCmd = &cobra.Command{ var usersImportCmd = &cobra.Command{
@ -32,10 +33,23 @@ var usersImportCmd = &cobra.Command{
checkErr(err) checkErr(err)
} }
if mustGetBool(cmd, "replace") {
oldUsers, err := d.store.Users.Gets("")
checkErr(err)
err = marshal("users.backup.json", list)
checkErr(err)
for _, user := range oldUsers {
err = d.store.Users.Delete(user.ID)
checkErr(err)
}
}
overwrite := mustGetBool(cmd, "overwrite") overwrite := mustGetBool(cmd, "overwrite")
for _, user := range list { for _, user := range list {
old, err := d.store.Users.Get("", user.ID) onDB, err := d.store.Users.Get("", user.ID)
// User exists in DB. // User exists in DB.
if err == nil { if err == nil {
@ -46,7 +60,7 @@ var usersImportCmd = &cobra.Command{
// If the usernames mismatch, check if there is another one in the DB // If the usernames mismatch, check if there is another one in the DB
// with the new username. If there is, print an error and cancel the // with the new username. If there is, print an error and cancel the
// operation // operation
if user.Username != old.Username { if user.Username != onDB.Username {
conflictuous, err := d.store.Users.Get("", user.Username) conflictuous, err := d.store.Users.Get("", user.Username)
if err == nil { if err == nil {
checkErr(usernameConflictError(user.Username, conflictuous.ID, user.ID)) checkErr(usernameConflictError(user.Username, conflictuous.ID, user.ID))