PIbd-21_Potapov_N.S._Stormtrooper_Hard/ProjectStormtrooper/FormPlaneCollection.java

package ProjectStormtrooper;

import javax.swing.*;
import javax.swing.event.ListSelectionEvent;
import javax.swing.filechooser.FileNameExtensionFilter;
import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.File;
import java.util.Objects;
import java.util.Stack;

public class FormPlaneCollection {
    final PlanesGenericStorage _storage;
    FrameDoubleParametrized _frameDoubleParametrized;
    FrameStormtrooper _frameRemovedPlanes;
    private JPanel PanelWrapper;
    private JPanel GroupBoxInstruments;
    private JPanel PictureBoxCollection;
    private JButton buttonAddPlane;
    private JTextField textFieldNumber;
    private JButton buttonRemovePlane;
    private JButton buttonRefreshCollection;
    private JButton buttonOpenGenerateWindow;
    private JTextField textFieldStorageName;
    private JButton buttonAddStorage;
    private JList listBoxStorages;
    private JButton buttonRemoveStorage;
    private JPanel storagesPanel;
    private JButton buttonShowRemovedPlanes;
    public DrawingPlane SelectedPlane;
    Stack<DrawingPlane> _removedPlanes;


    public JPanel getPanelWrapper() {
        return PanelWrapper;
    }

    public FormPlaneCollection() {
        PictureBoxCollection.setPreferredSize(new Dimension(800, 600));
        _storage = new PlanesGenericStorage(800, 600);
        _removedPlanes = new Stack<>();
        buttonAddPlane.addActionListener(this::buttonAddPlaneClicked);
        buttonRemovePlane.addActionListener(this::buttonRemovePlaneClicked);
        buttonRefreshCollection.addActionListener(this::buttonRefreshCollectionClicked);
        buttonOpenGenerateWindow.addActionListener(this::buttonOpenGenerateWindowClicked);
        buttonAddStorage.addActionListener(this::buttonAddStorageClicked);
        buttonRemoveStorage.addActionListener(this::buttonRemoveStorageClicked);
        listBoxStorages.addListSelectionListener(this::listBoxObjectsSelectedIndexChanged);
        buttonShowRemovedPlanes.addActionListener(this::buttonShowRemovedPlanesClicked);
    }

