search.js 3.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. const {MessageEmbed, Util} = require('discord.js');
  2. function gamepedia_search(lang, msg, searchterm, wiki, query, reaction, spoiler) {
  3. if ( searchterm.length > 250 ) {
  4. searchterm = searchterm.substring(0, 250);
  5. msg.reactEmoji('⚠️');
  6. }
  7. var pagelink = wiki.toLink('Special:Search', 'search=' + searchterm.toSearch() + '&fulltext=1', '', query.general);
  8. var embed = new MessageEmbed().setAuthor( query.general.sitename ).setTitle( '`' + searchterm + '`' ).setURL( pagelink );
  9. if ( !searchterm.trim() ) {
  10. pagelink = wiki.toLink('Special:Search', '', '', query.general);
  11. embed.setTitle( 'Special:Search' ).setURL( pagelink );
  12. }
  13. var description = [];
  14. got.get( wiki + 'api.php?action=query&titles=Special:Search&list=search&srinfo=totalhits&srprop=redirecttitle|sectiontitle&srnamespace=4|12|14|' + Object.values(query.namespaces).filter( ns => ns.content !== undefined ).map( ns => ns.id ).join('|') + '&srlimit=10&srsearch=' + encodeURIComponent( ( searchterm || ' ' ) ) + '&format=json', {
  15. responseType: 'json'
  16. } ).then( response => {
  17. var body = response.body;
  18. if ( body && body.warnings ) log_warn(body.warnings);
  19. if ( response.statusCode !== 200 || !body || !body.query || !body.query.search || body.batchcomplete === undefined ) {
  20. return console.log( '- ' + response.statusCode + ': Error while getting the search results: ' + ( body && body.error && body.error.info ) );
  21. }
  22. if ( body.query.pages && body.query.pages['-1'] && body.query.pages['-1'].title ) {
  23. if ( searchterm.trim() ) {
  24. pagelink = wiki.toLink(body.query.pages['-1'].title, 'search=' + searchterm.toSearch() + '&fulltext=1', '', query.general);
  25. embed.setURL( pagelink );
  26. }
  27. else {
  28. pagelink = wiki.toLink(body.query.pages['-1'].title, '', '', query.general);
  29. embed.setTitle( body.query.pages['-1'].title ).setURL( pagelink );
  30. }
  31. }
  32. if ( searchterm.trim() ) {
  33. body.query.search.forEach( result => {
  34. description.push( '• [' + result.title + '](' + wiki.toLink(result.title, '', '', query.general, true) + ')' + ( result.sectiontitle ? ' § [' + result.sectiontitle + '](' + wiki.toLink(result.title, '', result.sectiontitle, query.general, true) + ')' : '' ) + ( result.redirecttitle ? ' (⤷ [' + result.redirecttitle + '](' + wiki.toLink(result.redirecttitle, '', '', query.general, true) + '))' : '' ) );
  35. } );
  36. let count = lang.get('search.results');
  37. embed.setFooter( ( count[body.query.searchinfo.totalhits] || count['*' + body.query.searchinfo.totalhits % 100] || count['*' + body.query.searchinfo.totalhits % 10] || lang.get('search.results.default') ).replaceSave( '%s', body.query.searchinfo.totalhits ) );
  38. }
  39. }, error => {
  40. console.log( '- Error while getting the search results.' + error );
  41. } ).finally( () => {
  42. embed.setDescription( Util.splitMessage( description.join('\n') )[0] );
  43. msg.sendChannel( spoiler + '<' + pagelink + '>' + spoiler, {embed} );
  44. if ( reaction ) reaction.removeEmoji();
  45. } );
  46. }
  47. module.exports = {
  48. name: 'search',
  49. run: gamepedia_search
  50. };