125 lines
3.3 KiB
Go
125 lines
3.3 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
|
|
dbgen "github.com/SamuelTariku/FortuneBet-Backend/gen/db"
|
|
"github.com/SamuelTariku/FortuneBet-Backend/internal/domain"
|
|
"github.com/jackc/pgx/v5/pgtype"
|
|
)
|
|
|
|
func convertCreateCompany(company domain.CreateCompany) dbgen.CreateCompanyParams {
|
|
return dbgen.CreateCompanyParams{
|
|
Name: company.Name,
|
|
AdminID: company.AdminID,
|
|
WalletID: company.WalletID,
|
|
}
|
|
}
|
|
|
|
func convertDBCompany(dbCompany dbgen.Company) domain.Company {
|
|
return domain.Company{
|
|
ID: dbCompany.ID,
|
|
Name: dbCompany.Name,
|
|
AdminID: dbCompany.AdminID,
|
|
WalletID: dbCompany.WalletID,
|
|
}
|
|
}
|
|
|
|
func convertDBCompanyDetails(dbCompany dbgen.CompaniesDetail) domain.GetCompany {
|
|
return domain.GetCompany{
|
|
ID: dbCompany.ID,
|
|
Name: dbCompany.Name,
|
|
AdminID: dbCompany.AdminID,
|
|
WalletID: dbCompany.WalletID,
|
|
WalletBalance: domain.Currency(dbCompany.Balance),
|
|
IsWalletActive: dbCompany.IsActive,
|
|
AdminFirstName: dbCompany.AdminFirstName,
|
|
AdminLastName: dbCompany.AdminLastName,
|
|
AdminPhoneNumber: dbCompany.AdminPhoneNumber.String,
|
|
}
|
|
}
|
|
|
|
func convertUpdateCompany(updateCompany domain.UpdateCompany) dbgen.UpdateCompanyParams {
|
|
var newUpdateCompany dbgen.UpdateCompanyParams
|
|
|
|
newUpdateCompany.ID = updateCompany.ID
|
|
|
|
if updateCompany.Name != nil {
|
|
newUpdateCompany.Name = pgtype.Text{
|
|
String: *updateCompany.Name,
|
|
Valid: true,
|
|
}
|
|
}
|
|
|
|
if updateCompany.AdminID != nil {
|
|
newUpdateCompany.AdminID = pgtype.Int8{
|
|
Int64: *updateCompany.AdminID,
|
|
Valid: true,
|
|
}
|
|
}
|
|
|
|
return newUpdateCompany
|
|
}
|
|
|
|
func (s *Store) CreateCompany(ctx context.Context, company domain.CreateCompany) (domain.Company, error) {
|
|
dbCompany, err := s.queries.CreateCompany(ctx, convertCreateCompany(company))
|
|
if err != nil {
|
|
return domain.Company{}, err
|
|
}
|
|
return convertDBCompany(dbCompany), nil
|
|
}
|
|
|
|
func (s *Store) GetAllCompanies(ctx context.Context) ([]domain.GetCompany, error) {
|
|
dbCompanies, err := s.queries.GetAllCompanies(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var companies []domain.GetCompany = make([]domain.GetCompany, 0, len(dbCompanies))
|
|
for _, dbCompany := range dbCompanies {
|
|
companies = append(companies, convertDBCompanyDetails(dbCompany))
|
|
}
|
|
|
|
return companies, nil
|
|
}
|
|
|
|
func (s *Store) SearchCompanyByName(ctx context.Context, name string) ([]domain.GetCompany, error) {
|
|
dbCompanies, err := s.queries.SearchCompanyByName(ctx, pgtype.Text{
|
|
String: name,
|
|
Valid: true,
|
|
})
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var companies []domain.GetCompany = make([]domain.GetCompany, 0, len(dbCompanies))
|
|
|
|
for _, dbCompany := range dbCompanies {
|
|
companies = append(companies, convertDBCompanyDetails(dbCompany))
|
|
}
|
|
return companies, nil
|
|
}
|
|
|
|
func (s *Store) GetCompanyByID(ctx context.Context, id int64) (domain.GetCompany, error) {
|
|
dbCompany, err := s.queries.GetCompanyByID(ctx, id)
|
|
|
|
if err != nil {
|
|
return domain.GetCompany{}, err
|
|
}
|
|
return convertDBCompanyDetails(dbCompany), nil
|
|
}
|
|
|
|
func (s *Store) UpdateCompany(ctx context.Context, company domain.UpdateCompany) (domain.Company, error) {
|
|
dbCompany, err := s.queries.UpdateCompany(ctx, convertUpdateCompany(company))
|
|
|
|
if err != nil {
|
|
return domain.Company{}, err
|
|
}
|
|
|
|
return convertDBCompany(dbCompany), nil
|
|
}
|
|
|
|
func (s *Store) DeleteCompany(ctx context.Context, id int64) error {
|
|
return s.queries.DeleteCompany(ctx, id)
|
|
}
|