company.go

package repository

import (
	"fmt"
	"github.com/hojabri/backend/models"
	log "github.com/sirupsen/logrus"
	"gorm.io/gorm"
	"gorm.io/gorm/clause"

)

type CompanyRepository interface {
	Save(company models.Company) (uint, error)
	Update(models.Company) error
	Delete(models.Company) error
	FindAll() []*models.Company
	FindByID(companyID uint) (*models.Company, error)
	DeleteByID(companyID uint) error
	FindByName(name string) (*models.Company, error)
	FindByField(fieldName, fieldValue string) (*models.Company, error)
	UpdateSingleField(company models.Company, fieldName, fieldValue string) error
}
type companyDatabase struct {
	connection *gorm.DB
}

func NewCompanyRepository() CompanyRepository {
	if DB == nil {
		_, err = Connect()
		if err != nil {
			log.Error(err)
		}
	}
	return &companyDatabase{
		connection: DB,
	}
}

func (db companyDatabase) DeleteByID(companyID uint) error {
	company := models.Company{}
	company.ID = companyID
	result := db.connection.Delete(&company)
	return result.Error
}

func (db companyDatabase) Save(company models.Company) (uint, error) {
	result := db.connection.Create(&company)
	if result.Error != nil {
		return 0, result.Error
	}
	return company.ID, nil
}

func (db companyDatabase) Update(company models.Company) error {
	result := db.connection.Save(&company)
	return result.Error
}

func (db companyDatabase) Delete(company models.Company) error {
	result := db.connection.Delete(&company)
	return result.Error
}

func (db companyDatabase) FindAll() []*models.Company {
	var companies []*models.Company
	db.connection.Preload(clause.Associations).Find(&companies)
	return companies
}

func (db companyDatabase) FindByID(companyID uint) (*models.Company, error) {
	var company models.Company
	result := db.connection.Preload(clause.Associations).Find(&company, "id = ?", companyID)
	
	if result.Error!=nil {
		return nil, result.Error
	}
	if result.RowsAffected > 0 {
		return &company, nil
	}
	return nil, nil
}


func (db companyDatabase) FindByName(name string) (*models.Company, error) {
	var company models.Company
	result := db.connection.Preload(clause.Associations).Find(&company, "name = ?", name)
	
	if result.Error!=nil {
		return nil, result.Error
	}
	if result.RowsAffected > 0 {
		return &company, nil
	}
	return nil, nil
}

func (db companyDatabase) FindByField(fieldName, fieldValue string) (*models.Company, error) {
	var company models.Company
	result := db.connection.Preload(clause.Associations).Find(&company, fmt.Sprintf("%s = ?", fieldName) ,fieldValue)
	
	if result.Error!=nil {
		return nil, result.Error
	}
	if result.RowsAffected > 0 {
		return &company, nil
	}
	return nil, nil
}

func (db companyDatabase) UpdateSingleField(company models.Company, fieldName, fieldValue string) error {
	result := db.connection.Model(&company).Update(fieldName, fieldValue)
	return result.Error
}