AutowiredFXMLStage.java

package ui;

import javafx.scene.Scene;
import javafx.stage.Modality;
import javafx.stage.Stage;
import javafx.stage.StageStyle;
import javafx.stage.Window;
import org.springframework.beans.factory.annotation.Autowired;

import java.io.IOException;
import java.net.URL;

public class AutowiredFXMLStage extends Stage implements FXMLStage {

    @Autowired
    private FXMLLoader fxmlLoader;

    private URL fxml;

    public AutowiredFXMLStage(URL fxml, Window owner){
        this(fxml, owner, StageStyle.DECORATED);
    }

    public AutowiredFXMLStage(URL fxml, Window owner, StageStyle style){
        super(style);
        this.fxml = fxml;
        initOwner(owner);
        initModality(Modality.NONE);
    }

    @Override public void loadFXML() throws IOException {
        resetLoader();
        this.fxmlLoader.setLocation(this.fxml);
        setScene(new Scene(this.fxmlLoader.load()));
    }

    @Override public void setFXMLLoader(FXMLLoader loader){
        this.fxmlLoader = loader;
    }

    private void resetLoader(){
        this.fxmlLoader.setRoot(null);
        this.fxmlLoader.setLocation(null);
    }
}