Merge pull request #1846 from TheHive-Project/issue-1843

Issue 1843

thehive/app/org/thp/thehive/controllers/v1/MonitoringCtrl.scala

@@ -0,0 +1,47 @@
+package org.thp.thehive.controllers.v1
+
+import org.thp.scalligraph.controllers.Entrypoint
+import org.thp.scalligraph.models.Database
+import org.thp.scalligraph.services.config.{ApplicationConfig, ConfigItem}
+import org.thp.thehive.models.Permissions
+import play.api.libs.json.{Format, JsArray, Json}
+import play.api.mvc.{Action, AnyContent, Results}
+
+import java.io.File
+import javax.inject.{Inject, Singleton}
+import scala.util.Success
+
+@Singleton
+class MonitoringCtrl @Inject() (
+    appConfig: ApplicationConfig,
+    entrypoint: Entrypoint,
+    db: Database
+) {
+  case class PartitionConfig(location: String)
+
+  object PartitionConfig {
+    implicit val format: Format[PartitionConfig] = Json.format[PartitionConfig]
+  }
+
+  val diskLocationsConfig: ConfigItem[Seq[PartitionConfig], Seq[PartitionConfig]] =
+    appConfig.item[Seq[PartitionConfig]]("monitor.disk", "disk locations to monitor")
+  def diskLocations: Seq[PartitionConfig] = diskLocationsConfig.get
+
+  def diskUsage: Action[AnyContent] =
+    entrypoint("monitor disk usage")
+      .authPermittedTransaction(db, Permissions.managePlatform)(implicit request =>
+        implicit graph =>
+          for {
+            _ <- Success(())
+            locations = diskLocations.map { dl =>
+              val file = new File(dl.location)
+              Json.obj(
+                "location"   -> dl.location,
+                "freeSpace"  -> file.getFreeSpace,
+                "totalSpace" -> file.getTotalSpace
+              )
+            }
+          } yield Results.Ok(JsArray(locations))
+      )
+
+}

thehive/app/org/thp/thehive/controllers/v1/Router.scala

@@ -20,6 +20,7 @@ class Router @Inject() (
     // dashboardCtrl: DashboardCtrl,
     describeCtrl: DescribeCtrl,
     logCtrl: LogCtrl,
+    monitoringCtrl: MonitoringCtrl,
     observableCtrl: ObservableCtrl,
     observableTypeCtrl: ObservableTypeCtrl,
     organisationCtrl: OrganisationCtrl,
@@ -177,5 +178,7 @@ class Router @Inject() (
     case GET(p"/observable/type/$idOrName")    => observableTypeCtrl.get(idOrName)
     case POST(p"/observable/type")             => observableTypeCtrl.create
     case DELETE(p"/observable/type/$idOrName") => observableTypeCtrl.delete(idOrName)
+
+    case GET(p"/monitor/disk") => monitoringCtrl.diskUsage
   }
 }