Inicial

backend/internal/invoice/repository.go

@@ -0,0 +1,78 @@
+package invoice
+
+import (
+	"context"
+	"fmt"
+	"time"
+
+	"github.com/jackc/pgx/v5"
+	"github.com/jackc/pgx/v5/pgxpool"
+)
+
+type Invoice struct {
+	ID          string    `json:"id"`
+	WorkOrderID string    `json:"work_order_id"`
+	Type        string    `json:"type"`
+	Number      int       `json:"number"`
+	PDFPath     string    `json:"pdf_path"`
+	IssuedAt    time.Time `json:"issued_at"`
+	CreatedAt   time.Time `json:"created_at"`
+}
+
+func ListInvoices(ctx context.Context, conn *pgxpool.Conn) ([]*Invoice, error) {
+	rows, err := conn.Query(ctx,
+		`SELECT id, work_order_id, type, number, COALESCE(pdf_path,''), issued_at, created_at
+		 FROM invoices ORDER BY issued_at DESC`)
+	if err != nil {
+		return nil, fmt.Errorf("invoice: list: %w", err)
+	}
+	defer rows.Close()
+	var list []*Invoice
+	for rows.Next() {
+		var inv Invoice
+		if err := rows.Scan(&inv.ID, &inv.WorkOrderID, &inv.Type, &inv.Number,
+			&inv.PDFPath, &inv.IssuedAt, &inv.CreatedAt); err != nil {
+			return nil, fmt.Errorf("invoice: scan: %w", err)
+		}
+		list = append(list, &inv)
+	}
+	return list, rows.Err()
+}
+
+func GetInvoice(ctx context.Context, conn *pgxpool.Conn, id string) (*Invoice, error) {
+	row := conn.QueryRow(ctx,
+		`SELECT id, work_order_id, type, number, COALESCE(pdf_path,''), issued_at, created_at
+		 FROM invoices WHERE id = $1`, id)
+	var inv Invoice
+	err := row.Scan(&inv.ID, &inv.WorkOrderID, &inv.Type, &inv.Number,
+		&inv.PDFPath, &inv.IssuedAt, &inv.CreatedAt)
+	if err == pgx.ErrNoRows {
+		return nil, nil
+	}
+	if err != nil {
+		return nil, fmt.Errorf("invoice: get: %w", err)
+	}
+	return &inv, nil
+}
+
+func CreateInvoice(ctx context.Context, conn *pgxpool.Conn, woID, docType string) (*Invoice, error) {
+	row := conn.QueryRow(ctx,
+		`INSERT INTO invoices (work_order_id, type, issued_at)
+		 VALUES ($1, $2, NOW())
+		 RETURNING id, work_order_id, type, number, COALESCE(pdf_path,''), issued_at, created_at`,
+		woID, docType)
+	var inv Invoice
+	if err := row.Scan(&inv.ID, &inv.WorkOrderID, &inv.Type, &inv.Number,
+		&inv.PDFPath, &inv.IssuedAt, &inv.CreatedAt); err != nil {
+		return nil, fmt.Errorf("invoice: create: %w", err)
+	}
+	return &inv, nil
+}
+
+func SetPDFPath(ctx context.Context, conn *pgxpool.Conn, id, path string) error {
+	_, err := conn.Exec(ctx, `UPDATE invoices SET pdf_path = $2 WHERE id = $1`, id, path)
+	if err != nil {
+		return fmt.Errorf("invoice: set pdf path: %w", err)
+	}
+	return nil
+}