[31/8/22] add user restore

Author: ockibagusp

controllers/admin_controller.go

@@ -150,3 +150,64 @@ func (controller *Controller) DeletePermanentlyByID(c echo.Context) error {
 	log.Info("END request method GET for admin delete permanently by id: [+]success")
 	return c.Redirect(http.StatusMovedPermanently, "/admin/delete-permanently")
 }
+
+/*
+ * Restore User
+ *
+ * @target: [Admin] Restore User
+ * @method: GET
+ * @route: /admin/restore/:id
+ */
+func (controller *Controller) RestoreUser(c echo.Context) error {
+	session, _ := middleware.GetAuth(c)
+	log := log.WithFields(log.Fields{
+		"username": session.Values["username"],
+		"route":    c.Path(),
+	})
+	log.Info("START request method GET for admin restore")
+
+	is_auth_type := session.Values["is_auth_type"]
+	if is_auth_type == -1 {
+		log.Warn("for GET to admin restore without no-session [@route: /login]")
+		middleware.SetFlashError(c, "login process failed!")
+		log.Warn("END request method GET for admin restore: [-]failure")
+		return c.Redirect(http.StatusFound, "/login")
+	}
+
+	id, _ := strconv.Atoi(c.Param("id"))
+
+	// why?
+	// delete permanently not for admin
+	if id == 1 {
+		log.Warn("END request method GET for admin restore [admin]: [-]failure")
+		// HTTP response status: 403 Forbidden
+		return c.HTML(http.StatusForbidden, "403 Forbidden")
+	}
+
+	if !middleware.IsAdmin(is_auth_type) {
+		log.Warn("END request method GET for admin restore: [-]failure")
+		// HTTP response status: 404 Not Found
+		return c.HTML(http.StatusNotFound, "404 Not Found")
+	}
+
+	user, err := (models.User{}).UnscopedFirstUserByID(controller.DB, id)
+	if err != nil {
+		log.Warnf("for GET to admin restore without models.User{}.FirstByID() errors: `%v`", err)
+		log.Warn("END request method GET for admin restore: [-]failure")
+		// HTTP response status: 404 Not Found
+		return c.HTML(http.StatusNotFound, err.Error())
+	}
+
+	if err := user.Restore(controller.DB, id); err != nil {
+		log.Warnf("for GET to admin restore without models.User{}.Restore() errors: `%v`", err)
+		log.Warn("END request method GET for admin restore: [-]failure")
+		// HTTP response status: 403 Forbidden
+		return c.HTML(http.StatusForbidden, err.Error())
+	}
+
+	middleware.SetFlashSuccess(c, fmt.Sprintf("success restore user: %s!", user.Username))
+
+	// restore admin
+	log.Info("END request method GET for admin restore: [+]success")
+	return c.Redirect(http.StatusMovedPermanently, "/admin/delete-permanently")
+}

models/user.go

@@ -217,6 +217,26 @@ func (user User) FindDeleteAll(db *gorm.DB, admin_or_user ...string) ([]User, er
 	return users, nil
 }
 
+// User: Restore
+func (user User) Restore(db *gorm.DB, id int) error {
+	tx := db.Begin()
+	var count int64
+	// if tx.Unscoped().Select("id").First(&user).Error != nil {}
+	if tx.Unscoped().Select("id").First(&user).Count(&count); count != 1 {
+		tx.Rollback()
+		return errors.New("User Not Found")
+	}
+
+	// if tx.Model(&user).Unscoped().Where("id = ?", id).Update(...).Error; err != nil {}
+	if err := tx.Model(&user).Unscoped().Where("id = ?", id).Update("deleted_at", nil).First(&user).Error; err != nil {
+		tx.Rollback()
+		return err
+	}
+	tx.Commit()
+
+	return nil
+}
+
 // User: Delete Permanently
 func (user User) DeletePermanently(db *gorm.DB, id int) error {
 	tx := db.Begin()

router/router.go

@@ -69,6 +69,8 @@ func New(controllers *controllers.Controller) (router *echo.Echo) {
 	// admin
 	router.GET("/admin/delete-permanently", controllers.DeletePermanently).
 		Name = "/admin/delete-permanently get"
+	router.GET("/admin/restore/:id", controllers.RestoreUser).
+		Name = "/admin/restore/:id get"
 	// "/admin/delete-permanently/:id" can not
 	// "/admin/delete/permanently/:id" can
 	router.GET("/admin/delete/permanently/:id", controllers.DeletePermanentlyByID).