package com.company;

import javax.swing.*;
import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.File;
import java.io.FileNotFoundException;
import java.util.*;

public class WindowApp extends JFrame {
    private final static int LIST_X=300;
    private final static int LIST_Y=500;

    JScrollPane scrollPaneStudents;
    JScrollPane scrollPaneTask;
    JPanel mainPanel;
    JPanel northPanel;
    JButton addDataButton;
    JButton chooseFileButton;
    JButton sortButton;
    JList<Student> listStudents;
    JList<Student> listTask;

    public ArrayList<Student> getModelStudents() {
        return studentArrayList;
    }

    DefaultListModel<Student> modelStudents;
    ArrayList<Student> studentArrayList;
    ArrayList<Student> studentTempArrayList;

    WindowApp(){
        //panel
        mainPanel=new JPanel();
        mainPanel.setLayout(new BorderLayout());
        //buttons
        addDataButton=new JButton("Add new students...");
        chooseFileButton=new JButton("Choose text file...");
        sortButton=new JButton("Sort");
        addDataButton.setPreferredSize(new Dimension(LIST_X,20));
        sortButton.setPreferredSize(new Dimension(LIST_X,20));
        //
        northPanel=new JPanel();
        northPanel.add(addDataButton, BorderLayout.WEST);
        mainPanel.add(chooseFileButton,BorderLayout.SOUTH);
        northPanel.add(sortButton,BorderLayout.EAST);
        mainPanel.add(northPanel,BorderLayout.NORTH);
        //model
        studentArrayList=new ArrayList<>();
        //lists
        listStudents=new JList<>();
        scrollPaneStudents= new JScrollPane(listStudents);
        scrollPaneStudents.setPreferredSize(new Dimension(LIST_X,LIST_Y));
        mainPanel.add(scrollPaneStudents,BorderLayout.WEST);
        listTask=new JList<>();
        scrollPaneTask=new JScrollPane(listTask);
        scrollPaneTask.setPreferredSize(new Dimension(LIST_X,LIST_Y));
        mainPanel.add(scrollPaneTask,BorderLayout.EAST);
        //window
        setDefaultCloseOperation(WindowConstants.EXIT_ON_CLOSE);
        setResizable(false);
        setContentPane(mainPanel);
        pack();
        //listeners
        sortButton.addActionListener(new ActionListener() {
            @Override
            public void actionPerformed(ActionEvent e) {
                studentTempArrayList=new ArrayList<>();
                Iterator<Student> interalIt;
                Iterator<Student> externalIt=studentArrayList.iterator();
                while (externalIt.hasNext()){
                    Student studentExternal=externalIt.next();
                    interalIt=studentArrayList.iterator();
                    while (interalIt.hasNext()){
                        Student internalStudent=interalIt.next();
                        if(internalStudent.equals(studentExternal)){
                                studentTempArrayList.add(internalStudent);
                        }
                    }
                }
                MySort mySort=new MySort();
                Collections.sort(studentTempArrayList,mySort);
                updateTaskList();
            }
        });
        addDataButton.addActionListener(new ActionListener() {
            @Override
            public void actionPerformed(ActionEvent e) {
                AddDataDialog dialog= new AddDataDialog(WindowApp.this);
                dialog.setVisible(true);
            }
        });
        chooseFileButton.addActionListener(new ActionListener() {
            @Override
            public void actionPerformed(ActionEvent e) {
                JFileChooser fileOpen = new JFileChooser();
                int ret = fileOpen.showOpenDialog(null);
                if(ret ==JFileChooser.APPROVE_OPTION) {
                    try {
                        int recBookNum;
                        String surname;
                        int course;
                        int group;
                        Scanner scanner = new Scanner(new File(fileOpen.getSelectedFile().getAbsolutePath()));
                        while (scanner.hasNext()){
                            recBookNum=scanner.nextInt();
                            surname=scanner.next();
                            course=scanner.nextInt();
                            group=scanner.nextInt();
                            studentArrayList.add(new Student(recBookNum,surname,course,group));
                            updateList();
                        }
                    }
                    catch (FileNotFoundException exception){
                        JOptionPane.showMessageDialog(WindowApp.this,
                                "Cannot open the file.",
                                "Inane warning",
                                JOptionPane.WARNING_MESSAGE);
                    }
                }
            }
        });
    }
    public void updateList(){
        modelStudents = new DefaultListModel<>();
        Iterator<Student> iterator=studentArrayList.iterator();
        while (iterator.hasNext()){
            modelStudents.addElement(iterator.next());
        }
        listStudents.setModel(modelStudents);
    }
    public void updateTaskList(){
        modelStudents = new DefaultListModel<>();
        Iterator<Student> iterator=studentTempArrayList.iterator();
        while (iterator.hasNext()){
            modelStudents.addElement(iterator.next());
        }
        listTask.setModel(modelStudents);
    }
}
class MySort implements Comparator<Student> {

    public int compare(Student a, Student b) {
        if(a.getCourse()>b.getCourse()){
            return 1;
        }
        else if (a.getCourse()<b.getCourse()) {
            return -1;
        }
        else{
            if(a.getGroup()>b.getGroup()){
                return 1;
            }
            else if (a.getGroup()<b.getGroup()) {
                return -1;
            }
            else{
                return a.getSurname().compareTo(b.getSurname());
            }
        }
    }
}