matrix-react-sdk
Version:
SDK for matrix.org using React
220 lines (173 loc) • 23.5 kB
JavaScript
"use strict";
var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
Object.defineProperty(exports, "__esModule", {
value: true
});
exports.MessagePreviewStore = void 0;
var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
var _AsyncStoreWithClient = require("../AsyncStoreWithClient");
var _dispatcher = _interopRequireDefault(require("../../dispatcher/dispatcher"));
var _MessageEventPreview = require("./previews/MessageEventPreview");
var _utils = require("matrix-js-sdk/src/utils");
var _CallInviteEventPreview = require("./previews/CallInviteEventPreview");
var _CallAnswerEventPreview = require("./previews/CallAnswerEventPreview");
var _CallHangupEvent = require("./previews/CallHangupEvent");
var _StickerEventPreview = require("./previews/StickerEventPreview");
var _ReactionEventPreview = require("./previews/ReactionEventPreview");
var _AsyncStore = require("../AsyncStore");
/*
Copyright 2020 The Matrix.org Foundation C.I.C.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
// Emitted event for when a room's preview has changed. First argument will the room for which
// the change happened.
const ROOM_PREVIEW_CHANGED = "room_preview_changed";
const PREVIEWS = {
'm.room.message': {
isState: false,
previewer: new _MessageEventPreview.MessageEventPreview()
},
'm.call.invite': {
isState: false,
previewer: new _CallInviteEventPreview.CallInviteEventPreview()
},
'm.call.answer': {
isState: false,
previewer: new _CallAnswerEventPreview.CallAnswerEventPreview()
},
'm.call.hangup': {
isState: false,
previewer: new _CallHangupEvent.CallHangupEvent()
},
'm.sticker': {
isState: false,
previewer: new _StickerEventPreview.StickerEventPreview()
},
'm.reaction': {
isState: false,
previewer: new _ReactionEventPreview.ReactionEventPreview()
}
}; // The maximum number of events we're willing to look back on to get a preview.
const MAX_EVENTS_BACKWARDS = 50; // type merging ftw
const TAG_ANY
/*: TAG_ANY*/
= "im.vector.any";
class MessagePreviewStore extends _AsyncStoreWithClient.AsyncStoreWithClient
/*:: <IState>*/
{
// null indicates the preview is empty / irrelevant
constructor() {
super(_dispatcher.default, {});
(0, _defineProperty2.default)(this, "previews", new Map());
}
static get instance()
/*: MessagePreviewStore*/
{
return MessagePreviewStore.internalInstance;
}
static getPreviewChangedEventName(room
/*: Room*/
)
/*: string*/
{
return `${ROOM_PREVIEW_CHANGED}:${room?.roomId}`;
}
/**
* Gets the pre-translated preview for a given room
* @param room The room to get the preview for.
* @param inTagId The tag ID in which the room resides
* @returns The preview, or null if none present.
*/
async getPreviewForRoom(room
/*: Room*/
, inTagId
/*: TagID*/
)
/*: Promise<string>*/
{
if (!room) return null; // invalid room, just return nothing
if (!this.previews.has(room.roomId)) await this.generatePreview(room, inTagId);
const previews = this.previews.get(room.roomId);
if (!previews) return null;
if (!previews.has(inTagId)) {
return previews.get(TAG_ANY);
}
return previews.get(inTagId);
}
async generatePreview(room
/*: Room*/
, tagId
/*: TagID*/
) {
const events = room.timeline;
if (!events) return; // should only happen in tests
let map = this.previews.get(room.roomId);
if (!map) {
map = new Map();
this.previews.set(room.roomId, map);
} // Set the tags so we know what to generate
if (!map.has(TAG_ANY)) map.set(TAG_ANY, null);
if (tagId && !map.has(tagId)) map.set(tagId, null);
let changed = false;
for (let i = events.length - 1; i >= 0; i--) {
if (i === events.length - MAX_EVENTS_BACKWARDS) {
// limit reached - clear the preview by breaking out of the loop
break;
}
const event = events[i];
await this.matrixClient.decryptEventIfNeeded(event);
const previewDef = PREVIEWS[event.getType()];
if (!previewDef) continue;
if (previewDef.isState && (0, _utils.isNullOrUndefined)(event.getStateKey())) continue;
const anyPreview = previewDef.previewer.getTextFor(event, null);
if (!anyPreview) continue; // not previewable for some reason
changed = changed || anyPreview !== map.get(TAG_ANY);
map.set(TAG_ANY, anyPreview);
const tagsToGenerate = Array.from(map.keys()).filter(t => t !== TAG_ANY); // we did the any tag above
for (const genTagId of tagsToGenerate) {
const realTagId
/*: TagID*/
= genTagId === TAG_ANY ? null : genTagId;
const preview = previewDef.previewer.getTextFor(event, realTagId);
if (preview === anyPreview) {
changed = changed || anyPreview !== map.get(genTagId);
map.delete(genTagId);
} else {
changed = changed || preview !== map.get(genTagId);
map.set(genTagId, preview);
}
}
if (changed) {
// We've muted the underlying Map, so just emit that we've changed.
this.previews.set(room.roomId, map);
this.emit(_AsyncStore.UPDATE_EVENT, this);
this.emit(MessagePreviewStore.getPreviewChangedEventName(room), room);
}
return; // we're done
} // At this point, we didn't generate a preview so clear it
this.previews.set(room.roomId, new Map());
this.emit(_AsyncStore.UPDATE_EVENT, this);
this.emit(MessagePreviewStore.getPreviewChangedEventName(room), room);
}
async onAction(payload
/*: ActionPayload*/
) {
if (!this.matrixClient) return;
if (payload.action === 'MatrixActions.Room.timeline' || payload.action === 'MatrixActions.Event.decrypted') {
const event = payload.event; // TODO: Type out the dispatcher
if (!this.previews.has(event.getRoomId())) return; // not important
await this.generatePreview(this.matrixClient.getRoom(event.getRoomId()), TAG_ANY);
}
}
}
exports.MessagePreviewStore = MessagePreviewStore;
(0, _defineProperty2.default)(MessagePreviewStore, "internalInstance", new MessagePreviewStore());
//# sourceMappingURL=data:application/json;charset=utf-8;base64,eyJ2ZXJzaW9uIjozLCJzb3VyY2VzIjpbIi4uLy4uLy4uL3NyYy9zdG9yZXMvcm9vbS1saXN0L01lc3NhZ2VQcmV2aWV3U3RvcmUudHMiXSwibmFtZXMiOlsiUk9PTV9QUkVWSUVXX0NIQU5HRUQiLCJQUkVWSUVXUyIsImlzU3RhdGUiLCJwcmV2aWV3ZXIiLCJNZXNzYWdlRXZlbnRQcmV2aWV3IiwiQ2FsbEludml0ZUV2ZW50UHJldmlldyIsIkNhbGxBbnN3ZXJFdmVudFByZXZpZXciLCJDYWxsSGFuZ3VwRXZlbnQiLCJTdGlja2VyRXZlbnRQcmV2aWV3IiwiUmVhY3Rpb25FdmVudFByZXZpZXciLCJNQVhfRVZFTlRTX0JBQ0tXQVJEUyIsIlRBR19BTlkiLCJNZXNzYWdlUHJldmlld1N0b3JlIiwiQXN5bmNTdG9yZVdpdGhDbGllbnQiLCJjb25zdHJ1Y3RvciIsImRlZmF1bHREaXNwYXRjaGVyIiwiTWFwIiwiaW5zdGFuY2UiLCJpbnRlcm5hbEluc3RhbmNlIiwiZ2V0UHJldmlld0NoYW5nZWRFdmVudE5hbWUiLCJyb29tIiwicm9vbUlkIiwiZ2V0UHJldmlld0ZvclJvb20iLCJpblRhZ0lkIiwicHJldmlld3MiLCJoYXMiLCJnZW5lcmF0ZVByZXZpZXciLCJnZXQiLCJ0YWdJZCIsImV2ZW50cyIsInRpbWVsaW5lIiwibWFwIiwic2V0IiwiY2hhbmdlZCIsImkiLCJsZW5ndGgiLCJldmVudCIsIm1hdHJpeENsaWVudCIsImRlY3J5cHRFdmVudElmTmVlZGVkIiwicHJldmlld0RlZiIsImdldFR5cGUiLCJnZXRTdGF0ZUtleSIsImFueVByZXZpZXciLCJnZXRUZXh0Rm9yIiwidGFnc1RvR2VuZXJhdGUiLCJBcnJheSIsImZyb20iLCJrZXlzIiwiZmlsdGVyIiwidCIsImdlblRhZ0lkIiwicmVhbFRhZ0lkIiwicHJldmlldyIsImRlbGV0ZSIsImVtaXQiLCJVUERBVEVfRVZFTlQiLCJvbkFjdGlvbiIsInBheWxvYWQiLCJhY3Rpb24iLCJnZXRSb29tSWQiLCJnZXRSb29tIl0sIm1hcHBpbmdzIjoiOzs7Ozs7Ozs7OztBQWtCQTs7QUFDQTs7QUFDQTs7QUFFQTs7QUFDQTs7QUFDQTs7QUFDQTs7QUFDQTs7QUFDQTs7QUFDQTs7QUE1QkE7QUFDQTtBQUNBO0FBQ0E7QUFDQTtBQUNBO0FBQ0E7QUFDQTtBQUNBO0FBQ0E7QUFDQTtBQUNBO0FBQ0E7QUFDQTtBQUNBO0FBZ0JBO0FBQ0E7QUFDQSxNQUFNQSxvQkFBb0IsR0FBRyxzQkFBN0I7QUFFQSxNQUFNQyxRQUFRLEdBQUc7QUFDYixvQkFBa0I7QUFDZEMsSUFBQUEsT0FBTyxFQUFFLEtBREs7QUFFZEMsSUFBQUEsU0FBUyxFQUFFLElBQUlDLHdDQUFKO0FBRkcsR0FETDtBQUtiLG1CQUFpQjtBQUNiRixJQUFBQSxPQUFPLEVBQUUsS0FESTtBQUViQyxJQUFBQSxTQUFTLEVBQUUsSUFBSUUsOENBQUo7QUFGRSxHQUxKO0FBU2IsbUJBQWlCO0FBQ2JILElBQUFBLE9BQU8sRUFBRSxLQURJO0FBRWJDLElBQUFBLFNBQVMsRUFBRSxJQUFJRyw4Q0FBSjtBQUZFLEdBVEo7QUFhYixtQkFBaUI7QUFDYkosSUFBQUEsT0FBTyxFQUFFLEtBREk7QUFFYkMsSUFBQUEsU0FBUyxFQUFFLElBQUlJLGdDQUFKO0FBRkUsR0FiSjtBQWlCYixlQUFhO0FBQ1RMLElBQUFBLE9BQU8sRUFBRSxLQURBO0FBRVRDLElBQUFBLFNBQVMsRUFBRSxJQUFJSyx3Q0FBSjtBQUZGLEdBakJBO0FBcUJiLGdCQUFjO0FBQ1ZOLElBQUFBLE9BQU8sRUFBRSxLQURDO0FBRVZDLElBQUFBLFNBQVMsRUFBRSxJQUFJTSwwQ0FBSjtBQUZEO0FBckJELENBQWpCLEMsQ0EyQkE7O0FBQ0EsTUFBTUMsb0JBQW9CLEdBQUcsRUFBN0IsQyxDQUVBOztBQUVBLE1BQU1DO0FBQWdCO0FBQUEsRUFBRyxlQUF6Qjs7QUFNTyxNQUFNQyxtQkFBTixTQUFrQ0M7QUFBbEM7QUFBK0Q7QUFHbEU7QUFHUUMsRUFBQUEsV0FBUixHQUFzQjtBQUNsQixVQUFNQyxtQkFBTixFQUF5QixFQUF6QjtBQURrQixvREFGSCxJQUFJQyxHQUFKLEVBRUc7QUFFckI7O0FBRUQsYUFBa0JDLFFBQWxCO0FBQUE7QUFBa0Q7QUFDOUMsV0FBT0wsbUJBQW1CLENBQUNNLGdCQUEzQjtBQUNIOztBQUVELFNBQWNDLDBCQUFkLENBQXlDQztBQUF6QztBQUFBO0FBQUE7QUFBNkQ7QUFDekQsV0FBUSxHQUFFcEIsb0JBQXFCLElBQUdvQixJQUFJLEVBQUVDLE1BQU8sRUFBL0M7QUFDSDtBQUVEO0FBQ0o7QUFDQTtBQUNBO0FBQ0E7QUFDQTs7O0FBQ0ksUUFBYUMsaUJBQWIsQ0FBK0JGO0FBQS9CO0FBQUEsSUFBMkNHO0FBQTNDO0FBQUE7QUFBQTtBQUE0RTtBQUN4RSxRQUFJLENBQUNILElBQUwsRUFBVyxPQUFPLElBQVAsQ0FENkQsQ0FDaEQ7O0FBRXhCLFFBQUksQ0FBQyxLQUFLSSxRQUFMLENBQWNDLEdBQWQsQ0FBa0JMLElBQUksQ0FBQ0MsTUFBdkIsQ0FBTCxFQUFxQyxNQUFNLEtBQUtLLGVBQUwsQ0FBcUJOLElBQXJCLEVBQTJCRyxPQUEzQixDQUFOO0FBRXJDLFVBQU1DLFFBQVEsR0FBRyxLQUFLQSxRQUFMLENBQWNHLEdBQWQsQ0FBa0JQLElBQUksQ0FBQ0MsTUFBdkIsQ0FBakI7QUFDQSxRQUFJLENBQUNHLFFBQUwsRUFBZSxPQUFPLElBQVA7O0FBRWYsUUFBSSxDQUFDQSxRQUFRLENBQUNDLEdBQVQsQ0FBYUYsT0FBYixDQUFMLEVBQTRCO0FBQ3hCLGFBQU9DLFFBQVEsQ0FBQ0csR0FBVCxDQUFhaEIsT0FBYixDQUFQO0FBQ0g7O0FBQ0QsV0FBT2EsUUFBUSxDQUFDRyxHQUFULENBQWFKLE9BQWIsQ0FBUDtBQUNIOztBQUVELFFBQWNHLGVBQWQsQ0FBOEJOO0FBQTlCO0FBQUEsSUFBMENRO0FBQTFDO0FBQUEsSUFBeUQ7QUFDckQsVUFBTUMsTUFBTSxHQUFHVCxJQUFJLENBQUNVLFFBQXBCO0FBQ0EsUUFBSSxDQUFDRCxNQUFMLEVBQWEsT0FGd0MsQ0FFaEM7O0FBRXJCLFFBQUlFLEdBQUcsR0FBRyxLQUFLUCxRQUFMLENBQWNHLEdBQWQsQ0FBa0JQLElBQUksQ0FBQ0MsTUFBdkIsQ0FBVjs7QUFDQSxRQUFJLENBQUNVLEdBQUwsRUFBVTtBQUNOQSxNQUFBQSxHQUFHLEdBQUcsSUFBSWYsR0FBSixFQUFOO0FBQ0EsV0FBS1EsUUFBTCxDQUFjUSxHQUFkLENBQWtCWixJQUFJLENBQUNDLE1BQXZCLEVBQStCVSxHQUEvQjtBQUNILEtBUm9ELENBVXJEOzs7QUFDQSxRQUFJLENBQUNBLEdBQUcsQ0FBQ04sR0FBSixDQUFRZCxPQUFSLENBQUwsRUFBdUJvQixHQUFHLENBQUNDLEdBQUosQ0FBUXJCLE9BQVIsRUFBaUIsSUFBakI7QUFDdkIsUUFBSWlCLEtBQUssSUFBSSxDQUFDRyxHQUFHLENBQUNOLEdBQUosQ0FBUUcsS0FBUixDQUFkLEVBQThCRyxHQUFHLENBQUNDLEdBQUosQ0FBUUosS0FBUixFQUFlLElBQWY7QUFFOUIsUUFBSUssT0FBTyxHQUFHLEtBQWQ7O0FBQ0EsU0FBSyxJQUFJQyxDQUFDLEdBQUdMLE1BQU0sQ0FBQ00sTUFBUCxHQUFnQixDQUE3QixFQUFnQ0QsQ0FBQyxJQUFJLENBQXJDLEVBQXdDQSxDQUFDLEVBQXpDLEVBQTZDO0FBQ3pDLFVBQUlBLENBQUMsS0FBS0wsTUFBTSxDQUFDTSxNQUFQLEdBQWdCekIsb0JBQTFCLEVBQWdEO0FBQzVDO0FBQ0E7QUFDSDs7QUFFRCxZQUFNMEIsS0FBSyxHQUFHUCxNQUFNLENBQUNLLENBQUQsQ0FBcEI7QUFFQSxZQUFNLEtBQUtHLFlBQUwsQ0FBa0JDLG9CQUFsQixDQUF1Q0YsS0FBdkMsQ0FBTjtBQUVBLFlBQU1HLFVBQVUsR0FBR3RDLFFBQVEsQ0FBQ21DLEtBQUssQ0FBQ0ksT0FBTixFQUFELENBQTNCO0FBQ0EsVUFBSSxDQUFDRCxVQUFMLEVBQWlCO0FBQ2pCLFVBQUlBLFVBQVUsQ0FBQ3JDLE9BQVgsSUFBc0IsOEJBQWtCa0MsS0FBSyxDQUFDSyxXQUFOLEVBQWxCLENBQTFCLEVBQWtFO0FBRWxFLFlBQU1DLFVBQVUsR0FBR0gsVUFBVSxDQUFDcEMsU0FBWCxDQUFxQndDLFVBQXJCLENBQWdDUCxLQUFoQyxFQUF1QyxJQUF2QyxDQUFuQjtBQUNBLFVBQUksQ0FBQ00sVUFBTCxFQUFpQixTQWZ3QixDQWVkOztBQUUzQlQsTUFBQUEsT0FBTyxHQUFHQSxPQUFPLElBQUlTLFVBQVUsS0FBS1gsR0FBRyxDQUFDSixHQUFKLENBQVFoQixPQUFSLENBQXBDO0FBQ0FvQixNQUFBQSxHQUFHLENBQUNDLEdBQUosQ0FBUXJCLE9BQVIsRUFBaUIrQixVQUFqQjtBQUVBLFlBQU1FLGNBQWMsR0FBR0MsS0FBSyxDQUFDQyxJQUFOLENBQVdmLEdBQUcsQ0FBQ2dCLElBQUosRUFBWCxFQUF1QkMsTUFBdkIsQ0FBOEJDLENBQUMsSUFBSUEsQ0FBQyxLQUFLdEMsT0FBekMsQ0FBdkIsQ0FwQnlDLENBb0JpQzs7QUFDMUUsV0FBSyxNQUFNdUMsUUFBWCxJQUF1Qk4sY0FBdkIsRUFBdUM7QUFDbkMsY0FBTU87QUFBZ0I7QUFBQSxVQUFHRCxRQUFRLEtBQUt2QyxPQUFiLEdBQXVCLElBQXZCLEdBQThCdUMsUUFBdkQ7QUFDQSxjQUFNRSxPQUFPLEdBQUdiLFVBQVUsQ0FBQ3BDLFNBQVgsQ0FBcUJ3QyxVQUFyQixDQUFnQ1AsS0FBaEMsRUFBdUNlLFNBQXZDLENBQWhCOztBQUNBLFlBQUlDLE9BQU8sS0FBS1YsVUFBaEIsRUFBNEI7QUFDeEJULFVBQUFBLE9BQU8sR0FBR0EsT0FBTyxJQUFJUyxVQUFVLEtBQUtYLEdBQUcsQ0FBQ0osR0FBSixDQUFRdUIsUUFBUixDQUFwQztBQUNBbkIsVUFBQUEsR0FBRyxDQUFDc0IsTUFBSixDQUFXSCxRQUFYO0FBQ0gsU0FIRCxNQUdPO0FBQ0hqQixVQUFBQSxPQUFPLEdBQUdBLE9BQU8sSUFBSW1CLE9BQU8sS0FBS3JCLEdBQUcsQ0FBQ0osR0FBSixDQUFRdUIsUUFBUixDQUFqQztBQUNBbkIsVUFBQUEsR0FBRyxDQUFDQyxHQUFKLENBQVFrQixRQUFSLEVBQWtCRSxPQUFsQjtBQUNIO0FBQ0o7O0FBRUQsVUFBSW5CLE9BQUosRUFBYTtBQUNUO0FBQ0EsYUFBS1QsUUFBTCxDQUFjUSxHQUFkLENBQWtCWixJQUFJLENBQUNDLE1BQXZCLEVBQStCVSxHQUEvQjtBQUNBLGFBQUt1QixJQUFMLENBQVVDLHdCQUFWLEVBQXdCLElBQXhCO0FBQ0EsYUFBS0QsSUFBTCxDQUFVMUMsbUJBQW1CLENBQUNPLDBCQUFwQixDQUErQ0MsSUFBL0MsQ0FBVixFQUFnRUEsSUFBaEU7QUFDSDs7QUFDRCxhQXZDeUMsQ0F1Q2pDO0FBQ1gsS0F2RG9ELENBeURyRDs7O0FBQ0EsU0FBS0ksUUFBTCxDQUFjUSxHQUFkLENBQWtCWixJQUFJLENBQUNDLE1BQXZCLEVBQStCLElBQUlMLEdBQUosRUFBL0I7QUFDQSxTQUFLc0MsSUFBTCxDQUFVQyx3QkFBVixFQUF3QixJQUF4QjtBQUNBLFNBQUtELElBQUwsQ0FBVTFDLG1CQUFtQixDQUFDTywwQkFBcEIsQ0FBK0NDLElBQS9DLENBQVYsRUFBZ0VBLElBQWhFO0FBQ0g7O0FBRUQsUUFBZ0JvQyxRQUFoQixDQUF5QkM7QUFBekI7QUFBQSxJQUFpRDtBQUM3QyxRQUFJLENBQUMsS0FBS3BCLFlBQVYsRUFBd0I7O0FBRXhCLFFBQUlvQixPQUFPLENBQUNDLE1BQVIsS0FBbUIsNkJBQW5CLElBQW9ERCxPQUFPLENBQUNDLE1BQVIsS0FBbUIsK0JBQTNFLEVBQTRHO0FBQ3hHLFlBQU10QixLQUFLLEdBQUdxQixPQUFPLENBQUNyQixLQUF0QixDQUR3RyxDQUMzRTs7QUFDN0IsVUFBSSxDQUFDLEtBQUtaLFFBQUwsQ0FBY0MsR0FBZCxDQUFrQlcsS0FBSyxDQUFDdUIsU0FBTixFQUFsQixDQUFMLEVBQTJDLE9BRjZELENBRXJEOztBQUNuRCxZQUFNLEtBQUtqQyxlQUFMLENBQXFCLEtBQUtXLFlBQUwsQ0FBa0J1QixPQUFsQixDQUEwQnhCLEtBQUssQ0FBQ3VCLFNBQU4sRUFBMUIsQ0FBckIsRUFBbUVoRCxPQUFuRSxDQUFOO0FBQ0g7QUFDSjs7QUE3R2lFOzs7OEJBQXpEQyxtQixzQkFDeUIsSUFBSUEsbUJBQUosRSIsInNvdXJjZXNDb250ZW50IjpbIi8qXG5Db3B5cmlnaHQgMjAyMCBUaGUgTWF0cml4Lm9yZyBGb3VuZGF0aW9uIEMuSS5DLlxuXG5MaWNlbnNlZCB1bmRlciB0aGUgQXBhY2hlIExpY2Vuc2UsIFZlcnNpb24gMi4wICh0aGUgXCJMaWNlbnNlXCIpO1xueW91IG1heSBub3QgdXNlIHRoaXMgZmlsZSBleGNlcHQgaW4gY29tcGxpYW5jZSB3aXRoIHRoZSBMaWNlbnNlLlxuWW91IG1heSBvYnRhaW4gYSBjb3B5IG9mIHRoZSBMaWNlbnNlIGF0XG5cbiAgICBodHRwOi8vd3d3LmFwYWNoZS5vcmcvbGljZW5zZXMvTElDRU5TRS0yLjBcblxuVW5sZXNzIHJlcXVpcmVkIGJ5IGFwcGxpY2FibGUgbGF3IG9yIGFncmVlZCB0byBpbiB3cml0aW5nLCBzb2Z0d2FyZVxuZGlzdHJpYnV0ZWQgdW5kZXIgdGhlIExpY2Vuc2UgaXMgZGlzdHJpYnV0ZWQgb24gYW4gXCJBUyBJU1wiIEJBU0lTLFxuV0lUSE9VVCBXQVJSQU5USUVTIE9SIENPTkRJVElPTlMgT0YgQU5ZIEtJTkQsIGVpdGhlciBleHByZXNzIG9yIGltcGxpZWQuXG5TZWUgdGhlIExpY2Vuc2UgZm9yIHRoZSBzcGVjaWZpYyBsYW5ndWFnZSBnb3Zlcm5pbmcgcGVybWlzc2lvbnMgYW5kXG5saW1pdGF0aW9ucyB1bmRlciB0aGUgTGljZW5zZS5cbiovXG5cbmltcG9ydCB7IFJvb20gfSBmcm9tIFwibWF0cml4LWpzLXNkay9zcmMvbW9kZWxzL3Jvb21cIjtcbmltcG9ydCB7IEFjdGlvblBheWxvYWQgfSBmcm9tIFwiLi4vLi4vZGlzcGF0Y2hlci9wYXlsb2Fkc1wiO1xuaW1wb3J0IHsgQXN5bmNTdG9yZVdpdGhDbGllbnQgfSBmcm9tIFwiLi4vQXN5bmNTdG9yZVdpdGhDbGllbnRcIjtcbmltcG9ydCBkZWZhdWx0RGlzcGF0Y2hlciBmcm9tIFwiLi4vLi4vZGlzcGF0Y2hlci9kaXNwYXRjaGVyXCI7XG5pbXBvcnQgeyBNZXNzYWdlRXZlbnRQcmV2aWV3IH0gZnJvbSBcIi4vcHJldmlld3MvTWVzc2FnZUV2ZW50UHJldmlld1wiO1xuaW1wb3J0IHsgVGFnSUQgfSBmcm9tIFwiLi9tb2RlbHNcIjtcbmltcG9ydCB7IGlzTnVsbE9yVW5kZWZpbmVkIH0gZnJvbSBcIm1hdHJpeC1qcy1zZGsvc3JjL3V0aWxzXCI7XG5pbXBvcnQgeyBDYWxsSW52aXRlRXZlbnRQcmV2aWV3IH0gZnJvbSBcIi4vcHJldmlld3MvQ2FsbEludml0ZUV2ZW50UHJldmlld1wiO1xuaW1wb3J0IHsgQ2FsbEFuc3dlckV2ZW50UHJldmlldyB9IGZyb20gXCIuL3ByZXZpZXdzL0NhbGxBbnN3ZXJFdmVudFByZXZpZXdcIjtcbmltcG9ydCB7IENhbGxIYW5ndXBFdmVudCB9IGZyb20gXCIuL3ByZXZpZXdzL0NhbGxIYW5ndXBFdmVudFwiO1xuaW1wb3J0IHsgU3RpY2tlckV2ZW50UHJldmlldyB9IGZyb20gXCIuL3ByZXZpZXdzL1N0aWNrZXJFdmVudFByZXZpZXdcIjtcbmltcG9ydCB7IFJlYWN0aW9uRXZlbnRQcmV2aWV3IH0gZnJvbSBcIi4vcHJldmlld3MvUmVhY3Rpb25FdmVudFByZXZpZXdcIjtcbmltcG9ydCB7IFVQREFURV9FVkVOVCB9IGZyb20gXCIuLi9Bc3luY1N0b3JlXCI7XG5cbi8vIEVtaXR0ZWQgZXZlbnQgZm9yIHdoZW4gYSByb29tJ3MgcHJldmlldyBoYXMgY2hhbmdlZC4gRmlyc3QgYXJndW1lbnQgd2lsbCB0aGUgcm9vbSBmb3Igd2hpY2hcbi8vIHRoZSBjaGFuZ2UgaGFwcGVuZWQuXG5jb25zdCBST09NX1BSRVZJRVdfQ0hBTkdFRCA9IFwicm9vbV9wcmV2aWV3X2NoYW5nZWRcIjtcblxuY29uc3QgUFJFVklFV1MgPSB7XG4gICAgJ20ucm9vbS5tZXNzYWdlJzoge1xuICAgICAgICBpc1N0YXRlOiBmYWxzZSxcbiAgICAgICAgcHJldmlld2VyOiBuZXcgTWVzc2FnZUV2ZW50UHJldmlldygpLFxuICAgIH0sXG4gICAgJ20uY2FsbC5pbnZpdGUnOiB7XG4gICAgICAgIGlzU3RhdGU6IGZhbHNlLFxuICAgICAgICBwcmV2aWV3ZXI6IG5ldyBDYWxsSW52aXRlRXZlbnRQcmV2aWV3KCksXG4gICAgfSxcbiAgICAnbS5jYWxsLmFuc3dlcic6IHtcbiAgICAgICAgaXNTdGF0ZTogZmFsc2UsXG4gICAgICAgIHByZXZpZXdlcjogbmV3IENhbGxBbnN3ZXJFdmVudFByZXZpZXcoKSxcbiAgICB9LFxuICAgICdtLmNhbGwuaGFuZ3VwJzoge1xuICAgICAgICBpc1N0YXRlOiBmYWxzZSxcbiAgICAgICAgcHJldmlld2VyOiBuZXcgQ2FsbEhhbmd1cEV2ZW50KCksXG4gICAgfSxcbiAgICAnbS5zdGlja2VyJzoge1xuICAgICAgICBpc1N0YXRlOiBmYWxzZSxcbiAgICAgICAgcHJldmlld2VyOiBuZXcgU3RpY2tlckV2ZW50UHJldmlldygpLFxuICAgIH0sXG4gICAgJ20ucmVhY3Rpb24nOiB7XG4gICAgICAgIGlzU3RhdGU6IGZhbHNlLFxuICAgICAgICBwcmV2aWV3ZXI6IG5ldyBSZWFjdGlvbkV2ZW50UHJldmlldygpLFxuICAgIH0sXG59O1xuXG4vLyBUaGUgbWF4aW11bSBudW1iZXIgb2YgZXZlbnRzIHdlJ3JlIHdpbGxpbmcgdG8gbG9vayBiYWNrIG9uIHRvIGdldCBhIHByZXZpZXcuXG5jb25zdCBNQVhfRVZFTlRTX0JBQ0tXQVJEUyA9IDUwO1xuXG4vLyB0eXBlIG1lcmdpbmcgZnR3XG50eXBlIFRBR19BTlkgPSBcImltLnZlY3Rvci5hbnlcIjtcbmNvbnN0IFRBR19BTlk6IFRBR19BTlkgPSBcImltLnZlY3Rvci5hbnlcIjtcblxuaW50ZXJmYWNlIElTdGF0ZSB7XG4gICAgLy8gRW1wdHkgYmVjYXVzZSB3ZSBkb24ndCBhY3R1YWxseSB1c2UgdGhlIHN0YXRlXG59XG5cbmV4cG9ydCBjbGFzcyBNZXNzYWdlUHJldmlld1N0b3JlIGV4dGVuZHMgQXN5bmNTdG9yZVdpdGhDbGllbnQ8SVN0YXRlPiB7XG4gICAgcHJpdmF0ZSBzdGF0aWMgaW50ZXJuYWxJbnN0YW5jZSA9IG5ldyBNZXNzYWdlUHJldmlld1N0b3JlKCk7XG5cbiAgICAvLyBudWxsIGluZGljYXRlcyB0aGUgcHJldmlldyBpcyBlbXB0eSAvIGlycmVsZXZhbnRcbiAgICBwcml2YXRlIHByZXZpZXdzID0gbmV3IE1hcDxzdHJpbmcsIE1hcDxUYWdJRHxUQUdfQU5ZLCBzdHJpbmd8bnVsbD4+KCk7XG5cbiAgICBwcml2YXRlIGNvbnN0cnVjdG9yKCkge1xuICAgICAgICBzdXBlcihkZWZhdWx0RGlzcGF0Y2hlciwge30pO1xuICAgIH1cblxuICAgIHB1YmxpYyBzdGF0aWMgZ2V0IGluc3RhbmNlKCk6IE1lc3NhZ2VQcmV2aWV3U3RvcmUge1xuICAgICAgICByZXR1cm4gTWVzc2FnZVByZXZpZXdTdG9yZS5pbnRlcm5hbEluc3RhbmNlO1xuICAgIH1cblxuICAgIHB1YmxpYyBzdGF0aWMgZ2V0UHJldmlld0NoYW5nZWRFdmVudE5hbWUocm9vbTogUm9vbSk6IHN0cmluZyB7XG4gICAgICAgIHJldHVybiBgJHtST09NX1BSRVZJRVdfQ0hBTkdFRH06JHtyb29tPy5yb29tSWR9YDtcbiAgICB9XG5cbiAgICAvKipcbiAgICAgKiBHZXRzIHRoZSBwcmUtdHJhbnNsYXRlZCBwcmV2aWV3IGZvciBhIGdpdmVuIHJvb21cbiAgICAgKiBAcGFyYW0gcm9vbSBUaGUgcm9vbSB0byBnZXQgdGhlIHByZXZpZXcgZm9yLlxuICAgICAqIEBwYXJhbSBpblRhZ0lkIFRoZSB0YWcgSUQgaW4gd2hpY2ggdGhlIHJvb20gcmVzaWRlc1xuICAgICAqIEByZXR1cm5zIFRoZSBwcmV2aWV3LCBvciBudWxsIGlmIG5vbmUgcHJlc2VudC5cbiAgICAgKi9cbiAgICBwdWJsaWMgYXN5bmMgZ2V0UHJldmlld0ZvclJvb20ocm9vbTogUm9vbSwgaW5UYWdJZDogVGFnSUQpOiBQcm9taXNlPHN0cmluZz4ge1xuICAgICAgICBpZiAoIXJvb20pIHJldHVybiBudWxsOyAvLyBpbnZhbGlkIHJvb20sIGp1c3QgcmV0dXJuIG5vdGhpbmdcblxuICAgICAgICBpZiAoIXRoaXMucHJldmlld3MuaGFzKHJvb20ucm9vbUlkKSkgYXdhaXQgdGhpcy5nZW5lcmF0ZVByZXZpZXcocm9vbSwgaW5UYWdJZCk7XG5cbiAgICAgICAgY29uc3QgcHJldmlld3MgPSB0aGlzLnByZXZpZXdzLmdldChyb29tLnJvb21JZCk7XG4gICAgICAgIGlmICghcHJldmlld3MpIHJldHVybiBudWxsO1xuXG4gICAgICAgIGlmICghcHJldmlld3MuaGFzKGluVGFnSWQpKSB7XG4gICAgICAgICAgICByZXR1cm4gcHJldmlld3MuZ2V0KFRBR19BTlkpO1xuICAgICAgICB9XG4gICAgICAgIHJldHVybiBwcmV2aWV3cy5nZXQoaW5UYWdJZCk7XG4gICAgfVxuXG4gICAgcHJpdmF0ZSBhc3luYyBnZW5lcmF0ZVByZXZpZXcocm9vbTogUm9vbSwgdGFnSWQ/OiBUYWdJRCkge1xuICAgICAgICBjb25zdCBldmVudHMgPSByb29tLnRpbWVsaW5lO1xuICAgICAgICBpZiAoIWV2ZW50cykgcmV0dXJuOyAvLyBzaG91bGQgb25seSBoYXBwZW4gaW4gdGVzdHNcblxuICAgICAgICBsZXQgbWFwID0gdGhpcy5wcmV2aWV3cy5nZXQocm9vbS5yb29tSWQpO1xuICAgICAgICBpZiAoIW1hcCkge1xuICAgICAgICAgICAgbWFwID0gbmV3IE1hcDxUYWdJRCB8IFRBR19BTlksIHN0cmluZyB8IG51bGw+KCk7XG4gICAgICAgICAgICB0aGlzLnByZXZpZXdzLnNldChyb29tLnJvb21JZCwgbWFwKTtcbiAgICAgICAgfVxuXG4gICAgICAgIC8vIFNldCB0aGUgdGFncyBzbyB3ZSBrbm93IHdoYXQgdG8gZ2VuZXJhdGVcbiAgICAgICAgaWYgKCFtYXAuaGFzKFRBR19BTlkpKSBtYXAuc2V0KFRBR19BTlksIG51bGwpO1xuICAgICAgICBpZiAodGFnSWQgJiYgIW1hcC5oYXModGFnSWQpKSBtYXAuc2V0KHRhZ0lkLCBudWxsKTtcblxuICAgICAgICBsZXQgY2hhbmdlZCA9IGZhbHNlO1xuICAgICAgICBmb3IgKGxldCBpID0gZXZlbnRzLmxlbmd0aCAtIDE7IGkgPj0gMDsgaS0tKSB7XG4gICAgICAgICAgICBpZiAoaSA9PT0gZXZlbnRzLmxlbmd0aCAtIE1BWF9FVkVOVFNfQkFDS1dBUkRTKSB7XG4gICAgICAgICAgICAgICAgLy8gbGltaXQgcmVhY2hlZCAtIGNsZWFyIHRoZSBwcmV2aWV3IGJ5IGJyZWFraW5nIG91dCBvZiB0aGUgbG9vcFxuICAgICAgICAgICAgICAgIGJyZWFrO1xuICAgICAgICAgICAgfVxuXG4gICAgICAgICAgICBjb25zdCBldmVudCA9IGV2ZW50c1tpXTtcblxuICAgICAgICAgICAgYXdhaXQgdGhpcy5tYXRyaXhDbGllbnQuZGVjcnlwdEV2ZW50SWZOZWVkZWQoZXZlbnQpO1xuXG4gICAgICAgICAgICBjb25zdCBwcmV2aWV3RGVmID0gUFJFVklFV1NbZXZlbnQuZ2V0VHlwZSgpXTtcbiAgICAgICAgICAgIGlmICghcHJldmlld0RlZikgY29udGludWU7XG4gICAgICAgICAgICBpZiAocHJldmlld0RlZi5pc1N0YXRlICYmIGlzTnVsbE9yVW5kZWZpbmVkKGV2ZW50LmdldFN0YXRlS2V5KCkpKSBjb250aW51ZTtcblxuICAgICAgICAgICAgY29uc3QgYW55UHJldmlldyA9IHByZXZpZXdEZWYucHJldmlld2VyLmdldFRleHRGb3IoZXZlbnQsIG51bGwpO1xuICAgICAgICAgICAgaWYgKCFhbnlQcmV2aWV3KSBjb250aW51ZTsgLy8gbm90IHByZXZpZXdhYmxlIGZvciBzb21lIHJlYXNvblxuXG4gICAgICAgICAgICBjaGFuZ2VkID0gY2hhbmdlZCB8fCBhbnlQcmV2aWV3ICE9PSBtYXAuZ2V0KFRBR19BTlkpO1xuICAgICAgICAgICAgbWFwLnNldChUQUdfQU5ZLCBhbnlQcmV2aWV3KTtcblxuICAgICAgICAgICAgY29uc3QgdGFnc1RvR2VuZXJhdGUgPSBBcnJheS5mcm9tKG1hcC5rZXlzKCkpLmZpbHRlcih0ID0+IHQgIT09IFRBR19BTlkpOyAvLyB3ZSBkaWQgdGhlIGFueSB0YWcgYWJvdmVcbiAgICAgICAgICAgIGZvciAoY29uc3QgZ2VuVGFnSWQgb2YgdGFnc1RvR2VuZXJhdGUpIHtcbiAgICAgICAgICAgICAgICBjb25zdCByZWFsVGFnSWQ6IFRhZ0lEID0gZ2VuVGFnSWQgPT09IFRBR19BTlkgPyBudWxsIDogZ2VuVGFnSWQ7XG4gICAgICAgICAgICAgICAgY29uc3QgcHJldmlldyA9IHByZXZpZXdEZWYucHJldmlld2VyLmdldFRleHRGb3IoZXZlbnQsIHJlYWxUYWdJZCk7XG4gICAgICAgICAgICAgICAgaWYgKHByZXZpZXcgPT09IGFueVByZXZpZXcpIHtcbiAgICAgICAgICAgICAgICAgICAgY2hhbmdlZCA9IGNoYW5nZWQgfHwgYW55UHJldmlldyAhPT0gbWFwLmdldChnZW5UYWdJZCk7XG4gICAgICAgICAgICAgICAgICAgIG1hcC5kZWxldGUoZ2VuVGFnSWQpO1xuICAgICAgICAgICAgICAgIH0gZWxzZSB7XG4gICAgICAgICAgICAgICAgICAgIGNoYW5nZWQgPSBjaGFuZ2VkIHx8IHByZXZpZXcgIT09IG1hcC5nZXQoZ2VuVGFnSWQpO1xuICAgICAgICAgICAgICAgICAgICBtYXAuc2V0KGdlblRhZ0lkLCBwcmV2aWV3KTtcbiAgICAgICAgICAgICAgICB9XG4gICAgICAgICAgICB9XG5cbiAgICAgICAgICAgIGlmIChjaGFuZ2VkKSB7XG4gICAgICAgICAgICAgICAgLy8gV2UndmUgbXV0ZWQgdGhlIHVuZGVybHlpbmcgTWFwLCBzbyBqdXN0IGVtaXQgdGhhdCB3ZSd2ZSBjaGFuZ2VkLlxuICAgICAgICAgICAgICAgIHRoaXMucHJldmlld3Muc2V0KHJvb20ucm9vbUlkLCBtYXApO1xuICAgICAgICAgICAgICAgIHRoaXMuZW1pdChVUERBVEVfRVZFTlQsIHRoaXMpO1xuICAgICAgICAgICAgICAgIHRoaXMuZW1pdChNZXNzYWdlUHJldmlld1N0b3JlLmdldFByZXZpZXdDaGFuZ2VkRXZlbnROYW1lKHJvb20pLCByb29tKTtcbiAgICAgICAgICAgIH1cbiAgICAgICAgICAgIHJldHVybjsgLy8gd2UncmUgZG9uZVxuICAgICAgICB9XG5cbiAgICAgICAgLy8gQXQgdGhpcyBwb2ludCwgd2UgZGlkbid0IGdlbmVyYXRlIGEgcHJldmlldyBzbyBjbGVhciBpdFxuICAgICAgICB0aGlzLnByZXZpZXdzLnNldChyb29tLnJvb21JZCwgbmV3IE1hcDxUYWdJRHxUQUdfQU5ZLCBzdHJpbmd8bnVsbD4oKSk7XG4gICAgICAgIHRoaXMuZW1pdChVUERBVEVfRVZFTlQsIHRoaXMpO1xuICAgICAgICB0aGlzLmVtaXQoTWVzc2FnZVByZXZpZXdTdG9yZS5nZXRQcmV2aWV3Q2hhbmdlZEV2ZW50TmFtZShyb29tKSwgcm9vbSk7XG4gICAgfVxuXG4gICAgcHJvdGVjdGVkIGFzeW5jIG9uQWN0aW9uKHBheWxvYWQ6IEFjdGlvblBheWxvYWQpIHtcbiAgICAgICAgaWYgKCF0aGlzLm1hdHJpeENsaWVudCkgcmV0dXJuO1xuXG4gICAgICAgIGlmIChwYXlsb2FkLmFjdGlvbiA9PT0gJ01hdHJpeEFjdGlvbnMuUm9vbS50aW1lbGluZScgfHwgcGF5bG9hZC5hY3Rpb24gPT09ICdNYXRyaXhBY3Rpb25zLkV2ZW50LmRlY3J5cHRlZCcpIHtcbiAgICAgICAgICAgIGNvbnN0IGV2ZW50ID0gcGF5bG9hZC5ldmVudDsgLy8gVE9ETzogVHlwZSBvdXQgdGhlIGRpc3BhdGNoZXJcbiAgICAgICAgICAgIGlmICghdGhpcy5wcmV2aWV3cy5oYXMoZXZlbnQuZ2V0Um9vbUlkKCkpKSByZXR1cm47IC8vIG5vdCBpbXBvcnRhbnRcbiAgICAgICAgICAgIGF3YWl0IHRoaXMuZ2VuZXJhdGVQcmV2aWV3KHRoaXMubWF0cml4Q2xpZW50LmdldFJvb20oZXZlbnQuZ2V0Um9vbUlkKCkpLCBUQUdfQU5ZKTtcbiAgICAgICAgfVxuICAgIH1cbn1cbiJdfQ==