Merge 1ed0fe3 into 83e9692

cmd/cli/main.go

@@ -41,6 +41,12 @@ var (
 	plotting = flag.Bool("plotting", false, "enable images in all notifications")
 )
 
+var (
+	userDel  = flag.String("user-del", "", "Delete all contacts and subscriptions for a user")
+	fromUser = flag.String("from-user", "", "Transfer subscriptions and contacts from user.")
+	toUser   = flag.String("to-user", "", "Transfer subscriptions and contacts to user.")
+)
+
 func main() {
 	logger, dataBase := initApp()
 
@@ -71,6 +77,17 @@ func main() {
 			logger.Errorf("Failed to enable images in all notifications")
 		}
 	}
+	if *toUser != "" {
+		if err := transferUserSubscriptionsAndContacts(dataBase, *fromUser, *toUser); err != nil {
+			logger.Error(err)
+		}
+	}
+
+	if *userDel != "" {
+		if err := deleteUser(dataBase, *userDel); err != nil {
+			logger.Error(err)
+		}
+	}
 }
 
 func initApp() (moira.Logger, moira.Database) {

cmd/cli/user.go

@@ -0,0 +1,58 @@
+package main
+
+import (
+	"github.com/moira-alert/moira"
+)
+
+func transferUserSubscriptionsAndContacts(database moira.Database, from, to string) error {
+	contactIDs, err := database.GetUserContactIDs(from)
+	if err != nil {
+		return err
+	}
+	contacts, err := database.GetContacts(contactIDs)
+	if err != nil {
+		return err
+	}
+	for _, contact := range contacts {
+		contact.User = to
+		if err = database.SaveContact(contact); err != nil {
+			return err
+		}
+	}
+
+	subscriptionIDs, err := database.GetUserSubscriptionIDs(from)
+	if err != nil {
+		return err
+	}
+	subscriptions, err := database.GetSubscriptions(subscriptionIDs)
+	if err != nil {
+		return err
+	}
+	for _, subscription := range subscriptions {
+		subscription.User = to
+	}
+	return database.SaveSubscriptions(subscriptions)
+}
+
+func deleteUser(database moira.Database, user string) error {
+	subscriptionIDs, err := database.GetUserSubscriptionIDs(user)
+	if err != nil {
+		return err
+	}
+	for _, subscriptionID := range subscriptionIDs {
+		if err = database.RemoveSubscription(subscriptionID); err != nil {
+			return err
+		}
+	}
+
+	contactIDs, err := database.GetUserContactIDs(user)
+	if err != nil {
+		return err
+	}
+	for _, contactID := range contactIDs {
+		if err := database.RemoveContact(contactID); err != nil {
+			return err
+		}
+	}
+	return nil
+}