search.js 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. const {MessageEmbed, Util} = require('discord.js');
  2. /**
  3. * Searches a Fandom wiki.
  4. * @param {import('../../../util/i18n.js')} lang - The user language.
  5. * @param {import('discord.js').Message} msg - The Discord message.
  6. * @param {String} searchterm - The searchterm.
  7. * @param {String} wiki - The wiki for the search.
  8. * @param {Object} query - The siteinfo from the wiki.
  9. * @param {import('discord.js').MessageReaction} reaction - The reaction on the message.
  10. * @param {String} spoiler - If the response is in a spoiler.
  11. */
  12. function fandom_search(lang, msg, searchterm, wiki, query, reaction, spoiler) {
  13. if ( searchterm.length > 250 ) {
  14. searchterm = searchterm.substring(0, 250);
  15. msg.reactEmoji('⚠️');
  16. }
  17. var pagelink = wiki.toLink('Special:Search', 'search=' + searchterm.toSearch(), '', query.general);
  18. var embed = new MessageEmbed().setAuthor( query.general.sitename ).setTitle( '`' + searchterm + '`' ).setURL( pagelink );
  19. if ( !searchterm.trim() ) {
  20. pagelink = wiki.toLink('Special:Search', '', '', query.general);
  21. embed.setTitle( 'Special:Search' ).setURL( pagelink );
  22. msg.sendChannel( spoiler + '<' + pagelink + '>' + spoiler, {embed} );
  23. if ( reaction ) reaction.removeEmoji();
  24. return;
  25. }
  26. var description = [];
  27. got.get( wiki + 'api/v1/Search/List?minArticleQuality=0&namespaces=4,12,14,' + Object.values(query.namespaces).filter( ns => ns.content !== undefined ).map( ns => ns.id ).join(',') + '&limit=10&query=' + encodeURIComponent( searchterm ) + '&format=json', {
  28. responseType: 'json'
  29. } ).then( response => {
  30. var body = response.body;
  31. if ( response.statusCode !== 200 || !body || body.exception || !body.items ) {
  32. if ( !( body && body.exception && body.exception.code === 404 ) ) {
  33. console.log( '- ' + response.statusCode + ': Error while getting the search results: ' + ( body && body.exception && body.exception.details ) );
  34. }
  35. return;
  36. }
  37. body.items.forEach( result => {
  38. description.push( '• [' + result.title + '](' + wiki.toLink(result.title, '', '', query.general, true) + ')' );
  39. } );
  40. embed.setFooter( lang.get('search.results', body.total) );
  41. }, error => {
  42. console.log( '- Error while getting the search results.' + error );
  43. } ).finally( () => {
  44. embed.setDescription( Util.splitMessage( description.join('\n') )[0] );
  45. msg.sendChannel( spoiler + '<' + pagelink + '>' + spoiler, {embed} );
  46. if ( reaction ) reaction.removeEmoji();
  47. } );
  48. }
  49. module.exports = {
  50. name: 'search',
  51. run: fandom_search
  52. };