integrations.discord.events.comparison

  1"""
  2integrations/discord/events/comparison.py
  3"""
  4
  5import asyncio
  6import logging
  7from typing import TYPE_CHECKING, cast
  8
  9from discord import Message
 10from discord.channel import TextChannel
 11
 12import libs.global_value as g
 13from libs.domain import modify
 14from libs.domain.datamodels import ComparisonResults
 15from libs.domain.score import GameResult
 16from libs.functions import search
 17from libs.types import ActionStatus, CommandType, RemarkDict, StyleOptions
 18from libs.utils import formatter, validator
 19from libs.utils.timekit import ExtendedDatetime as ExtDt
 20
 21if TYPE_CHECKING:
 22    from integrations.discord.adapter import ServiceAdapter
 23    from integrations.protocols import MessageParserProtocol
 24
 25
 26def main(m: "MessageParserProtocol") -> None:
 27    """
 28    突合処理(非同期関数呼び出しラッパー)
 29
 30    Args:
 31        m (MessageParserProtocol): メッセージデータ
 32
 33    """
 34    asyncio.create_task(_wrapper(m))
 35
 36
 37async def _wrapper(m: "MessageParserProtocol") -> None:
 38    g.adapter = cast("ServiceAdapter", g.adapter)
 39    results = ComparisonResults(search_after=-g.adapter.conf.search_after)
 40    messages_list: list["MessageParserProtocol"] = []
 41
 42    await search_messages(results, messages_list)
 43    await check_omission(results, messages_list)
 44    await check_remarks(results, messages_list)
 45    await check_total_score(results, messages_list)
 46
 47    m.set_headline(results.output("headline"), StyleOptions(title=m.keyword))
 48    m.set_message(results.output("mismatch"), StyleOptions(title="不一致", key_title=False))
 49    m.set_message(results.output("missing"), StyleOptions(title="取りこぼし", key_title=False))
 50    m.set_message(results.output("delete"), StyleOptions(title="削除漏れ", key_title=False))
 51    m.set_message(results.output("remark_mod"), StyleOptions(title="メモ更新", key_title=False))
 52    m.set_message(results.output("remark_del"), StyleOptions(title="メモ削除", key_title=False))
 53    if results.invalid_score:
 54        m.set_message(results.output("invalid_score"), StyleOptions(title="供託残り", key_title=False))
 55
 56    m.post.thread = True
 57    m.status.action = ActionStatus.NOTHING
 58    g.adapter.api.post(m)
 59
 60
 61async def search_messages(results: ComparisonResults, messages_list: list["MessageParserProtocol"]) -> None:
 62    """
 63    メッセージ全検索
 64
 65    Args:
 66        results (ComparisonResults): 結果格納データクラス
 67        messages_list (list[MessageParserProtocol]): 検索結果
 68
 69    """
 70    g.adapter = cast("ServiceAdapter", g.adapter)
 71
 72    for ch in g.adapter.api.bot.get_all_channels():
 73        # アクセス権がないチャンネルはスキップ
 74        if not ch.permissions_for(ch.guild.me).read_messages:
 75            continue
 76
 77        if isinstance(ch, TextChannel):
 78            channel = g.adapter.api.bot.get_channel(ch.id)
 79            if not isinstance(channel, TextChannel):
 80                continue
 81
 82            logging.debug("channel: %s, after: %s", ch.name, results.after.format(ExtDt.FMT.YMDHMS))
 83
 84            messages = await channel.history(after=results.after.dt, oldest_first=True).flatten()
 85            for message in messages:
 86                if not isinstance(message, Message):
 87                    continue
 88                if message.author.bot:
 89                    continue
 90
 91                work_m = g.adapter.parser()  # 検索結果格納用
 92                work_m.parser(message)
 93                if not work_m.check_updatable:  # DB更新不可チャンネルは対象外
 94                    logging.debug("skip limited channel.")
 95                    break
 96
 97                messages_list.append(work_m)
 98
 99
