Inicial
This commit is contained in:
@@ -0,0 +1,183 @@
|
||||
package client
|
||||
|
||||
import (
|
||||
"errors"
|
||||
|
||||
"github.com/gofiber/fiber/v2"
|
||||
"github.com/jackc/pgx/v5"
|
||||
"github.com/techxcar/backend/internal/auth"
|
||||
"github.com/techxcar/backend/pkg/database"
|
||||
)
|
||||
|
||||
func RegisterRoutes(app *fiber.App, db *database.DB, secret string) {
|
||||
rw := []fiber.Handler{
|
||||
auth.RequireAuth(secret),
|
||||
auth.RequireRole("tenant_admin", "manager", "technician"),
|
||||
auth.TenantMiddleware(db),
|
||||
}
|
||||
write := []fiber.Handler{
|
||||
auth.RequireAuth(secret),
|
||||
auth.RequireRole("tenant_admin", "manager"),
|
||||
auth.TenantMiddleware(db),
|
||||
}
|
||||
|
||||
app.Get("/api/v1/clients", append(rw, listClientsH())...)
|
||||
app.Post("/api/v1/clients", append(write, createClientH())...)
|
||||
app.Get("/api/v1/clients/:id", append(rw, getClientH())...)
|
||||
app.Put("/api/v1/clients/:id", append(write, updateClientH())...)
|
||||
app.Delete("/api/v1/clients/:id", append(write, deleteClientH())...)
|
||||
|
||||
app.Get("/api/v1/clients/:id/vehicles", append(rw, listVehiclesH())...)
|
||||
app.Post("/api/v1/clients/:id/vehicles", append(write, createVehicleH())...)
|
||||
app.Put("/api/v1/vehicles/:id", append(write, updateVehicleH())...)
|
||||
}
|
||||
|
||||
func listClientsH() fiber.Handler {
|
||||
return func(c *fiber.Ctx) error {
|
||||
conn := auth.GetConn(c)
|
||||
list, err := ListClients(c.Context(), conn)
|
||||
if err != nil {
|
||||
return fiber.NewError(500, "erro ao listar clientes")
|
||||
}
|
||||
if list == nil {
|
||||
list = []*Client{}
|
||||
}
|
||||
return c.JSON(fiber.Map{"data": list, "error": nil})
|
||||
}
|
||||
}
|
||||
|
||||
func getClientH() fiber.Handler {
|
||||
return func(c *fiber.Ctx) error {
|
||||
conn := auth.GetConn(c)
|
||||
cl, err := GetClient(c.Context(), conn, c.Params("id"))
|
||||
if err != nil {
|
||||
if errors.Is(err, pgx.ErrNoRows) {
|
||||
return fiber.NewError(404, "cliente não encontrado")
|
||||
}
|
||||
return fiber.NewError(500, "erro ao obter cliente")
|
||||
}
|
||||
return c.JSON(fiber.Map{"data": cl, "error": nil})
|
||||
}
|
||||
}
|
||||
|
||||
type clientBody struct {
|
||||
Name string `json:"name"`
|
||||
NIF string `json:"nif"`
|
||||
Phone string `json:"phone"`
|
||||
Email string `json:"email"`
|
||||
Address string `json:"address"`
|
||||
Notes string `json:"notes"`
|
||||
}
|
||||
|
||||
func createClientH() fiber.Handler {
|
||||
return func(c *fiber.Ctx) error {
|
||||
var b clientBody
|
||||
if err := c.BodyParser(&b); err != nil {
|
||||
return fiber.NewError(400, "corpo inválido")
|
||||
}
|
||||
if b.Name == "" {
|
||||
return fiber.NewError(400, "nome é obrigatório")
|
||||
}
|
||||
conn := auth.GetConn(c)
|
||||
cl, err := CreateClient(c.Context(), conn, b.Name, b.NIF, b.Phone, b.Email, b.Address, b.Notes)
|
||||
if err != nil {
|
||||
return fiber.NewError(500, "erro ao criar cliente")
|
||||
}
|
||||
return c.Status(201).JSON(fiber.Map{"data": cl, "error": nil})
|
||||
}
|
||||
}
|
||||
|
||||
func updateClientH() fiber.Handler {
|
||||
return func(c *fiber.Ctx) error {
|
||||
var b clientBody
|
||||
if err := c.BodyParser(&b); err != nil {
|
||||
return fiber.NewError(400, "corpo inválido")
|
||||
}
|
||||
if b.Name == "" {
|
||||
return fiber.NewError(400, "nome é obrigatório")
|
||||
}
|
||||
conn := auth.GetConn(c)
|
||||
cl, err := UpdateClient(c.Context(), conn, c.Params("id"), b.Name, b.NIF, b.Phone, b.Email, b.Address, b.Notes)
|
||||
if err != nil {
|
||||
if errors.Is(err, pgx.ErrNoRows) {
|
||||
return fiber.NewError(404, "cliente não encontrado")
|
||||
}
|
||||
return fiber.NewError(500, "erro ao actualizar cliente")
|
||||
}
|
||||
return c.JSON(fiber.Map{"data": cl, "error": nil})
|
||||
}
|
||||
}
|
||||
|
||||
func deleteClientH() fiber.Handler {
|
||||
return func(c *fiber.Ctx) error {
|
||||
conn := auth.GetConn(c)
|
||||
if err := DeleteClient(c.Context(), conn, c.Params("id")); err != nil {
|
||||
return fiber.NewError(500, "erro ao eliminar cliente")
|
||||
}
|
||||
return c.SendStatus(204)
|
||||
}
|
||||
}
|
||||
|
||||
func listVehiclesH() fiber.Handler {
|
||||
return func(c *fiber.Ctx) error {
|
||||
conn := auth.GetConn(c)
|
||||
list, err := ListVehiclesByClient(c.Context(), conn, c.Params("id"))
|
||||
if err != nil {
|
||||
return fiber.NewError(500, "erro ao listar veículos")
|
||||
}
|
||||
if list == nil {
|
||||
list = []*Vehicle{}
|
||||
}
|
||||
return c.JSON(fiber.Map{"data": list, "error": nil})
|
||||
}
|
||||
}
|
||||
|
||||
type vehicleBody struct {
|
||||
Plate string `json:"plate"`
|
||||
Brand string `json:"brand"`
|
||||
Model string `json:"model"`
|
||||
Year int `json:"year"`
|
||||
VIN string `json:"vin"`
|
||||
FuelType string `json:"fuel_type"`
|
||||
Mileage int `json:"mileage"`
|
||||
Notes string `json:"notes"`
|
||||
}
|
||||
|
||||
func createVehicleH() fiber.Handler {
|
||||
return func(c *fiber.Ctx) error {
|
||||
var b vehicleBody
|
||||
if err := c.BodyParser(&b); err != nil {
|
||||
return fiber.NewError(400, "corpo inválido")
|
||||
}
|
||||
if b.Plate == "" {
|
||||
return fiber.NewError(400, "matrícula é obrigatória")
|
||||
}
|
||||
conn := auth.GetConn(c)
|
||||
v, err := CreateVehicle(c.Context(), conn, c.Params("id"), b.Plate, b.Brand, b.Model, b.Year, b.VIN, b.FuelType, b.Notes, b.Mileage)
|
||||
if err != nil {
|
||||
return fiber.NewError(500, "erro ao criar veículo")
|
||||
}
|
||||
return c.Status(201).JSON(fiber.Map{"data": v, "error": nil})
|
||||
}
|
||||
}
|
||||
|
||||
func updateVehicleH() fiber.Handler {
|
||||
return func(c *fiber.Ctx) error {
|
||||
var b vehicleBody
|
||||
if err := c.BodyParser(&b); err != nil {
|
||||
return fiber.NewError(400, "corpo inválido")
|
||||
}
|
||||
if b.Plate == "" {
|
||||
return fiber.NewError(400, "matrícula é obrigatória")
|
||||
}
|
||||
conn := auth.GetConn(c)
|
||||
v, err := UpdateVehicle(c.Context(), conn, c.Params("id"), b.Plate, b.Brand, b.Model, b.Year, b.VIN, b.FuelType, b.Notes, b.Mileage)
|
||||
if err != nil {
|
||||
if errors.Is(err, pgx.ErrNoRows) {
|
||||
return fiber.NewError(404, "veículo não encontrado")
|
||||
}
|
||||
return fiber.NewError(500, "erro ao actualizar veículo")
|
||||
}
|
||||
return c.JSON(fiber.Map{"data": v, "error": nil})
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,166 @@
|
||||
package client
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
"github.com/jackc/pgx/v5/pgxpool"
|
||||
)
|
||||
|
||||
type Client struct {
|
||||
ID string `json:"id"`
|
||||
Name string `json:"name"`
|
||||
NIF string `json:"nif"`
|
||||
Phone string `json:"phone"`
|
||||
Email string `json:"email"`
|
||||
Address string `json:"address"`
|
||||
Notes string `json:"notes"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
}
|
||||
|
||||
type Vehicle struct {
|
||||
ID string `json:"id"`
|
||||
ClientID *string `json:"client_id"`
|
||||
Plate string `json:"plate"`
|
||||
Brand string `json:"brand"`
|
||||
Model string `json:"model"`
|
||||
Year *int `json:"year"`
|
||||
VIN string `json:"vin"`
|
||||
Mileage *int `json:"mileage"`
|
||||
FuelType string `json:"fuel_type"`
|
||||
Notes string `json:"notes"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
}
|
||||
|
||||
func ListClients(ctx context.Context, conn *pgxpool.Conn) ([]*Client, error) {
|
||||
rows, err := conn.Query(ctx, `
|
||||
SELECT id, name, COALESCE(nif,''), COALESCE(phone,''), COALESCE(email,''),
|
||||
COALESCE(address,''), COALESCE(notes,''), created_at, updated_at
|
||||
FROM clients ORDER BY name`)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
var list []*Client
|
||||
for rows.Next() {
|
||||
var c Client
|
||||
if err := rows.Scan(&c.ID, &c.Name, &c.NIF, &c.Phone, &c.Email,
|
||||
&c.Address, &c.Notes, &c.CreatedAt, &c.UpdatedAt); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
list = append(list, &c)
|
||||
}
|
||||
return list, rows.Err()
|
||||
}
|
||||
|
||||
func GetClient(ctx context.Context, conn *pgxpool.Conn, id string) (*Client, error) {
|
||||
var c Client
|
||||
err := conn.QueryRow(ctx, `
|
||||
SELECT id, name, COALESCE(nif,''), COALESCE(phone,''), COALESCE(email,''),
|
||||
COALESCE(address,''), COALESCE(notes,''), created_at, updated_at
|
||||
FROM clients WHERE id = $1`, id).
|
||||
Scan(&c.ID, &c.Name, &c.NIF, &c.Phone, &c.Email,
|
||||
&c.Address, &c.Notes, &c.CreatedAt, &c.UpdatedAt)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &c, nil
|
||||
}
|
||||
|
||||
func CreateClient(ctx context.Context, conn *pgxpool.Conn, name, nif, phone, email, address, notes string) (*Client, error) {
|
||||
var c Client
|
||||
err := conn.QueryRow(ctx, `
|
||||
INSERT INTO clients (name, nif, phone, email, address, notes)
|
||||
VALUES ($1, NULLIF($2,''), NULLIF($3,''), NULLIF($4,''), NULLIF($5,''), NULLIF($6,''))
|
||||
RETURNING id, name, COALESCE(nif,''), COALESCE(phone,''), COALESCE(email,''),
|
||||
COALESCE(address,''), COALESCE(notes,''), created_at, updated_at`,
|
||||
name, nif, phone, email, address, notes).
|
||||
Scan(&c.ID, &c.Name, &c.NIF, &c.Phone, &c.Email,
|
||||
&c.Address, &c.Notes, &c.CreatedAt, &c.UpdatedAt)
|
||||
return &c, err
|
||||
}
|
||||
|
||||
func UpdateClient(ctx context.Context, conn *pgxpool.Conn, id, name, nif, phone, email, address, notes string) (*Client, error) {
|
||||
var c Client
|
||||
err := conn.QueryRow(ctx, `
|
||||
UPDATE clients SET name=$2, nif=NULLIF($3,''), phone=NULLIF($4,''), email=NULLIF($5,''),
|
||||
address=NULLIF($6,''), notes=NULLIF($7,''), updated_at=NOW()
|
||||
WHERE id=$1
|
||||
RETURNING id, name, COALESCE(nif,''), COALESCE(phone,''), COALESCE(email,''),
|
||||
COALESCE(address,''), COALESCE(notes,''), created_at, updated_at`,
|
||||
id, name, nif, phone, email, address, notes).
|
||||
Scan(&c.ID, &c.Name, &c.NIF, &c.Phone, &c.Email,
|
||||
&c.Address, &c.Notes, &c.CreatedAt, &c.UpdatedAt)
|
||||
return &c, err
|
||||
}
|
||||
|
||||
func DeleteClient(ctx context.Context, conn *pgxpool.Conn, id string) error {
|
||||
_, err := conn.Exec(ctx, `DELETE FROM clients WHERE id = $1`, id)
|
||||
return err
|
||||
}
|
||||
|
||||
func ListVehiclesByClient(ctx context.Context, conn *pgxpool.Conn, clientID string) ([]*Vehicle, error) {
|
||||
rows, err := conn.Query(ctx, `
|
||||
SELECT id, client_id, plate, brand, model, year, COALESCE(vin,''), mileage,
|
||||
COALESCE(fuel_type,''), COALESCE(notes,''), created_at, updated_at
|
||||
FROM vehicles WHERE client_id = $1 ORDER BY plate`, clientID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
var list []*Vehicle
|
||||
for rows.Next() {
|
||||
var v Vehicle
|
||||
if err := rows.Scan(&v.ID, &v.ClientID, &v.Plate, &v.Brand, &v.Model,
|
||||
&v.Year, &v.VIN, &v.Mileage, &v.FuelType, &v.Notes, &v.CreatedAt, &v.UpdatedAt); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
list = append(list, &v)
|
||||
}
|
||||
return list, rows.Err()
|
||||
}
|
||||
|
||||
func CreateVehicle(ctx context.Context, conn *pgxpool.Conn, clientID, plate, brand, model string, year int, vin, fuelType, notes string, mileage int) (*Vehicle, error) {
|
||||
var v Vehicle
|
||||
var yearPtr *int
|
||||
if year != 0 {
|
||||
yearPtr = &year
|
||||
}
|
||||
var mileagePtr *int
|
||||
if mileage != 0 {
|
||||
mileagePtr = &mileage
|
||||
}
|
||||
err := conn.QueryRow(ctx, `
|
||||
INSERT INTO vehicles (client_id, plate, brand, model, year, vin, fuel_type, mileage, notes)
|
||||
VALUES (NULLIF($1,'')::uuid, $2, $3, $4, $5, NULLIF($6,''), NULLIF($7,''), $8, NULLIF($9,''))
|
||||
RETURNING id, client_id, plate, brand, model, year, COALESCE(vin,''), mileage,
|
||||
COALESCE(fuel_type,''), COALESCE(notes,''), created_at, updated_at`,
|
||||
clientID, plate, brand, model, yearPtr, vin, fuelType, mileagePtr, notes).
|
||||
Scan(&v.ID, &v.ClientID, &v.Plate, &v.Brand, &v.Model,
|
||||
&v.Year, &v.VIN, &v.Mileage, &v.FuelType, &v.Notes, &v.CreatedAt, &v.UpdatedAt)
|
||||
return &v, err
|
||||
}
|
||||
|
||||
func UpdateVehicle(ctx context.Context, conn *pgxpool.Conn, id, plate, brand, model string, year int, vin, fuelType, notes string, mileage int) (*Vehicle, error) {
|
||||
var v Vehicle
|
||||
var yearPtr *int
|
||||
if year != 0 {
|
||||
yearPtr = &year
|
||||
}
|
||||
var mileagePtr *int
|
||||
if mileage != 0 {
|
||||
mileagePtr = &mileage
|
||||
}
|
||||
err := conn.QueryRow(ctx, `
|
||||
UPDATE vehicles SET plate=$2, brand=$3, model=$4,
|
||||
year=$5, vin=NULLIF($6,''), fuel_type=NULLIF($7,''), mileage=$8, notes=NULLIF($9,''), updated_at=NOW()
|
||||
WHERE id=$1
|
||||
RETURNING id, client_id, plate, brand, model, year, COALESCE(vin,''), mileage,
|
||||
COALESCE(fuel_type,''), COALESCE(notes,''), created_at, updated_at`,
|
||||
id, plate, brand, model, yearPtr, vin, fuelType, mileagePtr, notes).
|
||||
Scan(&v.ID, &v.ClientID, &v.Plate, &v.Brand, &v.Model,
|
||||
&v.Year, &v.VIN, &v.Mileage, &v.FuelType, &v.Notes, &v.CreatedAt, &v.UpdatedAt)
|
||||
return &v, err
|
||||
}
|
||||
@@ -0,0 +1,78 @@
|
||||
package client_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
"github.com/jackc/pgx/v5/pgxpool"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
"github.com/techxcar/backend/internal/client"
|
||||
)
|
||||
|
||||
func getTestConn(t *testing.T) *pgxpool.Conn {
|
||||
t.Helper()
|
||||
dsn := os.Getenv("TEST_DATABASE_URL")
|
||||
if dsn == "" {
|
||||
t.Skip("TEST_DATABASE_URL not set")
|
||||
}
|
||||
pool, err := pgxpool.New(context.Background(), dsn)
|
||||
require.NoError(t, err)
|
||||
t.Cleanup(func() { pool.Close() })
|
||||
|
||||
conn, err := pool.Acquire(context.Background())
|
||||
require.NoError(t, err)
|
||||
t.Cleanup(func() { conn.Release() })
|
||||
|
||||
_, err = conn.Exec(context.Background(), "SET search_path = tenant_test, public")
|
||||
require.NoError(t, err)
|
||||
return conn
|
||||
}
|
||||
|
||||
func TestClientCRUD(t *testing.T) {
|
||||
conn := getTestConn(t)
|
||||
ctx := context.Background()
|
||||
|
||||
c, err := client.CreateClient(ctx, conn, "João Silva", "123456789", "912345678", "joao@example.com", "Rua A", "")
|
||||
require.NoError(t, err)
|
||||
assert.NotEmpty(t, c.ID)
|
||||
assert.Equal(t, "João Silva", c.Name)
|
||||
|
||||
list, err := client.ListClients(ctx, conn)
|
||||
require.NoError(t, err)
|
||||
assert.GreaterOrEqual(t, len(list), 1)
|
||||
|
||||
got, err := client.GetClient(ctx, conn, c.ID)
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, c.ID, got.ID)
|
||||
|
||||
updated, err := client.UpdateClient(ctx, conn, c.ID, "João Santos", "987654321", "921000000", "joao2@example.com", "Rua B", "nota")
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, "João Santos", updated.Name)
|
||||
|
||||
err = client.DeleteClient(ctx, conn, c.ID)
|
||||
require.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestVehicleCRUD(t *testing.T) {
|
||||
conn := getTestConn(t)
|
||||
ctx := context.Background()
|
||||
|
||||
c, err := client.CreateClient(ctx, conn, "Test Client", "", "", "", "", "")
|
||||
require.NoError(t, err)
|
||||
|
||||
v, err := client.CreateVehicle(ctx, conn, c.ID, "AA-00-AA", "Toyota", "Yaris", 2020, "", 50000, "")
|
||||
require.NoError(t, err)
|
||||
assert.NotEmpty(t, v.ID)
|
||||
assert.Equal(t, "AA-00-AA", v.Plate)
|
||||
|
||||
list, err := client.ListVehiclesByClient(ctx, conn, c.ID)
|
||||
require.NoError(t, err)
|
||||
assert.Len(t, list, 1)
|
||||
|
||||
updated, err := client.UpdateVehicle(ctx, conn, v.ID, "BB-11-BB", "Toyota", "Yaris", 2021, "", 60000, "nota")
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, "BB-11-BB", updated.Plate)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user