DataInitializer.java

package com.spms.config;

import com.spms.enums.VehicleType;
import com.spms.model.Administrator;
import com.spms.model.Guard;
import com.spms.model.ParkingLot;
import com.spms.model.Supervisor;
import com.spms.model.Tariff;
import com.spms.repository.ParkingLotRepository;
import com.spms.repository.TariffRepository;
import com.spms.repository.UserRepository;
import lombok.RequiredArgsConstructor;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.CommandLineRunner;
import org.springframework.stereotype.Component;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.security.crypto.password.PasswordEncoder;

import java.util.List;

@Component
@RequiredArgsConstructor
@ConditionalOnProperty(name = "app.seed.enabled", havingValue = "true")
public class DataInitializer implements CommandLineRunner {

    private final ParkingLotRepository parkingLotRepository;
    private final TariffRepository tariffRepository;
    private final UserRepository userRepository;
    private final PasswordEncoder passwordEncoder;

    @Override
    @Transactional
    public void run(String... args) {
        seedLots();
        seedTariffs();
        seedUsers();
    }

    private void seedLots() {
        if (parkingLotRepository.count() > 0) {
            return;
        }

        parkingLotRepository.saveAll(List.of(
                new ParkingLot(null, "Lot A - North", "North Block, Gate 1", 3, 3, 28.6139, 77.2090),
                new ParkingLot(null, "Lot B - South", "South Block, Gate 2", 50, 50, 28.6100, 77.2085),
                new ParkingLot(null, "Lot C - East", "East Wing, Gate 3", 75, 75, 28.6145, 77.2130),
                new ParkingLot(null, "Lot D - West", "West Wing, Gate 4", 60, 60, 28.6135, 77.2045)
        ));
    }

    private void seedTariffs() {
        if (tariffRepository.count() > 0) {
            return;
        }

        tariffRepository.saveAll(List.of(
                new Tariff(null, VehicleType.CAR, 2.0),
                new Tariff(null, VehicleType.MOTORCYCLE, 1.0),
                new Tariff(null, VehicleType.TRUCK, 4.0),
                new Tariff(null, VehicleType.VAN, 3.0)
        ));
    }

    private void seedUsers() {
        List<ParkingLot> lots = parkingLotRepository.findAll();
        seedAdmin();
        seedLotUsers(lots);
    }

    private void seedAdmin() {
        if (userRepository.findByUsername("admin").isPresent()) {
            return;
        }

        Administrator admin = new Administrator();
        admin.setName("System Admin");
        admin.setUsername("admin");
        admin.setPassword(passwordEncoder.encode("changeme"));
        userRepository.save(admin);
    }

    private void seedLotUsers(List<ParkingLot> lots) {
        for (int i = 0; i < lots.size(); i++) {
            ParkingLot lot = lots.get(i);
            int lotNumber = i + 1;

            seedSupervisorForLot(lot, lotNumber);
            seedGuardForLot(lot, lotNumber);
        }
    }

    private void seedSupervisorForLot(ParkingLot assignedLot, int lotNumber) {
        String username = "sup_lot" + lotNumber;
        Supervisor supervisor = (Supervisor) userRepository.findByUsername(username).orElse(null);
        if (supervisor == null) {
            supervisor = new Supervisor();
            supervisor.setName("Supervisor Lot " + lotNumber);
            supervisor.setUsername(username);
            supervisor.setPassword(passwordEncoder.encode("sup" + lotNumber + "pass"));
        }
        supervisor.setAssignedLot(assignedLot);
        userRepository.save(supervisor);
    }

    private void seedGuardForLot(ParkingLot assignedLot, int lotNumber) {
        String username = "guard_lot" + lotNumber;
        Guard guard = (Guard) userRepository.findByUsername(username).orElse(null);
        if (guard == null) {
            guard = new Guard();
            guard.setName("Guard Lot " + lotNumber);
            guard.setUsername(username);
            guard.setPassword(passwordEncoder.encode("guard" + lotNumber + "pass"));
        }
        guard.setAssignedLot(assignedLot);
        userRepository.save(guard);
    }
}