510 lines
20 KiB
Java
510 lines
20 KiB
Java
package ctbrec.ui;
|
|
|
|
import java.io.IOException;
|
|
import java.util.Objects;
|
|
import java.util.concurrent.ExecutionException;
|
|
import java.util.function.Function;
|
|
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
import ctbrec.ChaturbateModel;
|
|
import ctbrec.Config;
|
|
import ctbrec.Model;
|
|
import ctbrec.io.HttpClient;
|
|
import ctbrec.recorder.Recorder;
|
|
import ctbrec.recorder.StreamInfo;
|
|
import javafx.animation.FadeTransition;
|
|
import javafx.animation.FillTransition;
|
|
import javafx.animation.ParallelTransition;
|
|
import javafx.animation.Transition;
|
|
import javafx.application.Platform;
|
|
import javafx.beans.property.SimpleBooleanProperty;
|
|
import javafx.beans.value.ChangeListener;
|
|
import javafx.beans.value.ObservableValue;
|
|
import javafx.collections.ObservableList;
|
|
import javafx.geometry.Insets;
|
|
import javafx.geometry.Pos;
|
|
import javafx.scene.Cursor;
|
|
import javafx.scene.Node;
|
|
import javafx.scene.control.Alert;
|
|
import javafx.scene.control.ContextMenu;
|
|
import javafx.scene.image.Image;
|
|
import javafx.scene.image.ImageView;
|
|
import javafx.scene.layout.StackPane;
|
|
import javafx.scene.paint.Color;
|
|
import javafx.scene.paint.Paint;
|
|
import javafx.scene.shape.Circle;
|
|
import javafx.scene.shape.Rectangle;
|
|
import javafx.scene.shape.Shape;
|
|
import javafx.scene.text.Font;
|
|
import javafx.scene.text.Text;
|
|
import javafx.scene.text.TextAlignment;
|
|
import javafx.util.Duration;
|
|
import okhttp3.Request;
|
|
import okhttp3.RequestBody;
|
|
import okhttp3.Response;
|
|
|
|
public class ThumbCell extends StackPane {
|
|
|
|
private static final transient Logger LOG = LoggerFactory.getLogger(ThumbCell.class);
|
|
public static int width = 180;
|
|
private static final Duration ANIMATION_DURATION = new Duration(250);
|
|
|
|
private ChaturbateModel model;
|
|
private ImageView iv;
|
|
private Rectangle resolutionBackground;
|
|
private final Paint resolutionOnlineColor = new Color(0.22, 0.8, 0.29, 1);
|
|
private final Color resolutionOfflineColor = new Color(0.8, 0.28, 0.28, 1);
|
|
private Rectangle nameBackground;
|
|
private Rectangle topicBackground;
|
|
private Rectangle selectionOverlay;
|
|
private Text name;
|
|
private Text topic;
|
|
private Text resolutionTag;
|
|
private Recorder recorder;
|
|
private Circle recordingIndicator;
|
|
private int index = 0;
|
|
ContextMenu popup;
|
|
private final Color colorNormal = Color.BLACK;
|
|
private final Color colorHighlight = Color.WHITE;
|
|
private final Color colorRecording = new Color(0.8, 0.28, 0.28, 1);
|
|
private SimpleBooleanProperty selectionProperty = new SimpleBooleanProperty(false);
|
|
|
|
private HttpClient client;
|
|
|
|
private ObservableList<Node> thumbCellList;
|
|
private boolean mouseHovering = false;
|
|
private boolean recording = false;
|
|
|
|
public ThumbCell(ThumbOverviewTab parent, ChaturbateModel model, Recorder recorder, HttpClient client) {
|
|
this.thumbCellList = parent.grid.getChildren();
|
|
this.model = model;
|
|
this.recorder = recorder;
|
|
this.client = client;
|
|
recording = recorder.isRecording(model);
|
|
|
|
iv = new ImageView();
|
|
setImage(model.getPreview());
|
|
iv.setSmooth(true);
|
|
getChildren().add(iv);
|
|
|
|
nameBackground = new Rectangle();
|
|
nameBackground.setFill(recording ? colorRecording : colorNormal);
|
|
nameBackground.setOpacity(.7);
|
|
StackPane.setAlignment(nameBackground, Pos.BOTTOM_CENTER);
|
|
getChildren().add(nameBackground);
|
|
|
|
topicBackground = new Rectangle();
|
|
topicBackground.setFill(Color.BLACK);
|
|
topicBackground.setOpacity(0);
|
|
StackPane.setAlignment(topicBackground, Pos.TOP_LEFT);
|
|
getChildren().add(topicBackground);
|
|
|
|
resolutionBackground = new Rectangle(34, 16);
|
|
resolutionBackground.setFill(resolutionOnlineColor );
|
|
resolutionBackground.setVisible(false);
|
|
resolutionBackground.setArcHeight(5);
|
|
resolutionBackground.setArcWidth(resolutionBackground.getArcHeight());
|
|
StackPane.setAlignment(resolutionBackground, Pos.TOP_RIGHT);
|
|
StackPane.setMargin(resolutionBackground, new Insets(2));
|
|
getChildren().add(resolutionBackground);
|
|
|
|
name = new Text(model.getName());
|
|
name.setFill(Color.WHITE);
|
|
name.setFont(new Font("Sansserif", 16));
|
|
name.setTextAlignment(TextAlignment.CENTER);
|
|
name.prefHeight(25);
|
|
StackPane.setAlignment(name, Pos.BOTTOM_CENTER);
|
|
getChildren().add(name);
|
|
|
|
topic = new Text(model.getDescription());
|
|
|
|
topic.setFill(Color.WHITE);
|
|
topic.setFont(new Font("Sansserif", 13));
|
|
topic.setTextAlignment(TextAlignment.LEFT);
|
|
topic.setOpacity(0);
|
|
int margin = 4;
|
|
StackPane.setMargin(topic, new Insets(margin));
|
|
StackPane.setAlignment(topic, Pos.TOP_CENTER);
|
|
getChildren().add(topic);
|
|
|
|
resolutionTag = new Text();
|
|
resolutionTag.setFill(Color.WHITE);
|
|
resolutionTag.setVisible(false);
|
|
StackPane.setAlignment(resolutionTag, Pos.TOP_RIGHT);
|
|
StackPane.setMargin(resolutionTag, new Insets(2, 4, 2, 2));
|
|
getChildren().add(resolutionTag);
|
|
|
|
recordingIndicator = new Circle(8);
|
|
recordingIndicator.setFill(colorRecording);
|
|
StackPane.setMargin(recordingIndicator, new Insets(3));
|
|
StackPane.setAlignment(recordingIndicator, Pos.TOP_LEFT);
|
|
getChildren().add(recordingIndicator);
|
|
|
|
selectionOverlay = new Rectangle();
|
|
selectionOverlay.setOpacity(0);
|
|
StackPane.setAlignment(selectionOverlay, Pos.TOP_LEFT);
|
|
getChildren().add(selectionOverlay);
|
|
|
|
setOnMouseEntered((e) -> {
|
|
mouseHovering = true;
|
|
Color normal = recording ? colorRecording : colorNormal;
|
|
new ParallelTransition(changeColor(nameBackground, normal, colorHighlight), changeColor(name, colorHighlight, normal)).playFromStart();
|
|
new ParallelTransition(changeOpacity(topicBackground, 0.7), changeOpacity(topic, 0.7)).playFromStart();
|
|
if(Config.getInstance().getSettings().determineResolution) {
|
|
resolutionBackground.setVisible(false);
|
|
resolutionTag.setVisible(false);
|
|
}
|
|
});
|
|
setOnMouseExited((e) -> {
|
|
mouseHovering = false;
|
|
Color normal = recording ? colorRecording : colorNormal;
|
|
new ParallelTransition(changeColor(nameBackground, colorHighlight, normal), changeColor(name, normal, colorHighlight)).playFromStart();
|
|
new ParallelTransition(changeOpacity(topicBackground, 0), changeOpacity(topic, 0)).playFromStart();
|
|
if(Config.getInstance().getSettings().determineResolution && !resolutionTag.getText().isEmpty()) {
|
|
resolutionBackground.setVisible(true);
|
|
resolutionTag.setVisible(true);
|
|
}
|
|
});
|
|
setThumbWidth(width);
|
|
|
|
setRecording(recording);
|
|
if(Config.getInstance().getSettings().determineResolution) {
|
|
determineResolution();
|
|
}
|
|
}
|
|
|
|
public void setSelected(boolean selected) {
|
|
selectionProperty.set(selected);
|
|
selectionOverlay.getStyleClass().add("selection-background");
|
|
selectionOverlay.setOpacity(selected ? .75 : 0);
|
|
}
|
|
|
|
public boolean isSelected() {
|
|
return selectionProperty.get();
|
|
}
|
|
|
|
public ObservableValue<Boolean> selectionProperty() {
|
|
return selectionProperty;
|
|
}
|
|
|
|
private void determineResolution() {
|
|
if(ThumbOverviewTab.resolutionProcessing.contains(model)) {
|
|
LOG.trace("Already fetching resolution for model {}. Queue size {}", model.getName(), ThumbOverviewTab.resolutionProcessing.size());
|
|
return;
|
|
}
|
|
|
|
ThumbOverviewTab.threadPool.submit(() -> {
|
|
try {
|
|
ThumbOverviewTab.resolutionProcessing.add(model);
|
|
int[] resolution = model.getStreamResolution();
|
|
updateResolutionTag(resolution);
|
|
|
|
// the model is online, but the resolution is 0. probably something went wrong
|
|
// when we first requested the stream info, so we remove this invalid value from the "cache"
|
|
// so that it is requested again
|
|
try {
|
|
if (model.isOnline() && resolution[1] == 0) {
|
|
LOG.debug("Removing invalid resolution value for {}", model.getName());
|
|
model.invalidateCacheEntries();
|
|
}
|
|
} catch (IOException | ExecutionException | InterruptedException e) {
|
|
LOG.error("Coulnd't get resolution for model {}", model, e);
|
|
}
|
|
} catch (ExecutionException e1) {
|
|
LOG.warn("Couldn't update resolution tag for model {}", model.getName(), e1);
|
|
} catch (IOException e1) {
|
|
LOG.warn("Couldn't update resolution tag for model {}", model.getName(), e1);
|
|
} finally {
|
|
ThumbOverviewTab.resolutionProcessing.remove(model);
|
|
}
|
|
});
|
|
}
|
|
|
|
private void updateResolutionTag(int[] resolution) throws IOException, ExecutionException {
|
|
String _res = "n/a";
|
|
Paint resolutionBackgroundColor = resolutionOnlineColor;
|
|
String state = model.getOnlineState();
|
|
if ("public".equals(state)) {
|
|
LOG.trace("Model resolution {} {}x{}", model.getName(), resolution[0], resolution[1]);
|
|
LOG.trace("Resolution queue size: {}", ThumbOverviewTab.queue.size());
|
|
final int w = resolution[1];
|
|
_res = w > 0 ? Integer.toString(w) : state;
|
|
} else {
|
|
_res = model.getOnlineState();
|
|
resolutionBackgroundColor = resolutionOfflineColor;
|
|
}
|
|
final String resText = _res;
|
|
final Paint c = resolutionBackgroundColor;
|
|
Platform.runLater(() -> {
|
|
resolutionTag.setText(resText);
|
|
if(!mouseHovering) {
|
|
resolutionTag.setVisible(true);
|
|
resolutionBackground.setVisible(true);
|
|
}
|
|
resolutionBackground.setWidth(resolutionTag.getBoundsInLocal().getWidth() + 4);
|
|
resolutionBackground.setFill(c);
|
|
});
|
|
}
|
|
|
|
private void setImage(String url) {
|
|
if(!Objects.equals(System.getenv("CTBREC_DEV"), "1")) {
|
|
Image img = new Image(url, true);
|
|
|
|
// wait for the image to load, otherwise the ImageView replaces the current image with an "empty" image,
|
|
// which causes to show the grey background until the image is loaded
|
|
img.progressProperty().addListener(new ChangeListener<Number>() {
|
|
@Override
|
|
public void changed(ObservableValue<? extends Number> observable, Number oldValue, Number newValue) {
|
|
if(newValue.doubleValue() == 1.0) {
|
|
iv.setImage(img);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
}
|
|
|
|
private Transition changeColor(Shape shape, Color from, Color to) {
|
|
FillTransition transition = new FillTransition(ANIMATION_DURATION, from, to);
|
|
transition.setShape(shape);
|
|
return transition;
|
|
}
|
|
|
|
private Transition changeOpacity(Shape shape, double opacity) {
|
|
FadeTransition transition = new FadeTransition(ANIMATION_DURATION, shape);
|
|
transition.setFromValue(shape.getOpacity());
|
|
transition.setToValue(opacity);
|
|
return transition;
|
|
}
|
|
|
|
void startPlayer() {
|
|
try {
|
|
if(model.isOnline(true)) {
|
|
StreamInfo streamInfo = model.getStreamInfo();
|
|
LOG.debug("Playing {}", streamInfo.url);
|
|
Player.play(streamInfo.url);
|
|
} else {
|
|
Alert alert = new AutosizeAlert(Alert.AlertType.INFORMATION);
|
|
alert.setTitle("Room not public");
|
|
alert.setHeaderText("Room is currently not public");
|
|
alert.showAndWait();
|
|
}
|
|
} catch (IOException | ExecutionException | InterruptedException e1) {
|
|
LOG.error("Couldn't get stream information for model {}", model, e1);
|
|
Alert alert = new AutosizeAlert(Alert.AlertType.ERROR);
|
|
alert.setTitle("Error");
|
|
alert.setHeaderText("Couldn't determine stream URL");
|
|
alert.showAndWait();
|
|
}
|
|
}
|
|
|
|
private void setRecording(boolean recording) {
|
|
this.recording = recording;
|
|
if(recording) {
|
|
Color c = mouseHovering ? colorHighlight : colorRecording;
|
|
nameBackground.setFill(c);
|
|
} else {
|
|
Color c = mouseHovering ? colorHighlight : colorNormal;
|
|
nameBackground.setFill(c);
|
|
}
|
|
recordingIndicator.setVisible(recording);
|
|
}
|
|
|
|
void startStopAction(boolean start) {
|
|
setCursor(Cursor.WAIT);
|
|
|
|
boolean selectSource = Config.getInstance().getSettings().chooseStreamQuality;
|
|
if(selectSource && start) {
|
|
Function<Model, Void> onSuccess = (model) -> {
|
|
_startStopAction(model, start);
|
|
return null;
|
|
};
|
|
Function<Throwable, Void> onFail = (throwable) -> {
|
|
Alert alert = new AutosizeAlert(Alert.AlertType.ERROR);
|
|
alert.setTitle("Error");
|
|
alert.setHeaderText("Couldn't start/stop recording");
|
|
alert.setContentText("I/O error while starting/stopping the recording: " + throwable.getLocalizedMessage());
|
|
alert.showAndWait();
|
|
return null;
|
|
};
|
|
StreamSourceSelectionDialog.show(model, client, onSuccess, onFail);
|
|
} else {
|
|
_startStopAction(model, start);
|
|
}
|
|
}
|
|
|
|
private void _startStopAction(Model model, boolean start) {
|
|
new Thread() {
|
|
@Override
|
|
public void run() {
|
|
try {
|
|
if(start) {
|
|
recorder.startRecording(model);
|
|
setRecording(true);
|
|
} else {
|
|
recorder.stopRecording(model);
|
|
setRecording(false);
|
|
}
|
|
} catch (Exception e1) {
|
|
LOG.error("Couldn't start/stop recording", e1);
|
|
Platform.runLater(() -> {
|
|
Alert alert = new AutosizeAlert(Alert.AlertType.ERROR);
|
|
alert.setTitle("Error");
|
|
alert.setHeaderText("Couldn't start/stop recording");
|
|
alert.setContentText("I/O error while starting/stopping the recording: " + e1.getLocalizedMessage());
|
|
alert.showAndWait();
|
|
});
|
|
} finally {
|
|
setCursor(Cursor.DEFAULT);
|
|
}
|
|
}
|
|
}.start();
|
|
}
|
|
|
|
void follow(boolean follow) {
|
|
setCursor(Cursor.WAIT);
|
|
new Thread() {
|
|
@Override
|
|
public void run() {
|
|
try {
|
|
Request req = new Request.Builder().url(model.getUrl()).build();
|
|
Response resp = HttpClient.getInstance().execute(req);
|
|
resp.close();
|
|
|
|
String url = null;
|
|
if(follow) {
|
|
url = CtbrecApplication.BASE_URI + "/follow/follow/" + model.getName() + "/";
|
|
} else {
|
|
url = CtbrecApplication.BASE_URI + "/follow/unfollow/" + model.getName() + "/";
|
|
}
|
|
|
|
RequestBody body = RequestBody.create(null, new byte[0]);
|
|
req = new Request.Builder()
|
|
.url(url)
|
|
.method("POST", body)
|
|
.header("Accept", "*/*")
|
|
.header("Accept-Language", "en-US,en;q=0.5")
|
|
.header("Referer", model.getUrl())
|
|
.header("User-Agent", "Mozilla/5.0 (X11; Linux x86_64; rv:59.0) Gecko/20100101 Firefox/59.0")
|
|
.header("X-CSRFToken", HttpClient.getInstance().getToken())
|
|
.header("X-Requested-With", "XMLHttpRequest")
|
|
.build();
|
|
resp = HttpClient.getInstance().execute(req, true);
|
|
if(resp.isSuccessful()) {
|
|
String msg = resp.body().string();
|
|
if(!msg.equalsIgnoreCase("ok")) {
|
|
LOG.debug(msg);
|
|
throw new IOException("Response was " + msg.substring(0, Math.min(msg.length(), 500)));
|
|
} else {
|
|
LOG.debug("Follow/Unfollow -> {}", msg);
|
|
if(!follow) {
|
|
Platform.runLater(() -> thumbCellList.remove(ThumbCell.this));
|
|
}
|
|
}
|
|
} else {
|
|
resp.close();
|
|
throw new IOException("HTTP status " + resp.code() + " " + resp.message());
|
|
}
|
|
} catch (Exception e1) {
|
|
LOG.error("Couldn't follow/unfollow model {}", model.getName(), e1);
|
|
Platform.runLater(() -> {
|
|
Alert alert = new AutosizeAlert(Alert.AlertType.ERROR);
|
|
alert.setTitle("Error");
|
|
alert.setHeaderText("Couldn't follow/unfollow model");
|
|
alert.setContentText("I/O error while following/unfollowing model " + model.getName() + ": " + e1.getLocalizedMessage());
|
|
alert.showAndWait();
|
|
});
|
|
} finally {
|
|
setCursor(Cursor.DEFAULT);
|
|
}
|
|
}
|
|
}.start();
|
|
}
|
|
|
|
public ChaturbateModel getModel() {
|
|
return model;
|
|
}
|
|
|
|
public void setModel(Model model) {
|
|
//this.model = model;
|
|
this.model.setName(model.getName());
|
|
this.model.setDescription(model.getDescription());
|
|
this.model.setPreview(model.getPreview());
|
|
this.model.setTags(model.getTags());
|
|
this.model.setUrl(model.getUrl());
|
|
|
|
update();
|
|
}
|
|
|
|
public int getIndex() {
|
|
return index;
|
|
}
|
|
|
|
public void setIndex(int index) {
|
|
this.index = index;
|
|
}
|
|
|
|
private void update() {
|
|
setRecording(recorder.isRecording(model));
|
|
setImage(model.getPreview());
|
|
topic.setText(model.getDescription());
|
|
|
|
if(Config.getInstance().getSettings().determineResolution) {
|
|
determineResolution();
|
|
} else {
|
|
resolutionBackground.setVisible(false);
|
|
resolutionTag.setVisible(false);
|
|
}
|
|
requestLayout();
|
|
}
|
|
|
|
@Override
|
|
public int hashCode() {
|
|
final int prime = 31;
|
|
int result = 1;
|
|
result = prime * result + ((model == null) ? 0 : model.hashCode());
|
|
return result;
|
|
}
|
|
|
|
@Override
|
|
public boolean equals(Object obj) {
|
|
if (this == obj)
|
|
return true;
|
|
if (obj == null)
|
|
return false;
|
|
if (getClass() != obj.getClass())
|
|
return false;
|
|
ThumbCell other = (ThumbCell) obj;
|
|
if (model == null) {
|
|
if (other.model != null)
|
|
return false;
|
|
} else if (!model.equals(other.model))
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
public void setThumbWidth(int width) {
|
|
int height = width * 3 / 4;
|
|
setSize(width, height);
|
|
}
|
|
|
|
private void setSize(int w, int h) {
|
|
iv.setFitWidth(w);
|
|
iv.setFitHeight(h);
|
|
setMinSize(w, h);
|
|
setPrefSize(w, h);
|
|
nameBackground.setWidth(w);
|
|
nameBackground.setHeight(20);
|
|
topicBackground.setWidth(w);
|
|
topicBackground.setHeight(h-nameBackground.getHeight());
|
|
topic.prefHeight(h-25);
|
|
topic.maxHeight(h-25);
|
|
int margin = 4;
|
|
topic.maxWidth(w-margin*2);
|
|
topic.setWrappingWidth(w-margin*2);
|
|
selectionOverlay.setWidth(w);
|
|
selectionOverlay.setHeight(h);
|
|
}
|
|
}
|