package pl.knjpjatk.controller;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
import org.springframework.security.crypto.password.PasswordEncoder;
import org.springframework.stereotype.Component;
import pl.knjpjatk.entity.Inspectorate;
import pl.knjpjatk.entity.Role;
import pl.knjpjatk.entity.User;
import pl.knjpjatk.repositorium.InspectorateRepositorium;
import pl.knjpjatk.repositorium.RoleRepository;
import pl.knjpjatk.repositorium.UserRepositorium;
import java.util.HashSet;
import java.util.Set;

/**
 * Created by vadik on 19-Apr-17.
 */
@Component
public class DataLoader implements ApplicationRunner{

    private UserRepositorium userRepositorium;
    private InspectorateRepositorium inspectorateRepositorium;
    private RoleRepository roleRepository;

    private String USERNAME = "UserName";
    private String AUTHORITY_ADMIN = "Admin";
    private String AUTHORITY_USER = "User";
    private String INSPECTORATE_FULL_NAME = "Inspectorate";

    @Autowired
    public DataLoader(InspectorateRepositorium inspectorateRepositorium,
                      UserRepositorium userRepositorium,
                      RoleRepository roleRepository){
        this.inspectorateRepositorium = inspectorateRepositorium;
        this.userRepositorium = userRepositorium;
        this.roleRepository = roleRepository;
    }

    @Autowired
    private PasswordEncoder passwordEncoder;

    @Override
    public void run(ApplicationArguments applicationArguments) throws Exception {

        if(roleRepository.getRoleByAuthority(AUTHORITY_ADMIN) == null){
            Role roleAdmin = new Role();
            roleAdmin.setAuthority("Admin");
            roleRepository.save(roleAdmin);
        }

        if(roleRepository.getRoleByAuthority(AUTHORITY_USER) == null){
            Role roleUser = new Role();
            roleUser.setAuthority("User");
            roleRepository.save(roleUser);
        }

        if(inspectorateRepositorium.getInspectorateByFullName(INSPECTORATE_FULL_NAME) == null){
            Inspectorate inspectorate = new Inspectorate();
            inspectorate.setFullName("Inspectorate");
            inspectorate.setShortName("insp");
            inspectorateRepositorium.save(inspectorate);
        }

        if(userRepositorium.findPrincipalByUsername(USERNAME) == null){
            User user = new User();
            user.setEmail("kowalski@gmail.com");
            user.setEnabled(false);
            user.setFirstName("Jan");
            user.setLastName("Kowalski");
            user.setInspectorate(inspectorateRepositorium.getInspectorateByFullName(INSPECTORATE_FULL_NAME));
            user.setPassword(passwordEncoder.encode("password"));
            user.setPhone("555-555-555");
            user.setUsername(USERNAME);
            userRepositorium.save(user);
            Set set = new HashSet<Role>();
            set.add(roleRepository.getRoleByAuthority(AUTHORITY_ADMIN));
            set.add(roleRepository.getRoleByAuthority(AUTHORITY_USER));
            user.setAuthorities(set);
        }
    }
}