Fix ProgramFinishedException color.

PokemonAutomation · Dec 8, 2024 · 5f89e8f · 5f89e8f
1 parent 7df8d30
commit 5f89e8f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/SerialPrograms/Source/CommonFramework/Exceptions/ProgramFinishedException.h b/SerialPrograms/Source/CommonFramework/Exceptions/ProgramFinishedException.h
@@ -51,6 +51,7 @@ class ProgramFinishedException : public ScreenshotException{
         std::shared_ptr<const ImageRGB32> screenshot
     );
 
+    virtual Color color() const override{ return COLOR_GREEN; }
 
 public:
     virtual void log(Logger& logger) const override;

diff --git a/SerialPrograms/Source/CommonFramework/Exceptions/ScreenshotException.cpp b/SerialPrograms/Source/CommonFramework/Exceptions/ScreenshotException.cpp
@@ -100,7 +100,7 @@ void ScreenshotException::send_notification(ProgramEnvironment& env, EventNotifi
 
     send_program_notification(
         env, notification,
-        COLOR_RED,
+        color(),
         name(),
         std::move(embeds), "",
         screenshot()

diff --git a/SerialPrograms/Source/CommonFramework/Exceptions/ScreenshotException.h b/SerialPrograms/Source/CommonFramework/Exceptions/ScreenshotException.h
@@ -69,6 +69,7 @@ class ScreenshotException : public Exception{
     virtual std::string message() const override{ return m_message; }
     ImageViewRGB32 screenshot() const;
 
+    virtual Color color() const{ return COLOR_RED; }
     virtual void send_notification(ProgramEnvironment& env, EventNotificationOption& notification) const;
 
 public:
-Original file line number
+Diff line change
@@ Expand Up / @@ -51,6 +51,7 @@ class ProgramFinishedException : public ScreenshotException{ @@
             std::shared_ptr<const ImageRGB32> screenshot
         );
+        virtual Color color() const override{ return COLOR_GREEN; }
     public:
         virtual void log(Logger& logger) const override;
@@ Expand Down @@