From 8f1719a17f53ea357c9ebc51d18eb1fe50840afb Mon Sep 17 00:00:00 2001
From: Radek Golan <shield@shielddagger.com>
Date: Mon, 22 Apr 2024 16:45:05 +0200
Subject: [PATCH] Fix package + add trivyignore

---
 .trivyignore                     |  8 ++++++
 src/discord_notifier/__init__.py | 42 --------------------------------
 src/discord_notifier/__main__.py | 42 ++++++++++++++++++++++++++++++++
 3 files changed, 50 insertions(+), 42 deletions(-)
 create mode 100644 .trivyignore
 create mode 100644 src/discord_notifier/__main__.py

diff --git a/.trivyignore b/.trivyignore
new file mode 100644
index 0000000..a8ae57e
--- /dev/null
+++ b/.trivyignore
@@ -0,0 +1,8 @@
+CVE-2023-52425
+CVE-2024-26462
+CVE-2023-7104
+CVE-2023-50387
+CVE-2023-50868
+CVE-2023-31484
+CVE-2023-45853
+CVE-2024-2961
diff --git a/src/discord_notifier/__init__.py b/src/discord_notifier/__init__.py
index 38055b6..e69de29 100644
--- a/src/discord_notifier/__init__.py
+++ b/src/discord_notifier/__init__.py
@@ -1,42 +0,0 @@
-from os import environ
-from discord_webhook import DiscordEmbed, DiscordWebhook
-
-webhook = DiscordWebhook(environ["PLUGIN_WEBHOOK_URL"])
-
-success = environ["CI_PIPELINE_STATUS"] != "failed"
-
-webhook.username = "Woodpecker CI"
-webhook.avatar_url = "https://ci.shielddagger.com/favicons/favicon-dark-success.png" if success else \
-                     "https://ci.shielddagger.com/favicons/favicon-dark-error.png"
-
-webhook.add_embed(DiscordEmbed(
-    "Pipeline Failed" if not success else "Pipeline Succeeded",
-    color="ED4245" if environ["CI_PIPELINE_STATUS"] == "failed" else "57F287",
-    url=environ["CI_PIPELINE_URL"],
-    author={
-        "name": environ["CI_COMMIT_AUTHOR"],
-        "icon_url": environ["CI_COMMIT_AUTHOR_AVATAR"]
-    },
-    footer={
-        "text": environ["CI_REPO"],
-        "icon_url": environ["CI_REPO_URL"]
-    },
-    provider={
-        "name": "Woodpecker CI",
-        "url": environ["PLUGIN_WOODPECKER_URL"]
-    },
-    fields=[{
-        "name": "Commit",
-        "value": environ["CI_COMMIT_SHA"][0:8],
-        "inline": True
-    },{
-        "name": "Branch",
-        "value": environ["CI_COMMIT_BRANCH"],
-        "inline": True
-    },{
-        "name": "Commit Message",
-        "value": environ["CI_COMMIT_MESSAGE"]
-    }]
-))
-
-webhook.execute()    
diff --git a/src/discord_notifier/__main__.py b/src/discord_notifier/__main__.py
new file mode 100644
index 0000000..38055b6
--- /dev/null
+++ b/src/discord_notifier/__main__.py
@@ -0,0 +1,42 @@
+from os import environ
+from discord_webhook import DiscordEmbed, DiscordWebhook
+
+webhook = DiscordWebhook(environ["PLUGIN_WEBHOOK_URL"])
+
+success = environ["CI_PIPELINE_STATUS"] != "failed"
+
+webhook.username = "Woodpecker CI"
+webhook.avatar_url = "https://ci.shielddagger.com/favicons/favicon-dark-success.png" if success else \
+                     "https://ci.shielddagger.com/favicons/favicon-dark-error.png"
+
+webhook.add_embed(DiscordEmbed(
+    "Pipeline Failed" if not success else "Pipeline Succeeded",
+    color="ED4245" if environ["CI_PIPELINE_STATUS"] == "failed" else "57F287",
+    url=environ["CI_PIPELINE_URL"],
+    author={
+        "name": environ["CI_COMMIT_AUTHOR"],
+        "icon_url": environ["CI_COMMIT_AUTHOR_AVATAR"]
+    },
+    footer={
+        "text": environ["CI_REPO"],
+        "icon_url": environ["CI_REPO_URL"]
+    },
+    provider={
+        "name": "Woodpecker CI",
+        "url": environ["PLUGIN_WOODPECKER_URL"]
+    },
+    fields=[{
+        "name": "Commit",
+        "value": environ["CI_COMMIT_SHA"][0:8],
+        "inline": True
+    },{
+        "name": "Branch",
+        "value": environ["CI_COMMIT_BRANCH"],
+        "inline": True
+    },{
+        "name": "Commit Message",
+        "value": environ["CI_COMMIT_MESSAGE"]
+    }]
+))
+
+webhook.execute()