src/contentScripts/communityConsole/avatars.js - infinitegforums - Gitiles

 import {waitFor} from 'poll-until-promise';

 import {CCApi} from '../../common/api.js';
 import {parseUrl} from '../../common/commonUtils.js';

 import AvatarsDB from './utils/AvatarsDB.js'

 export default class AvatarsHandler {
   constructor() {
     this.isFilterSetUp = false;
     this.privateForums = [];
     this.db = new AvatarsDB();
   }

   // Gets a list of private forums. If it is already cached, the cached list is
   // returned; otherwise it is also computed and cached.
   getPrivateForums() {
     return new Promise((resolve, reject) => {
       if (this.isFilterSetUp) return resolve(this.privateForums);

       if (!document.documentElement.hasAttribute('data-startup'))
         return reject('[threadListAvatars] Couldn\'t get startup data.');

       var startupData =
           JSON.parse(document.documentElement.getAttribute('data-startup'));
       var forums = startupData?.['1']?.['2'];
       if (forums === undefined)
         return reject(
             '[threadListAvatars] Couldn\'t retrieve forums from startup data.');

       for (var f of forums) {
         var forumId = f?.['2']?.['1']?.['1'];
         var forumVisibility = f?.['2']?.['18'];
         if (forumId === undefined || forumVisibility === undefined) {
           console.warn(
               '[threadListAvatars] Coudln\'t retrieve forum id and/or forum visibility for the following forum:',
               f);
           continue;
         }

         // forumVisibility's value 1 means "PUBLIC".
         if (forumVisibility != 1) this.privateForums.push(forumId);
       }

       // Forum 51488989 is marked as public but it is in fact private.
       this.privateForums.push('51488989');

       this.isFilterSetUp = true;
       return resolve(this.privateForums);
     });
   }

   // Some threads belong to private forums, and this feature will not be able to
   // get its avatars since it makes an anonymomus call to get the contents of
   // the thread.
   //
   // This function returns whether avatars should be retrieved depending on if
   // the thread belongs to a known private forum.
   shouldRetrieveAvatars(thread) {
     return this.getPrivateForums().then(privateForums => {
       if (privateForums.includes(thread.forum)) return false;

       return this.db.isForumUnauthorized(thread.forum).then(res => !res);
     });
   }

   // Get an object with the author of the thread, an array of the first |num|
   // replies from the thread |thread|, and additional information about the
   // thread.
   getFirstMessages(thread, num = 15) {
     return CCApi(
                'ViewThread', {
                  1: thread.forum,
                  2: thread.thread,
                  // options
                  3: {
                    // pagination
                    1: {
                      2: num,  // maxNum
                    },
                    3: true,    // withMessages
                    5: true,    // withUserProfile
                    10: false,  // withPromotedMessages
                    16: false,  // withThreadNotes
                    18: true,   // sendNewThreadIfMoved
                  }
                },
                // |authentication| is false because otherwise this would mark
                // the thread as read as a side effect, and that would mark all
                // threads in the list as read.
                //
                // Due to the fact that we have to call this endpoint
                // anonymously, this means we can't retrieve information about
                // threads in private forums.
                /* authentication = */ false, /* authuser = */ 0,
                /* returnUnauthorizedStatus = */ true)
         .then(response => {
           if (response.unauthorized)
             return this.db.putUnauthorizedForum(thread.forum).then(() => {
               throw new Error('Permission denied to load thread.');
             });

           var data = response.body;

           var numMessages = data?.['1']?.['8'];
           if (numMessages === undefined)
             throw new Error(
                 'Request to view thread doesn\'t include the number of messages');

           var messages = numMessages == 0 ? [] : data?.['1']['3'];
           if (messages === undefined)
             throw new Error(
                 'numMessages was ' + numMessages +
                 ' but the response didn\'t include any message.');

           var author = data?.['1']?.['4'];
           if (author === undefined)
             throw new Error(
                 'Author isn\'t included in the ViewThread response.');

           return {
             messages,
             author,

             // The following fields are useful for the cache and can be
             // undefined, but this is checked before adding an entry to the
             // cache.
             lastMessageId: data?.['1']?.['2']?.['10'],
           };
         });
   }

   // Get a list of at most |num| avatars for thread |thread| by calling the API
   getVisibleAvatarsFromServer(thread, num) {
     return this.getFirstMessages(thread).then(result => {
       var messages = result.messages;
       var author = result.author;
       var lastMessageId = result.lastMessageId;

       var avatarUrls = [];

       var authorUrl = author?.['1']?.['2'];
       if (authorUrl !== undefined) avatarUrls.push(authorUrl);

       for (var m of messages) {
         var url = m?.['3']?.['1']?.['2'];

         if (url === undefined) continue;
         if (!avatarUrls.includes(url)) avatarUrls.push(url);
         if (avatarUrls.length == 3) break;
       }

       // Add entry to cache if all the extra metadata could be retrieved.
       if (lastMessageId !== undefined)
         this.db.putCacheEntry({
           threadId: thread.thread,
           lastMessageId,
           avatarUrls,
           num,
           lastUsedTimestamp: Math.floor(Date.now() / 1000),
         });

       return avatarUrls;
     });
   }

   // Returns an object with a cache entry that matches the request if found (via
   // the |entry| property). The property |found| indicates whether the cache
   // entry was found.
   //
   // The |checkRecent| parameter is used to indicate whether lastUsedTimestamp
   // must be within the last 30 seconds (which means that the thread has been
   // checked for a potential invalidation).
   getVisibleAvatarsFromCache(thread, num, checkRecent) {
     return this.db.getCacheEntry(thread.thread).then(entry => {
       if (entry === undefined || entry.num < num)
         return {
           found: false,
         };

       if (checkRecent) {
         var now = Math.floor(Date.now() / 1000);
         var diff = now - entry.lastUsedTimestamp;
         if (diff > 30)
           throw new Error(
               'lastUsedTimestamp isn\'t within the last 30 seconds (id: ' +
               thread.thread + ' the difference is: ' + diff + ').');
       }

       return {
         found: true,
         entry,
       };
     });
   }

   // Waits for the XHR interceptor to invalidate any outdated threads and
   // returns what getVisibleAvatarsFromCache returns. If this times out, it
   // returns the current cache entry anyways if it exists.
   getVisibleAvatarsFromCacheAfterInvalidations(thread, num) {
     return waitFor(
                () => this.getVisibleAvatarsFromCache(
                    thread, num, /* checkRecent = */ true),
                {
                  interval: 450,
                  timeout: 2 * 1000,
                })
         .catch(err => {
           console.debug(
               '[threadListAvatars] Error while retrieving avatars from cache ' +
                   '(probably timed out waiting for lastUsedTimestamp to change):',
               err);

           // Sometimes when going back to a thread list, the API call to load
           // the thread list is not made, and so the previous piece of code
           // times out waiting to intercept that API call and handle thread
           // invalidations.
           //
           // If this is the case, this point will be reached. We'll assume we
           // intercept all API calls, so reaching this point means that an API
           // call wasn't made. Therefore, try again to get visible avatars from
           // the cache without checking whether the entry has been checked for
           // potential invalidation.
           //
           // See https://bugs.avm99963.com/p/twpowertools/issues/detail?id=10.
           return this.getVisibleAvatarsFromCache(
               thread, num, /* checkRecent = */ false);
         });
   }

   // Get a list of at most |num| avatars for thread |thread|
   getVisibleAvatars(thread, num = 3) {
     return this.shouldRetrieveAvatars(thread).then(shouldRetrieve => {
       if (!shouldRetrieve) {
         console.debug('[threadListAvatars] Skipping thread', thread);
         return [];
       }

       return this.getVisibleAvatarsFromCacheAfterInvalidations(thread, num)
           .then(res => {
             if (!res.found) {
               var err = new Error('Cache entry doesn\'t exist.');
               err.name = 'notCached';
               throw err;
             }
             return res.entry.avatarUrls;
           })
           .catch(err => {
             // If the name is "notCached", then this is not an actual error so
             // don't log an error, but still get avatars from the server.
             if (err?.name !== 'notCached')
               console.error(
                   '[threadListAvatars] Error while accessing avatars cache:',
                   err);

             return this.getVisibleAvatarsFromServer(thread, num);
           });
     });
   }

   // Inject avatars for thread summary (thread item) |node| in a thread list.
   inject(node) {
     var header = node.querySelector(
         'ec-thread-summary .main-header .panel-description a.header');
     if (header === null) {
       console.error(
           '[threadListAvatars] Header is not present in the thread item\'s DOM.');
       return;
     }

     var thread = parseUrl(header.href);
     if (thread === false) {
       console.error('[threadListAvatars] Thread\'s link cannot be parsed.');
       return;
     }

     this.getVisibleAvatars(thread)
         .then(avatarUrls => {
           var avatarsContainer = document.createElement('div');
           avatarsContainer.classList.add('TWPT-avatars');

           var count = Math.floor(Math.random() * 4);

           for (var i = 0; i < avatarUrls.length; ++i) {
             var avatar = document.createElement('div');
             avatar.classList.add('TWPT-avatar');
             avatar.style.backgroundImage = 'url(\'' + avatarUrls[i] + '\')';
             avatarsContainer.appendChild(avatar);
           }

           header.appendChild(avatarsContainer);
         })
         .catch(err => {
           console.error(
               '[threadListAvatars] Could not retrieve avatars for thread',
               thread, err);
         });
   }
 };
	import {waitFor} from 'poll-until-promise';

	import {CCApi} from '../../common/api.js';
	import {parseUrl} from '../../common/commonUtils.js';

	import AvatarsDB from './utils/AvatarsDB.js'

	export default class AvatarsHandler {
	constructor() {
	this.isFilterSetUp = false;
	this.privateForums = [];
	this.db = new AvatarsDB();
	}

	// Gets a list of private forums. If it is already cached, the cached list is
	// returned; otherwise it is also computed and cached.
	getPrivateForums() {
	return new Promise((resolve, reject) => {
	if (this.isFilterSetUp) return resolve(this.privateForums);

	if (!document.documentElement.hasAttribute('data-startup'))
	return reject('[threadListAvatars] Couldn\'t get startup data.');

	var startupData =
	JSON.parse(document.documentElement.getAttribute('data-startup'));
	var forums = startupData?.['1']?.['2'];
	if (forums === undefined)
	return reject(
	'[threadListAvatars] Couldn\'t retrieve forums from startup data.');

	for (var f of forums) {
	var forumId = f?.['2']?.['1']?.['1'];
	var forumVisibility = f?.['2']?.['18'];
	if (forumId === undefined \|\| forumVisibility === undefined) {
	console.warn(
	'[threadListAvatars] Coudln\'t retrieve forum id and/or forum visibility for the following forum:',
	f);
	continue;
	}

	// forumVisibility's value 1 means "PUBLIC".
	if (forumVisibility != 1) this.privateForums.push(forumId);
	}

	// Forum 51488989 is marked as public but it is in fact private.
	this.privateForums.push('51488989');

	this.isFilterSetUp = true;
	return resolve(this.privateForums);
	});
	}

	// Some threads belong to private forums, and this feature will not be able to
	// get its avatars since it makes an anonymomus call to get the contents of
	// the thread.
	//
	// This function returns whether avatars should be retrieved depending on if
	// the thread belongs to a known private forum.
	shouldRetrieveAvatars(thread) {
	return this.getPrivateForums().then(privateForums => {
	if (privateForums.includes(thread.forum)) return false;

	return this.db.isForumUnauthorized(thread.forum).then(res => !res);
	});
	}

	// Get an object with the author of the thread, an array of the first \|num\|
	// replies from the thread \|thread\|, and additional information about the
	// thread.
	getFirstMessages(thread, num = 15) {
	return CCApi(
	'ViewThread', {
	1: thread.forum,
	2: thread.thread,
	// options
	3: {
	// pagination
	1: {
	2: num, // maxNum
	},
	3: true, // withMessages
	5: true, // withUserProfile
	10: false, // withPromotedMessages
	16: false, // withThreadNotes
	18: true, // sendNewThreadIfMoved
	}
	},
	// \|authentication\| is false because otherwise this would mark
	// the thread as read as a side effect, and that would mark all
	// threads in the list as read.
	//
	// Due to the fact that we have to call this endpoint
	// anonymously, this means we can't retrieve information about
	// threads in private forums.
	/* authentication = / false, / authuser = */ 0,
	/* returnUnauthorizedStatus = */ true)
	.then(response => {
	if (response.unauthorized)
	return this.db.putUnauthorizedForum(thread.forum).then(() => {
	throw new Error('Permission denied to load thread.');
	});

	var data = response.body;

	var numMessages = data?.['1']?.['8'];
	if (numMessages === undefined)
	throw new Error(
	'Request to view thread doesn\'t include the number of messages');

	var messages = numMessages == 0 ? [] : data?.['1']['3'];
	if (messages === undefined)
	throw new Error(
	'numMessages was ' + numMessages +
	' but the response didn\'t include any message.');

	var author = data?.['1']?.['4'];
	if (author === undefined)
	throw new Error(
	'Author isn\'t included in the ViewThread response.');

	return {
	messages,
	author,

	// The following fields are useful for the cache and can be
	// undefined, but this is checked before adding an entry to the
	// cache.
	lastMessageId: data?.['1']?.['2']?.['10'],
	};
	});
	}

	// Get a list of at most \|num\| avatars for thread \|thread\| by calling the API
	getVisibleAvatarsFromServer(thread, num) {
	return this.getFirstMessages(thread).then(result => {
	var messages = result.messages;
	var author = result.author;
	var lastMessageId = result.lastMessageId;

	var avatarUrls = [];

	var authorUrl = author?.['1']?.['2'];
	if (authorUrl !== undefined) avatarUrls.push(authorUrl);

	for (var m of messages) {
	var url = m?.['3']?.['1']?.['2'];

	if (url === undefined) continue;
	if (!avatarUrls.includes(url)) avatarUrls.push(url);
	if (avatarUrls.length == 3) break;
	}

	// Add entry to cache if all the extra metadata could be retrieved.
	if (lastMessageId !== undefined)
	this.db.putCacheEntry({
	threadId: thread.thread,
	lastMessageId,
	avatarUrls,
	num,
	lastUsedTimestamp: Math.floor(Date.now() / 1000),
	});

	return avatarUrls;
	});
	}

	// Returns an object with a cache entry that matches the request if found (via
	// the \|entry\| property). The property \|found\| indicates whether the cache
	// entry was found.
	//
	// The \|checkRecent\| parameter is used to indicate whether lastUsedTimestamp
	// must be within the last 30 seconds (which means that the thread has been
	// checked for a potential invalidation).
	getVisibleAvatarsFromCache(thread, num, checkRecent) {
	return this.db.getCacheEntry(thread.thread).then(entry => {
	if (entry === undefined \|\| entry.num < num)
	return {
	found: false,
	};

	if (checkRecent) {
	var now = Math.floor(Date.now() / 1000);
	var diff = now - entry.lastUsedTimestamp;
	if (diff > 30)
	throw new Error(
	'lastUsedTimestamp isn\'t within the last 30 seconds (id: ' +
	thread.thread + ' the difference is: ' + diff + ').');
	}

	return {
	found: true,
	entry,
	};
	});
	}

	// Waits for the XHR interceptor to invalidate any outdated threads and
	// returns what getVisibleAvatarsFromCache returns. If this times out, it
	// returns the current cache entry anyways if it exists.
	getVisibleAvatarsFromCacheAfterInvalidations(thread, num) {
	return waitFor(
	() => this.getVisibleAvatarsFromCache(
	thread, num, /* checkRecent = */ true),
	{
	interval: 450,
	timeout: 2 * 1000,
	})
	.catch(err => {
	console.debug(
	'[threadListAvatars] Error while retrieving avatars from cache ' +
	'(probably timed out waiting for lastUsedTimestamp to change):',
	err);

	// Sometimes when going back to a thread list, the API call to load
	// the thread list is not made, and so the previous piece of code
	// times out waiting to intercept that API call and handle thread
	// invalidations.
	//
	// If this is the case, this point will be reached. We'll assume we
	// intercept all API calls, so reaching this point means that an API
	// call wasn't made. Therefore, try again to get visible avatars from
	// the cache without checking whether the entry has been checked for
	// potential invalidation.
	//
	// See https://bugs.avm99963.com/p/twpowertools/issues/detail?id=10.
	return this.getVisibleAvatarsFromCache(
	thread, num, /* checkRecent = */ false);
	});
	}

	// Get a list of at most \|num\| avatars for thread \|thread\|
	getVisibleAvatars(thread, num = 3) {
	return this.shouldRetrieveAvatars(thread).then(shouldRetrieve => {
	if (!shouldRetrieve) {
	console.debug('[threadListAvatars] Skipping thread', thread);
	return [];
	}

	return this.getVisibleAvatarsFromCacheAfterInvalidations(thread, num)
	.then(res => {
	if (!res.found) {
	var err = new Error('Cache entry doesn\'t exist.');
	err.name = 'notCached';
	throw err;
	}
	return res.entry.avatarUrls;
	})
	.catch(err => {
	// If the name is "notCached", then this is not an actual error so
	// don't log an error, but still get avatars from the server.
	if (err?.name !== 'notCached')
	console.error(
	'[threadListAvatars] Error while accessing avatars cache:',
	err);

	return this.getVisibleAvatarsFromServer(thread, num);
	});
	});
	}

	// Inject avatars for thread summary (thread item) \|node\| in a thread list.
	inject(node) {
	var header = node.querySelector(
	'ec-thread-summary .main-header .panel-description a.header');
	if (header === null) {
	console.error(
	'[threadListAvatars] Header is not present in the thread item\'s DOM.');
	return;
	}

	var thread = parseUrl(header.href);
	if (thread === false) {
	console.error('[threadListAvatars] Thread\'s link cannot be parsed.');
	return;
	}

	this.getVisibleAvatars(thread)
	.then(avatarUrls => {
	var avatarsContainer = document.createElement('div');
	avatarsContainer.classList.add('TWPT-avatars');

	var count = Math.floor(Math.random() * 4);

	for (var i = 0; i < avatarUrls.length; ++i) {
	var avatar = document.createElement('div');
	avatar.classList.add('TWPT-avatar');
	avatar.style.backgroundImage = 'url(\'' + avatarUrls[i] + '\')';
	avatarsContainer.appendChild(avatar);
	}

	header.appendChild(avatarsContainer);
	})
	.catch(err => {
	console.error(
	'[threadListAvatars] Could not retrieve avatars for thread',
	thread, err);
	});
	}
	};