19/12/2021

package com.example.revision_timer;

import javafx.application.Platform;
import javafx.fxml.FXML;
import javafx.fxml.Initializable;
import javafx.scene.control.Label;
import javafx.scene.control.ProgressBar;
import javafx.scene.control.Slider;
import javafx.scene.media.Media;
import javafx.scene.media.MediaPlayer;

import java.awt.*;
import java.io.File;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.*;

public class Controller implements Initializable {

    // Creates FXML variables for all elements in the UI that need to be referenced.

    @FXML
    private Label songText;

    @FXML
    private Slider volume;

    @FXML
    private ProgressBar progressBar;

    private File directory;
    private ArrayList<File> songs;

    private MediaPlayer mediaPlayer;

    private Timer revisionTimer;

    private final int reviDuration = 1800;
    private final int restDuration = 300;

    protected boolean running;
    private boolean revision = false;

    private int duration;
    private double current = 0;

    /**
     * Initialised at the beginning of the program, initialises variables.
     * @param url The location on disk used to work from when importing.
     * @param resourceBundle N/A, contains null.
     */

    @Override
    public void initialize(URL url, ResourceBundle resourceBundle) {

        songs = new ArrayList<>();
        directory = new File("src/main/resources/Music");
        File[] files = directory.listFiles();

        if (files != null){
            Collections.addAll(songs, files);
        }

        switchTimers();

        //Creates a listener for the volume slider being changed.

        volume.valueProperty().addListener((observableValue, number, t1) -> mediaPlayer.setVolume(volume.getValue() * 0.01));

    }

    //Randomly selects a new song from the library and plays it.

    public void startNewSong(){
        int amountSongs = Objects.requireNonNull(directory.listFiles()).length;

        Random ran = new Random();
        int songNumber = ran.nextInt(amountSongs);

        Media media = new Media(songs.get(songNumber).toURI().toString());
        mediaPlayer = new MediaPlayer(media);
        mediaPlayer.setCycleCount(MediaPlayer.INDEFINITE);

        String songName = songs.get(songNumber).getName();

        Platform.runLater(() -> songText.setText(songName));


        playPauseMedia();
    }

    public void playPauseMedia() {

        if (running) {
            mediaPlayer.pause();
            running = false;
            revisionTimer.cancel();
        } else {
            mediaPlayer.play();
            beginTimer();
        }

    }

    public void beginTimer(){

        revisionTimer = new Timer();

        mediaPlayer.setVolume(volume.getValue()*0.01);

        TimerTask task = new TimerTask() {

            public void run() {

                running = true;

                if (revision) {
                    current = mediaPlayer.getCurrentTime().toSeconds();
                    duration = reviDuration;
                } else {
                    current += 1;
                    duration = restDuration;
                }

                progressBar.setProgress(current / duration);

                if (current / duration > 1) {
                    revisionTimer.cancel();
                    switchTimers();
                }
            }
        };

        revisionTimer.scheduleAtFixedRate(task,1000,1000);

    }

    public void switchTimers(){
        if (revision){
            current = 0;
            revision = false;
            mediaPlayer.stop();
            displayNotification();
            beginTimer();
        } else{
            running = false;
            revision = true;
            startNewSong();
            displayNotification();
        }

    }

    public void displayNotification(){
        if (SystemTray.isSupported()) {
            try{
                displayTray();
            }catch(AWTException | MalformedURLException ignored){}
        } else {
            System.err.println("System tray not supported!");
        }
    }

    public void displayTray() throws AWTException, MalformedURLException {
        //Obtain only one instance of the SystemTray object
        SystemTray tray = SystemTray.getSystemTray();

        //If the icon is a file
        Image image = Toolkit.getDefaultToolkit().createImage("src/main/resources/com/example/revision_timer/icon.png");
        //Alternative (if the icon is on the classpath):
        //Image image = Toolkit.getDefaultToolkit().createImage(getClass().getResource("icon.png"));

        TrayIcon trayIcon = new TrayIcon(image, "Revision Timer");
        //Let the system resize the image if needed
        trayIcon.setImageAutoSize(true);
        //Set tooltip text for the tray icon
        trayIcon.setToolTip("Revision Timer");
        tray.add(trayIcon);

        if (revision) {
            trayIcon.displayMessage("Get Revising", "30 minutes starting now", TrayIcon.MessageType.INFO);
        } else {
            trayIcon.displayMessage("Get Relaxing", "5 minutes starting now", TrayIcon.MessageType.INFO);
        }

    }

}