add warehouse_request_source and warehouse_request_source_2 fields to deliveries table with validation and normalization logic

2026-05-04 17:32:03 +06:00
parent f9c54b5172
commit c39bde0b10
13 changed files with 305 additions and 165 deletions
--- a/backend/internal/db/migrations/000005_add_warehouse_request_source.down.sql
+++ b/backend/internal/db/migrations/000005_add_warehouse_request_source.down.sql
@@ -0,0 +1,2 @@
+ALTER TABLE deliveries DROP COLUMN warehouse_request_source_2;
+ALTER TABLE deliveries DROP COLUMN warehouse_request_source;
--- a/backend/internal/db/migrations/000005_add_warehouse_request_source.up.sql
+++ b/backend/internal/db/migrations/000005_add_warehouse_request_source.up.sql
@@ -0,0 +1,2 @@
+ALTER TABLE deliveries ADD COLUMN warehouse_request_source varchar(20);
+ALTER TABLE deliveries ADD COLUMN warehouse_request_source_2 varchar(20);
--- a/backend/internal/db/queries/query.sql
+++ b/backend/internal/db/queries/query.sql
@@ -30,11 +30,11 @@ SELECT * FROM deliveries WHERE date = $1;
 
 -- name: CreateDelivery :one
 INSERT INTO deliveries (
-    date, pickup_location, pickup_location_2, product_name, product_name_2,
+    date, pickup_location, pickup_location_2, warehouse_request_source, warehouse_request_source_2, product_name, product_name_2,
    customer_name, address, street, house, apartment, entrance, floor,
    phone, additional_phone, has_elevator, service_info, comment
 )
-VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17)
+VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18, $19)
 RETURNING *;

 -- name: GetDeliveryByID :one
@@ -48,22 +48,24 @@ UPDATE deliveries SET
    date = $1,
    pickup_location = $2,
    pickup_location_2 = $3,
-    product_name = $4,
-    product_name_2 = $5,
-    customer_name = $6,
-    address = $7,
-    street = $8,
-    house = $9,
-    apartment = $10,
-    entrance = $11,
-    floor = $12,
-    phone = $13,
-    additional_phone = $14,
-    has_elevator = $15,
-    service_info = $16,
-    comment = $17,
+    warehouse_request_source = $4,
+    warehouse_request_source_2 = $5,
+    product_name = $6,
+    product_name_2 = $7,
+    customer_name = $8,
+    address = $9,
+    street = $10,
+    house = $11,
+    apartment = $12,
+    entrance = $13,
+    floor = $14,
+    phone = $15,
+    additional_phone = $16,
+    has_elevator = $17,
+    service_info = $18,
+    comment = $19,
    updated_at = NOW()
-WHERE id = $18;
+WHERE id = $20;

 -- name: GetDeliveryCount :many
 SELECT COUNT(*) as count, date FROM deliveries
