source: back_end/src/main/java/it/finki/tinki/service/impl/AccountServiceImpl.java

package it.finki.tinki.service.impl;

import it.finki.tinki.model.Address;
import it.finki.tinki.model.Work.Internship;
import it.finki.tinki.model.Work.Job;
import it.finki.tinki.model.Work.Project;
import it.finki.tinki.model.Skill;
import it.finki.tinki.model.Users.Account;
import it.finki.tinki.model.Users.Company;
import it.finki.tinki.model.Users.Team;
import it.finki.tinki.model.Users.User;
import it.finki.tinki.model.enumerator.AccountType;
import it.finki.tinki.model.exception.InvalidArgumentsException;
import it.finki.tinki.model.exception.UserExistsException;
import it.finki.tinki.repository.*;
import it.finki.tinki.service.AccountService;
import it.finki.tinki.service.MatchmakerService;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;
import java.util.Optional;

@Service
public class AccountServiceImpl implements AccountService {

    AddressRepository addressRepository;
    UserRepository userRepository;
    TeamRepository teamRepository;
    CompanyRepository companyRepository;
    JobRepository jobRepository;
    ProjectRepository projectRepository;
    InternshipRepository internshipRepository;
    MatchmakerService matchmakerService;

    public AccountServiceImpl(AddressRepository addressRepository,
                              UserRepository userRepository,
                              TeamRepository teamRepository,
                              CompanyRepository companyRepository,
                              MatchmakerService matchmakerService,
                              JobRepository jobRepository,
                              ProjectRepository projectRepository,
                              InternshipRepository internshipRepository) {
        this.addressRepository = addressRepository;
        this.userRepository = userRepository;
        this.teamRepository = teamRepository;
        this.companyRepository = companyRepository;
        this.jobRepository = jobRepository;
        this.projectRepository = projectRepository;
        this.internshipRepository = internshipRepository;
        this.matchmakerService = matchmakerService;
    }


    @Override
    public boolean hasData() {

        List<User> users = this.userRepository.findAll();
        List<Company> companies = this.companyRepository.findAll();
        List<Team> teams = this.teamRepository.findAll();

        return users.size() > 0 || companies.size() > 0 || teams.size() > 0;
    }

    @Override
    public Account findUser(String email, String password, AccountType type) {

        if(type.equals(AccountType.USER)){
            User u1 = this.userRepository.findByEmailAndPassword(email, password);
            return u1;
        }
        else if(type.equals(AccountType.TEAM)){
            Team t1 = this.teamRepository.findByEmailAndPassword(email, password);
            return t1;
        }
        else if(type.equals(AccountType.COMPANY)){
            Company c1 = this.companyRepository.findByEmailAndPassword(email, password);
            return c1;
        }

        return null;
    }

    public Account registerUser(String email, String password, String name, String surname, List<Skill> retainedSkills, List<Skill> skillsToLearn){
        if(email==null || email.isEmpty() || password==null || password.isEmpty() || name==null || name.isEmpty() || surname==null || surname.isEmpty()){
            throw new InvalidArgumentsException();
        }

        if(this.userRepository.findByEmail(email).isPresent()){
            throw new UserExistsException();
        }

        User u = new User(email, password, name, AccountType.USER, surname, retainedSkills, skillsToLearn);
        User ru = this.userRepository.save(u);

        setUpUser(ru);

        return ru;
    }

    public Account registerTeam(String email, String password, String name, int members){
        if(email==null || email.isEmpty() || password==null || password.isEmpty() || name==null || name.isEmpty()){
            throw new InvalidArgumentsException();
        }

        if(this.teamRepository.findByEmail(email).isPresent()){
            throw new UserExistsException();
        }

        Team t = new Team(email, password, name, AccountType.TEAM, members);
        return this.teamRepository.save(t);
    }