    public JMenuBar getMenuBar() {
        JMenuBar menuBar = new JMenuBar();
        JMenu fileMenu = new JMenu("Файл");
        JMenuItem openItem = new JMenuItem("Загрузить");
        openItem.addActionListener(
                e -> {
                    JFileChooser fileChooser = new JFileChooser();
                    fileChooser.setFileFilter(new FileNameExtensionFilter("Текстовые файлы (*.txt)", "txt"));
                    fileChooser.setDialogTitle("Выберите файл для загрузки данных");
                    if (fileChooser.showOpenDialog(null) == JFileChooser.APPROVE_OPTION) {
                        File selectedFile = fileChooser.getSelectedFile();
                        try {
                            if (_storage.LoadData(selectedFile.getAbsolutePath())) {
                                JOptionPane.showMessageDialog(null, "Загрузка прошла успешно", "Результат", JOptionPane.INFORMATION_MESSAGE);
                            } else {
                                JOptionPane.showMessageDialog(null, "Не загрузилось", "Результат", JOptionPane.ERROR_MESSAGE);
                            }
                        } catch (Exception ex) {
                            JOptionPane.showMessageDialog(
                                    null,
                                    "Ошибка при загрузке объектов" + ex.getMessage(),
                                    "Результат",
                                    JOptionPane.ERROR_MESSAGE
                            );
                        }
                    }
                    ReloadObjects();
                }
        );
        JMenuItem saveItem = new JMenuItem("Сохранить");
        saveItem.addActionListener(
                e -> {
                    JFileChooser fileChooser = new JFileChooser();
                    fileChooser.setDialogTitle("Выберите файл для сохранения данных");
                    fileChooser.setFileFilter(new FileNameExtensionFilter("Текстовые файлы (*.txt)", "txt"));
                    if (fileChooser.showSaveDialog(null) == JFileChooser.APPROVE_OPTION) {
                        File selectedFile = fileChooser.getSelectedFile();
                        try {
                            if (_storage.SaveData(selectedFile.getAbsolutePath()))
                                JOptionPane.showMessageDialog(null, "Сохранение прошло успешно", "Результат", JOptionPane.INFORMATION_MESSAGE);
                            else
                                JOptionPane.showMessageDialog(null, "Не сохранилось", "Результат", JOptionPane.ERROR_MESSAGE);
                        } catch (Exception ex) {
                            JOptionPane.showMessageDialog(
                                    null,
                                    "Ошибка при сохранении объектов" + ex.getMessage(),
                                    "Результат",
                                    JOptionPane.ERROR_MESSAGE
                            );
                        }
                    }
                }
        );
        JMenuItem openItemSingle = new JMenuItem("Загрузить коллекцию");
        openItemSingle.addActionListener(
                e -> {
                    JFileChooser fileChooser = new JFileChooser();
                    fileChooser.setFileFilter(new FileNameExtensionFilter("Текстовые файлы (*.txt)", "txt"));
                    fileChooser.setDialogTitle("Выберите файл для загрузки данных");
                    if (fileChooser.showOpenDialog(null) == JFileChooser.APPROVE_OPTION) {
                        File selectedFile = fileChooser.getSelectedFile();
                        if (_storage.LoadDataSingle(selectedFile.getAbsolutePath())) {
                            JOptionPane.showMessageDialog(null, "Загрузка прошла успешно", "Результат", JOptionPane.INFORMATION_MESSAGE);
                        } else {
                            JOptionPane.showMessageDialog(null, "Не загрузилось", "Результат", JOptionPane.ERROR_MESSAGE);
                        }
                    }
                    ReloadObjects();
                }
        );
        JMenuItem saveItemSingle = new JMenuItem("Сохранить коллекцию");
        saveItemSingle.addActionListener(
                e -> {
                    if (listBoxStorages.getSelectedValue() == null) {
                        JOptionPane.showMessageDialog(null, "Коллекция не выбрана", "Ошибка", JOptionPane.ERROR_MESSAGE);
                        return;
                    }
                    JFileChooser fileChooser = new JFileChooser();
                    fileChooser.setFileFilter(new FileNameExtensionFilter("Текстовые файлы (*.txt)", "txt"));
                    fileChooser.setDialogTitle("Выберите файл для сохранения данных");
                    if (fileChooser.showSaveDialog(null) == JFileChooser.APPROVE_OPTION) {
                        File selectedFile = fileChooser.getSelectedFile();
                        if (_storage.SaveDataSingle(selectedFile.getAbsolutePath(), (String) listBoxStorages.getSelectedValue()))
                            JOptionPane.showMessageDialog(null, "Сохранение прошло успешно", "Результат", JOptionPane.INFORMATION_MESSAGE);
                        else
                            JOptionPane.showMessageDialog(null, "Не сохранилось", "Результат", JOptionPane.ERROR_MESSAGE);
                    }
                }
        );
        fileMenu.add(openItem);
        fileMenu.add(saveItem);
        fileMenu.add(openItemSingle);
        fileMenu.add(saveItemSingle);
        menuBar.add(fileMenu);
        return menuBar;
    }

    private void ReloadObjects() {
        int index = listBoxStorages.getSelectedIndex();
        listBoxStorages.setListData(_storage.Keys().toArray());
        if (listBoxStorages.getModel().getSize() > 0 && (index == -1 || index >= listBoxStorages.getModel().getSize())) {
            listBoxStorages.setSelectedIndex(0);
        } else if (listBoxStorages.getModel().getSize() > 0 && index > -1 && index < listBoxStorages.getModel().getSize()) {
            listBoxStorages.setSelectedIndex(index);
        }
        listBoxStorages.invalidate();
    }

    private void buttonAddStorageClicked(ActionEvent e) {
        String storageName = textFieldStorageName.getText();
        if (Objects.equals(storageName, "")) {
            JOptionPane.showMessageDialog(this.getPanelWrapper(),
                    "Введите название",
                    "Ошибка",
                    JOptionPane.ERROR_MESSAGE);
            return;
        }
        _storage.AddSet(storageName);
        ReloadObjects();
    }

    private void listBoxObjectsSelectedIndexChanged(ListSelectionEvent e) {
        refreshPictureBox();
    }

    private void buttonRemoveStorageClicked(ActionEvent e) {
        if (listBoxStorages.getSelectedIndex() == -1) {
            return;
        }
        Object[] options = {"Да", "Нет"};
        int n = JOptionPane.showOptionDialog(this.getPanelWrapper(),
                "Удалить объект?",
                "Все серьезно",
                JOptionPane.YES_NO_OPTION,
                JOptionPane.QUESTION_MESSAGE,
                null,
                options,
                options[0]
        );
        if (n == 1) {
            return;
        }
        _storage.DelSet(listBoxStorages.getSelectedValue().toString());
        ReloadObjects();
    }

