gamepedia.js 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477
  1. const htmlparser = require('htmlparser2');
  2. const {MessageEmbed} = require('discord.js');
  3. const extract_desc = require('../../util/extract_desc.js');
  4. const {limit: {interwiki: interwikiLimit}, wikiProjects} = require('../../util/default.json');
  5. const fs = require('fs');
  6. var fn = {
  7. special_page: require('../../functions/special_page.js'),
  8. discussion: require('../../functions/discussion.js')
  9. };
  10. fs.readdir( './cmds/wiki/gamepedia', (error, files) => {
  11. if ( error ) return error;
  12. files.filter( file => file.endsWith('.js') ).forEach( file => {
  13. var command = require('./gamepedia/' + file);
  14. fn[command.name] = command.run;
  15. } );
  16. } );
  17. var minecraft = {};
  18. fs.readdir( './cmds/minecraft', (error, files) => {
  19. if ( error ) return error;
  20. files.filter( file => file.endsWith('.js') ).forEach( file => {
  21. var command = require('../minecraft/' + file);
  22. minecraft[command.name] = command.run;
  23. } );
  24. } );
  25. /**
  26. * Checks a Gamepedia wiki.
  27. * @param {import('../../util/i18n.js')} lang - The user language.
  28. * @param {import('discord.js').Message} msg - The Discord message.
  29. * @param {String} title - The page title.
  30. * @param {String} wiki - The wiki for the page.
  31. * @param {String} cmd - The command at this point.
  32. * @param {import('discord.js').MessageReaction} reaction - The reaction on the message.
  33. * @param {String} [spoiler] - If the response is in a spoiler.
  34. * @param {String} [querystring] - The querystring for the link.
  35. * @param {String} [fragment] - The section for the link.
  36. * @param {String} [interwiki] - The fallback interwiki link.
  37. * @param {Number} [selfcall] - The amount of followed interwiki links.
  38. */
  39. function gamepedia_check_wiki(lang, msg, title, wiki, cmd, reaction, spoiler = '', querystring = '', fragment = '', interwiki = '', selfcall = 0) {
  40. var full_title = title;
  41. if ( title.includes( '#' ) ) {
  42. fragment = title.split('#').slice(1).join('#');
  43. title = title.split('#')[0];
  44. }
  45. if ( /\?\w+=/.test(title) ) {
  46. var querystart = title.search(/\?\w+=/);
  47. querystring = title.substring(querystart + 1) + ( querystring ? '&' + querystring : '' );
  48. title = title.substring(0, querystart);
  49. }
  50. if ( title.length > 250 ) {
  51. title = title.substring(0, 250);
  52. msg.reactEmoji('⚠️');
  53. }
  54. var invoke = title.split(' ')[0].toLowerCase();
  55. var aliasInvoke = ( lang.aliases[invoke] || invoke );
  56. var args = title.split(' ').slice(1);
  57. if ( !msg.notMinecraft && wiki === lang.get('minecraft.link') && ( aliasInvoke in minecraft || invoke.startsWith( '/' ) ) ) {
  58. minecraft.WIKI = this;
  59. if ( aliasInvoke in minecraft ) minecraft[aliasInvoke](lang, msg, args, title, cmd, querystring, fragment, reaction, spoiler);
  60. else minecraft.SYNTAX(lang, msg, invoke.substring(1), args, title, cmd, querystring, fragment, reaction, spoiler);
  61. }
  62. else if ( aliasInvoke === 'random' && !args.join('') && !querystring && !fragment ) fn.random(lang, msg, wiki, reaction, spoiler);
  63. else if ( aliasInvoke === 'overview' && !args.join('') && !querystring && !fragment ) fn.overview(lang, msg, wiki, reaction, spoiler);
  64. else if ( aliasInvoke === 'test' && !args.join('') && !querystring && !fragment ) {
  65. this.test(lang, msg, args, '', wiki);
  66. if ( reaction ) reaction.removeEmoji();
  67. }
  68. else if ( aliasInvoke === 'page' ) {
  69. msg.sendChannel( spoiler + '<' + wiki.toLink(args.join('_'), querystring.toTitle(), fragment) + '>' + spoiler );
  70. if ( reaction ) reaction.removeEmoji();
  71. }
  72. else if ( aliasInvoke === 'diff' && args.join('') && !querystring && !fragment ) fn.diff(lang, msg, args, wiki, reaction, spoiler);
  73. else {
  74. var noRedirect = ( /(?:^|&)redirect=no(?:&|$)/.test(querystring) || /(?:^|&)action=(?!view(?:&|$))/.test(querystring) );
  75. got.get( wiki + 'api.php?action=query&meta=siteinfo&siprop=general|namespaces|specialpagealiases&iwurl=true' + ( noRedirect ? '' : '&redirects=true' ) + '&prop=pageimages|categoryinfo|pageprops|extracts&piprop=original|name&ppprop=description|displaytitle&explaintext=true&exsectionformat=raw&exlimit=1&titles=%1F' + encodeURIComponent( title.replace( /\x1F/g, '\ufffd' ) ) + '&format=json', {
  76. responseType: 'json'
  77. } ).then( response => {
  78. var body = response.body;
  79. if ( body && body.warnings ) log_warn(body.warnings);
  80. if ( response.statusCode !== 200 || !body || body.batchcomplete === undefined || !body.query ) {
  81. if ( interwiki ) msg.sendChannel( spoiler + ' ' + interwiki.replace( /@(here|everyone)/g, '%40$1' ) + fragment + ' ' + spoiler );
  82. else if ( wiki.noWiki(response.url) || response.statusCode === 410 ) {
  83. console.log( '- This wiki doesn\'t exist!' );
  84. msg.reactEmoji('nowiki');
  85. }
  86. else {
  87. console.log( '- ' + response.statusCode + ': Error while getting the search results: ' + ( body && body.error && body.error.info ) );
  88. msg.sendChannelError( spoiler + '<' + wiki.toLink( ( querystring || fragment || !title ? title : 'Special:Search' ), ( querystring || fragment || !title ? querystring.toTitle() : 'search=' + title.toSearch() ), fragment) + '>' + spoiler );
  89. }
  90. if ( reaction ) reaction.removeEmoji();
  91. }
  92. else if ( aliasInvoke === 'search' ) {
  93. fn.search(lang, msg, full_title.split(' ').slice(1).join(' '), wiki, body.query, reaction, spoiler);
  94. }
  95. else if ( aliasInvoke === 'discussion' && wiki.isFandom() && !querystring && !fragment ) {
  96. fn.discussion(lang, msg, wiki, args.join(' '), body.query, reaction, spoiler);
  97. }
  98. else {
  99. if ( body.query.pages ) {
  100. var querypages = Object.values(body.query.pages);
  101. var querypage = querypages[0];
  102. if ( body.query.redirects && body.query.redirects[0].from.split(':')[0] === body.query.namespaces['-1']['*'] && body.query.specialpagealiases.filter( sp => ['Mypage','Mytalk','MyLanguage'].includes( sp.realname ) ).map( sp => sp.aliases[0] ).includes( body.query.redirects[0].from.split(':').slice(1).join(':').split('/')[0].replace( / /g, '_' ) ) ) {
  103. querypage.title = body.query.redirects[0].from;
  104. delete body.query.redirects[0].tofragment;
  105. delete querypage.missing;
  106. querypage.ns = -1;
  107. querypage.special = '';
  108. }
  109. var contribs = body.query.namespaces['-1']['*'] + ':' + body.query.specialpagealiases.find( sp => sp.realname === 'Contributions' ).aliases[0] + '/';
  110. if ( ( querypage.ns === 2 || querypage.ns === 202 || querypage.ns === 1200 ) && ( !querypage.title.includes( '/' ) || /^[^:]+:(?:(?:\d{1,3}\.){3}\d{1,3}\/\d{2}|(?:[\dA-F]{1,4}:){7}[\dA-F]{1,4}\/\d{2,3})$/.test(querypage.title) ) ) {
  111. var userparts = querypage.title.split(':');
  112. querypage.noRedirect = noRedirect;
  113. fn.user(lang, msg, userparts[0].toTitle() + ':', userparts.slice(1).join(':'), wiki, querystring, fragment, querypage, contribs.toTitle(), reaction, spoiler);
  114. }
  115. else if ( querypage.ns === -1 && querypage.title.startsWith( contribs ) && querypage.title.length > contribs.length ) {
  116. var username = querypage.title.split('/').slice(1).join('/');
  117. got.get( wiki + 'api.php?action=query&titles=User:' + encodeURIComponent( username ) + '&format=json', {
  118. responseType: 'json'
  119. } ).then( uresponse => {
  120. var ubody = uresponse.body;
  121. if ( uresponse.statusCode !== 200 || !ubody || ubody.batchcomplete === undefined || !ubody.query ) {
  122. console.log( '- ' + uresponse.statusCode + ': Error while getting the user: ' + ( ubody && ubody.error && ubody.error.info ) );
  123. msg.sendChannelError( spoiler + '<' + wiki.toLink(contribs + username, querystring.toTitle(), fragment, body.query.general) + '>' + spoiler );
  124. if ( reaction ) reaction.removeEmoji();
  125. }
  126. else {
  127. querypage = Object.values(ubody.query.pages)[0];
  128. if ( querypage.ns === 2 ) {
  129. username = querypage.title.split(':').slice(1).join(':');
  130. querypage.title = contribs + username;
  131. delete querypage.missing;
  132. querypage.ns = -1;
  133. querypage.special = '';
  134. querypage.noRedirect = noRedirect;
  135. fn.user(lang, msg, contribs.toTitle(), username, wiki, querystring, fragment, querypage, contribs.toTitle(), reaction, spoiler);
  136. }
  137. else {
  138. msg.reactEmoji('error');
  139. if ( reaction ) reaction.removeEmoji();
  140. }
  141. }
  142. }, error => {
  143. console.log( '- Error while getting the user: ' + error );
  144. msg.sendChannelError( spoiler + '<' + wiki.toLink(contribs + username, querystring.toTitle(), fragment, body.query.general) + '>' + spoiler );
  145. if ( reaction ) reaction.removeEmoji();
  146. } );
  147. }
  148. else if ( ( querypage.missing !== undefined && querypage.known === undefined && !( noRedirect || querypage.categoryinfo ) ) || querypage.invalid !== undefined ) {
  149. got.get( wiki + 'api.php?action=query&prop=pageimages|categoryinfo|pageprops|extracts&piprop=original|name&ppprop=description|displaytitle&explaintext=true&exsectionformat=raw&exlimit=1&generator=search&gsrnamespace=4|12|14|' + Object.values(body.query.namespaces).filter( ns => ns.content !== undefined ).map( ns => ns.id ).join('|') + '&gsrlimit=1&gsrsearch=' + encodeURIComponent( title ) + '&format=json', {
  150. responseType: 'json'
  151. } ).then( srresponse => {
  152. var srbody = srresponse.body;
  153. if ( srbody && srbody.warnings ) log_warn(srbody.warnings);
  154. if ( srresponse.statusCode !== 200 || !srbody || srbody.batchcomplete === undefined ) {
  155. console.log( '- ' + srresponse.statusCode + ': Error while getting the search results: ' + ( srbody && srbody.error && srbody.error.info ) );
  156. msg.sendChannelError( spoiler + '<' + wiki.toLink('Special:Search', 'search=' + title.toSearch(), '', body.query.general) + '>' + spoiler );
  157. }
  158. else {
  159. if ( !srbody.query ) {
  160. msg.reactEmoji('🤷');
  161. }
  162. else {
  163. querypage = Object.values(srbody.query.pages)[0];
  164. var pagelink = wiki.toLink(querypage.title, querystring.toTitle(), fragment, body.query.general);
  165. var text = '';
  166. var embed = new MessageEmbed().setAuthor( body.query.general.sitename ).setTitle( querypage.title.escapeFormatting() ).setURL( pagelink );
  167. if ( querypage.pageprops && querypage.pageprops.displaytitle ) {
  168. var displaytitle = htmlToDiscord( querypage.pageprops.displaytitle );
  169. if ( displaytitle.length > 250 ) displaytitle = displaytitle.substring(0, 250) + '\u2026';
  170. embed.setTitle( displaytitle );
  171. }
  172. if ( querypage.extract ) {
  173. var extract = extract_desc(querypage.extract, fragment);
  174. embed.setDescription( extract[0] );
  175. if ( extract[2].length ) embed.addField( extract[1], extract[2] );
  176. }
  177. if ( querypage.pageprops && querypage.pageprops.description ) {
  178. var description = htmlToPlain( querypage.pageprops.description );
  179. if ( description.length > 2000 ) description = description.substring(0, 2000) + '\u2026';
  180. embed.setDescription( description );
  181. }
  182. if ( querypage.pageimage && querypage.original && querypage.title !== body.query.general.mainpage ) {
  183. var pageimage = querypage.original.source;
  184. if ( querypage.ns === 6 ) {
  185. if ( msg.showEmbed() && /\.(?:png|jpg|jpeg|gif)$/.test(querypage.pageimage.toLowerCase()) ) embed.setImage( pageimage );
  186. else if ( msg.uploadFiles() ) embed.attachFiles( [{attachment:pageimage,name:( spoiler ? 'SPOILER ' : '' ) + querypage.pageimage}] );
  187. } else embed.setThumbnail( pageimage );
  188. } else embed.setThumbnail( logoToURL(body.query.general) );
  189. var prefix = ( msg.channel.type === 'text' && patreons[msg.guild.id] || process.env.prefix );
  190. var linksuffix = ( querystring ? '?' + querystring : '' ) + ( fragment ? '#' + fragment : '' );
  191. if ( title.replace( /\-/g, ' ' ).toTitle().toLowerCase() === querypage.title.replace( /\-/g, ' ' ).toTitle().toLowerCase() ) {
  192. text = '';
  193. }
  194. else if ( !srbody.continue ) {
  195. text = '\n' + lang.get('search.infopage', '`' + prefix + cmd + ( lang.localNames.page || 'page' ) + ' ' + title + linksuffix + '`');
  196. }
  197. else {
  198. text = '\n' + lang.get('search.infosearch', '`' + prefix + cmd + ( lang.localNames.page || 'page' ) + ' ' + title + linksuffix + '`', '`' + prefix + cmd + ( lang.localNames.search || 'search' ) + ' ' + title + linksuffix + '`');
  199. }
  200. if ( querypage.categoryinfo ) {
  201. var category = [lang.get('search.category.content')];
  202. if ( querypage.categoryinfo.size === 0 ) {
  203. category.push(lang.get('search.category.empty'));
  204. }
  205. if ( querypage.categoryinfo.pages > 0 ) {
  206. category.push(lang.get('search.category.pages', querypage.categoryinfo.pages));
  207. }
  208. if ( querypage.categoryinfo.files > 0 ) {
  209. category.push(lang.get('search.category.files', querypage.categoryinfo.files));
  210. }
  211. if ( querypage.categoryinfo.subcats > 0 ) {
  212. category.push(lang.get('search.category.subcats', querypage.categoryinfo.subcats));
  213. }
  214. if ( msg.showEmbed() ) embed.addField( category[0], category.slice(1).join('\n') );
  215. else text += '\n\n' + category.join('\n');
  216. }
  217. msg.sendChannel( spoiler + '<' + pagelink + '>' + text + spoiler, {embed} );
  218. }
  219. }
  220. }, error => {
  221. console.log( '- Error while getting the search results: ' + error );
  222. msg.sendChannelError( spoiler + '<' + wiki.toLink('Special:Search', 'search=' + title.toSearch(), '', body.query.general) + '>' + spoiler );
  223. } ).finally( () => {
  224. if ( reaction ) reaction.removeEmoji();
  225. } );
  226. }
  227. else if ( querypage.ns === -1 ) {
  228. var pagelink = wiki.toLink(querypage.title, querystring.toTitle(), fragment, body.query.general);
  229. var embed = new MessageEmbed().setAuthor( body.query.general.sitename ).setTitle( querypage.title.escapeFormatting() ).setURL( pagelink ).setThumbnail( logoToURL(body.query.general) );
  230. var specialpage = body.query.specialpagealiases.find( sp => body.query.namespaces['-1']['*'] + ':' + sp.aliases[0].replace( /\_/g, ' ' ) === querypage.title.split('/')[0] );
  231. specialpage = ( specialpage ? specialpage.realname : querypage.title.replace( body.query.namespaces['-1']['*'] + ':', '' ).split('/')[0] ).toLowerCase();
  232. fn.special_page(lang, msg, querypage.title, specialpage, embed, wiki, reaction, spoiler);
  233. }
  234. else {
  235. var pagelink = wiki.toLink(querypage.title, querystring.toTitle(), ( fragment || ( body.query.redirects && body.query.redirects[0].tofragment ) || '' ), body.query.general);
  236. var text = '';
  237. var embed = new MessageEmbed().setAuthor( body.query.general.sitename ).setTitle( querypage.title.escapeFormatting() ).setURL( pagelink );
  238. if ( querypage.pageprops && querypage.pageprops.displaytitle ) {
  239. var displaytitle = htmlToDiscord( querypage.pageprops.displaytitle );
  240. if ( displaytitle.length > 250 ) displaytitle = displaytitle.substring(0, 250) + '\u2026';
  241. embed.setTitle( displaytitle );
  242. }
  243. if ( querypage.extract ) {
  244. var extract = extract_desc(querypage.extract, ( fragment || ( body.query.redirects && body.query.redirects[0].tofragment ) || '' ));
  245. embed.setDescription( extract[0] );
  246. if ( extract[2].length ) embed.addField( extract[1], extract[2] );
  247. }
  248. if ( querypage.pageprops && querypage.pageprops.description ) {
  249. var description = htmlToPlain( querypage.pageprops.description );
  250. if ( description.length > 2000 ) description = description.substring(0, 2000) + '\u2026';
  251. embed.setDescription( description );
  252. }
  253. if ( querypage.pageimage && querypage.original && querypage.title !== body.query.general.mainpage ) {
  254. var pageimage = querypage.original.source;
  255. if ( querypage.ns === 6 ) {
  256. if ( msg.showEmbed() && /\.(?:png|jpg|jpeg|gif)$/.test(querypage.pageimage.toLowerCase()) ) embed.setImage( pageimage );
  257. else if ( msg.uploadFiles() ) embed.attachFiles( [{attachment:pageimage,name:( spoiler ? 'SPOILER ' : '' ) + querypage.pageimage}] );
  258. } else embed.setThumbnail( pageimage );
  259. } else embed.setThumbnail( logoToURL(body.query.general) );
  260. if ( querypage.categoryinfo ) {
  261. var category = [lang.get('search.category.content')];
  262. if ( querypage.categoryinfo.size === 0 ) {
  263. category.push(lang.get('search.category.empty'));
  264. }
  265. if ( querypage.categoryinfo.pages > 0 ) {
  266. category.push(lang.get('search.category.pages', querypage.categoryinfo.pages));
  267. }
  268. if ( querypage.categoryinfo.files > 0 ) {
  269. category.push(lang.get('search.category.files', querypage.categoryinfo.files));
  270. }
  271. if ( querypage.categoryinfo.subcats > 0 ) {
  272. category.push(lang.get('search.category.subcats', querypage.categoryinfo.subcats));
  273. }
  274. if ( msg.showEmbed() ) embed.addField( category[0], category.slice(1).join('\n') );
  275. else text += '\n\n' + category.join('\n');
  276. }
  277. msg.sendChannel( spoiler + '<' + pagelink + '>' + text + spoiler, {embed} );
  278. if ( reaction ) reaction.removeEmoji();
  279. }
  280. }
  281. else if ( body.query.interwiki ) {
  282. if ( msg.channel.type === 'text' && pause[msg.guild.id] ) {
  283. if ( reaction ) reaction.removeEmoji();
  284. console.log( '- Aborted, paused.' );
  285. return;
  286. }
  287. interwiki = body.query.interwiki[0].url;
  288. var maxselfcall = interwikiLimit[( msg?.guild?.id in patreons ? 'patreon' : 'default' )];
  289. if ( selfcall < maxselfcall && /^(?:https?:)?\/\//.test(interwiki) ) {
  290. selfcall++;
  291. var regex = interwiki.match( /^(?:https?:)?\/\/([a-z\d-]{1,50})\.gamepedia\.com(?:\/|$)/ );
  292. if ( regex ) {
  293. let iwtitle = decodeURIComponent( interwiki.replace( regex[0], '' ) ).replace( /\_/g, ' ' );
  294. this.gamepedia(lang, msg, iwtitle, 'https://' + regex[1] + '.gamepedia.com/', '!' + regex[1] + ' ', reaction, spoiler, querystring, fragment, interwiki, selfcall);
  295. return;
  296. }
  297. regex = interwiki.match( /^(?:https?:)?\/\/(([a-z\d-]{1,50})\.(?:fandom\.com|wikia\.org)(?:(?!\/wiki\/)\/([a-z-]{2,12}))?)(?:\/wiki\/|\/?$)/ );
  298. if ( regex ) {
  299. let iwtitle = decodeURIComponent( interwiki.replace( regex[0], '' ) ).replace( /\_/g, ' ' );
  300. this.fandom(lang, msg, iwtitle, 'https://' + regex[1] + '/', ( regex[1].includes( '.wikia.org' ) ? '??' : '?' ) + ( regex[3] ? regex[3] + '.' : '' ) + regex[2] + ' ', reaction, spoiler, querystring, fragment, interwiki, selfcall);
  301. return;
  302. }
  303. let project = wikiProjects.find( project => interwiki.split('/')[2].endsWith( project.name ) );
  304. if ( project ) {
  305. regex = interwiki.match( new RegExp( '^(?:https?:)?//' + project.regex + `(?:${project.articlePath}|/?$)` ) );
  306. if ( regex ) {
  307. let iwtitle = decodeURIComponent( interwiki.replace( regex[0], '' ) ).replace( /\_/g, ' ' );
  308. this.gamepedia(lang, msg, iwtitle, 'https://' + regex[1] + project.scriptPath, cmd + body.query.interwiki[0].iw + ':', reaction, spoiler, querystring, fragment, interwiki, selfcall);
  309. return;
  310. }
  311. }
  312. }
  313. if ( fragment ) fragment = '#' + fragment.toSection();
  314. if ( interwiki.includes( '#' ) ) {
  315. if ( !fragment ) fragment = '#' + interwiki.split('#').slice(1).join('#');
  316. interwiki = interwiki.split('#')[0];
  317. }
  318. if ( querystring ) interwiki += ( interwiki.includes( '?' ) ? '&' : '?' ) + querystring.toTitle();
  319. msg.sendChannel( spoiler + ' ' + interwiki.replace( /@(here|everyone)/g, '%40$1' ) + fragment + ' ' + spoiler ).then( message => {
  320. if ( message && selfcall === maxselfcall ) message.reactEmoji('⚠️');
  321. } );
  322. if ( reaction ) reaction.removeEmoji();
  323. }
  324. else if ( body.query.redirects ) {
  325. var pagelink = wiki.toLink(body.query.redirects[0].to, querystring.toTitle(), ( fragment || body.query.redirects[0].tofragment || '' ), body.query.general);
  326. var embed = new MessageEmbed().setAuthor( body.query.general.sitename ).setTitle( body.query.redirects[0].to.escapeFormatting() ).setURL( pagelink ).setThumbnail( logoToURL(body.query.general) );
  327. msg.sendChannel( spoiler + '<' + pagelink + '>' + spoiler, {embed} );
  328. if ( reaction ) reaction.removeEmoji();;
  329. }
  330. else {
  331. var pagelink = wiki.toLink(body.query.general.mainpage, querystring.toTitle(), fragment, body.query.general);
  332. var embed = new MessageEmbed().setAuthor( body.query.general.sitename ).setTitle( body.query.general.mainpage.escapeFormatting() ).setURL( pagelink ).setThumbnail( logoToURL(body.query.general) );
  333. got.get( wiki + 'api.php?action=query' + ( noRedirect ? '' : '&redirects=true' ) + '&prop=pageprops|extracts&ppprop=description|displaytitle&explaintext=true&exsectionformat=raw&exlimit=1&titles=' + encodeURIComponent( body.query.general.mainpage ) + '&format=json', {
  334. responseType: 'json'
  335. } ).then( mpresponse => {
  336. var mpbody = mpresponse.body;
  337. if ( mpbody && mpbody.warnings ) log_warn(body.warnings);
  338. if ( mpresponse.statusCode !== 200 || !mpbody || mpbody.batchcomplete === undefined || !mpbody.query ) {
  339. console.log( '- ' + mpresponse.statusCode + ': Error while getting the main page: ' + ( mpbody && mpbody.error && mpbody.error.info ) );
  340. } else {
  341. var querypage = Object.values(mpbody.query.pages)[0];
  342. if ( querypage.pageprops && querypage.pageprops.displaytitle ) {
  343. var displaytitle = htmlToDiscord( querypage.pageprops.displaytitle );
  344. if ( displaytitle.length > 250 ) displaytitle = displaytitle.substring(0, 250) + '\u2026';
  345. embed.setTitle( displaytitle );
  346. }
  347. if ( querypage.extract ) {
  348. var extract = extract_desc(querypage.extract, fragment);
  349. embed.setDescription( extract[0] );
  350. if ( extract[2].length ) embed.addField( extract[1], extract[2] );
  351. }
  352. if ( querypage.pageprops && querypage.pageprops.description ) {
  353. var description = htmlToPlain( querypage.pageprops.description );
  354. if ( description.length > 2000 ) description = description.substring(0, 2000) + '\u2026';
  355. embed.setDescription( description );
  356. }
  357. }
  358. }, error => {
  359. console.log( '- Error while getting the main page: ' + error );
  360. } ).finally( () => {
  361. msg.sendChannel( spoiler + '<' + pagelink + '>' + spoiler, {embed} );
  362. if ( reaction ) reaction.removeEmoji();
  363. } );
  364. }
  365. }
  366. }, error => {
  367. if ( interwiki ) msg.sendChannel( spoiler + ' ' + interwiki.replace( /@(here|everyone)/g, '%40$1' ) + fragment + ' ' + spoiler );
  368. else if ( wiki.noWiki(error.message) ) {
  369. console.log( '- This wiki doesn\'t exist!' );
  370. msg.reactEmoji('nowiki');
  371. }
  372. else {
  373. console.log( '- Error while getting the search results: ' + error );
  374. msg.sendChannelError( spoiler + '<' + wiki.toLink( ( querystring || fragment || !title ? title : 'Special:Search' ), ( querystring || fragment || !title ? querystring.toTitle() : 'search=' + title.toSearch() ), fragment) + '>' + spoiler );
  375. }
  376. if ( reaction ) reaction.removeEmoji();
  377. } );
  378. }
  379. }
  380. /**
  381. * Turns the siteinfo logo into an URL.
  382. * @param {Object} arg - The siteinfo from the wiki.
  383. * @param {String} arg.logo - The logo from the wiki.
  384. * @param {String} arg.server - The server URL from the wiki.
  385. * @returns {String}
  386. */
  387. function logoToURL({logo, server: serverURL}) {
  388. if ( /^(?:https?:)?\/\//.test(logo) ) logo = logo.replace( /^(?:https?:)?\/\//, 'https://' );
  389. else logo = serverURL + ( logo.startsWith( '/' ) ? '' : '/' ) + logo;
  390. return logo;
  391. }
  392. /**
  393. * Change HTML text to plain text.
  394. * @param {String} html - The text in HTML.
  395. * @returns {String}
  396. */
  397. function htmlToPlain(html) {
  398. var text = '';
  399. var parser = new htmlparser.Parser( {
  400. ontext: (htmltext) => {
  401. text += htmltext.escapeFormatting();
  402. }
  403. }, {decodeEntities:true} );
  404. parser.write( html );
  405. parser.end();
  406. return text;
  407. };
  408. /**
  409. * Change HTML text to markdown text.
  410. * @param {String} html - The text in HTML.
  411. * @returns {String}
  412. */
  413. function htmlToDiscord(html) {
  414. var text = '';
  415. var parser = new htmlparser.Parser( {
  416. onopentag: (tagname, attribs) => {
  417. switch (tagname) {
  418. case 'b':
  419. text += '**';
  420. break;
  421. case 'i':
  422. text += '*';
  423. break;
  424. case 's':
  425. text += '~~';
  426. break;
  427. case 'u':
  428. text += '__';
  429. break;
  430. }
  431. },
  432. ontext: (htmltext) => {
  433. text += htmltext.escapeFormatting();
  434. },
  435. onclosetag: (tagname) => {
  436. switch (tagname) {
  437. case 'b':
  438. text += '**';
  439. break;
  440. case 'i':
  441. text += '*';
  442. break;
  443. case 's':
  444. text += '~~';
  445. break;
  446. case 'u':
  447. text += '__';
  448. break;
  449. }
  450. }
  451. }, {decodeEntities:true} );
  452. parser.write( html );
  453. parser.end();
  454. return text;
  455. };
  456. module.exports = gamepedia_check_wiki;