rocks.xmpp.debug.gui.DebugController Maven / Gradle / Ivy
/*
* The MIT License (MIT)
*
* Copyright (c) 2014 Christian Schudt
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
package rocks.xmpp.debug.gui;
import javafx.application.Platform;
import javafx.beans.binding.StringBinding;
import javafx.beans.property.SimpleObjectProperty;
import javafx.beans.value.ChangeListener;
import javafx.beans.value.ObservableValue;
import javafx.collections.transformation.FilteredList;
import javafx.collections.transformation.SortedList;
import javafx.event.ActionEvent;
import javafx.fxml.FXML;
import javafx.fxml.Initializable;
import javafx.geometry.Pos;
import javafx.scene.control.*;
import javafx.scene.layout.Region;
import javafx.scene.shape.Circle;
import javafx.scene.text.Text;
import javafx.util.Callback;
import org.xml.sax.ErrorHandler;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;
import org.xml.sax.SAXParseException;
import rocks.xmpp.core.Jid;
import rocks.xmpp.core.stanza.model.client.IQ;
import rocks.xmpp.core.stanza.model.client.Presence;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.parsers.SAXParser;
import javax.xml.parsers.SAXParserFactory;
import javax.xml.transform.*;
import javax.xml.transform.sax.SAXSource;
import javax.xml.transform.stream.StreamResult;
import java.io.StringReader;
import java.io.StringWriter;
import java.net.URL;
import java.text.DateFormat;
import java.util.Date;
import java.util.ResourceBundle;
import java.util.function.Predicate;
/**
* @author Christian Schudt
*/
public final class DebugController implements Initializable {
private static final String CSS_INCOMING_STANZA = "incoming-stanza";
private static final String CSS_OUTGOING_STANZA = "outgoing-stanza";
private static final String CSS_ERROR_STANZA = "error-stanza";
private static final String CSS_HIGHLIGHT_ROW = "highlight-row";
private static final String CSS_PRESENCE = "presence";
private static final String CSS_AVAILABLE = "available";
private static final String CSS_UNAVAILABLE = "unavailable";
private static final String CSS_AWAY = "away";
final DebugViewModel viewModel;
private final Transformer transformer;
private final SAXParser parser;
@FXML
private Text lblStatus;
@FXML
private Text lblServer;
@FXML
private Text lblPort;
@FXML
private Circle circlePresence;
private FilteredList filteredList;
@FXML
private CheckBox cbIgnoreCase;
@FXML
private TextField searchField;
@FXML
private CheckBox cbIncoming;
@FXML
private CheckBox cbOutgoing;
@FXML
private TextArea stanzaView;
@FXML
private TextArea txtOutgoing;
@FXML
private TextArea txtIncoming;
@FXML
private TableView stanzaTableView;
@FXML
private TableColumn columnIncoming;
@FXML
private TableColumn columnDate;
@FXML
private TableColumn columnStanza;
@FXML
private TableColumn columnFrom;
@FXML
private TableColumn columnTo;
public DebugController() {
viewModel = new DebugViewModel();
final TransformerFactory transformerFactory = TransformerFactory.newInstance();
transformerFactory.setAttribute("indent-number", 4);
try {
// http://stackoverflow.com/questions/21208325/how-do-i-change-the-default-logging-in-java-transformer
SAXParserFactory parserFactory = SAXParserFactory.newInstance();
parser = parserFactory.newSAXParser();
parser.getXMLReader().setErrorHandler(new ErrorHandler() {
@Override
public void warning(SAXParseException exception) throws SAXException {
throw exception;
}
@Override
public void error(SAXParseException exception) throws SAXException {
throw exception;
}
@Override
public void fatalError(SAXParseException exception) throws SAXException {
throw exception;
}
});
transformer = transformerFactory.newTransformer();
transformer.setOutputProperty(OutputKeys.INDENT, "yes");
transformer.setOutputProperty(OutputKeys.ENCODING, "UTF-8");
transformer.setErrorListener(new ErrorListener() {
@Override
public void warning(TransformerException exception) throws TransformerException {
throw exception;
}
@Override
public void error(TransformerException exception) throws TransformerException {
throw exception;
}
@Override
public void fatalError(TransformerException exception) throws TransformerException {
throw exception;
}
});
} catch (TransformerConfigurationException | ParserConfigurationException | SAXException e) {
throw new RuntimeException(e);
}
}
private static boolean containsIgnoreCase(String src, String input) {
final int length = input.length();
if (length == 0) {
return true; // Empty string is contained
}
final char firstLo = Character.toLowerCase(input.charAt(0));
final char firstUp = Character.toUpperCase(input.charAt(0));
for (int i = src.length() - length; i >= 0; i--) {
// Quick check before calling the more expensive regionMatches() method:
final char ch = src.charAt(i);
if (ch != firstLo && ch != firstUp)
continue;
if (src.regionMatches(true, i, input, 0, length))
return true;
}
return false;
}
@Override
public void initialize(URL location, ResourceBundle resources) {
lblServer.textProperty().bind(viewModel.server);
lblPort.textProperty().bind(new StringBinding() {
{
super.bind(viewModel.port);
}
@Override
protected String computeValue() {
return viewModel.port.getValue() != null ? Integer.toString(viewModel.port.get()) : "";
}
});
lblStatus.textProperty().bind(new StringBinding() {
{
super.bind(viewModel.status);
}
@Override
protected String computeValue() {
return viewModel.status.getValue() != null ? viewModel.status.getValue().name().toLowerCase() : "";
}
});
circlePresence.setRadius(10);
circlePresence.getStyleClass().addAll(CSS_PRESENCE, CSS_UNAVAILABLE);
viewModel.presence.addListener(new ChangeListener() {
@Override
public void changed(ObservableValue observable, Presence oldValue, Presence newValue) {
circlePresence.getStyleClass().removeAll(CSS_UNAVAILABLE, CSS_AVAILABLE);
Presence presence = viewModel.presence.get();
if (presence != null) {
if (presence.isAvailable()) {
if (presence.getShow() != null) {
switch (presence.getShow()) {
case AWAY:
circlePresence.getStyleClass().add(CSS_AWAY);
break;
default:
circlePresence.getStyleClass().add(CSS_AVAILABLE);
break;
}
} else {
circlePresence.getStyleClass().add(CSS_AVAILABLE);
}
} else {
circlePresence.getStyleClass().add(CSS_UNAVAILABLE);
}
} else {
circlePresence.getStyleClass().add(CSS_UNAVAILABLE);
}
}
});
// circlePresence.fillProperty().bind(new ObjectBinding() {
// {
// super.bind(viewModel.presence);
// }
//
// @Override
// protected Paint computeValue() {
// if (viewModel.presence.get() != null) {
// if (viewModel.presence.get().isAvailable()) {
// return Color.GREEN;
// }
// return Color.GRAY;
// }
// return Color.GRAY;
// }
// });
filteredList = new FilteredList<>(viewModel.stanzas, new Predicate() {
@Override
public boolean test(StanzaEntry stanzaEntry) {
return isVisible(stanzaEntry);
}
});
searchField.textProperty().addListener(new ChangeListener() {
@Override
public void changed(ObservableValue observable, String oldValue, String newValue) {
filter();
}
});
SortedList sortedList = new SortedList<>(filteredList);
stanzaTableView.setItems(sortedList);
sortedList.comparatorProperty().bind(stanzaTableView.comparatorProperty());
stanzaTableView.getSelectionModel().selectedItemProperty().addListener(new ChangeListener() {
@Override
public void changed(ObservableValue observable, StanzaEntry oldValue, StanzaEntry newValue) {
viewModel.highlightedItems.clear();
stanzaView.getStyleClass().removeAll(CSS_INCOMING_STANZA, CSS_OUTGOING_STANZA, CSS_ERROR_STANZA);
if (newValue != null) {
if (newValue.isIncoming()) {
stanzaView.getStyleClass().add(CSS_INCOMING_STANZA);
} else {
stanzaView.getStyleClass().add(CSS_OUTGOING_STANZA);
}
try {
if (newValue.getStanza() != null) {
StreamResult result = new StreamResult(new StringWriter());
Source source = new SAXSource(parser.getXMLReader(), new InputSource(new StringReader(newValue.getXml())));
transformer.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, "yes");
transformer.transform(source, result);
stanzaView.setText(result.getWriter().toString());
} else {
stanzaView.setText(newValue.getXml());
}
} catch (TransformerException | SAXException e) {
try {
StreamResult result = new StreamResult(new StringWriter());
String streamEndTag = "";
SAXSource source = new SAXSource(parser.getXMLReader(), new InputSource(new StringReader(newValue.getXml() + streamEndTag)));
transformer.setOutputProperty(OutputKeys.OMIT_XML_DECLARATION, "no");
transformer.transform(source, result);
String stream = result.getWriter().toString().trim();
if (stream.endsWith("/>")) {
stanzaView.setText(stream);
} else {
stanzaView.setText(stream.substring(0, stream.length() - 1 - streamEndTag.length()));
}
} catch (SAXException | TransformerException e1) {
stanzaView.setText(newValue.getXml());
}
}
for (StanzaEntry entry : stanzaTableView.getItems()) {
if (newValue.getStanza() instanceof IQ && entry.getStanza() instanceof IQ) {
IQ selectedIQ = (IQ) newValue.getStanza();
IQ otherIQ = (IQ) entry.getStanza();
if (otherIQ.getId() != null && otherIQ.getId().equals(selectedIQ.getId())
&& ((selectedIQ.isRequest() && otherIQ.isResponse())
|| selectedIQ.isResponse() && otherIQ.isRequest())) {
// Add the highlighted items.
viewModel.highlightedItems.add(entry);
}
}
}
// Workaround to refresh table:
// http://stackoverflow.com/questions/11065140/javafx-2-1-tableview-refresh-items
stanzaTableView.getColumns().get(0).setVisible(false);
stanzaTableView.getColumns().get(0).setVisible(true);
} else {
stanzaView.setText(null);
}
}
});
stanzaTableView.setRowFactory(new Callback, TableRow>() {
@Override
public TableRow call(TableView param) {
return new TableRow() {
@Override
protected void updateItem(StanzaEntry item, boolean empty) {
super.updateItem(item, empty);
getStyleClass().removeAll(CSS_INCOMING_STANZA, CSS_OUTGOING_STANZA, CSS_ERROR_STANZA, CSS_HIGHLIGHT_ROW);
setContextMenu(null);
if (!empty) {
if (item.isError()) {
getStyleClass().add(CSS_ERROR_STANZA);
}
if (item.isIncoming()) {
getStyleClass().add(CSS_INCOMING_STANZA);
} else {
getStyleClass().add(CSS_OUTGOING_STANZA);
}
if (item.getStanza() instanceof IQ && ((IQ) item.getStanza()).isRequest()) {
setContextMenu(iqContextMenu);
iqContextMenu.update(item);
}
if (viewModel.highlightedItems.contains(item)) {
getStyleClass().add(CSS_HIGHLIGHT_ROW);
}
}
}
private final IQContextMenu iqContextMenu = new IQContextMenu(stanzaTableView);
};
}
});
// Do not use PropertyValueFactory for columns, because it requires the item class to be public
columnIncoming.setCellValueFactory(new Callback, ObservableValue>() {
@Override
public ObservableValue call(TableColumn.CellDataFeatures param) {
return new SimpleObjectProperty<>(param.getValue().isIncoming());
}
});
columnIncoming.setCellFactory(new Callback, TableCell>() {
@Override
public TableCell call(TableColumn booleanStanzaEntryTableColumn) {
TableCell cell = new TableCell() {
@Override
protected void updateItem(Boolean item, boolean empty) {
super.updateItem(item, empty);
setText(null);
setGraphic(null);
if (!empty && item != null) {
// Make a region, so that -fx-shape can be applied from CSS.
Region rectangle = new Region();
rectangle.setMaxWidth(Control.USE_PREF_SIZE);
rectangle.setMaxHeight(Control.USE_PREF_SIZE);
rectangle.setRotate(item ? 90 : 270);
rectangle.getStyleClass().add("arrow");
setGraphic(rectangle);
}
}
};
cell.setAlignment(Pos.CENTER);
return cell;
}
});
columnDate.setCellValueFactory(new Callback, ObservableValue>() {
@Override
public ObservableValue call(TableColumn.CellDataFeatures param) {
return new SimpleObjectProperty<>(param.getValue().getDate());
}
});
columnDate.setCellFactory(new Callback, TableCell>() {
@Override
public TableCell call(TableColumn dateStanzaEntryTableColumn) {
TableCell cell = new TableCell() {
@Override
protected void updateItem(Date item, boolean empty) {
super.updateItem(item, empty);
setText(null);
setTooltip(null);
if (!empty) {
DateFormat timeFormat = DateFormat.getTimeInstance(DateFormat.SHORT);
DateFormat dateFormat = DateFormat.getDateTimeInstance(DateFormat.SHORT, DateFormat.SHORT);
setText(timeFormat.format(item));
setTooltip(new Tooltip(dateFormat.format(item)));
}
}
};
cell.setAlignment(Pos.CENTER);
return cell;
}
});
columnStanza.setMaxWidth(Double.MAX_VALUE);
columnStanza.setCellValueFactory(new Callback, ObservableValue>() {
@Override
public ObservableValue call(TableColumn.CellDataFeatures param) {
return new SimpleObjectProperty<>(param.getValue().getXml());
}
});
columnStanza.setCellFactory(new Callback, TableCell>() {
@Override
public TableCell call(TableColumn columnStanza) {
TableCell cell = new TableCell() {
@Override
protected void updateItem(String item, boolean empty) {
super.updateItem(item, empty);
setText(null);
setTooltip(null);
if (!empty) {
setText(item);
// StreamResult result = new StreamResult(new StringWriter());
// try {
// Source source = new SAXSource(debugView.parser.getXMLReader(), new InputSource(new StringReader(item)));
// debugView.transformer.transform(source, result);
// setTooltip(new Tooltip(result.getWriter().toString()));
// } catch (TransformerException | SAXException e) {
// setTooltip(new Tooltip(item));
// }
}
}
};
cell.getStyleClass().add("stanza-cell");
return cell;
}
});
columnFrom.setCellValueFactory(new Callback, ObservableValue>() {
@Override
public ObservableValue call(TableColumn.CellDataFeatures param) {
return new SimpleObjectProperty<>(param.getValue().getFrom());
}
});
columnFrom.setCellFactory(new Callback, TableCell>() {
@Override
public TableCell call(TableColumn columnStanza) {
return new TableCell() {
@Override
protected void updateItem(Jid item, boolean empty) {
super.updateItem(item, empty);
setText(null);
if (!empty && item != null) {
setText(item.toString());
}
}
};
}
});
columnTo.setCellValueFactory(new Callback, ObservableValue>() {
@Override
public ObservableValue call(TableColumn.CellDataFeatures param) {
return new SimpleObjectProperty<>(param.getValue().getTo());
}
});
columnTo.setCellFactory(new Callback, TableCell>() {
@Override
public TableCell call(TableColumn columnStanza) {
return new TableCell() {
@Override
protected void updateItem(Jid item, boolean empty) {
super.updateItem(item, empty);
setText(null);
if (!empty && item != null) {
setText(item.toString());
}
}
};
}
});
txtIncoming.setText("");
}
@FXML
public void filter() {
filteredList.setPredicate(new Predicate() {
@Override
public boolean test(StanzaEntry stanzaEntry) {
return isVisible(stanzaEntry);
}
});
}
@FXML
public void clear() {
viewModel.stanzas.clear();
}
void addStanza(StanzaEntry stanzaEntry) {
viewModel.stanzas.add(stanzaEntry);
}
private boolean isVisible(StanzaEntry stanzaEntry) {
return (cbIncoming.isSelected() && stanzaEntry.isIncoming()
|| cbOutgoing.isSelected() && !stanzaEntry.isIncoming())
&& (searchField.getText() == null || searchField.getText().equals("")
|| stanzaEntry.getXml().contains(searchField.getText()) && !cbIgnoreCase.isSelected()
|| containsIgnoreCase(stanzaEntry.getXml(), searchField.getText()) && cbIgnoreCase.isSelected());
}
void appendTextIncoming(final String s) {
Platform.runLater(new Runnable() {
@Override
public void run() {
txtIncoming.appendText(s);
}
});
}
void appendTextOutgoing(final String s) {
Platform.runLater(new Runnable() {
@Override
public void run() {
txtOutgoing.appendText(s);
}
});
}
public void clearOutgoing(ActionEvent actionEvent) {
txtOutgoing.clear();
}
public void clearIncoming(ActionEvent actionEvent) {
txtIncoming.clear();
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy