From bf8c744d8dc98def867ebb4df99b30ef6198df86 Mon Sep 17 00:00:00 2001 From: Koki Takahashi Date: Tue, 15 Oct 2024 06:10:28 +0000 Subject: [PATCH] city-symbol: Add AI hint sources section to answerMessage --- city-symbol/index.ts | 74 ++++++++++++++++++++++++++++++++++++++------ 1 file changed, 64 insertions(+), 10 deletions(-) diff --git a/city-symbol/index.ts b/city-symbol/index.ts index a683f1da..5a11c26e 100644 --- a/city-symbol/index.ts +++ b/city-symbol/index.ts @@ -82,12 +82,19 @@ interface CityInformation { placeImage: string; plainText: string; ruby: string; + kotobankUrl: string | null; kotobankDescription: string | null; + chakuwikiUrl: string | null; chakuwikiDescription: string | null; } type City = CitySymbol & CityInformation; +interface DictionaryInformation { + url: string; + description: string; +} + const extractWikipediaPlaintext = (content: string) => { const normalizedContent = content .replaceAll(/]*>.*?<\/ref>/g, '') @@ -201,7 +208,7 @@ const getPlaintextWikipedia = async (title: string): Promise => { return content; }; -const getKotobankCityDescription = async (city: CitySymbol) => { +const getKotobankCityDescription = async (city: CitySymbol): Promise => { log.info(`Getting kotobank britannica information for ${city.cityName}...`); const response = await fetch(`https://kotobank.jp/word/${city.cityName}`); @@ -229,14 +236,21 @@ const getKotobankCityDescription = async (city: CitySymbol) => { continue; } - return normalizedDescription; + const id = $britannica.attr('id') ?? ''; + const url = new URL(response.url); + url.hash = id; + + return { + url: url.toString(), + description: normalizedDescription, + }; } log.warn(`Kotobank britannica information for ${city.cityName} not found`); return null; }; -const getChakuwikiCityDescription = async (city: CitySymbol) => { +const getChakuwikiCityDescription = async (city: CitySymbol): Promise => { log.info(`Getting chakuwiki information for ${city.cityName}...`); const fullCityName = `${city.prefectureName}${city.cityName}`; const title = chakuwikiTitleMap.get(fullCityName); @@ -245,7 +259,7 @@ const getChakuwikiCityDescription = async (city: CitySymbol) => { return null; } - const url = `https://chakuwiki.org/w/api.php?${qs.encode({ + const chakuwikiApiUrl = `https://chakuwiki.org/w/api.php?${qs.encode({ format: 'json', action: 'query', prop: 'revisions', @@ -253,7 +267,7 @@ const getChakuwikiCityDescription = async (city: CitySymbol) => { titles: title, })}`; - const response = await fetch(url); + const response = await fetch(chakuwikiApiUrl); const json = await response.json(); const pages = json?.query?.pages; @@ -264,11 +278,15 @@ const getChakuwikiCityDescription = async (city: CitySymbol) => { } let isRumorSection = false; + let header = null; const rumors: string[] = []; for (const line of content.split('\n')) { if (line.startsWith('=')) { if (line.includes(city.cityName)) { isRumorSection = true; + if (header === null) { + header = line.replaceAll('=', '').trim(); + } } else { isRumorSection = false; } @@ -280,7 +298,14 @@ const getChakuwikiCityDescription = async (city: CitySymbol) => { } } - return rumors.map((rumor) => `* ${rumor}`).join('\n'); + const description = rumors.map((rumor) => `* ${rumor}`).join('\n'); + const url = new URL(`https://chakuwiki.org/wiki/${title}`); + url.hash = header; + + return { + url: url.toString(), + description, + }; }; const getCityInformation = async (city: CitySymbol): Promise => { @@ -310,10 +335,18 @@ const getCityInformation = async (city: CitySymbol): Promise => const rubyMatches = plainText.match(/（(.+?)）は/); const ruby = rubyMatches?.[1] ?? ''; - const kotobankDescription = await getKotobankCityDescription(city); - const chakuwikiDescription = await getChakuwikiCityDescription(city); - - return {placeImage, ruby, plainText, kotobankDescription, chakuwikiDescription}; + const kotobankInformation = await getKotobankCityDescription(city); + const chakuwikiInformation = await getChakuwikiCityDescription(city); + + return { + placeImage, + plainText, + ruby, + kotobankUrl: kotobankInformation?.url ?? null, + kotobankDescription: kotobankInformation?.description ?? null, + chakuwikiUrl: chakuwikiInformation?.url ?? null, + chakuwikiDescription: chakuwikiInformation?.description ?? null, + }; }; const getRandomCitySymbol = async (): Promise => { @@ -325,7 +358,9 @@ const getRandomCitySymbol = async (): Promise => { reason: '伯方の塩のパッケージに描かれている赤と青のストライプを直方体にあしらったもの', plainText: '博多市は、TSG CTF や TSG LIVE! の開催などを行ったTSG部員である。', kotobankDescription: '博多市は、JavaScriptやTypeScriptなどのプログラミング言語を得意とするプログラマであり、AtCoderの最高レーティングは2309である。', + kotobankUrl: 'https://hakatashi.com', chakuwikiDescription: '博多市は、加熱した玉ねぎを食べることを苦手としており、麦から作られたウイスキーやビールなどの飲み物を嫌っている。', + chakuwikiUrl: 'https://hakatashi.com', date: '2020年6月21日', notes: 'なし', files: ['https://raw.githubusercontent.com/hakatashi/icon/master/images/icon_480px.png'], @@ -449,6 +484,16 @@ export default (slackClients: SlackInterface) => { const aiHintsLoader = new Loader(() => generateAiHints(city)); aiHintsLoader.load(); + const aiHintSources: string[] = []; + const wikipediaUrl = `https://ja.wikipedia.org/wiki/${encodeURIComponent(city.cityWikipediaName)}`; + aiHintSources.push(`<${wikipediaUrl}|Wikipedia>`); + if (city.kotobankUrl) { + aiHintSources.push(`<${city.kotobankUrl}|コトバンク>`); + } + if (city.chakuwikiUrl) { + aiHintSources.push(`<${city.chakuwikiUrl}|Chakuwiki>`); + } + const problem = { problemMessage: { channel: message.channel, @@ -575,6 +620,15 @@ export default (slackClients: SlackInterface) => { }, ], }, + { + type: 'context', + elements: [ + { + type: 'mrkdwn', + text: `ソース: ${aiHintSources.join(', ')}`, + }, + ], + }, ]), ], };