    public void buttonAddPlaneClicked(ActionEvent e) {
        FrameStormtrooper frameConfig = new FrameStormtrooper();
        FormPlaneConfig formConfig = new FormPlaneConfig();
        frameConfig.setContentPane(formConfig.panelWrapper);
        formConfig.buttonApply.addActionListener(ev -> {
            AddPlaneListener(formConfig.selectedPlane);
            frameConfig.dispose();
        });
        formConfig.buttonCancel.addActionListener(ev -> {
            frameConfig.dispose();
        });
        frameConfig.setVisible(true);
    }

    public void AddPlaneListener(DrawingPlane plane) {
        if (listBoxStorages.getSelectedIndex() == -1) {
            return;
        }
        plane.SetDrawingBounds(PictureBoxCollection.getWidth(), PictureBoxCollection.getHeight());
        var obj = _storage.Get(listBoxStorages.getSelectedValue().toString());
        if (obj == null) {
            return;
        }

        SelectedPlane = plane;
        if (SelectedPlane != null) {
            if (obj.Add(SelectedPlane) > -1) {
                refreshPictureBox();
                JOptionPane.showMessageDialog(this.getPanelWrapper(),
                        "Объект добавлен",
                        "Успех",
                        JOptionPane.INFORMATION_MESSAGE);
            } else {
                JOptionPane.showMessageDialog(this.getPanelWrapper(),
                        "Не удалось добавить объект",
                        "Ошибка",
                        JOptionPane.ERROR_MESSAGE);
            }
        }
    }

    public void buttonRemovePlaneClicked(ActionEvent e) {
        if (listBoxStorages.getSelectedIndex() == -1) {
            return;
        }
        var obj = _storage.Get(listBoxStorages.getSelectedValue().toString());
        if (obj == null) {
            return;
        }
        int pos;
        try {
            pos = Integer.parseInt(textFieldNumber.getText());
        } catch (NumberFormatException ex) {
            JOptionPane.showMessageDialog(this.getPanelWrapper(),
                    "Неверное значение",
                    "Ошибка",
                    JOptionPane.ERROR_MESSAGE);
            return;
        }
        Object[] options = {"Да", "Нет"};
        int n = JOptionPane.showOptionDialog(this.getPanelWrapper(),
                "Удалить объект?",
                "Все серьезно",
                JOptionPane.YES_NO_OPTION,
                JOptionPane.QUESTION_MESSAGE,
                null,
                options,
                options[0]
        );
        if (n == 1) {
            return;
        }
        DrawingPlane removedPlane = obj.Sub(pos);
        if (removedPlane != null) {
            _removedPlanes.push(removedPlane);
            refreshPictureBox();
            JOptionPane.showMessageDialog(this.getPanelWrapper(),
                    "Объект удален",
                    "Успех",
                    JOptionPane.INFORMATION_MESSAGE);
        } else {
            JOptionPane.showMessageDialog(this.getPanelWrapper(),
                    "Не удалось удалить объект",
                    "Ошибка",
                    JOptionPane.ERROR_MESSAGE);
        }
    }

    public void buttonRefreshCollectionClicked(ActionEvent e) {
        refreshPictureBox();
    }

    public void buttonOpenGenerateWindowClicked(ActionEvent e) {
        if (_frameDoubleParametrized != null) {
            _frameDoubleParametrized.dispose();
        }
        _frameDoubleParametrized = new FrameDoubleParametrized();
        _frameDoubleParametrized.setVisible(true);
    }

    public void buttonShowRemovedPlanesClicked(ActionEvent e) {
        if (_removedPlanes.empty()) {
            JOptionPane.showMessageDialog(this.getPanelWrapper(),
                    "Нет удаленных объектов",
                    "Инфо",
                    JOptionPane.INFORMATION_MESSAGE);
            return;
        }
        _frameRemovedPlanes = new FrameStormtrooper();
        _frameRemovedPlanes._formStromtrooper._drawingPlane = _removedPlanes.pop();
        _frameRemovedPlanes.setVisible(true);
        _frameRemovedPlanes._formStromtrooper.Draw();
    }

    public void refreshPictureBox() {
        if (listBoxStorages.getSelectedIndex() == -1) {
            return;
        }
        var obj = _storage.Get(listBoxStorages.getSelectedValue().toString());
        if (obj == null) {
            return;
        }
        Graphics g = PictureBoxCollection.getGraphics();
        PictureBoxCollection.paint(g);
        obj.ShowPlanes(g);
    }
}