Adrià Vilanova Martínez | 412b758 | 2022-12-30 01:35:30 +0100 | [diff] [blame] | 1 | import {kAdditionalInfoPrefix} from '../../contentScripts/communityConsole/flattenThreads/flattenThreads.js'; |
Adrià Vilanova Martínez | 2d9be8d | 2022-12-28 00:50:14 +0100 | [diff] [blame] | 2 | import GapModel from '../../models/Gap.js'; |
| 3 | import MessageModel from '../../models/Message.js'; |
| 4 | |
Adrià Vilanova Martínez | 4fb615f | 2023-02-04 18:55:16 +0100 | [diff] [blame^] | 5 | const flattenThread = { |
Adrià Vilanova Martínez | 2d9be8d | 2022-12-28 00:50:14 +0100 | [diff] [blame] | 6 | urlRegex: /api\/ViewThread/i, |
| 7 | featureGated: true, |
| 8 | features: ['flattenthreads', 'flattenthreads_switch_enabled'], |
| 9 | isEnabled(options) { |
| 10 | return options['flattenthreads'] && |
| 11 | options['flattenthreads_switch_enabled']; |
| 12 | }, |
| 13 | async interceptor(_request, response) { |
| 14 | if (!response[1]?.[40]) return response; |
| 15 | |
Adrià Vilanova Martínez | 412b758 | 2022-12-30 01:35:30 +0100 | [diff] [blame] | 16 | // Do the actual flattening |
Adrià Vilanova Martínez | 2d9be8d | 2022-12-28 00:50:14 +0100 | [diff] [blame] | 17 | const originalMogs = |
| 18 | MessageModel.mapToMessageOrGapModels(response[1][40] ?? []); |
| 19 | let extraMogs = []; |
| 20 | originalMogs.forEach(mog => { |
| 21 | if (mog instanceof GapModel) return; |
| 22 | const cogs = mog.getCommentsAndGaps(); |
| 23 | extraMogs = extraMogs.concat(cogs); |
| 24 | mog.clearCommentsAndGaps(); |
| 25 | }); |
| 26 | const mogs = originalMogs.concat(extraMogs); |
Adrià Vilanova Martínez | 412b758 | 2022-12-30 01:35:30 +0100 | [diff] [blame] | 27 | |
| 28 | // Add some message data to the payload so the extension can show the parent |
| 29 | // comment/reply in the case of comments. |
Adrià Vilanova Martínez | 115e3d8 | 2023-01-10 21:50:06 +0100 | [diff] [blame] | 30 | let prevReplyId; |
| 31 | let prevReplyParentId; |
Adrià Vilanova Martínez | 412b758 | 2022-12-30 01:35:30 +0100 | [diff] [blame] | 32 | mogs.forEach(m => { |
Adrià Vilanova Martínez | 115e3d8 | 2023-01-10 21:50:06 +0100 | [diff] [blame] | 33 | const info = this.getAdditionalInformation( |
| 34 | m, mogs, prevReplyId, prevReplyParentId); |
| 35 | prevReplyId = m.getId(); |
| 36 | prevReplyParentId = info.parentId; |
| 37 | |
Adrià Vilanova Martínez | 412b758 | 2022-12-30 01:35:30 +0100 | [diff] [blame] | 38 | const span = document.createElement('span'); |
| 39 | span.textContent = kAdditionalInfoPrefix + JSON.stringify(info); |
| 40 | span.setAttribute('style', 'display: none'); |
| 41 | m.newPayload = m.getPayload() + span.outerHTML; |
| 42 | }); |
| 43 | mogs.forEach(m => m.setPayload(m.newPayload)); |
| 44 | |
Adrià Vilanova Martínez | 4fb615f | 2023-02-04 18:55:16 +0100 | [diff] [blame^] | 45 | // Clear parent_message_id fields |
| 46 | mogs.forEach(m => m.clearParentMessageId()); |
| 47 | |
Adrià Vilanova Martínez | 412b758 | 2022-12-30 01:35:30 +0100 | [diff] [blame] | 48 | // Sort the messages by date |
Adrià Vilanova Martínez | 2d9be8d | 2022-12-28 00:50:14 +0100 | [diff] [blame] | 49 | mogs.sort((a, b) => { |
| 50 | const c = a instanceof MessageModel ? a.getCreatedMicroseconds() : |
| 51 | a.getStartTimestamp(); |
| 52 | const d = b instanceof MessageModel ? b.getCreatedMicroseconds() : |
| 53 | b.getStartTimestamp(); |
| 54 | const diff = c - d; |
| 55 | return diff > 0 ? 1 : diff < 0 ? -1 : 0; |
| 56 | }); |
Adrià Vilanova Martínez | 412b758 | 2022-12-30 01:35:30 +0100 | [diff] [blame] | 57 | |
Adrià Vilanova Martínez | 2d9be8d | 2022-12-28 00:50:14 +0100 | [diff] [blame] | 58 | response[1][40] = mogs.map(mog => mog.toRawMessageOrGap()); |
Adrià Vilanova Martínez | 412b758 | 2022-12-30 01:35:30 +0100 | [diff] [blame] | 59 | |
Adrià Vilanova Martínez | 4fb615f | 2023-02-04 18:55:16 +0100 | [diff] [blame^] | 60 | // Set last_message to the last message after sorting |
| 61 | if (response[1]?.[17]?.[3]) |
| 62 | response[1][17][3] = response[1][40].slice(-1)?.[1]; |
| 63 | |
Adrià Vilanova Martínez | 2d9be8d | 2022-12-28 00:50:14 +0100 | [diff] [blame] | 64 | // Set num_messages to the updated value, since we've flattened the replies. |
| 65 | response[1][8] = response[1][40].length; |
| 66 | return response; |
| 67 | }, |
Adrià Vilanova Martínez | 115e3d8 | 2023-01-10 21:50:06 +0100 | [diff] [blame] | 68 | getAdditionalInformation(message, mogs, prevReplyId, prevReplyParentId) { |
Adrià Vilanova Martínez | 412b758 | 2022-12-30 01:35:30 +0100 | [diff] [blame] | 69 | const id = message.getId(); |
| 70 | const parentId = message.getParentMessageId(); |
Adrià Vilanova Martínez | 115e3d8 | 2023-01-10 21:50:06 +0100 | [diff] [blame] | 71 | const authorName = message.getAuthor()?.[1]?.[1]; |
| 72 | if (!parentId) { |
Adrià Vilanova Martínez | 412b758 | 2022-12-30 01:35:30 +0100 | [diff] [blame] | 73 | return { |
| 74 | isComment: false, |
| 75 | id, |
Adrià Vilanova Martínez | 115e3d8 | 2023-01-10 21:50:06 +0100 | [diff] [blame] | 76 | authorName, |
Adrià Vilanova Martínez | 412b758 | 2022-12-30 01:35:30 +0100 | [diff] [blame] | 77 | }; |
| 78 | } |
| 79 | |
Adrià Vilanova Martínez | 115e3d8 | 2023-01-10 21:50:06 +0100 | [diff] [blame] | 80 | let prevId; |
Adrià Vilanova Martínez | b47ec06 | 2023-01-15 17:43:26 +0100 | [diff] [blame] | 81 | if (parentId == prevReplyParentId && prevReplyParentId) |
Adrià Vilanova Martínez | 115e3d8 | 2023-01-10 21:50:06 +0100 | [diff] [blame] | 82 | prevId = prevReplyId; |
| 83 | else |
| 84 | prevId = parentId; |
| 85 | |
Adrià Vilanova Martínez | 4fb615f | 2023-02-04 18:55:16 +0100 | [diff] [blame^] | 86 | const prevMessage = prevId ? mogs.find(m => m.getId() == prevId) : null; |
Adrià Vilanova Martínez | 115e3d8 | 2023-01-10 21:50:06 +0100 | [diff] [blame] | 87 | |
Adrià Vilanova Martínez | 412b758 | 2022-12-30 01:35:30 +0100 | [diff] [blame] | 88 | return { |
| 89 | isComment: true, |
| 90 | id, |
Adrià Vilanova Martínez | 115e3d8 | 2023-01-10 21:50:06 +0100 | [diff] [blame] | 91 | authorName, |
| 92 | parentId, |
| 93 | prevMessage: { |
| 94 | id: prevId, |
Adrià Vilanova Martínez | b47ec06 | 2023-01-15 17:43:26 +0100 | [diff] [blame] | 95 | payload: prevMessage?.getPayload(), |
| 96 | author: prevMessage?.getAuthor(), |
Adrià Vilanova Martínez | 412b758 | 2022-12-30 01:35:30 +0100 | [diff] [blame] | 97 | }, |
| 98 | }; |
| 99 | } |
Adrià Vilanova Martínez | 2d9be8d | 2022-12-28 00:50:14 +0100 | [diff] [blame] | 100 | }; |
| 101 | |
Adrià Vilanova Martínez | 4fb615f | 2023-02-04 18:55:16 +0100 | [diff] [blame^] | 102 | export default flattenThread; |