aswi2022code-of-duty-gitlab/backend/src/main/java/cz/zcu/kiv/backendapi/config/DataInitiator.java @ bf502b70

package cz.zcu.kiv.backendapi.config;

import cz.zcu.kiv.backendapi.catalog.CatalogItem;
import cz.zcu.kiv.backendapi.catalog.CatalogItemRepository;
import cz.zcu.kiv.backendapi.catalog.ICatalogItemService;
import cz.zcu.kiv.backendapi.sources.ISourcesService;
import cz.zcu.kiv.backendapi.sources.Sources;
import cz.zcu.kiv.backendapi.titlepage.ITitlePageService;
import cz.zcu.kiv.backendapi.titlepage.TitlePage;
import cz.zcu.kiv.backendapi.type.ITypeService;
import cz.zcu.kiv.backendapi.type.Type;
import cz.zcu.kiv.backendapi.user.UserEntity;
import cz.zcu.kiv.backendapi.user.UserRepository;
import lombok.RequiredArgsConstructor;
import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVParser;
import org.apache.commons.csv.CSVRecord;
import org.springframework.context.ApplicationListener;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
import org.springframework.stereotype.Component;
import org.springframework.transaction.annotation.Transactional;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;

//TODO comment or delete
@Component
@RequiredArgsConstructor
public class DataInitiator implements ApplicationListener<ContextRefreshedEvent> {

    private final ICatalogItemService catalogService;

    private final ITypeService typeService;

    private final CatalogItemRepository catalogItemRepository;

    private final UserRepository userRepository;

    private final ITitlePageService titlePageService;
    private final ISourcesService sourcesService;

    private final BCryptPasswordEncoder encoder;

    @Override
    @Transactional
    public void onApplicationEvent(ContextRefreshedEvent event) {
        List<Type> types = loadTypes();
        for (Type type : types) {
            typeService.saveType(type);
        }
        List<CatalogItem> catalog = loadCatalog();
        catalogService.saveCatalog(catalog);

        titlePageService.updateTitlePage(new TitlePage(1L, ""));
        sourcesService.updateSources(new Sources(1L, ""));

        UserEntity user1 = new UserEntity("admin", "admin", encoder.encode("password"), (byte) 7, true);
        userRepository.save(user1);

        UserEntity user2 = new UserEntity("normal", "normal", encoder.encode("password"), (byte) 1, false);
        userRepository.save(user2);

    }

    private List<CatalogItem> loadCatalog() {
        List<CatalogItem> catalogEntities = new ArrayList<>();
        InputStream csvData = getClass().getClassLoader().getResourceAsStream("22-05-09-AssyrianProject-AllNoDupl.csv");
//        File csvData = new File("AssyrianProject-AllNoDupl-22-03-28.csv");
        CSVParser parser;
        try {
            parser = CSVParser.parse(csvData, StandardCharsets.UTF_8, CSVFormat.Builder.create(CSVFormat.DEFAULT)
                    .setHeader()
                    .setSkipHeaderRecord(true)
                    .build());
            for (CSVRecord csvRecord : parser) {
                catalogEntities.add(new CatalogItem(csvRecord.toList()));
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return catalogEntities;

    }

    private List<Type> loadTypes() {
        List<Type> types = new ArrayList<>();
        CSVParser parser;
        File csvData = new File("AssyrianProject-All.csv");
        try {
            parser = CSVParser.parse(csvData, Charset.defaultCharset(), CSVFormat.DEFAULT.builder().setSkipHeaderRecord(true).build());
            for (CSVRecord csvRecord : parser) {
                if (csvRecord.get(0).equals("10000")) {
                    types.add(new Type(csvRecord.get(2)));
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return types;
    }
}