129 lines
3.1 KiB
Go
129 lines
3.1 KiB
Go
package postgres
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"errors"
|
|
"time"
|
|
|
|
"github.com/rede5/gohorsejobs/backend/internal/core/domain/entity"
|
|
)
|
|
|
|
type CompanyRepository struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
func NewCompanyRepository(db *sql.DB) *CompanyRepository {
|
|
return &CompanyRepository{db: db}
|
|
}
|
|
|
|
func (r *CompanyRepository) Save(ctx context.Context, company *entity.Company) (*entity.Company, error) {
|
|
// companies table uses UUID id, DB generates it
|
|
query := `
|
|
INSERT INTO companies (name, slug, type, document, email, phone, website, address, description, verified, active, created_at, updated_at)
|
|
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13)
|
|
RETURNING id
|
|
`
|
|
|
|
slug := company.Slug
|
|
// Fallback slug generation if empty
|
|
if slug == "" {
|
|
slug = company.Name
|
|
}
|
|
|
|
var compType = company.Type
|
|
if compType == "" {
|
|
compType = "company"
|
|
}
|
|
|
|
var id string
|
|
err := r.db.QueryRowContext(ctx, query,
|
|
company.Name,
|
|
slug,
|
|
compType,
|
|
company.Document,
|
|
company.Contact, // email
|
|
company.Phone,
|
|
company.Website,
|
|
company.Address,
|
|
company.Description,
|
|
true, // verified
|
|
company.Status == "ACTIVE",
|
|
company.CreatedAt,
|
|
company.UpdatedAt,
|
|
).Scan(&id)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
company.ID = id
|
|
return company, nil
|
|
}
|
|
|
|
func (r *CompanyRepository) FindByID(ctx context.Context, id string) (*entity.Company, error) {
|
|
query := `SELECT id, name, document, email,
|
|
CASE WHEN active THEN 'ACTIVE' ELSE 'INACTIVE' END as status,
|
|
created_at, updated_at
|
|
FROM companies WHERE id = $1`
|
|
|
|
row := r.db.QueryRowContext(ctx, query, id)
|
|
|
|
c := &entity.Company{}
|
|
var dbID string
|
|
err := row.Scan(&dbID, &c.Name, &c.Document, &c.Contact, &c.Status, &c.CreatedAt, &c.UpdatedAt)
|
|
if err == sql.ErrNoRows {
|
|
return nil, errors.New("company not found")
|
|
}
|
|
c.ID = dbID
|
|
return c, err
|
|
}
|
|
|
|
func (r *CompanyRepository) Update(ctx context.Context, company *entity.Company) (*entity.Company, error) {
|
|
company.UpdatedAt = time.Now()
|
|
|
|
query := `
|
|
UPDATE companies
|
|
SET name=$1, document=$2, email=$3, active=$4, updated_at=$5
|
|
WHERE id=$6
|
|
`
|
|
_, err := r.db.ExecContext(ctx, query,
|
|
company.Name,
|
|
company.Document,
|
|
company.Contact,
|
|
company.Status == "ACTIVE",
|
|
company.UpdatedAt,
|
|
company.ID,
|
|
)
|
|
return company, err
|
|
}
|
|
|
|
func (r *CompanyRepository) Delete(ctx context.Context, id string) error {
|
|
query := `DELETE FROM companies WHERE id = $1`
|
|
_, err := r.db.ExecContext(ctx, query, id)
|
|
return err
|
|
}
|
|
|
|
func (r *CompanyRepository) FindAll(ctx context.Context) ([]*entity.Company, error) {
|
|
query := `SELECT id, name, document, email,
|
|
CASE WHEN active THEN 'ACTIVE' ELSE 'INACTIVE' END as status,
|
|
created_at, updated_at
|
|
FROM companies
|
|
WHERE type = 'company' OR type = 'COMPANY'`
|
|
rows, err := r.db.QueryContext(ctx, query)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
var companies []*entity.Company
|
|
for rows.Next() {
|
|
c := &entity.Company{}
|
|
var dbID string
|
|
if err := rows.Scan(&dbID, &c.Name, &c.Document, &c.Contact, &c.Status, &c.CreatedAt, &c.UpdatedAt); err != nil {
|
|
return nil, err
|
|
}
|
|
c.ID = dbID
|
|
companies = append(companies, c)
|
|
}
|
|
return companies, nil
|
|
}
|