push-status-to-discord-action/main.py

import sys
import typing
import datetime
import os

import httpx

COLORS = {
    "failure": 0xCB2431,
    "success": 0x28A745,
    "info": 0x0000FF,
}


class EmbedField(typing.TypedDict):
    name: str
    value: str
    inline: bool


class Embed(typing.TypedDict):
    color: str
    title: str
    fields: list[EmbedField]


class MessageState(typing.TypedDict):
    """
    Partial mapping of Discord message information.

    See: https://discord.com/developers/docs/resources/channel#message-object
    """

    id: int
    timestamp: str
    edited_timestamp: str
    embeds: list[Embed]


class Context(typing.TypedDict):
    message: MessageState


if __name__ == "__main__":
    (
        webhook_url,
        status,
        variant,
        title,
    ) = sys.argv[1:5]

    message_id = sys.argv[5] if len(sys.argv) == 6 else None

    previous_message_data = (
        httpx.get(f"{webhook_url}/messages/{message_id}").json()
        if message_id is not None
        else None
    )

    embed_data = Embed(
        color=COLORS[variant],
        title=title,
        fields=[
            EmbedField(name="Status", value=status, inline=True),
            EmbedField(
                name="Updated at",
                value=str(datetime.datetime.now().ctime()),
                inline=True,
            ),
        ],
    )

    if message_id:
        previous_message_data = httpx.get(f"{webhook_url}/messages/{message_id}").json()

        response = httpx.patch(
            f"{webhook_url}/messages/{message_id}?wait=true",
            json={"embeds": [*previous_message_data["embeds"], embed_data]},
        )
        message_id = response.json()["id"]
    else:
        response = httpx.post(f"{webhook_url}?wait=true", json={"embeds": [embed_data]})
        message_id = response.json()["id"]

    outputs_path = os.environ.get("GITHUB_OUTPUT")

    if not outputs_path:
        raise RuntimeError("Cannot write to outputs, no GITHUB_ENV set.")

    with open(outputs_path, "a", encoding="utf-8") as outputs_file:
        outputs_file.write(f"message-id={message_id}\n")