CRUD for Posts(Final Part)

Author: Euler-B

database/postgres.go

@@ -112,6 +112,12 @@ func (repo *PostgresRepository) GetUserByEmail(ctx context.Context, email string
 func (repo *PostgresRepository) UpdatePost(ctx context.Context, post *models.Post) error {
 	_, err := repo.db.ExecContext(ctx, 
 		"UPDATE posts SET post_content = $1 WHERE id = $2 and user_id = $3", 
-		post.PostContent, post.Id, post.UserId )
+		post.PostContent, post.Id, post.UserId)
+	return err
+}
+
+func (repo *PostgresRepository) DeletePost(ctx context.Context, id string, userId string) error {
+	_, err := repo.db.ExecContext(ctx, "DELETE FROM posts WHERE id = $1 AND user_id = $2", 
+		id, userId)
 	return err
 }

handlers/posts.go

@@ -87,6 +87,7 @@ func GetPostByIdHandler(s server.Server) http.HandlerFunc {
 
 func UpdatePostHandler(s server.Server) http.HandlerFunc {
 	return func(w http.ResponseWriter, r *http.Request) {
+		params    := mux.Vars(r)
 		tokenString := strings.TrimSpace(r.Header.Get("Authorization")) // reto propuesto:
 		token, err := jwt.ParseWithClaims(tokenString, &models.AppClaims{}, // buscar una forma de no repetir tanto el mismo codigo para validar tokens
 			func(token *jwt.Token) (interface{}, error) {
@@ -97,7 +98,6 @@ func UpdatePostHandler(s server.Server) http.HandlerFunc {
 				return
 			}
 			if claims, ok := token.Claims.(*models.AppClaims); ok && token.Valid {
-				params    := mux.Vars(r)
 				var postRequest = UpsertPostRequest{}
 				err := json.NewDecoder(r.Body).Decode(&postRequest); 
 				if err != nil {
@@ -123,4 +123,33 @@ func UpdatePostHandler(s server.Server) http.HandlerFunc {
 				return
 			}
 	}
+}
+
+func DeletePostHandler(s server.Server) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		params    := mux.Vars(r)
+		tokenString := strings.TrimSpace(r.Header.Get("Authorization")) 
+		token, err := jwt.ParseWithClaims(tokenString, &models.AppClaims{},
+			func(token *jwt.Token) (interface{}, error) {
+				return []byte(s.Config().JWTSecret), nil
+			})
+			if err != nil {
+				http.Error(w, err.Error(), http.StatusUnauthorized)
+				return
+			}
+			if claims, ok := token.Claims.(*models.AppClaims); ok && token.Valid {
+				err = repository.DeletePost(r.Context(), params["id"], claims.UserId)
+				if err != nil {
+					http.Error(w, err.Error(), http.StatusInternalServerError)
+					return
+				}
+				w.Header().Set("Content-Type", "application/json")
+				json.NewEncoder(w).Encode(PostUpdateResponse{
+					Message: "Post Deleted",
+				})
+			} else {
+				http.Error(w, err.Error(), http.StatusInternalServerError)
+				return
+			}
+	}
 }

main.go

@@ -49,4 +49,5 @@ func BindRoutes(s server.Server, r *mux.Router) {
 	r.HandleFunc("/posts", handlers.InsertPostHandler(s)).Methods(http.MethodPost)
 	r.HandleFunc("/posts/{id}", handlers.GetPostByIdHandler(s)).Methods(http.MethodGet)
 	r.HandleFunc("/posts/{id}", handlers.UpdatePostHandler(s)).Methods(http.MethodPut)
+	r.HandleFunc("/posts/{id}", handlers.DeletePostHandler(s)).Methods(http.MethodDelete)
 }

repository/repository.go

@@ -14,6 +14,7 @@ type Repository interface {
 	InsertPost(ctx context.Context, post *models.Post) error
 	GetPostById(ctx context.Context, id string) (*models.Post, error)
 	UpdatePost(ctx context.Context, post *models.Post) error
+	DeletePost(ctx context.Context, id string, userId string) error
 	Close() error
 }
 
@@ -49,4 +50,8 @@ func GetPostById(ctx context.Context, id string) (*models.Post, error) {
 
 func UpdatePost(ctx context.Context, post *models.Post) error {
 	return implementation.UpdatePost(ctx, post)
+}
+
+func DeletePost(ctx context.Context, id string, userId string) error {
+	return implementation.DeletePost(ctx, id, userId) 
 }