src/main/java/events/controllers/EventController.java - develop - arseny_go/EventTracker

src/main/java/events/controllers/EventController.java
83 строки3 KB
ArsenyGolovin
More info is shown, some new features
12 янв 2025, 15:13
12 янв 2025, 15:1392a6576
package events.controllers;

import java.util.List;
import java.util.Optional;

import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.ModelAttribute;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.SessionAttributes;

import events.dataClasses.Entry;
import events.dataClasses.Event;
import events.dataClasses.User;
import events.repositories.EntryRepository;
import events.repositories.EventRepository;
import events.repositories.UserRepository;

@Controller
@SessionAttributes("user")
@RequestMapping("/events")
public class EventController {

	private final UserRepository userRepo;
	private final EventRepository eventRepo;
	private final EntryRepository entryRepo;

	public EventController(UserRepository userRepo, EventRepository eventRepo, EntryRepository entryRepo) {
		this.userRepo = userRepo;
		this.eventRepo = eventRepo;
		this.entryRepo = entryRepo;
	}

	@GetMapping("/{eventId}")
	public String showEvent(Model model, @PathVariable int eventId) {
		Optional<Event> e = eventRepo.findById(eventId);
		if (e.isPresent()) {
			Event event = e.get();
			model.addAttribute("organizer", userRepo.findById(event.getCreatorId()).get());
			model.addAttribute("event", event);
			List<User> parttakers = entryRepo.findParttakersByEventId(event.getId());
			model.addAttribute("parttakers", parttakers);
			return "event";
		} else {
			model.addAttribute("message", "Мероприятие не найдено.");
			return "error";
		}
	}

	@GetMapping("/create")
	public String showEventCreationForm(Model model) {
		model.addAttribute("event", new Event());
		return "event_creation_form";
	}

	@PostMapping("/create")
	public String createEvent(Model model, @ModelAttribute Event event) {
		User user = (User) model.getAttribute("user"); // Don't add via @ModelAttribute -
														// Event.name and User.name will be mixed up.
		event.setCreatorId(user.getId());
		event = eventRepo.save(event);
		return "redirect:/events/" + event.getId();
	}

	@PostMapping(path = "/add-or-delete-parttaker", headers = "hx-request=true")
	private String addOrDeleteParttaker(Model model, @RequestParam int eventId, @ModelAttribute User user) {
		List<User> parttakers = entryRepo.findParttakersByEventId(eventId);
		if (parttakers.contains(user)) {
			entryRepo.deleteByParttakerIdAndEventId(user.getId(), eventId);
			parttakers.remove(user);
		} else {
			entryRepo.save(new Entry(user.getId(), eventId));
			parttakers.add(user);
		}
		model.addAttribute("parttakers", parttakers);
		model.addAttribute("eventId", eventId);
		return "_fragments :: btn-and-parttakers";
	}
}