MyINPulse/MyINPulse-back/src/main/java/enseirb/myinpulse/service/database/EntrepreneurService.java

package enseirb.myinpulse.service.database;

import enseirb.myinpulse.model.Entrepreneur;
import enseirb.myinpulse.model.Project;
import enseirb.myinpulse.repository.EntrepreneurRepository;

import org.springframework.http.HttpStatus;
import org.springframework.stereotype.Service;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.server.ResponseStatusException;

import java.util.Optional;

@Service
public class EntrepreneurService {

    private final EntrepreneurRepository entrepreneurRepository;

    EntrepreneurService(EntrepreneurRepository entrepreneurRepository) {
        this.entrepreneurRepository = entrepreneurRepository;
    }

    public Iterable<Entrepreneur> getAllEntrepreneurs() {
        return this.entrepreneurRepository.findAll();
    }

    public Entrepreneur getEntrepreneurById(Long id) {
        Optional<Entrepreneur> entrepreneur = entrepreneurRepository.findById(id);
        if (entrepreneur.isEmpty()) {
            throw new ResponseStatusException(
                    HttpStatus.NOT_FOUND, "Cet entrepreneur n'existe pas");
        }
        return entrepreneur.get();
    }

    public Entrepreneur addEntrepreneur(Entrepreneur entrepreneur) {
        return this.entrepreneurRepository.save(entrepreneur);
    }

    public Entrepreneur updateEntrepreneur(
            Long id, String school, String course, Boolean sneeStatus) {
        Optional<Entrepreneur> entrepreneur = entrepreneurRepository.findById(id);
        if (entrepreneur.isEmpty()) {
            throw new ResponseStatusException(
                    HttpStatus.NOT_FOUND, "Cet entrepreneur n'existe pas");
        }
        if (school != null) {
            entrepreneur.get().setSchool(school);
        }
        if (course != null) {
            entrepreneur.get().setCourse(course);
        }
        if (sneeStatus != null) {
            entrepreneur.get().setSneeStatus(sneeStatus);
        }
        return this.entrepreneurRepository.save(entrepreneur.get());
    }

    public Iterable<Entrepreneur> GetEntrepreneurByProject(Project project) {
        return this.entrepreneurRepository.getEntrepreneurByProjectParticipation(project);
    }
}