package ui; import models.User; import util.Session; import javax.swing.*; import java.awt.*; import java.util.function.Consumer; public class HomePanel extends BasePanel { public HomePanel(Consumer changePanel) { super(changePanel); JButton logoutButton = new JButton("Logout"); logoutButton.addActionListener(e -> { Session.getInstance().logout(); changePanel.accept("auth"); }); JButton addTodoButton = new JButton("Create Todo"); addTodoButton.addActionListener(e -> { User user = Session.getInstance().getUser(); if (user == null) { JOptionPane.showMessageDialog(this, "You must be logged in"); changePanel.accept("auth"); return; } changePanel.accept("create_todo"); }); JButton showTodoButton = new JButton("Show all Todo"); showTodoButton.addActionListener(e -> { changePanel.accept("show_all_todo"); }); add(logoutButton); add(addTodoButton, BorderLayout.NORTH); add(showTodoButton, BorderLayout.NORTH); } }