Files

115 lines
2.5 KiB
Go

// TODO: READ AND UNDERSTAND
package main
import (
"context"
"encoding/json"
"net/http"
"strconv"
"github.com/go-chi/chi/v5"
)
// GET /tickets
func getTickets(w http.ResponseWriter, r *http.Request) {
rows, err := db.Query(context.Background(), "SELECT id, title, description, status, created_at, updated_at FROM tickets ORDER BY id;")
if err != nil {
http.Error(w, err.Error(), 500)
return
}
defer rows.Close()
var tickets []Ticket
for rows.Next() {
var t Ticket
rows.Scan(&t.Id, &t.Title, &t.Description, &t.Status, &t.CreatedAt, &t.UpdatedAt)
tickets = append(tickets, t)
}
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(tickets)
}
// POST /tickets
func createTicket(w http.ResponseWriter, r *http.Request) {
var t Ticket
err := json.NewDecoder(r.Body).Decode(&t)
if err != nil {
http.Error(w, "Invalid JSON", 400)
return
}
err = db.QueryRow(
context.Background(),
"INSERT INTO tickets(title, description) VALUES($1, $2) RETURNING id, status, created_at, updated_at;",
t.Title, t.Description,
).Scan(&t.Id, &t.Status, &t.CreatedAt, &t.UpdatedAt)
if err != nil {
http.Error(w, err.Error(), 500)
return
}
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(t)
}
// PATCH /tickets/{id}
func updateTicket(w http.ResponseWriter, r *http.Request) {
// get ID from URL
idParam := chi.URLParam(r, "id")
id, err := strconv.Atoi(idParam)
if err != nil {
http.Error(w, "Invalid ID", http.StatusBadRequest)
return
}
// parse request body
var status Status
err = json.NewDecoder(r.Body).Decode(&status)
if err != nil {
http.Error(w, "Invalid JSON", http.StatusBadRequest)
return
}
// update DB
commandTag, err := db.Exec(
context.Background(),
`UPDATE tickets
SET status=$1
WHERE id=$2`,
status.Status, id,
)
if err != nil {
http.Error(w, err.Error(), 500)
return
}
// check if row exists
if commandTag.RowsAffected() == 0 {
http.Error(w, "Ticket not found", http.StatusNotFound)
return
}
// fetch the updated row to return it
var t Ticket
err = db.QueryRow(
context.Background(),
`SELECT id, title, description, status, created_at, updated_at
FROM tickets
WHERE id=$1`,
id,
).Scan(&t.Id, &t.Title, &t.Description, &t.Status, &t.CreatedAt, &t.UpdatedAt)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
// return updated object
w.Header().Set("Content-Type", "application/json")
json.NewEncoder(w).Encode(t)
}