qrm2/exts/morse.py

"""
Morse Code extension for qrm
---
Copyright (C) 2019-2021 Abigail Gold, 0x5c

This file is part of qrm2 and is released under the terms of
the GNU General Public License, version 2.
"""


import json

import discord.ext.commands as commands

import common as cmn


class MorseCog(commands.Cog):
    def __init__(self, bot: commands.Bot):
        self.bot = bot
        with open(cmn.paths.resources / "morse.1.json") as file:
            d = json.load(file)
            self.morse: dict[str, str] = d["morse"]
            self.ascii: dict[str, int] = d["ascii"]

    @commands.command(name="morse", aliases=["cw"], category=cmn.cat.ref)
    async def _morse(self, ctx: commands.Context, *, msg: str):
        """Converts ASCII to international morse code."""
        result = ""
        for char in msg.upper():
            try:
                result += self.morse[char]
            except KeyError:
                result += "<?>"
            result += " "
        embed = cmn.embed_factory(ctx)
        embed.title = f"Morse Code for {msg}"
        embed.description = "**" + result + "**"
        embed.colour = cmn.colours.good
        await ctx.send(embed=embed)

    @commands.command(name="unmorse", aliases=["demorse", "uncw", "decw"], category=cmn.cat.ref)
    async def _unmorse(self, ctx: commands.Context, *, msg: str):
        """Converts international morse code to ASCII."""
        result = ""
        msg0 = msg
        msg = msg.split("/")
        msg = [m.split() for m in msg]
        for word in msg:
            for char in word:
                try:
                    result += self.ascii[char]
                except KeyError:
                    result += "<?>"
            result += " "
        embed = cmn.embed_factory(ctx)
        embed.title = f"ASCII for {msg0}"
        embed.description = result
        embed.colour = cmn.colours.good
        await ctx.send(embed=embed)

    @commands.command(name="cwweight", aliases=["weight", "cww"], category=cmn.cat.ref)
    async def _weight(self, ctx: commands.Context, *, msg: str):
        """Calculates the CW weight of a callsign or message."""
        embed = cmn.embed_factory(ctx)
        msg = msg.upper()
        weight = 0
        for char in msg:
            try:
                cw_char = self.morse[char].replace("-", "==")
                weight += len(cw_char) * 2 + 2
            except KeyError:
                embed.title = "Error in calculation of CW weight"
                embed.description = f"Unknown character `{char}` in message"
                embed.colour = cmn.colours.bad
                await ctx.send(embed=embed)
                return
        embed.title = f"CW Weight of {msg}"
        embed.description = f"The CW weight is **{weight}**"
        embed.colour = cmn.colours.good
        await ctx.send(embed=embed)


def setup(bot: commands.Bot):
    bot.add_cog(MorseCog(bot))