TaskFlow/views/deleteViews.go

package views

/*
Holds the delete related view handlers
*/

import (
	"log"
	"net/http"
	"strconv"

	"github.com/thewhitetulip/Tasks/db"
	"github.com/thewhitetulip/Tasks/sessions"
	"github.com/thewhitetulip/Tasks/utils"
)

//TrashTaskFunc is used to populate the trash tasks
func TrashTaskFunc(w http.ResponseWriter, r *http.Request) {
	//for best UX we want the user to be returned to the page making
	//the delete transaction, we use the r.Referer() function to get the link
	redirectUrl := utils.GetRedirectUrl(r.Referer())

	if r.Method == "GET" && sessions.IsLoggedIn(r) {
		id, err := strconv.Atoi(r.URL.Path[len("/trash/"):])
		if err != nil {
			log.Println("TrashTaskFunc", err)
			message = "Incorrect command"
			http.Redirect(w, r, redirectUrl, http.StatusFound)
		} else {
			err = db.TrashTask(id)
			if err != nil {
				message = "Error trashing task"
			} else {
				message = "Task trashed"
			}
			http.Redirect(w, r, redirectUrl, http.StatusFound)
		}
	} else {
		message = "Method not allowed"
		http.Redirect(w, r, redirectUrl, http.StatusFound)
	}
}

//RestoreTaskFunc is used to restore task from trash, handles "/restore/" URL
func RestoreTaskFunc(w http.ResponseWriter, r *http.Request) {
	if r.Method == "GET" && sessions.IsLoggedIn(r) {
		id, err := strconv.Atoi(r.URL.Path[len("/restore/"):])
		if err != nil {
			log.Println(err)
			http.Redirect(w, r, "/deleted", http.StatusBadRequest)
		} else {
			err = db.RestoreTask(id)
			if err != nil {
				message = "Restore failed"
			} else {
				message = "Task restored"
			}
			http.Redirect(w, r, "/deleted/", http.StatusFound)
		}
	} else {
		message = "Method not allowed"
		http.Redirect(w, r, "/", http.StatusFound)
	}
}

//DeleteTaskFunc is used to delete a task, trash = move to recycle bin, delete = permanent delete
func DeleteTaskFunc(w http.ResponseWriter, r *http.Request) {
	if r.Method == "GET" && sessions.IsLoggedIn(r) {
		id := r.URL.Path[len("/delete/"):]
		if id == "all" {
			err := db.DeleteAll()
			if err != nil {
				message = "Error deleting tasks"
				http.Redirect(w, r, "/", http.StatusInternalServerError)
			}
			http.Redirect(w, r, "/", http.StatusFound)
		} else {
			id, err := strconv.Atoi(id)
			if err != nil {
				log.Println(err)
				http.Redirect(w, r, "/", http.StatusBadRequest)
			} else {
				err = db.DeleteTask(id)
				if err != nil {
					message = "Error deleting task"
				} else {
					message = "Task deleted"
				}
				http.Redirect(w, r, "/deleted", http.StatusFound)
			}
		}
	} else {
		message = "Method not allowed"
		http.Redirect(w, r, "/", http.StatusFound)
	}
}

//RestoreFromCompleteFunc restores the task from complete to pending
func RestoreFromCompleteFunc(w http.ResponseWriter, r *http.Request) {
	if r.Method == "GET" && sessions.IsLoggedIn(r) {
		id, err := strconv.Atoi(r.URL.Path[len("/incomplete/"):])
		if err != nil {
			log.Println(err)
			http.Redirect(w, r, "/completed", http.StatusBadRequest)
		} else {
			err = db.RestoreTaskFromComplete(id)
			if err != nil {
				message = "Restore failed"
			} else {
				message = "Task restored"
			}
			http.Redirect(w, r, "/completed", http.StatusFound)
		}
	} else {
		message = "Method not allowed"
		http.Redirect(w, r, "/completed", http.StatusFound)
	}
}

//DeleteCategoryFunc will delete any category
func DeleteCategoryFunc(w http.ResponseWriter, r *http.Request) {
	if r.Method == "GET" && sessions.IsLoggedIn(r) {
		categoryName := r.URL.Path[len("/del-category/"):]
		err := db.DeleteCategoryByName(categoryName)
		if err != nil {
			message = "error deleting category"
		} else {
			message = "Category " + categoryName + " deleted"
		}

		http.Redirect(w, r, "/", http.StatusFound)
	}
}

//DeleteCommentFunc will delete any category
func DeleteCommentFunc(w http.ResponseWriter, r *http.Request) {
	if r.Method == "GET" && sessions.IsLoggedIn(r) {
		id := r.URL.Path[len("/del-comment/"):]
		commentID, err := strconv.Atoi(id)
		if err != nil {
			http.Redirect(w, r, "/", http.StatusBadRequest)
			return
		}

		err = db.DeleteCommentByID(commentID)

		if err != nil {
			message = "comment not deleted"
		} else {
			message = "comment deleted"
		}

		http.Redirect(w, r, "/", http.StatusFound)
	}
}