source: src/main/java/edu/gjoko/schedlr/services/BusinessService.java@ 9050790

package edu.gjoko.schedlr.services;

import edu.gjoko.schedlr.entity.Business;
import edu.gjoko.schedlr.entity.ServiceType;
import edu.gjoko.schedlr.entity.StakeholderType;
import edu.gjoko.schedlr.repositories.BusinessRepository;
import edu.gjoko.schedlr.repositories.ServiceTypeRepository;
import lombok.AllArgsConstructor;
import org.springframework.stereotype.Service;

import java.util.List;

import static edu.gjoko.schedlr.entity.BusinessStatus.ACTIVE;
import static edu.gjoko.schedlr.entity.BusinessStatus.NEW;

@Service
@AllArgsConstructor
public class BusinessService {

    private final BusinessRepository businessRepository;
    private final ServiceTypeRepository serviceTypeRepository;

    public void saveBusiness(Business business) {
        saveNewServiceTypes(business);
        business.getOwner().setStakeholderType(StakeholderType.BUSINESS_OWNER);
        business.setBusinessStatus(NEW);
        businessRepository.save(business);
    }

    public void saveNewServiceTypes(Business business) {
        business.getServices()
                .stream()
                .forEach(service -> {
                    if (service.getServiceType().getId() == null) {
                        ServiceType serviceType = service.getServiceType();
                        serviceType.setBusinessType(business.getBusinessType());
                        service.setServiceType(serviceTypeRepository.save(serviceType));
                    }
                });
    }

    public List<Business> findAll() {
        return businessRepository.findAll();
    }

    public void updateBusinesses(List<Business> businessList) {
        businessList
                .stream()
                .forEach(business -> {
                    businessRepository.save(business);
                });
    }

}