import java.sql.Timestamp;
import java.util.List;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
import org.springframework.stereotype.Component;

import hr.gym.artu.domain.jpa.entity.ReservationEntity;
import hr.gym.artu.domain.jpa.entity.ServiceEntity;
import hr.gym.artu.domain.jpa.entity.WorkTimeEntity;
import hr.gym.artu.domain.jpa.entity.WorkerEntity;
import hr.gym.artu.domain.jpa.repository.ReservationRepository;
import hr.gym.artu.domain.jpa.repository.ServiceRepository;
import hr.gym.artu.domain.jpa.repository.WorkTimeRepository;
import hr.gym.artu.domain.jpa.repository.WorkerRepository;

@Component
public class DataLoader implements ApplicationRunner {

    private WorkerRepository workerRepository;
    private ServiceRepository serviceRepository;
    private WorkTimeRepository workTimeRepository;
    private ReservationRepository reservationRepository;

    @Autowired
    public DataLoader(WorkerRepository workerRepository,
    				  ServiceRepository serviceRepository,
    				  WorkTimeRepository workTimeRepository,
    				  ReservationRepository reservationRepository) {
        this.workerRepository = workerRepository;
        this.serviceRepository = serviceRepository;
        this.workTimeRepository = workTimeRepository;
        this.reservationRepository = reservationRepository;
    }

    public void run(ApplicationArguments args) {
    	List<ServiceEntity> services = this.serviceRepository.findAll();
    	if(services.size() == 0) {
	    	ServiceEntity serviceA = new ServiceEntity();
	    	serviceA.setActive(true);
	    	serviceA.setName("Brijanje");
	    	serviceA.setDescription("Gentleman");
	    	serviceA.setPrice(80l);
	    	this.serviceRepository.saveAndFlush(serviceA);

	    	ServiceEntity serviceB = new ServiceEntity();
	    	serviceB.setActive(true);
	    	serviceB.setName("Šišanje");
	    	serviceB.setDescription("Fudbalerka");
	    	serviceB.setPrice(100l);
	    	this.serviceRepository.saveAndFlush(serviceB);

	    	services.add(serviceA);
	    	services.add(serviceB);
    	}

    	if(this.workerRepository.findAll().size() == 0) {
    		WorkerEntity workerA = new WorkerEntity();
        	workerA.setActive(true);
        	workerA.setName("Janko");
        	workerA.setDescription("Janje");
        	workerA.setServices(services);
        	this.workerRepository.saveAndFlush(workerA);

        	WorkerEntity workerB = new WorkerEntity();
        	workerB.setActive(true);
        	workerB.setName("Teta Ruža");
        	workerB.setDescription("Kao i inače?");
        	workerB.setServices(services);
        	this.workerRepository.saveAndFlush(workerB);
    	}

    	if(this.workTimeRepository.findAll().size() == 0) {
    		WorkerEntity worker = this.workerRepository.findById(1l).get();
    		Timestamp timestamp = new Timestamp(2018, 7, 19, 8, 0, 0, 0);
    		ReservationEntity reservationA = new ReservationEntity();
    		reservationA.setClientMail("clinet@mail.com");
    		reservationA.setClientName("Clinet");
    		reservationA.setClientPhone("0991234567");
    		reservationA.setClientSurname("Klijentić");
    		reservationA.setService(this.serviceRepository.findById(1l).get());
    		reservationA.setWorker(worker);
    		this.reservationRepository.saveAndFlush(reservationA);

    		WorkTimeEntity workTimeA = new WorkTimeEntity();
    		workTimeA.setTime(timestamp);
    		workTimeA.setWorker(worker);
    		workTimeA.setReservation(reservationA);
    		this.workTimeRepository.saveAndFlush(workTimeA);

    		ReservationEntity reservation = new ReservationEntity();
    		reservation.setClientMail("luka.hrgovic2@fer.hr");
    		reservation.setClientName("Luka");
    		reservation.setClientPhone("0911941431");
    		reservation.setClientSurname("Hrgović");
    		reservation.setService(this.serviceRepository.findById(1l).get());
    		reservation.setWorker(worker);
    		this.reservationRepository.saveAndFlush(reservation);

    		WorkTimeEntity workTimeB = new WorkTimeEntity();
    		timestamp = new Timestamp(2018, 7, 19, 8, 30, 0, 0);
    		workTimeB.setTime(timestamp);
    		workTimeB.setWorker(worker);
    		workTimeB.setReservation(reservation);
    		this.workTimeRepository.saveAndFlush(workTimeB);
    	}
    }
}