Luciano Milani
79 lines
2.4 KiB
Go
79 lines
2.4 KiB
Go
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, issued_at AS 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, issued_at AS 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, issued_at AS 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
|
|
}
|