package treetableviewtest;

import com.sun.javafx.runtime.VersionInfo;
import javafx.application.Application;
import javafx.beans.property.SimpleIntegerProperty;
import javafx.beans.property.SimpleStringProperty;
import javafx.beans.value.ObservableValue;
import javafx.event.ActionEvent;
import javafx.event.EventHandler;
import javafx.scene.Scene;
import javafx.scene.control.Button;
import javafx.scene.control.Label;
import javafx.scene.control.TreeItem;
import javafx.scene.control.TreeTableColumn;
import javafx.scene.control.TreeTableColumn.CellDataFeatures;
import javafx.scene.control.TreeTableView;
import javafx.scene.control.cell.TextFieldTreeTableCell;
import javafx.scene.input.MouseButton;
import javafx.scene.input.MouseEvent;
import javafx.scene.layout.HBox;
import javafx.stage.Stage;
import javafx.util.Callback;

public class TreeTableViewTest extends Application {
    
    private SimpleIntegerProperty clickCounter = new SimpleIntegerProperty(0);
    
    @Override
    public void start(Stage primaryStage) {
        
        HBox root = new HBox(10.0);
        Scene scene = new Scene(root, 400, 250);
        
        final TreeTableView ttv = new TreeTableView();
        ttv.setRoot(new TreeItem(new Data("Root", "42")));
        ttv.getRoot().setExpanded(true);
        
        for (int i = 0; i < 10; i++) {
            ttv.getRoot().getChildren().add(new TreeItem(new Data("item - " + i, "id - " + i)));
        }
        
        final TreeTableColumn colA = new TreeTableColumn("Column A");
        colA.setCellValueFactory(new Callback<TreeTableColumn.CellDataFeatures<Data, String>, ObservableValue<String>>() {
            @Override
            public ObservableValue<String> call(CellDataFeatures<Data, String> p) {
                return p.getValue().getValue().name;
            }
        });
        
        final TreeTableColumn colB = new TreeTableColumn("Column B");
        colB.setCellValueFactory(new Callback<TreeTableColumn.CellDataFeatures<Data, String>, ObservableValue<String>>() {
            @Override
            public ObservableValue<String> call(CellDataFeatures<Data, String> p) {
                return p.getValue().getValue().id;
            }
        });
        
        ttv.getColumns().addAll(colA, colB);

        ttv.setOnMouseClicked(new EventHandler<MouseEvent>() {
            @Override
            public void handle(MouseEvent t) {
                if (t.getButton() == MouseButton.PRIMARY) {
                    clickCounter.set(clickCounter.get() + 1);
                }
            }
        });
        
        Label lblClick = new Label();
        lblClick.textProperty().bind(clickCounter.asString());
        
        Button btnSetEditable = new Button("Set editable");
        btnSetEditable.setOnAction(new EventHandler<ActionEvent>() {
            @Override
            public void handle(ActionEvent t) {
                ttv.setEditable(true);
                colA.setCellFactory(TextFieldTreeTableCell.forTreeTableColumn());
                colB.setCellFactory(TextFieldTreeTableCell.forTreeTableColumn());
            }
        });
        
        root.getChildren().addAll(ttv, btnSetEditable, lblClick);
        
        primaryStage.setTitle(VersionInfo.getRuntimeVersion());
        primaryStage.setScene(scene);
        primaryStage.show();
    }

    public static void main(String[] args) {
        launch(args);
    }
    
    private class Data {
        public SimpleStringProperty name;
        public SimpleStringProperty id;
        
        public Data(String _name, String _id) {
            name = new SimpleStringProperty(_name);
            id = new SimpleStringProperty(_id);
        }
    }
}