diff --git a/rest-api/src/main/java/lcsb/mapviewer/api/comment/CommentRestImpl.java b/rest-api/src/main/java/lcsb/mapviewer/api/comment/CommentRestImpl.java index 87b01dc6138a20ac74c83b979b7f6970cab1f974..b83178ebcd80078bd401738abadec0d3238a8b4f 100644 --- a/rest-api/src/main/java/lcsb/mapviewer/api/comment/CommentRestImpl.java +++ b/rest-api/src/main/java/lcsb/mapviewer/api/comment/CommentRestImpl.java @@ -26,6 +26,7 @@ import lcsb.mapviewer.services.interfaces.ICommentService; import lcsb.mapviewer.services.interfaces.IModelService; import lcsb.mapviewer.services.interfaces.IUserService; import lcsb.mapviewer.services.overlay.IconManager; +import lcsb.mapviewer.services.search.data.ElementIdentifier.ElementIdentifierType; import lcsb.mapviewer.services.view.AuthenticationToken; @Transactional(value = "txManager") @@ -138,6 +139,8 @@ public class CommentRestImpl { value = comment.isDeleted(); } else if (column.equals("coord")) { value = getCoordinates(comment); + } else if (column.equals("type")) { + value = getType(comment); } else if (column.equals("icon")) { value = IconManager.getInstance().getCommentIcon(); } else if (admin) { @@ -156,6 +159,18 @@ public class CommentRestImpl { return result; } + private Object getType(Comment comment) { + if (comment.getTableName() != null) { + if (comment.getTableName().getName().contains("Reaction")) { + return ElementIdentifierType.REACTION; + } else { + return ElementIdentifierType.ALIAS; + } + } else { + return ElementIdentifierType.POINT; + } + } + private String getTitle(Comment comment) { String title = ""; if (comment.getCoordinates() != null) {