    public Account registerCompany(String email, String password, String name, String country, String city, String street){
        if(email==null || email.isEmpty() || password==null || password.isEmpty() || name==null || name.isEmpty()
                || country==null || country.isEmpty() || city==null || city.isEmpty() || street==null || street.isEmpty()){
            throw new InvalidArgumentsException();
        }

        System.out.println(email);

        if(this.companyRepository.findByEmail(email).isPresent()){
            throw new UserExistsException();
        }

        Address a = new Address(country, city, street);
        this.addressRepository.save(a);
        Company c = new Company(email, password, name, AccountType.COMPANY, a);
        return this.companyRepository.save(c);
    }

    public Account findByIdAndType(Long id, AccountType type){
        switch (type){
            case COMPANY:
                return this.companyRepository.findById(id).get();
            case TEAM:
                return this.teamRepository.findById(id).get();
            case USER:
                return this.userRepository.findById(id).get();
        }

        return null;
    }

    public Optional<?> findByIdAndEmail(Long id, String email, AccountType accountType){

        switch (accountType){
            case USER:
                return this.userRepository.findByIdAndEmail(id, email);
            case TEAM:
                return this.teamRepository.findByIdAndEmail(id, email);
            case COMPANY:
                return this.companyRepository.findByIdAndEmail(id, email);
        }

        return Optional.empty();
    }

    public User editUser(Long id, String email, String name, String surname, List<Skill> retainedSkills, List<Skill> skillsToLearn){
        if(email==null || email.isEmpty() || name==null || name.isEmpty() || surname==null || surname.isEmpty()){
            throw new InvalidArgumentsException();
        }

        User u = this.userRepository.findById(id).get();
        Optional<User> t = this.userRepository.findByEmail(email);

        if(t.isPresent() && !t.get().equals(u)){
            throw new UserExistsException();
        }

        u.setEmail(email);
        u.setName(name);
        u.setSurname(surname);
        u.setRetainedSkills(retainedSkills);
        u.setSkillsToLearn(skillsToLearn);

        User m = this.userRepository.save(u);

        setUpUser(m);

        return m;
    }

    public Company editCompany(Long id, String email, String name, String country, String city, String street){
        if(email==null || email.isEmpty() || name==null || name.isEmpty()
                || country==null || country.isEmpty() || city==null || city.isEmpty() || street==null || street.isEmpty()){
            throw new InvalidArgumentsException();
        }

        Company c = this.companyRepository.findById(id).get();
        Optional<Company> t = this.companyRepository.findByEmail(email);

        if(t.isPresent() && !t.get().equals(c)){
            throw new UserExistsException();
        }

        c.setEmail(email);
        c.setName(name);

        Address ad = c.getAddress();
        ad.setCountry(country);
        ad.setCity(city);
        ad.setStreet(street);

        this.addressRepository.save(ad);
        return this.companyRepository.save(c);
    }

    public Team editTeam(Long id, String email, String name, int members){
        if(email==null || email.isEmpty() || name==null || name.isEmpty()){
            throw new InvalidArgumentsException();
        }

        Team t = this.teamRepository.findById(id).get();
        Optional<Team> tt = this.teamRepository.findByEmail(email);

        if(tt.isPresent() && !tt.get().equals(t)){
            throw new UserExistsException();
        }

        t.setEmail(email);
        t.setName(name);
        t.setMembers(members);

        return this.teamRepository.save(t);
    }


    private void setUpUser(User u){

        this.matchmakerService.removeByUserId(u.getId());

        List<Job> jobs = this.jobRepository.findAll();
        List<Project> projects = this.projectRepository.findAll();
        List<Internship> internships = this.internshipRepository.findAll();

        if(jobs.size()!=0){
            for (Job job : jobs) {
                this.matchmakerService.setUpUserJobMatches(job, u);
            }
        }

        if(projects.size()!=0){
            for (Project project : projects) {
                this.matchmakerService.setUpUserProjectMatches(project, u);
            }
        }

        if(internships.size()!=0){
            for(Internship internship : internships){
                this.matchmakerService.setUpUserInternshipMatches(internship, u);
            }
        }
    }
}