From a7ba75b1b6ca4faa392cb3e5655fc784687e02ac Mon Sep 17 00:00:00 2001
From: "Tyler St. Onge" <tylertstonge@gmail.com>
Date: Thu, 5 Nov 2020 15:16:00 -0500
Subject: added discord reporter

---
 src/main/scala/com/tylerstonge/honeypot/Supervisor.scala | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

(limited to 'src/main/scala/com/tylerstonge/honeypot/Supervisor.scala')

diff --git a/src/main/scala/com/tylerstonge/honeypot/Supervisor.scala b/src/main/scala/com/tylerstonge/honeypot/Supervisor.scala
index a6551e7..d5c59fe 100644
--- a/src/main/scala/com/tylerstonge/honeypot/Supervisor.scala
+++ b/src/main/scala/com/tylerstonge/honeypot/Supervisor.scala
@@ -6,7 +6,7 @@ import akka.actor.{Actor, Props}
 import akka.event.{Logging, LoggingAdapter}
 import com.tylerstonge.honeypot.ftp.FtpListener
 import com.tylerstonge.honeypot.messages.{MStartComponent, MStartReporter}
-import com.tylerstonge.honeypot.reporter.LogReporter
+import com.tylerstonge.honeypot.reporter.{LogReporter, DiscordReporter}
 import org.typelevel.jawn.ast.{JArray, JParser}
 
 
@@ -17,7 +17,7 @@ class Supervisor extends Actor {
   override def preStart: Unit = {
     val cfg = JParser.parseFromFile(Paths.get("config.json").toFile).get
     cfg.get("reporters").asInstanceOf[JArray].vs.foreach(r => {
-      self ! MStartReporter(r.get("name").asString, r.get("type").asString)
+      self ! MStartReporter(r.get("name").asString, r.get("type").asString, r.get("options"))
     })
     cfg.get("components").asInstanceOf[JArray].vs.foreach(c => {
       self ! MStartComponent(c.get("name").asString, c.get("type").asString, c.get("port").asInt)
@@ -38,7 +38,7 @@ class Supervisor extends Actor {
     log.info("starting component :: {}", msg.name)
     msg.ctype match {
       case "ftp" => context.actorOf(FtpListener.props(msg.port), name = msg.name)
-      case _ => log.error("unknown component type: {}", msg.ctype);
+      case _ => log.error("unknown component type: {}", msg.ctype)
     }
   }
 
@@ -46,7 +46,8 @@ class Supervisor extends Actor {
     log.info("starting reporter :: {}", msg.name)
     msg.rtype match {
       case "console-logger" => context.actorOf(Props[LogReporter], name = msg.name)
-      case _ => log.error("unknown reporter type: {}", msg.rtype);
+      case "discord-logger" => context.actorOf(DiscordReporter.props(msg.options.get("webhook").asString), name = msg.name)
+      case _ => log.error("unknown reporter type: {}", msg.rtype)
     }
   }
 }
\ No newline at end of file
-- 
cgit v1.1