--- a/backend/internal/db/sqlc/models.go
+++ b/backend/internal/db/sqlc/models.go
@@ -9,27 +9,29 @@ import (
 )

 type Delivery struct {
-	ID              pgtype.UUID        `db:"id" json:"id"`
-	Date            pgtype.Date        `db:"date" json:"date"`
-	PickupLocation  string             `db:"pickup_location" json:"pickup_location"`
-	ProductName     string             `db:"product_name" json:"product_name"`
-	Address         string             `db:"address" json:"address"`
-	Phone           string             `db:"phone" json:"phone"`
-	AdditionalPhone pgtype.Text        `db:"additional_phone" json:"additional_phone"`
-	HasElevator     bool               `db:"has_elevator" json:"has_elevator"`
-	Comment         pgtype.Text        `db:"comment" json:"comment"`
-	Status          string             `db:"status" json:"status"`
-	CreatedAt       pgtype.Timestamptz `db:"created_at" json:"created_at"`
-	UpdatedAt       pgtype.Timestamptz `db:"updated_at" json:"updated_at"`
-	CustomerName    string             `db:"customer_name" json:"customer_name"`
-	ServiceInfo     pgtype.Text        `db:"service_info" json:"service_info"`
-	PickupLocation2 pgtype.Text        `db:"pickup_location_2" json:"pickup_location_2"`
-	ProductName2    pgtype.Text        `db:"product_name_2" json:"product_name_2"`
-	Street          string             `db:"street" json:"street"`
-	House           string             `db:"house" json:"house"`
-	Apartment       pgtype.Text        `db:"apartment" json:"apartment"`
-	Entrance        pgtype.Text        `db:"entrance" json:"entrance"`
-	Floor           pgtype.Text        `db:"floor" json:"floor"`
+	ID                      pgtype.UUID        `db:"id" json:"id"`
+	Date                    pgtype.Date        `db:"date" json:"date"`
+	PickupLocation          string             `db:"pickup_location" json:"pickup_location"`
+	ProductName             string             `db:"product_name" json:"product_name"`
+	Address                 string             `db:"address" json:"address"`
+	Phone                   string             `db:"phone" json:"phone"`
+	AdditionalPhone         pgtype.Text        `db:"additional_phone" json:"additional_phone"`
+	HasElevator             bool               `db:"has_elevator" json:"has_elevator"`
+	Comment                 pgtype.Text        `db:"comment" json:"comment"`
+	Status                  string             `db:"status" json:"status"`
+	CreatedAt               pgtype.Timestamptz `db:"created_at" json:"created_at"`
+	UpdatedAt               pgtype.Timestamptz `db:"updated_at" json:"updated_at"`
+	CustomerName            string             `db:"customer_name" json:"customer_name"`
+	ServiceInfo             pgtype.Text        `db:"service_info" json:"service_info"`
+	PickupLocation2         pgtype.Text        `db:"pickup_location_2" json:"pickup_location_2"`
+	ProductName2            pgtype.Text        `db:"product_name_2" json:"product_name_2"`
+	Street                  string             `db:"street" json:"street"`
+	House                   string             `db:"house" json:"house"`
+	Apartment               pgtype.Text        `db:"apartment" json:"apartment"`
+	Entrance                pgtype.Text        `db:"entrance" json:"entrance"`
+	Floor                   pgtype.Text        `db:"floor" json:"floor"`
+	WarehouseRequestSource  pgtype.Text        `db:"warehouse_request_source" json:"warehouse_request_source"`
+	WarehouseRequestSource2 pgtype.Text        `db:"warehouse_request_source_2" json:"warehouse_request_source_2"`
 }

 type User struct {
--- a/backend/internal/db/sqlc/query.sql.go
+++ b/backend/internal/db/sqlc/query.sql.go
@@ -13,32 +13,34 @@ import (

 const createDelivery = `-- name: CreateDelivery :one
 INSERT INTO deliveries (
-    date, pickup_location, pickup_location_2, product_name, product_name_2,
+    date, pickup_location, pickup_location_2, warehouse_request_source, warehouse_request_source_2, product_name, product_name_2,
    customer_name, address, street, house, apartment, entrance, floor,
    phone, additional_phone, has_elevator, service_info, comment
 )
-VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17)
-RETURNING id, date, pickup_location, product_name, address, phone, additional_phone, has_elevator, comment, status, created_at, updated_at, customer_name, service_info, pickup_location_2, product_name_2, street, house, apartment, entrance, floor
+VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18, $19)
+RETURNING id, date, pickup_location, product_name, address, phone, additional_phone, has_elevator, comment, status, created_at, updated_at, customer_name, service_info, pickup_location_2, product_name_2, street, house, apartment, entrance, floor, warehouse_request_source, warehouse_request_source_2
 `

 type CreateDeliveryParams struct {
-	Date            pgtype.Date `db:"date" json:"date"`
-	PickupLocation  string      `db:"pickup_location" json:"pickup_location"`
-	PickupLocation2 pgtype.Text `db:"pickup_location_2" json:"pickup_location_2"`
-	ProductName     string      `db:"product_name" json:"product_name"`
-	ProductName2    pgtype.Text `db:"product_name_2" json:"product_name_2"`
-	CustomerName    string      `db:"customer_name" json:"customer_name"`
-	Address         string      `db:"address" json:"address"`
-	Street          string      `db:"street" json:"street"`
-	House           string      `db:"house" json:"house"`
-	Apartment       pgtype.Text `db:"apartment" json:"apartment"`
-	Entrance        pgtype.Text `db:"entrance" json:"entrance"`
-	Floor           pgtype.Text `db:"floor" json:"floor"`
-	Phone           string      `db:"phone" json:"phone"`
-	AdditionalPhone pgtype.Text `db:"additional_phone" json:"additional_phone"`
-	HasElevator     bool        `db:"has_elevator" json:"has_elevator"`
-	ServiceInfo     pgtype.Text `db:"service_info" json:"service_info"`
-	Comment         pgtype.Text `db:"comment" json:"comment"`
+	Date                    pgtype.Date `db:"date" json:"date"`
+	PickupLocation          string      `db:"pickup_location" json:"pickup_location"`
+	PickupLocation2         pgtype.Text `db:"pickup_location_2" json:"pickup_location_2"`
+	WarehouseRequestSource  pgtype.Text `db:"warehouse_request_source" json:"warehouse_request_source"`
+	WarehouseRequestSource2 pgtype.Text `db:"warehouse_request_source_2" json:"warehouse_request_source_2"`
+	ProductName             string      `db:"product_name" json:"product_name"`
+	ProductName2            pgtype.Text `db:"product_name_2" json:"product_name_2"`
+	CustomerName            string      `db:"customer_name" json:"customer_name"`
+	Address                 string      `db:"address" json:"address"`
+	Street                  string      `db:"street" json:"street"`
+	House                   string      `db:"house" json:"house"`
+	Apartment               pgtype.Text `db:"apartment" json:"apartment"`
+	Entrance                pgtype.Text `db:"entrance" json:"entrance"`
+	Floor                   pgtype.Text `db:"floor" json:"floor"`
+	Phone                   string      `db:"phone" json:"phone"`
+	AdditionalPhone         pgtype.Text `db:"additional_phone" json:"additional_phone"`
+	HasElevator             bool        `db:"has_elevator" json:"has_elevator"`
+	ServiceInfo             pgtype.Text `db:"service_info" json:"service_info"`
+	Comment                 pgtype.Text `db:"comment" json:"comment"`
 }

 func (q *Queries) CreateDelivery(ctx context.Context, arg CreateDeliveryParams) (Delivery, error) {
@@ -46,6 +48,8 @@ func (q *Queries) CreateDelivery(ctx context.Context, arg CreateDeliveryParams)
 		arg.Date,
 		arg.PickupLocation,
 		arg.PickupLocation2,
+		arg.WarehouseRequestSource,
+		arg.WarehouseRequestSource2,
 		arg.ProductName,
 		arg.ProductName2,
 		arg.CustomerName,
@@ -84,6 +88,8 @@ func (q *Queries) CreateDelivery(ctx context.Context, arg CreateDeliveryParams)
 		&i.Apartment,
 		&i.Entrance,
 		&i.Floor,
+		&i.WarehouseRequestSource,
+		&i.WarehouseRequestSource2,
 	)
 	return i, err
 }
@@ -123,7 +129,7 @@ func (q *Queries) DeleteDelivery(ctx context.Context, id pgtype.UUID) error {
 }

 const getDeliveriesByDate = `-- name: GetDeliveriesByDate :many
-SELECT id, date, pickup_location, product_name, address, phone, additional_phone, has_elevator, comment, status, created_at, updated_at, customer_name, service_info, pickup_location_2, product_name_2, street, house, apartment, entrance, floor FROM deliveries WHERE date = $1
+SELECT id, date, pickup_location, product_name, address, phone, additional_phone, has_elevator, comment, status, created_at, updated_at, customer_name, service_info, pickup_location_2, product_name_2, street, house, apartment, entrance, floor, warehouse_request_source, warehouse_request_source_2 FROM deliveries WHERE date = $1
 `

 func (q *Queries) GetDeliveriesByDate(ctx context.Context, date pgtype.Date) ([]Delivery, error) {
@@ -157,6 +163,8 @@ func (q *Queries) GetDeliveriesByDate(ctx context.Context, date pgtype.Date) ([]
 			&i.Apartment,
 			&i.Entrance,
 			&i.Floor,
+			&i.WarehouseRequestSource,
+			&i.WarehouseRequestSource2,
 		); err != nil {
 			return nil, err
 		}
@@ -169,7 +177,7 @@ func (q *Queries) GetDeliveriesByDate(ctx context.Context, date pgtype.Date) ([]
 }

 const getDeliveryByID = `-- name: GetDeliveryByID :one
-SELECT id, date, pickup_location, product_name, address, phone, additional_phone, has_elevator, comment, status, created_at, updated_at, customer_name, service_info, pickup_location_2, product_name_2, street, house, apartment, entrance, floor FROM deliveries WHERE id = $1
+SELECT id, date, pickup_location, product_name, address, phone, additional_phone, has_elevator, comment, status, created_at, updated_at, customer_name, service_info, pickup_location_2, product_name_2, street, house, apartment, entrance, floor, warehouse_request_source, warehouse_request_source_2 FROM deliveries WHERE id = $1
 `

 func (q *Queries) GetDeliveryByID(ctx context.Context, id pgtype.UUID) (Delivery, error) {
@@ -197,6 +205,8 @@ func (q *Queries) GetDeliveryByID(ctx context.Context, id pgtype.UUID) (Delivery
 		&i.Apartment,
 		&i.Entrance,
 		&i.Floor,
+		&i.WarehouseRequestSource,
+		&i.WarehouseRequestSource2,
 	)
 	return i, err
 }
@@ -293,43 +303,47 @@ UPDATE deliveries SET
    date = $1,
    pickup_location = $2,
    pickup_location_2 = $3,
-    product_name = $4,
-    product_name_2 = $5,
-    customer_name = $6,
-    address = $7,
-    street = $8,
-    house = $9,
-    apartment = $10,
-    entrance = $11,
-    floor = $12,
-    phone = $13,
-    additional_phone = $14,
-    has_elevator = $15,
-    service_info = $16,
-    comment = $17,
+    warehouse_request_source = $4,
+    warehouse_request_source_2 = $5,
+    product_name = $6,
+    product_name_2 = $7,
+    customer_name = $8,
+    address = $9,
+    street = $10,
+    house = $11,
+    apartment = $12,
+    entrance = $13,
+    floor = $14,
+    phone = $15,
+    additional_phone = $16,
+    has_elevator = $17,
+    service_info = $18,
+    comment = $19,
    updated_at = NOW()
-WHERE id = $18
+WHERE id = $20
 `

 type UpdateDeliveryParams struct {
-	Date            pgtype.Date `db:"date" json:"date"`
-	PickupLocation  string      `db:"pickup_location" json:"pickup_location"`
-	PickupLocation2 pgtype.Text `db:"pickup_location_2" json:"pickup_location_2"`
-	ProductName     string      `db:"product_name" json:"product_name"`
-	ProductName2    pgtype.Text `db:"product_name_2" json:"product_name_2"`
-	CustomerName    string      `db:"customer_name" json:"customer_name"`
-	Address         string      `db:"address" json:"address"`
-	Street          string      `db:"street" json:"street"`
-	House           string      `db:"house" json:"house"`
-	Apartment       pgtype.Text `db:"apartment" json:"apartment"`
-	Entrance        pgtype.Text `db:"entrance" json:"entrance"`
-	Floor           pgtype.Text `db:"floor" json:"floor"`
-	Phone           string      `db:"phone" json:"phone"`
-	AdditionalPhone pgtype.Text `db:"additional_phone" json:"additional_phone"`
-	HasElevator     bool        `db:"has_elevator" json:"has_elevator"`
-	ServiceInfo     pgtype.Text `db:"service_info" json:"service_info"`
-	Comment         pgtype.Text `db:"comment" json:"comment"`
-	ID              pgtype.UUID `db:"id" json:"id"`
+	Date                    pgtype.Date `db:"date" json:"date"`
+	PickupLocation          string      `db:"pickup_location" json:"pickup_location"`
+	PickupLocation2         pgtype.Text `db:"pickup_location_2" json:"pickup_location_2"`
+	WarehouseRequestSource  pgtype.Text `db:"warehouse_request_source" json:"warehouse_request_source"`
+	WarehouseRequestSource2 pgtype.Text `db:"warehouse_request_source_2" json:"warehouse_request_source_2"`
+	ProductName             string      `db:"product_name" json:"product_name"`
+	ProductName2            pgtype.Text `db:"product_name_2" json:"product_name_2"`
+	CustomerName            string      `db:"customer_name" json:"customer_name"`
+	Address                 string      `db:"address" json:"address"`
+	Street                  string      `db:"street" json:"street"`
+	House                   string      `db:"house" json:"house"`
+	Apartment               pgtype.Text `db:"apartment" json:"apartment"`
+	Entrance                pgtype.Text `db:"entrance" json:"entrance"`
+	Floor                   pgtype.Text `db:"floor" json:"floor"`
+	Phone                   string      `db:"phone" json:"phone"`
+	AdditionalPhone         pgtype.Text `db:"additional_phone" json:"additional_phone"`
+	HasElevator             bool        `db:"has_elevator" json:"has_elevator"`
+	ServiceInfo             pgtype.Text `db:"service_info" json:"service_info"`
+	Comment                 pgtype.Text `db:"comment" json:"comment"`
+	ID                      pgtype.UUID `db:"id" json:"id"`
 }

 func (q *Queries) UpdateDelivery(ctx context.Context, arg UpdateDeliveryParams) error {
@@ -337,6 +351,8 @@ func (q *Queries) UpdateDelivery(ctx context.Context, arg UpdateDeliveryParams)
 		arg.Date,
 		arg.PickupLocation,
 		arg.PickupLocation2,
+		arg.WarehouseRequestSource,
+		arg.WarehouseRequestSource2,
 		arg.ProductName,
 		arg.ProductName2,
 		arg.CustomerName,
--- a/backend/internal/delivery/handler.go
+++ b/backend/internal/delivery/handler.go
@@ -1,6 +1,7 @@
 package delivery

 import (
+	"errors"
 	"net/http"
 	"time"

@@ -16,23 +17,25 @@ type Handler struct {

 // DeliveryRequest represents the request body for creating or updating a delivery
 type DeliveryRequest struct {
-	Date            string  `json:"date" binding:"required"` // DD-MM-YYYY
-	PickupLocation  string  `json:"pickup_location" binding:"required,oneof=warehouse symbat nursaya galaktika"`
-	PickupLocation2 *string `json:"pickup_location_2" binding:"omitempty,oneof=warehouse symbat nursaya galaktika"`
-	ProductName     string  `json:"product_name" binding:"required"`
-	ProductName2    *string `json:"product_name_2"`
-	CustomerName    string  `json:"customer_name" binding:"required"`
-	Address         string  `json:"address" binding:"required"`
-	Street          string  `json:"street" binding:"required"`
-	House           string  `json:"house" binding:"required"`
-	Apartment       *string `json:"apartment"`
-	Entrance        *string `json:"entrance"`
-	Floor           *string `json:"floor"`
-	Phone           string  `json:"phone" binding:"required"`
-	AdditionalPhone *string `json:"additional_phone"`
-	HasElevator     bool    `json:"has_elevator"`
-	ServiceInfo     *string `json:"service_info"`
-	Comment         string  `json:"comment"`
+	Date                    string  `json:"date" binding:"required"` // DD-MM-YYYY
+	PickupLocation          string  `json:"pickup_location" binding:"required,oneof=warehouse symbat nursaya galaktika"`
+	PickupLocation2         *string `json:"pickup_location_2" binding:"omitempty,oneof=warehouse symbat nursaya galaktika"`
+	WarehouseRequestSource  *string `json:"warehouse_request_source" binding:"omitempty,oneof=symbat nursaya galaktika"`
+	WarehouseRequestSource2 *string `json:"warehouse_request_source_2" binding:"omitempty,oneof=symbat nursaya galaktika"`
+	ProductName             string  `json:"product_name" binding:"required"`
+	ProductName2            *string `json:"product_name_2"`
+	CustomerName            string  `json:"customer_name" binding:"required"`
+	Address                 string  `json:"address" binding:"required"`
+	Street                  string  `json:"street" binding:"required"`
+	House                   string  `json:"house" binding:"required"`
+	Apartment               *string `json:"apartment"`
+	Entrance                *string `json:"entrance"`
+	Floor                   *string `json:"floor"`
+	Phone                   string  `json:"phone" binding:"required"`
+	AdditionalPhone         *string `json:"additional_phone"`
+	HasElevator             bool    `json:"has_elevator"`
+	ServiceInfo             *string `json:"service_info"`
+	Comment                 string  `json:"comment"`
 }

 func NewHandler(queries *sqlc.Queries) *Handler {
@@ -97,6 +100,10 @@ func (h *Handler) CreateDelivery(c *gin.Context) {
 		c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid request body", "details": err.Error()})
 		return
 	}
+	if err := normalizeWarehouseRequestSources(&req); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}

 	// Parse date from DD-MM-YYYY
 	t, err := parseDate(req.Date)
@@ -106,23 +113,25 @@ func (h *Handler) CreateDelivery(c *gin.Context) {
 	}

 	params := sqlc.CreateDeliveryParams{
-		Date:            pgtype.Date{Time: t, Valid: true},
-		PickupLocation:  req.PickupLocation,
-		PickupLocation2: pgtype.Text{String: derefString(req.PickupLocation2), Valid: req.PickupLocation2 != nil},
-		ProductName:     req.ProductName,
-		ProductName2:    pgtype.Text{String: derefString(req.ProductName2), Valid: req.ProductName2 != nil},
-		CustomerName:    req.CustomerName,
-		Address:         req.Address,
-		Street:          req.Street,
-		House:           req.House,
-		Apartment:       pgtype.Text{String: derefString(req.Apartment), Valid: req.Apartment != nil},
-		Entrance:        pgtype.Text{String: derefString(req.Entrance), Valid: req.Entrance != nil},
-		Floor:           pgtype.Text{String: derefString(req.Floor), Valid: req.Floor != nil},
-		Phone:           req.Phone,
-		AdditionalPhone: pgtype.Text{String: derefString(req.AdditionalPhone), Valid: req.AdditionalPhone != nil},
-		HasElevator:     req.HasElevator,
-		ServiceInfo:     pgtype.Text{String: derefString(req.ServiceInfo), Valid: req.ServiceInfo != nil},
-		Comment:         pgtype.Text{String: req.Comment, Valid: true},
+		Date:                    pgtype.Date{Time: t, Valid: true},
+		PickupLocation:          req.PickupLocation,
+		PickupLocation2:         pgtype.Text{String: derefString(req.PickupLocation2), Valid: req.PickupLocation2 != nil},
+		WarehouseRequestSource:  pgtype.Text{String: derefString(req.WarehouseRequestSource), Valid: req.WarehouseRequestSource != nil},
+		WarehouseRequestSource2: pgtype.Text{String: derefString(req.WarehouseRequestSource2), Valid: req.WarehouseRequestSource2 != nil},
+		ProductName:             req.ProductName,
+		ProductName2:            pgtype.Text{String: derefString(req.ProductName2), Valid: req.ProductName2 != nil},
+		CustomerName:            req.CustomerName,
+		Address:                 req.Address,
+		Street:                  req.Street,
+		House:                   req.House,
+		Apartment:               pgtype.Text{String: derefString(req.Apartment), Valid: req.Apartment != nil},
+		Entrance:                pgtype.Text{String: derefString(req.Entrance), Valid: req.Entrance != nil},
+		Floor:                   pgtype.Text{String: derefString(req.Floor), Valid: req.Floor != nil},
+		Phone:                   req.Phone,
+		AdditionalPhone:         pgtype.Text{String: derefString(req.AdditionalPhone), Valid: req.AdditionalPhone != nil},
+		HasElevator:             req.HasElevator,
+		ServiceInfo:             pgtype.Text{String: derefString(req.ServiceInfo), Valid: req.ServiceInfo != nil},
+		Comment:                 pgtype.Text{String: req.Comment, Valid: true},
 	}
 	res, err := h.queries.CreateDelivery(c.Request.Context(), params)
 	if err != nil {
@@ -141,6 +150,10 @@ func (h *Handler) UpdateDelivery(c *gin.Context) {
 		c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid request body", "details": err.Error()})
 		return
 	}
+	if err := normalizeWarehouseRequestSources(&req); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}

 	id := c.Param("id")

@@ -161,24 +174,26 @@ func (h *Handler) UpdateDelivery(c *gin.Context) {
 	}

 	if err := h.queries.UpdateDelivery(c.Request.Context(), sqlc.UpdateDeliveryParams{
-		ID:              pgtype.UUID{Bytes: parsedID, Valid: true},
-		Date:            pgtype.Date{Time: t, Valid: true},
-		PickupLocation:  req.PickupLocation,
-		PickupLocation2: pgtype.Text{String: derefString(req.PickupLocation2), Valid: req.PickupLocation2 != nil},
-		ProductName:     req.ProductName,
-		ProductName2:    pgtype.Text{String: derefString(req.ProductName2), Valid: req.ProductName2 != nil},
-		CustomerName:    req.CustomerName,
-		Address:         req.Address,
-		Street:          req.Street,
-		House:           req.House,
-		Apartment:       pgtype.Text{String: derefString(req.Apartment), Valid: req.Apartment != nil},
-		Entrance:        pgtype.Text{String: derefString(req.Entrance), Valid: req.Entrance != nil},
-		Floor:           pgtype.Text{String: derefString(req.Floor), Valid: req.Floor != nil},
-		Phone:           req.Phone,
-		AdditionalPhone: pgtype.Text{String: derefString(req.AdditionalPhone), Valid: req.AdditionalPhone != nil},
-		HasElevator:     req.HasElevator,
-		ServiceInfo:     pgtype.Text{String: derefString(req.ServiceInfo), Valid: req.ServiceInfo != nil},
-		Comment:         pgtype.Text{String: req.Comment, Valid: true},
+		ID:                      pgtype.UUID{Bytes: parsedID, Valid: true},
+		Date:                    pgtype.Date{Time: t, Valid: true},
+		PickupLocation:          req.PickupLocation,
+		PickupLocation2:         pgtype.Text{String: derefString(req.PickupLocation2), Valid: req.PickupLocation2 != nil},
+		WarehouseRequestSource:  pgtype.Text{String: derefString(req.WarehouseRequestSource), Valid: req.WarehouseRequestSource != nil},
+		WarehouseRequestSource2: pgtype.Text{String: derefString(req.WarehouseRequestSource2), Valid: req.WarehouseRequestSource2 != nil},
+		ProductName:             req.ProductName,
+		ProductName2:            pgtype.Text{String: derefString(req.ProductName2), Valid: req.ProductName2 != nil},
+		CustomerName:            req.CustomerName,
+		Address:                 req.Address,
+		Street:                  req.Street,
+		House:                   req.House,
+		Apartment:               pgtype.Text{String: derefString(req.Apartment), Valid: req.Apartment != nil},
+		Entrance:                pgtype.Text{String: derefString(req.Entrance), Valid: req.Entrance != nil},
+		Floor:                   pgtype.Text{String: derefString(req.Floor), Valid: req.Floor != nil},
+		Phone:                   req.Phone,
+		AdditionalPhone:         pgtype.Text{String: derefString(req.AdditionalPhone), Valid: req.AdditionalPhone != nil},
+		HasElevator:             req.HasElevator,
+		ServiceInfo:             pgtype.Text{String: derefString(req.ServiceInfo), Valid: req.ServiceInfo != nil},
+		Comment:                 pgtype.Text{String: req.Comment, Valid: true},
 	}); err != nil {
 		c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update delivery", "details": err.Error()})
 		return
@@ -266,3 +281,19 @@ func derefString(s *string) string {
 	}
 	return *s
 }
+
+func normalizeWarehouseRequestSources(req *DeliveryRequest) error {
+	if req.PickupLocation == "warehouse" && req.WarehouseRequestSource == nil {
+		return errors.New("warehouse_request_source is required when pickup_location is warehouse")
+	}
+	if req.PickupLocation != "warehouse" {
+		req.WarehouseRequestSource = nil
+	}
+	if req.PickupLocation2 != nil && *req.PickupLocation2 == "warehouse" && req.WarehouseRequestSource2 == nil {
+		return errors.New("warehouse_request_source_2 is required when pickup_location_2 is warehouse")
+	}
+	if req.PickupLocation2 == nil || *req.PickupLocation2 != "warehouse" {
+		req.WarehouseRequestSource2 = nil
+	}
+	return nil
+}