100async def check_omission(results: ComparisonResults, messages_list: list["MessageParserProtocol"]) -> None:
101    """
102    スコア突合
103
104    Args:
105        results (ComparisonResults): 結果格納データクラス
106        messages_list (list[MessageParserProtocol]): 検索結果
107
108    """
109    g.adapter = cast("ServiceAdapter", g.adapter)
110    discord_score: list[GameResult] = []
111
112    for work_m in messages_list:
113        if work_m.keyword in g.keyword_dispatcher:  # コマンドキーワードはスキップ
114            continue
115        if detection := validator.check_score(work_m):
116            score = GameResult(**detection)
117            for k, v in score.to_dict().items():  # 名前の正規化
118                if str(k).endswith("_name"):
119                    score.set(**{k: formatter.name_replace(str(v), not_replace=True)})
120            discord_score.append(score)
121            results.score_list.update({work_m.data.event_ts: work_m})
122            logging.debug(score.to_text("logging"))
123
124    db_score = search.for_db_score(float(results.after.format(ExtDt.FMT.TS)))
125
126    # DISCORD -> DATABASE
127    ts_list = [x.ts for x in db_score]
128    for score in discord_score:
129        work_m = results.score_list[score.ts]
130        if score.ts in ts_list:
131            target = db_score[ts_list.index(score.ts)]
132            if score != target:  # 不一致(更新)
133                results.mismatch.append({"before": target, "after": score})
134                logging.info("mismatch: %s (%s)", score.ts, ExtDt(float(score.ts)).format(ExtDt.FMT.YMDHMS))
135                logging.debug("  * discord: %s", score.to_text("detail"))
136                logging.debug("  *      db: %s", target.to_text("detail"))
137                modify.db_update(score, work_m)
138        else:  # 取りこぼし(追加)
139            results.missing.append(score)
140            logging.info("missing: %s (%s)", score.ts, ExtDt(float(score.ts)).format(ExtDt.FMT.YMDHMS))
141            logging.debug(score.to_text("logging"))
142            modify.db_insert(score, work_m)
143
144    # DATABASE -> DISCORD
145    ts_list = [x.ts for x in discord_score]
146    work_m = g.adapter.parser()
147    for score in db_score:
148        if score.ts not in ts_list:  # 削除漏れ
149            results.delete.append(score)
150            work_m.data.event_ts = score.ts
151            if score.source:
152                work_m.data.channel_id = score.source.replace("discord_", "")
153            logging.info("delete (Only database): %s %s", ExtDt(float(score.ts)).format(ExtDt.FMT.YMDHMS), score.to_text("logging"))
154            work_m.status.command_type = CommandType.COMPARISON
155            modify.db_delete(work_m)
156
157
158async def check_remarks(results: ComparisonResults, messages_list: list["MessageParserProtocol"]) -> None:
159    """
160    メモ突合
161
162    Args:
163        results (ComparisonResults): 結果格納データクラス
164        messages_list (list[MessageParserProtocol]): 検索結果
165
166    """
167    g.adapter = cast("ServiceAdapter", g.adapter)
168    discord_remarks: list[RemarkDict] = []
169    score_list: dict[str, GameResult] = {}
170
171    for loop_m in messages_list:
172        if detection := validator.check_score(loop_m):
173            score = GameResult(**detection)
174            for k, v in score.to_dict().items():  # 名前の正規化
175                if str(k).endswith("_name"):
176                    score.set(**{k: formatter.name_replace(str(v), not_replace=True)})
177            score_list.update({loop_m.data.event_ts: score})
178
179        if loop_m.keyword in g.cfg.rule.remarks_words:
180            for name, matter in zip(loop_m.argument[0::2], loop_m.argument[1::2]):
181                # 対象外のメモはスキップ
182                if not float(loop_m.data.thread_ts):
183                    continue  # リプライになっていない
184                if loop_m.data.thread_ts not in score_list:
185                    continue  # ゲーム結果に紐付かない
186                pname = formatter.name_replace(str(name), not_replace=True)
187                if pname not in score_list[loop_m.data.thread_ts].to_list("name"):
188                    continue  # ゲーム結果に名前がない
189
190                discord_remarks.append(
191                    {
192                        "thread_ts": loop_m.data.thread_ts,
193                        "event_ts": loop_m.data.event_ts,
194                        "name": pname,
195                        "matter": matter,
196                        "source": loop_m.status.source,
197                    }
198                )
199
200    db_remarks = search.for_db_remarks(float(results.after.format(ExtDt.FMT.TS)))
201
202    # DISCORD -> DATABASE
203    work_m = g.adapter.parser()
204    work_m.status.command_type = CommandType.COMPARISON
205
206    for remark in discord_remarks:
207        if remark in db_remarks:  # 変化なし
208            continue
209        results.remark_mod.append(remark)
210
211    for event_ts in {x["event_ts"] for x in results.remark_mod}:
212        work_m.data.event_ts = event_ts
213        modify.remarks_delete(work_m)
214    modify.remarks_append(work_m, results.remark_mod)
215
216    # DATABASE -> DISCORD
217    work_remarks = [{k: str(v) for k, v in d.items() if k != "source"} for d in discord_remarks]  # sourceを除外したリスト
218    for remark in db_remarks:
219        check_remark = {k: str(v) for k, v in remark.items() if k != "source"}
220        if check_remark not in work_remarks:  # Discordに記録なし
221            results.remark_del.append(remark)
222            modify.remarks_delete_compar(work_m, remark)
223
224
225async def check_total_score(results: ComparisonResults, messages_list: list["MessageParserProtocol"]) -> None:
226    """
227    素点合計の再チェック
228
229    Args:
230        results (ComparisonResults): 結果格納データクラス
231        messages_list (list[MessageParserProtocol]): 検索結果
232
233    """
234    for work_m in messages_list:
235        if detection := validator.check_score(work_m):
236            score = GameResult(**detection)
237            for k, v in score.to_dict().items():  # 名前の正規化
238                if str(k).endswith("_name"):
239                    score.set(**{k: formatter.name_replace(str(v), not_replace=True)})
240            if score.deposit:
241                results.invalid_score.append(score)