@reltio/components 1.4.2154 → 1.4.2156

package/CollaborationItem/IntegrationCollaborationItem.test.js

@@ -118,6 +118,7 @@ var setUp = function (_a) {
         return (React.createElement(MdmModuleProvider, { values: mdmValues },
             React.createElement(CollaborationContext.Provider, { value: {
                     commentsMap: {},
+                    totalCommentsCount: null,
                     currentComment: currentComment,
                     createComment: function () { return Promise.resolve(); },
                     editComment: function () { return Promise.resolve(); },

package/CollaborationItem/components/Comment/Comment.test.js

@@ -126,6 +126,7 @@ describe('Comment tests', function () {
     };
     var defaultCollaborationContextValue = {
         commentsMap: defaultCommentsMap,
+        totalCommentsCount: null,
         currentComment: defaultCurrentComment,
         createComment: function () { return Promise.resolve(); },
         editComment: function () { return Promise.resolve(); },

package/CollaborationItem/components/SendMessageArea/SendMessageArea.test.js

@@ -65,7 +65,7 @@ describe('SendMessageArea tests', function () {
     };
     var createCollaborationContext = function (props) {
         if (props === void 0) { props = {}; }
-        return __assign({ comments: [], commentsMap: null, currentComment: null, getComment: identity, getComments: function () { return new Promise(identity); }, getCommentsCount: identity, resolveThread: identity, reOpenThread: identity, clearCurrentComment: identity, deleteComment: identity, sending: false, loading: false, pageToken: null, deleteReply: identity, createReply: function () { return new Promise(function (res) { return res(); }); }, createComment: function () { return new Promise(function (res) { return res(); }); }, editComment: function () { return new Promise(function (res) { return res(); }); }, editReply: function () { return new Promise(function (res) { return res(); }); }, getCommentState: function () { return ({}); }, clearCommentState: function () { }, updateCommentState: function () { } }, props);
+        return __assign({ comments: [], commentsMap: null, totalCommentsCount: null, currentComment: null, getComment: identity, getComments: function () { return new Promise(identity); }, getCommentsCount: identity, resolveThread: identity, reOpenThread: identity, clearCurrentComment: identity, deleteComment: identity, sending: false, loading: false, pageToken: null, deleteReply: identity, createReply: function () { return new Promise(function (res) { return res(); }); }, createComment: function () { return new Promise(function (res) { return res(); }); }, editComment: function () { return new Promise(function (res) { return res(); }); }, editReply: function () { return new Promise(function (res) { return res(); }); }, getCommentState: function () { return ({}); }, clearCommentState: function () { }, updateCommentState: function () { } }, props);
     };
     var createProps = function (comment, extraProps) {
         if (extraProps === void 0) { extraProps = {}; }

package/cjs/CollaborationItem/IntegrationCollaborationItem.test.js

@@ -123,6 +123,7 @@ var setUp = function (_a) {
         return (react_1.default.createElement(MdmModuleContext_1.MdmModuleProvider, { values: mdmValues },
             react_1.default.createElement(CollaborationContext_1.CollaborationContext.Provider, { value: {
                     commentsMap: {},
+                    totalCommentsCount: null,
                     currentComment: currentComment,
                     createComment: function () { return Promise.resolve(); },
                     editComment: function () { return Promise.resolve(); },

package/cjs/CollaborationItem/components/Comment/Comment.test.js

@@ -131,6 +131,7 @@ describe('Comment tests', function () {
     };
     var defaultCollaborationContextValue = {
         commentsMap: defaultCommentsMap,
+        totalCommentsCount: null,
         currentComment: defaultCurrentComment,
         createComment: function () { return Promise.resolve(); },
         editComment: function () { return Promise.resolve(); },

package/cjs/CollaborationItem/components/SendMessageArea/SendMessageArea.test.js

@@ -70,7 +70,7 @@ describe('SendMessageArea tests', function () {
     };
     var createCollaborationContext = function (props) {
         if (props === void 0) { props = {}; }
-        return __assign({ comments: [], commentsMap: null, currentComment: null, getComment: ramda_1.identity, getComments: function () { return new Promise(ramda_1.identity); }, getCommentsCount: ramda_1.identity, resolveThread: ramda_1.identity, reOpenThread: ramda_1.identity, clearCurrentComment: ramda_1.identity, deleteComment: ramda_1.identity, sending: false, loading: false, pageToken: null, deleteReply: ramda_1.identity, createReply: function () { return new Promise(function (res) { return res(); }); }, createComment: function () { return new Promise(function (res) { return res(); }); }, editComment: function () { return new Promise(function (res) { return res(); }); }, editReply: function () { return new Promise(function (res) { return res(); }); }, getCommentState: function () { return ({}); }, clearCommentState: function () { }, updateCommentState: function () { } }, props);
+        return __assign({ comments: [], commentsMap: null, totalCommentsCount: null, currentComment: null, getComment: ramda_1.identity, getComments: function () { return new Promise(ramda_1.identity); }, getCommentsCount: ramda_1.identity, resolveThread: ramda_1.identity, reOpenThread: ramda_1.identity, clearCurrentComment: ramda_1.identity, deleteComment: ramda_1.identity, sending: false, loading: false, pageToken: null, deleteReply: ramda_1.identity, createReply: function () { return new Promise(function (res) { return res(); }); }, createComment: function () { return new Promise(function (res) { return res(); }); }, editComment: function () { return new Promise(function (res) { return res(); }); }, editReply: function () { return new Promise(function (res) { return res(); }); }, getCommentState: function () { return ({}); }, clearCommentState: function () { }, updateCommentState: function () { } }, props);
     };
     var createProps = function (comment, extraProps) {
         if (extraProps === void 0) { extraProps = {}; }

package/cjs/contexts/CollaborationContext/index.d.ts

@@ -22,6 +22,7 @@ export declare const InitialCollaborationContextValue: {
     pageToken: any;
     deleteReply: any;
     editReply: () => Promise<void>;
+    totalCommentsCount: any;
 };
 export declare const CollaborationContext: React.Context<{
     objectTypes?: CollaborationObjectTypes[];
@@ -67,6 +68,7 @@ export declare const CollaborationContext: React.Context<{
     sending: boolean;
     loading: boolean;
     pageToken: string;
+    totalCommentsCount: number | null;
 }>;
 export declare const CollaborationContextProvider: ({ collaboration, children }: {
     collaboration: any;

package/cjs/contexts/CollaborationContext/index.js

@@ -29,7 +29,8 @@ exports.InitialCollaborationContextValue = {
     loading: false,
     pageToken: null,
     deleteReply: ramda_1.identity,
-    editReply: function () { return new Promise(ramda_1.identity); }
+    editReply: function () { return new Promise(ramda_1.identity); },
+    totalCommentsCount: null
 };
 exports.CollaborationContext = react_1.default.createContext(exports.InitialCollaborationContextValue);
 exports.CollaborationContext.displayName = 'CollaborationContext';

package/cjs/hooks/useCollaboration/useCollaboration.d.ts

@@ -3,8 +3,9 @@ type Props = {
     objectIds?: string[];
     objectTypes?: CollaborationObjectTypes[];
     enabled?: boolean;
+    needsTotalCount?: boolean;
 };
-export declare const useCollaboration: ({ objectIds, objectTypes, enabled }: Props) => {
+export declare const useCollaboration: ({ objectIds, objectTypes, enabled, needsTotalCount }: Props) => {
     clearCurrentComment: () => void;
     comments: Comment[];
     commentsMap: CommentsMap;
@@ -27,5 +28,6 @@ export declare const useCollaboration: ({ objectIds, objectTypes, enabled }: Pro
     sending: boolean;
     deleteReply: ({ uri, commentId, reply }: any) => void;
     editReply: ({ content, namedUsers, commentId, replyId }: any) => Promise<void>;
+    totalCommentsCount: number;
 };
 export {};

package/cjs/hooks/useCollaboration/useCollaboration.js

@@ -38,35 +38,57 @@ var getUpdatedReplies = function (comment, reply) {
 var defaultObjectIds = [];
 var catchErrorInRequest = function (e) { return console.warn('Collaboration error', e); };
 var useCollaboration = function (_a) {
-    var _b = _a.objectIds, objectIds = _b === void 0 ? defaultObjectIds : _b, objectTypes = _a.objectTypes, _c = _a.enabled, enabled = _c === void 0 ? true : _c;
+    var _b = _a.objectIds, objectIds = _b === void 0 ? defaultObjectIds : _b, objectTypes = _a.objectTypes, _c = _a.enabled, enabled = _c === void 0 ? true : _c, _d = _a.needsTotalCount, needsTotalCount = _d === void 0 ? false : _d;
     var entity = (0, MdmModuleContext_1.useMdmEntity)();
     var tenant = (0, MdmModuleContext_1.useMdmTenant)();
     var collaborationPath = (0, MdmModuleContext_1.useMdmCollaborationPath)();
     var isCollaborationEnabled = (0, MdmModuleContext_1.useMdmIsCollaborationEnabled)() && enabled;
-    var _d = (0, react_1.useState)(null), pageToken = _d[0], setPageToken = _d[1];
-    var _e = (0, react_1.useState)([]), comments = _e[0], setComments = _e[1];
-    var _f = (0, react_1.useState)(null), commentsMap = _f[0], setCommentsMap = _f[1];
-    var _g = (0, react_1.useState)(null), currentComment = _g[0], setCurrentComment = _g[1];
-    var _h = (0, react_1.useState)(false), sending = _h[0], setSending = _h[1];
-    var _j = (0, react_1.useState)(false), loading = _j[0], setLoading = _j[1];
-    var _k = (0, react_1.useState)({}), collaborationStateMap = _k[0], setCollaborationStateMap = _k[1];
+    var _e = (0, react_1.useState)(null), pageToken = _e[0], setPageToken = _e[1];
+    var _f = (0, react_1.useState)([]), comments = _f[0], setComments = _f[1];
+    var _g = (0, react_1.useState)(null), commentsMap = _g[0], setCommentsMap = _g[1];
+    var _h = (0, react_1.useState)(null), totalCommentsCount = _h[0], setTotalCommentsCount = _h[1];
+    var _j = (0, react_1.useState)(null), currentComment = _j[0], setCurrentComment = _j[1];
+    var _k = (0, react_1.useState)(false), sending = _k[0], setSending = _k[1];
+    var _l = (0, react_1.useState)(false), loading = _l[0], setLoading = _l[1];
+    var _m = (0, react_1.useState)({}), collaborationStateMap = _m[0], setCollaborationStateMap = _m[1];
     var addCommentsToCommentsMap = (0, react_1.useCallback)(function (commentsMap) {
         setCommentsMap(function (currentCommentsMap) { return (__assign(__assign({}, currentCommentsMap), commentsMap)); });
     }, []);
     var entityUri = entity === null || entity === void 0 ? void 0 : entity.uri;
+    var isCollaborationEnabledForEntity = isCollaborationEnabled && entityUri && !(0, mdm_sdk_1.isTempUri)(entityUri);
     var getCommentsCount = (0, react_1.useCallback)(function (objectIds, shouldResetCommentsMap) {
         if (shouldResetCommentsMap === void 0) { shouldResetCommentsMap = false; }
-        if (isCollaborationEnabled && entityUri && !(0, mdm_sdk_1.isTempUri)(entityUri)) {
+        if (!isCollaborationEnabledForEntity) {
+            setCommentsMap(null);
+            return;
+        }
+        var onError = function (e) {
+            console.warn('Collaboration error', e);
+            setCommentsMap(null);
+        };
+        (0, ramda_1.pipe)(mdm_sdk_1.getCommentsCount, (0, ramda_1.andThen)((0, ramda_1.map)((0, ramda_1.props)(['objectId', 'comments']))), (0, ramda_1.andThen)(ramda_1.fromPairs), (0, ramda_1.andThen)(shouldResetCommentsMap ? setCommentsMap : addCommentsToCommentsMap), (0, ramda_1.otherwise)(onError))({ uris: objectIds, tenant: tenant, collaborationPath: collaborationPath });
+    }, [addCommentsToCommentsMap, collaborationPath, isCollaborationEnabledForEntity, tenant]);
+    var getTotalCommentsCount = (0, react_1.useCallback)(function () {
+        if (isCollaborationEnabledForEntity) {
             var onError = function (e) {
                 console.warn('Collaboration error', e);
-                setCommentsMap(null);
+                setTotalCommentsCount(null);
             };
-            (0, ramda_1.pipe)(mdm_sdk_1.getCommentsCount, (0, ramda_1.andThen)((0, ramda_1.map)((0, ramda_1.props)(['objectId', 'comments']))), (0, ramda_1.andThen)(ramda_1.fromPairs), (0, ramda_1.andThen)(shouldResetCommentsMap ? setCommentsMap : addCommentsToCommentsMap), (0, ramda_1.otherwise)(onError))({ uris: objectIds, tenant: tenant, collaborationPath: collaborationPath });
+            (0, ramda_1.pipe)(mdm_sdk_1.getTotalCommentsCount, (0, ramda_1.andThen)((0, ramda_1.prop)('total')), (0, ramda_1.andThen)(setTotalCommentsCount), (0, ramda_1.otherwise)(onError))({
+                relatedObjectUri: entityUri,
+                collaborationPath: collaborationPath,
+                tenant: tenant
+            });
         }
         else {
-            setCommentsMap(null);
+            setTotalCommentsCount(null);
         }
-    }, [addCommentsToCommentsMap, collaborationPath, entityUri, isCollaborationEnabled, tenant]);
+    }, [isCollaborationEnabledForEntity, collaborationPath, entityUri, tenant]);
+    (0, react_1.useEffect)(function () {
+        if (needsTotalCount) {
+            getTotalCommentsCount();
+        }
+    }, [getTotalCommentsCount, needsTotalCount]);
     (0, react_1.useEffect)(function () {
         getCommentsCount(objectIds, true);
     }, [objectIds, getCommentsCount]);
@@ -98,21 +120,29 @@ var useCollaboration = function (_a) {
             return __assign(__assign({}, prevCommentsMap), (_a = {}, _a[uri] = __spreadArray(__spreadArray([], (prevCommentsMap[uri] || []), true), [{ commentId: comment.commentId, replies: 0, status: 'open' }], false), _a));
         });
         setComments(function (prevComments) { return __spreadArray([comment], prevComments, true); });
+        setTotalCommentsCount(function (prevCount) { return (prevCount || 0) + 1; });
     };
     var updateCommentInState = function (comment) {
         setCurrentComment(comment);
         setComments((0, ramda_1.map)((0, ramda_1.when)((0, ramda_1.propEq)('commentId', comment.commentId), (0, ramda_1.always)(comment))));
     };
     var deleteCommentFromState = function (commentId, uri) {
+        var repliesCountFromComments;
+        var repliesCountFromCommentsMap;
         setCurrentComment(null);
         setCommentsMap(function (prevCommentsMap) {
             var _a;
+            var _b, _c;
+            repliesCountFromCommentsMap = (_c = (_b = prevCommentsMap === null || prevCommentsMap === void 0 ? void 0 : prevCommentsMap[uri]) === null || _b === void 0 ? void 0 : _b.find(function (c) { return c.commentId === commentId; })) === null || _c === void 0 ? void 0 : _c.replies;
             var comments = prevCommentsMap[uri] || [];
             return __assign(__assign({}, prevCommentsMap), (_a = {}, _a[uri] = comments.filter(function (c) { return c.commentId !== commentId; }), _a));
         });
         setComments(function (prevComments) {
+            var _a;
+            repliesCountFromComments = (_a = prevComments.find(function (comment) { return comment.commentId === commentId; })) === null || _a === void 0 ? void 0 : _a.replies.length;
             return prevComments.filter(function (comment) { return comment.commentId !== commentId; });
         });
+        setTotalCommentsCount(function (prevCount) { var _a; return Math.max((prevCount || 0) - (1 + ((_a = repliesCountFromComments !== null && repliesCountFromComments !== void 0 ? repliesCountFromComments : repliesCountFromCommentsMap) !== null && _a !== void 0 ? _a : 0)), 0); });
     };
     var createComment = (0, react_1.useCallback)(function (_a) {
         var content = _a.content, namedUsers = _a.namedUsers, objectType = _a.objectType, uri = _a.uri, relatedObjectUris = _a.relatedObjectUris;
@@ -198,6 +228,12 @@ var useCollaboration = function (_a) {
                 return comment;
             });
         });
+        if (isDeleteReply) {
+            setTotalCommentsCount(function (prevCount) { return Math.max((prevCount || 0) - 1, 0); });
+        }
+        else {
+            setTotalCommentsCount(function (prevCount) { return (prevCount || 0) + 1; });
+        }
     }, []);
     var deleteReply = (0, react_1.useCallback)(function (_a) {
         var uri = _a.uri, commentId = _a.commentId, reply = _a.reply;
@@ -292,7 +328,8 @@ var useCollaboration = function (_a) {
         resolveThread: resolveThread,
         sending: sending,
         deleteReply: deleteReply,
-        editReply: editReply
+        editReply: editReply,
+        totalCommentsCount: totalCommentsCount
     }); }, [
         clearCurrentComment,
         comments,
@@ -315,7 +352,8 @@ var useCollaboration = function (_a) {
         resolveThread,
         sending,
         deleteReply,
-        editReply
+        editReply,
+        totalCommentsCount
     ]);
 };
 exports.useCollaboration = useCollaboration;

package/cjs/hooks/useCollaboration/useCollaboration.test.js

@@ -64,7 +64,7 @@ var react_hooks_1 = require("@testing-library/react-hooks");
 var mdm_sdk_1 = require("@reltio/mdm-sdk");
 var MdmModuleContext_1 = require("../../contexts/MdmModuleContext");
 var useCollaboration_1 = require("./useCollaboration");
-jest.mock('@reltio/mdm-sdk', function () { return (__assign(__assign({}, jest.requireActual('@reltio/mdm-sdk')), { getCommentsCount: jest.fn(), getComment: jest.fn(), createComment: jest.fn(), createReply: jest.fn(), updateComment: jest.fn(), deleteComment: jest.fn(), getComments: jest.fn(), deleteReply: jest.fn(), updateReply: jest.fn() })); });
+jest.mock('@reltio/mdm-sdk', function () { return (__assign(__assign({}, jest.requireActual('@reltio/mdm-sdk')), { getCommentsCount: jest.fn(), getTotalCommentsCount: jest.fn(), getComment: jest.fn(), createComment: jest.fn(), createReply: jest.fn(), updateComment: jest.fn(), deleteComment: jest.fn(), getComments: jest.fn(), deleteReply: jest.fn(), updateReply: jest.fn() })); });
 describe('useCollaboration behaviour', function () {
     var defaultMdmValues = {
         uiPath: 'https://localhost:3000/ui/alenat',
@@ -82,15 +82,20 @@ describe('useCollaboration behaviour', function () {
         return (0, react_hooks_1.renderHook)(useCollaboration_1.useCollaboration, { initialProps: props, wrapper: Providers });
     };
     beforeEach(function () {
-        jest.clearAllMocks();
         mdm_sdk_1.getComment.mockResolvedValue({ commentId: 'commentId' });
         mdm_sdk_1.getCommentsCount.mockResolvedValue([{ objectId: 'uri1', comments: [] }]);
+        mdm_sdk_1.getTotalCommentsCount.mockResolvedValue({ total: 1 });
+    });
+    afterEach(function () {
+        jest.resetAllMocks();
     });
     it('should not call getCommentsCount if collaboration is not enabled', function () {
         var mdmValues = __assign(__assign({}, defaultMdmValues), { collaborationPath: undefined, isCollaborationEnabled: false });
         var result = setUp({ mdmValues: mdmValues }).result;
         expect(result.current.commentsMap).toBe(null);
+        expect(result.current.totalCommentsCount).toBe(null);
         expect(mdm_sdk_1.getCommentsCount).not.toHaveBeenCalled();
+        expect(mdm_sdk_1.getTotalCommentsCount).not.toHaveBeenCalled();
     });
     it('should call getCommentsCount if collaboration is enabled and uris is undefined and update commentsMap value', function () { return __awaiter(void 0, void 0, void 0, function () {
         var result;
@@ -98,12 +103,8 @@ describe('useCollaboration behaviour', function () {
             switch (_a.label) {
                 case 0:
                     mdm_sdk_1.getCommentsCount.mockResolvedValueOnce([]);
+                    mdm_sdk_1.getTotalCommentsCount.mockResolvedValueOnce({ total: 1 });
                     result = setUp().result;
-                    expect(mdm_sdk_1.getCommentsCount).toHaveBeenCalledWith({
-                        uris: [],
-                        tenant: 'alenat',
-                        collaborationPath: defaultMdmValues.collaborationPath
-                    });
                     return [4 /*yield*/, (0, react_hooks_1.act)(function () { return __awaiter(void 0, void 0, void 0, function () {
                             return __generator(this, function (_a) {
                                 switch (_a.label) {
@@ -116,6 +117,11 @@ describe('useCollaboration behaviour', function () {
                         }); })];
                 case 1:
                     _a.sent();
+                    expect(mdm_sdk_1.getCommentsCount).toHaveBeenCalledWith({
+                        uris: [],
+                        tenant: 'alenat',
+                        collaborationPath: defaultMdmValues.collaborationPath
+                    });
                     expect(result.current.commentsMap).toEqual({});
                     return [2 /*return*/];
             }
@@ -1235,4 +1241,90 @@ describe('useCollaboration behaviour', function () {
             }
         });
     }); });
+    describe('totalCommentsCount behavior', function () {
+        it('should not call getTotalCommentsCount if collaboration is not enabled', function () {
+            var mdmValues = __assign(__assign({}, defaultMdmValues), { collaborationPath: undefined, isCollaborationEnabled: false });
+            var result = setUp({ mdmValues: mdmValues }).result;
+            expect(result.current.totalCommentsCount).toBe(null);
+            expect(mdm_sdk_1.getTotalCommentsCount).not.toHaveBeenCalled();
+        });
+        it('should not call getTotalCommentsCount and update totalCommentsCount value if collaboration is enabled but needsTotalCount is false', function () { return __awaiter(void 0, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        mdm_sdk_1.getTotalCommentsCount.mockResolvedValueOnce({ total: 15 });
+                        setUp();
+                        expect(mdm_sdk_1.getTotalCommentsCount).not.toHaveBeenCalled();
+                        return [4 /*yield*/, (0, react_hooks_1.act)(function () { return __awaiter(void 0, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, Promise.resolve()];
+                                        case 1:
+                                            _a.sent();
+                                            return [2 /*return*/];
+                                    }
+                                });
+                            }); })];
+                    case 1:
+                        _a.sent();
+                        expect(mdm_sdk_1.getTotalCommentsCount).not.toHaveBeenCalled();
+                        return [2 /*return*/];
+                }
+            });
+        }); });
+        it('should call getTotalCommentsCount and update totalCommentsCount value if collaboration is enabled and needsTotalCount is true', function () { return __awaiter(void 0, void 0, void 0, function () {
+            var result;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        mdm_sdk_1.getTotalCommentsCount.mockResolvedValueOnce({ total: 15 });
+                        result = setUp({ props: { needsTotalCount: true } }).result;
+                        expect(mdm_sdk_1.getTotalCommentsCount).toHaveBeenCalledWith({
+                            relatedObjectUri: '1',
+                            collaborationPath: '123',
+                            tenant: 'alenat'
+                        });
+                        return [4 /*yield*/, (0, react_hooks_1.act)(function () { return __awaiter(void 0, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, Promise.resolve()];
+                                        case 1:
+                                            _a.sent();
+                                            return [2 /*return*/];
+                                    }
+                                });
+                            }); })];
+                    case 1:
+                        _a.sent();
+                        expect(result.current.totalCommentsCount).toBe(15);
+                        return [2 /*return*/];
+                }
+            });
+        }); });
+        it('should set totalCommentsCount as null in case of unsuccessful request', function () { return __awaiter(void 0, void 0, void 0, function () {
+            var result;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        mdm_sdk_1.getTotalCommentsCount.mockRejectedValueOnce(new Error('Network error'));
+                        result = setUp({ props: { needsTotalCount: true } }).result;
+                        expect(mdm_sdk_1.getTotalCommentsCount).toHaveBeenCalled();
+                        return [4 /*yield*/, (0, react_hooks_1.act)(function () { return __awaiter(void 0, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, Promise.resolve()];
+                                        case 1:
+                                            _a.sent();
+                                            return [2 /*return*/];
+                                    }
+                                });
+                            }); })];
+                    case 1:
+                        _a.sent();
+                        expect(result.current.totalCommentsCount).toBe(null);
+                        return [2 /*return*/];
+                }
+            });
+        }); });
+    });
 });

package/contexts/CollaborationContext/index.d.ts

@@ -22,6 +22,7 @@ export declare const InitialCollaborationContextValue: {
     pageToken: any;
     deleteReply: any;
     editReply: () => Promise<void>;
+    totalCommentsCount: any;
 };
 export declare const CollaborationContext: React.Context<{
     objectTypes?: CollaborationObjectTypes[];
@@ -67,6 +68,7 @@ export declare const CollaborationContext: React.Context<{
     sending: boolean;
     loading: boolean;
     pageToken: string;
+    totalCommentsCount: number | null;
 }>;
 export declare const CollaborationContextProvider: ({ collaboration, children }: {
     collaboration: any;

package/contexts/CollaborationContext/index.js

@@ -23,7 +23,8 @@ export var InitialCollaborationContextValue = {
     loading: false,
     pageToken: null,
     deleteReply: identity,
-    editReply: function () { return new Promise(identity); }
+    editReply: function () { return new Promise(identity); },
+    totalCommentsCount: null
 };
 export var CollaborationContext = React.createContext(InitialCollaborationContextValue);
 CollaborationContext.displayName = 'CollaborationContext';

package/hooks/useCollaboration/useCollaboration.d.ts

@@ -3,8 +3,9 @@ type Props = {
     objectIds?: string[];
     objectTypes?: CollaborationObjectTypes[];
     enabled?: boolean;
+    needsTotalCount?: boolean;
 };
-export declare const useCollaboration: ({ objectIds, objectTypes, enabled }: Props) => {
+export declare const useCollaboration: ({ objectIds, objectTypes, enabled, needsTotalCount }: Props) => {
     clearCurrentComment: () => void;
     comments: Comment[];
     commentsMap: CommentsMap;
@@ -27,5 +28,6 @@ export declare const useCollaboration: ({ objectIds, objectTypes, enabled }: Pro
     sending: boolean;
     deleteReply: ({ uri, commentId, reply }: any) => void;
     editReply: ({ content, namedUsers, commentId, replyId }: any) => Promise<void>;
+    totalCommentsCount: number;
 };
 export {};

package/hooks/useCollaboration/useCollaboration.js

@@ -20,7 +20,7 @@ var __spreadArray = (this && this.__spreadArray) || function (to, from, pack) {
 };
 import { useCallback, useEffect, useState, useMemo } from 'react';
 import { andThen, fromPairs, map, otherwise, pipe, props, uniqBy, prop, propEq, when, always, assocPath, dissocPath, path, pathOr, mergeRight } from 'ramda';
-import { createComment as createCommentRequest, createReply as createReplyRequest, getComment as getCommentRequest, getCommentsCount as getCommentsCountRequest, updateComment as updateCommentRequest, updateReply as updateReplyRequest, deleteComment as deleteCommentRequest, deleteReply as deleteReplyRequest, getComments as getCommentsRequest, isTempUri } from '@reltio/mdm-sdk';
+import { createComment as createCommentRequest, createReply as createReplyRequest, getComment as getCommentRequest, getCommentsCount as getCommentsCountRequest, getTotalCommentsCount as getTotalCommentsCountRequest, updateComment as updateCommentRequest, updateReply as updateReplyRequest, deleteComment as deleteCommentRequest, deleteReply as deleteReplyRequest, getComments as getCommentsRequest, isTempUri } from '@reltio/mdm-sdk';
 import { useMdmCollaborationPath, useMdmEntity, useMdmIsCollaborationEnabled, useMdmTenant } from '../../contexts/MdmModuleContext';
 var filterReplies = function (replyId, replies) { return replies.filter(function (reply) { return reply.replyId !== replyId; }); };
 var getUpdatedReplies = function (comment, reply) {
@@ -35,35 +35,57 @@ var getUpdatedReplies = function (comment, reply) {
 var defaultObjectIds = [];
 var catchErrorInRequest = function (e) { return console.warn('Collaboration error', e); };
 export var useCollaboration = function (_a) {
-    var _b = _a.objectIds, objectIds = _b === void 0 ? defaultObjectIds : _b, objectTypes = _a.objectTypes, _c = _a.enabled, enabled = _c === void 0 ? true : _c;
+    var _b = _a.objectIds, objectIds = _b === void 0 ? defaultObjectIds : _b, objectTypes = _a.objectTypes, _c = _a.enabled, enabled = _c === void 0 ? true : _c, _d = _a.needsTotalCount, needsTotalCount = _d === void 0 ? false : _d;
     var entity = useMdmEntity();
     var tenant = useMdmTenant();
     var collaborationPath = useMdmCollaborationPath();
     var isCollaborationEnabled = useMdmIsCollaborationEnabled() && enabled;
-    var _d = useState(null), pageToken = _d[0], setPageToken = _d[1];
-    var _e = useState([]), comments = _e[0], setComments = _e[1];
-    var _f = useState(null), commentsMap = _f[0], setCommentsMap = _f[1];
-    var _g = useState(null), currentComment = _g[0], setCurrentComment = _g[1];
-    var _h = useState(false), sending = _h[0], setSending = _h[1];
-    var _j = useState(false), loading = _j[0], setLoading = _j[1];
-    var _k = useState({}), collaborationStateMap = _k[0], setCollaborationStateMap = _k[1];
+    var _e = useState(null), pageToken = _e[0], setPageToken = _e[1];
+    var _f = useState([]), comments = _f[0], setComments = _f[1];
+    var _g = useState(null), commentsMap = _g[0], setCommentsMap = _g[1];
+    var _h = useState(null), totalCommentsCount = _h[0], setTotalCommentsCount = _h[1];
+    var _j = useState(null), currentComment = _j[0], setCurrentComment = _j[1];
+    var _k = useState(false), sending = _k[0], setSending = _k[1];
+    var _l = useState(false), loading = _l[0], setLoading = _l[1];
+    var _m = useState({}), collaborationStateMap = _m[0], setCollaborationStateMap = _m[1];
     var addCommentsToCommentsMap = useCallback(function (commentsMap) {
         setCommentsMap(function (currentCommentsMap) { return (__assign(__assign({}, currentCommentsMap), commentsMap)); });
     }, []);
     var entityUri = entity === null || entity === void 0 ? void 0 : entity.uri;
+    var isCollaborationEnabledForEntity = isCollaborationEnabled && entityUri && !isTempUri(entityUri);
     var getCommentsCount = useCallback(function (objectIds, shouldResetCommentsMap) {
         if (shouldResetCommentsMap === void 0) { shouldResetCommentsMap = false; }
-        if (isCollaborationEnabled && entityUri && !isTempUri(entityUri)) {
+        if (!isCollaborationEnabledForEntity) {
+            setCommentsMap(null);
+            return;
+        }
+        var onError = function (e) {
+            console.warn('Collaboration error', e);
+            setCommentsMap(null);
+        };
+        pipe(getCommentsCountRequest, andThen(map(props(['objectId', 'comments']))), andThen(fromPairs), andThen(shouldResetCommentsMap ? setCommentsMap : addCommentsToCommentsMap), otherwise(onError))({ uris: objectIds, tenant: tenant, collaborationPath: collaborationPath });
+    }, [addCommentsToCommentsMap, collaborationPath, isCollaborationEnabledForEntity, tenant]);
+    var getTotalCommentsCount = useCallback(function () {
+        if (isCollaborationEnabledForEntity) {
             var onError = function (e) {
                 console.warn('Collaboration error', e);
-                setCommentsMap(null);
+                setTotalCommentsCount(null);
             };
-            pipe(getCommentsCountRequest, andThen(map(props(['objectId', 'comments']))), andThen(fromPairs), andThen(shouldResetCommentsMap ? setCommentsMap : addCommentsToCommentsMap), otherwise(onError))({ uris: objectIds, tenant: tenant, collaborationPath: collaborationPath });
+            pipe(getTotalCommentsCountRequest, andThen(prop('total')), andThen(setTotalCommentsCount), otherwise(onError))({
+                relatedObjectUri: entityUri,
+                collaborationPath: collaborationPath,
+                tenant: tenant
+            });
         }
         else {
-            setCommentsMap(null);
+            setTotalCommentsCount(null);
         }
-    }, [addCommentsToCommentsMap, collaborationPath, entityUri, isCollaborationEnabled, tenant]);
+    }, [isCollaborationEnabledForEntity, collaborationPath, entityUri, tenant]);
+    useEffect(function () {
+        if (needsTotalCount) {
+            getTotalCommentsCount();
+        }
+    }, [getTotalCommentsCount, needsTotalCount]);
     useEffect(function () {
         getCommentsCount(objectIds, true);
     }, [objectIds, getCommentsCount]);
@@ -95,21 +117,29 @@ export var useCollaboration = function (_a) {
             return __assign(__assign({}, prevCommentsMap), (_a = {}, _a[uri] = __spreadArray(__spreadArray([], (prevCommentsMap[uri] || []), true), [{ commentId: comment.commentId, replies: 0, status: 'open' }], false), _a));
         });
         setComments(function (prevComments) { return __spreadArray([comment], prevComments, true); });
+        setTotalCommentsCount(function (prevCount) { return (prevCount || 0) + 1; });
     };
     var updateCommentInState = function (comment) {
         setCurrentComment(comment);
         setComments(map(when(propEq('commentId', comment.commentId), always(comment))));
     };
     var deleteCommentFromState = function (commentId, uri) {
+        var repliesCountFromComments;
+        var repliesCountFromCommentsMap;
         setCurrentComment(null);
         setCommentsMap(function (prevCommentsMap) {
             var _a;
+            var _b, _c;
+            repliesCountFromCommentsMap = (_c = (_b = prevCommentsMap === null || prevCommentsMap === void 0 ? void 0 : prevCommentsMap[uri]) === null || _b === void 0 ? void 0 : _b.find(function (c) { return c.commentId === commentId; })) === null || _c === void 0 ? void 0 : _c.replies;
             var comments = prevCommentsMap[uri] || [];
             return __assign(__assign({}, prevCommentsMap), (_a = {}, _a[uri] = comments.filter(function (c) { return c.commentId !== commentId; }), _a));
         });
         setComments(function (prevComments) {
+            var _a;
+            repliesCountFromComments = (_a = prevComments.find(function (comment) { return comment.commentId === commentId; })) === null || _a === void 0 ? void 0 : _a.replies.length;
             return prevComments.filter(function (comment) { return comment.commentId !== commentId; });
         });
+        setTotalCommentsCount(function (prevCount) { var _a; return Math.max((prevCount || 0) - (1 + ((_a = repliesCountFromComments !== null && repliesCountFromComments !== void 0 ? repliesCountFromComments : repliesCountFromCommentsMap) !== null && _a !== void 0 ? _a : 0)), 0); });
     };
     var createComment = useCallback(function (_a) {
         var content = _a.content, namedUsers = _a.namedUsers, objectType = _a.objectType, uri = _a.uri, relatedObjectUris = _a.relatedObjectUris;
@@ -195,6 +225,12 @@ export var useCollaboration = function (_a) {
                 return comment;
             });
         });
+        if (isDeleteReply) {
+            setTotalCommentsCount(function (prevCount) { return Math.max((prevCount || 0) - 1, 0); });
+        }
+        else {
+            setTotalCommentsCount(function (prevCount) { return (prevCount || 0) + 1; });
+        }
     }, []);
     var deleteReply = useCallback(function (_a) {
         var uri = _a.uri, commentId = _a.commentId, reply = _a.reply;
@@ -289,7 +325,8 @@ export var useCollaboration = function (_a) {
         resolveThread: resolveThread,
         sending: sending,
         deleteReply: deleteReply,
-        editReply: editReply
+        editReply: editReply,
+        totalCommentsCount: totalCommentsCount
     }); }, [
         clearCurrentComment,
         comments,
@@ -312,6 +349,7 @@ export var useCollaboration = function (_a) {
         resolveThread,
         sending,
         deleteReply,
-        editReply
+        editReply,
+        totalCommentsCount
     ]);
 };

package/hooks/useCollaboration/useCollaboration.test.js

@@ -56,10 +56,10 @@ var __spreadArray = (this && this.__spreadArray) || function (to, from, pack) {
 };
 import React from 'react';
 import { renderHook, act } from '@testing-library/react-hooks';
-import { getCommentsCount, getComment, createComment, createReply, deleteComment, updateComment, getComments, deleteReply, updateReply } from '@reltio/mdm-sdk';
+import { getCommentsCount, getTotalCommentsCount, getComment, createComment, createReply, deleteComment, updateComment, getComments, deleteReply, updateReply } from '@reltio/mdm-sdk';
 import { MdmModuleProvider } from '../../contexts/MdmModuleContext';
 import { useCollaboration } from './useCollaboration';
-jest.mock('@reltio/mdm-sdk', function () { return (__assign(__assign({}, jest.requireActual('@reltio/mdm-sdk')), { getCommentsCount: jest.fn(), getComment: jest.fn(), createComment: jest.fn(), createReply: jest.fn(), updateComment: jest.fn(), deleteComment: jest.fn(), getComments: jest.fn(), deleteReply: jest.fn(), updateReply: jest.fn() })); });
+jest.mock('@reltio/mdm-sdk', function () { return (__assign(__assign({}, jest.requireActual('@reltio/mdm-sdk')), { getCommentsCount: jest.fn(), getTotalCommentsCount: jest.fn(), getComment: jest.fn(), createComment: jest.fn(), createReply: jest.fn(), updateComment: jest.fn(), deleteComment: jest.fn(), getComments: jest.fn(), deleteReply: jest.fn(), updateReply: jest.fn() })); });
 describe('useCollaboration behaviour', function () {
     var defaultMdmValues = {
         uiPath: 'https://localhost:3000/ui/alenat',
@@ -77,15 +77,20 @@ describe('useCollaboration behaviour', function () {
         return renderHook(useCollaboration, { initialProps: props, wrapper: Providers });
     };
     beforeEach(function () {
-        jest.clearAllMocks();
         getComment.mockResolvedValue({ commentId: 'commentId' });
         getCommentsCount.mockResolvedValue([{ objectId: 'uri1', comments: [] }]);
+        getTotalCommentsCount.mockResolvedValue({ total: 1 });
+    });
+    afterEach(function () {
+        jest.resetAllMocks();
     });
     it('should not call getCommentsCount if collaboration is not enabled', function () {
         var mdmValues = __assign(__assign({}, defaultMdmValues), { collaborationPath: undefined, isCollaborationEnabled: false });
         var result = setUp({ mdmValues: mdmValues }).result;
         expect(result.current.commentsMap).toBe(null);
+        expect(result.current.totalCommentsCount).toBe(null);
         expect(getCommentsCount).not.toHaveBeenCalled();
+        expect(getTotalCommentsCount).not.toHaveBeenCalled();
     });
     it('should call getCommentsCount if collaboration is enabled and uris is undefined and update commentsMap value', function () { return __awaiter(void 0, void 0, void 0, function () {
         var result;
@@ -93,12 +98,8 @@ describe('useCollaboration behaviour', function () {
             switch (_a.label) {
                 case 0:
                     getCommentsCount.mockResolvedValueOnce([]);
+                    getTotalCommentsCount.mockResolvedValueOnce({ total: 1 });
                     result = setUp().result;
-                    expect(getCommentsCount).toHaveBeenCalledWith({
-                        uris: [],
-                        tenant: 'alenat',
-                        collaborationPath: defaultMdmValues.collaborationPath
-                    });
                     return [4 /*yield*/, act(function () { return __awaiter(void 0, void 0, void 0, function () {
                             return __generator(this, function (_a) {
                                 switch (_a.label) {
@@ -111,6 +112,11 @@ describe('useCollaboration behaviour', function () {
                         }); })];
                 case 1:
                     _a.sent();
+                    expect(getCommentsCount).toHaveBeenCalledWith({
+                        uris: [],
+                        tenant: 'alenat',
+                        collaborationPath: defaultMdmValues.collaborationPath
+                    });
                     expect(result.current.commentsMap).toEqual({});
                     return [2 /*return*/];
             }
@@ -1230,4 +1236,90 @@ describe('useCollaboration behaviour', function () {
             }
         });
     }); });
+    describe('totalCommentsCount behavior', function () {
+        it('should not call getTotalCommentsCount if collaboration is not enabled', function () {
+            var mdmValues = __assign(__assign({}, defaultMdmValues), { collaborationPath: undefined, isCollaborationEnabled: false });
+            var result = setUp({ mdmValues: mdmValues }).result;
+            expect(result.current.totalCommentsCount).toBe(null);
+            expect(getTotalCommentsCount).not.toHaveBeenCalled();
+        });
+        it('should not call getTotalCommentsCount and update totalCommentsCount value if collaboration is enabled but needsTotalCount is false', function () { return __awaiter(void 0, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        getTotalCommentsCount.mockResolvedValueOnce({ total: 15 });
+                        setUp();
+                        expect(getTotalCommentsCount).not.toHaveBeenCalled();
+                        return [4 /*yield*/, act(function () { return __awaiter(void 0, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, Promise.resolve()];
+                                        case 1:
+                                            _a.sent();
+                                            return [2 /*return*/];
+                                    }
+                                });
+                            }); })];
+                    case 1:
+                        _a.sent();
+                        expect(getTotalCommentsCount).not.toHaveBeenCalled();
+                        return [2 /*return*/];
+                }
+            });
+        }); });
+        it('should call getTotalCommentsCount and update totalCommentsCount value if collaboration is enabled and needsTotalCount is true', function () { return __awaiter(void 0, void 0, void 0, function () {
+            var result;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        getTotalCommentsCount.mockResolvedValueOnce({ total: 15 });
+                        result = setUp({ props: { needsTotalCount: true } }).result;
+                        expect(getTotalCommentsCount).toHaveBeenCalledWith({
+                            relatedObjectUri: '1',
+                            collaborationPath: '123',
+                            tenant: 'alenat'
+                        });
+                        return [4 /*yield*/, act(function () { return __awaiter(void 0, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, Promise.resolve()];
+                                        case 1:
+                                            _a.sent();
+                                            return [2 /*return*/];
+                                    }
+                                });
+                            }); })];
+                    case 1:
+                        _a.sent();
+                        expect(result.current.totalCommentsCount).toBe(15);
+                        return [2 /*return*/];
+                }
+            });
+        }); });
+        it('should set totalCommentsCount as null in case of unsuccessful request', function () { return __awaiter(void 0, void 0, void 0, function () {
+            var result;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        getTotalCommentsCount.mockRejectedValueOnce(new Error('Network error'));
+                        result = setUp({ props: { needsTotalCount: true } }).result;
+                        expect(getTotalCommentsCount).toHaveBeenCalled();
+                        return [4 /*yield*/, act(function () { return __awaiter(void 0, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, Promise.resolve()];
+                                        case 1:
+                                            _a.sent();
+                                            return [2 /*return*/];
+                                    }
+                                });
+                            }); })];
+                    case 1:
+                        _a.sent();
+                        expect(result.current.totalCommentsCount).toBe(null);
+                        return [2 /*return*/];
+                }
+            });
+        }); });
+    });
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@reltio/components",
-  "version": "1.4.2154",
+  "version": "1.4.2156",
   "license": "SEE LICENSE IN LICENSE FILE",
   "main": "./cjs/index.js",
   "module": "./index.js",
@@ -11,7 +11,7 @@
     "@fluentui/react-context-selector": "^9.1.26",
     "@googlemaps/markerclusterer": "^2.5.3",
     "@react-sigma/core": "3.4.0",
-    "@reltio/mdm-sdk": "^1.4.1975",
+    "@reltio/mdm-sdk": "^1.4.1977",
     "@vis.gl/react-google-maps": "^1.3.0",
     "d3-cloud": "^1.2.5",
     "d3-geo": "^2.0.1",