source: src/main/java/tech/techharbor/Web/SingInRegisterController.java

package tech.techharbor.Web;

import jakarta.servlet.http.HttpSession;
import org.springframework.security.crypto.password.PasswordEncoder;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;
import tech.techharbor.Model.CustomerModel;
import tech.techharbor.Model.DeliveryManModel;
import tech.techharbor.Model.UserTableModel;
import tech.techharbor.Repository.CustomerRepository;
import tech.techharbor.Repository.DeliveryManRepository;
import tech.techharbor.Service.UserService;

import java.util.Optional;
import java.util.regex.Pattern;

@Controller
public class SingInRegisterController {

    private final UserService userService;
    private final PasswordEncoder passwordEncoder;

    private final DeliveryManRepository deliveryManRepository;

    private final CustomerRepository customerRepository;

    public SingInRegisterController(UserService userService, PasswordEncoder passwordEncoder, DeliveryManRepository deliveryManRepository, CustomerRepository customerRepository) {
        this.userService = userService;
        this.passwordEncoder = passwordEncoder;
        this.deliveryManRepository = deliveryManRepository;
        this.customerRepository = customerRepository;
    }

    @GetMapping("/register")
    public String registerUser() {

        return "register";
    }

    @PostMapping("/register")
    public String registerUser(@RequestParam String name,
                               @RequestParam String username,
                               @RequestParam String email,
                               @RequestParam String phoneNumber,
                               @RequestParam String password,
                               @RequestParam String confirmPassword) {

        String emailRegex = "^[a-zA-Z0-9_+&*-]+(?:\\.[a-zA-Z0-9_+&*-]+)*@(?:[a-zA-Z0-9-]+\\.)+[a-zA-Z]{2,7}$";

        // Check if email is valid
        if (!Pattern.matches(emailRegex, email))
            throw new IllegalArgumentException("Invalid email address");


        if (password.equals(confirmPassword)) {
            userService.create(name, username, email, passwordEncoder.encode(password), phoneNumber);
        }
        return "redirect:/login";
    }

    @GetMapping("/login")
    public String loginUser() {

        return "signIn";
    }

    @PostMapping("/login")
    public String loginUser(@RequestParam String username,
                            @RequestParam String password,
                            HttpSession session) {

        UserTableModel user = this.userService.findByUsername(username);
        if (user == null)
            throw new IllegalArgumentException("Invalid username");

        Optional<DeliveryManModel> deliveryMan = deliveryManRepository.findById(user.getUserId());

        if (passwordEncoder.matches(password, user.getPassword())) {
            if (deliveryMan.isEmpty()) {
                CustomerModel customer = new CustomerModel(user.getUserId());
                customerRepository.save(customer);
                session.setAttribute("user", user);
            } else {
                DeliveryManModel deliveryManUser = new DeliveryManModel(deliveryMan.get().getUserId());
                deliveryManRepository.save(deliveryManUser);
                session.setAttribute("deliveryMan", user);
            }
            return "redirect:/";
        }
        return "redirect:/login";
    }

    @GetMapping("/logout")
    public String logoutUser(HttpSession session) {
        session.invalidate();
        return "redirect:/";
    }


}