Skip to content

Commit

Permalink
#53 Create Cortex connector
Browse files Browse the repository at this point in the history
  • Loading branch information
To-om committed Dec 2, 2016
1 parent b862f87 commit 584d6d4
Show file tree
Hide file tree
Showing 23 changed files with 584 additions and 638 deletions.
11 changes: 7 additions & 4 deletions build.sbt
Original file line number Diff line number Diff line change
Expand Up @@ -11,10 +11,15 @@ lazy val thehiveMisp = (project in file("thehive-misp"))
.dependsOn(thehiveBackend)
.settings(publish := {})

lazy val thehiveCortex = (project in file("thehive-cortex"))
.dependsOn(thehiveBackend)
.settings(publish := {})
.settings(SbtScalariform.scalariformSettings: _*)

lazy val main = (project in file("."))
.enablePlugins(PlayScala)
.dependsOn(thehiveBackend, thehiveMetrics, thehiveMisp)
.aggregate(thehiveBackend, thehiveMetrics, thehiveMisp)
.dependsOn(thehiveBackend, thehiveMetrics, thehiveMisp, thehiveCortex)
.aggregate(thehiveBackend, thehiveMetrics, thehiveMisp, thehiveCortex)
.settings(aggregate in Docker := false)
.settings(PublishToBinTray.settings: _*)
.settings(Release.settings: _*)
Expand Down Expand Up @@ -113,8 +118,6 @@ import scalariform.formatter.preferences._
import com.typesafe.sbt.SbtScalariform
import com.typesafe.sbt.SbtScalariform.ScalariformKeys

SbtScalariform.defaultScalariformSettings

ScalariformKeys.preferences in ThisBuild := ScalariformKeys.preferences.value
.setPreference(AlignParameters, false)
// .setPreference(FirstParameterOnNewline, Force)
Expand Down
51 changes: 0 additions & 51 deletions thehive-backend/app/controllers/Analyzer.scala

This file was deleted.

13 changes: 4 additions & 9 deletions thehive-backend/app/controllers/Artifact.scala
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,7 @@ import javax.inject.{ Inject, Singleton }
import scala.concurrent.{ ExecutionContext, Future }

import play.api.http.Status
import play.api.libs.json.{ JsArray, Json }
import play.api.libs.json.Json.toJsFieldJsValueWrapper
import play.api.libs.json.JsArray
import play.api.mvc.Controller

import org.elastic4play.{ BadRequestError, Timed }
Expand All @@ -16,13 +15,11 @@ import org.elastic4play.services.{ Agg, AuxSrv }
import org.elastic4play.services.{ QueryDSL, QueryDef, Role }
import org.elastic4play.services.JsonFormat.{ aggReads, queryReads }

import models.JsonFormat.analyzerWrites
import services.{ AnalyzerSrv, ArtifactSrv }
import services.ArtifactSrv

@Singleton
class ArtifactCtrl @Inject() (
artifactSrv: ArtifactSrv,
analyzerSrv: AnalyzerSrv,
auxSrv: AuxSrv,
authenticated: Authenticated,
renderer: Renderer,
Expand Down Expand Up @@ -54,10 +51,8 @@ class ArtifactCtrl @Inject() (

@Timed
def get(id: String) = authenticated(Role.read).async(fieldsBodyParser) { implicit request
for {
artifact artifactSrv.get(id, request.body.getStrings("fields").map("dataType" +: _))
analyzers analyzerSrv.availableFor(artifact.dataType()).map(multiResult Json.toJson(multiResult))
} yield renderer.toOutput(OK, Json.obj("artifact" artifact, "analyzers" analyzers))
artifactSrv.get(id, request.body.getStrings("fields").map("dataType" +: _))
.map(artifact renderer.toOutput(OK, artifact))
}

@Timed
Expand Down
68 changes: 0 additions & 68 deletions thehive-backend/app/controllers/Job.scala

This file was deleted.

142 changes: 0 additions & 142 deletions thehive-backend/app/models/Analyzer.scala

This file was deleted.

9 changes: 0 additions & 9 deletions thehive-backend/app/models/JsonFormat.scala
Original file line number Diff line number Diff line change
Expand Up @@ -14,18 +14,9 @@ object JsonFormat {
implicit val caseResolutionStatusFormat = enumFormat(CaseResolutionStatus)
implicit val caseImpactStatusFormat = enumFormat(CaseImpactStatus)
implicit val artifactStatusFormat = enumFormat(ArtifactStatus)
implicit val jobStatusFormat = enumFormat(JobStatus)
implicit val taskStatusFormat = enumFormat(TaskStatus)
implicit val logStatusFormat = enumFormat(LogStatus)
implicit val caseTemplateStatusFormat = enumFormat(CaseTemplateStatus)

implicit val pathWrites: Writes[Path] = Writes((value: Path) JsString(value.toString))

implicit val analyzerWrites: Writes[Analyzer] = Writes((analyzer: Analyzer)
Json.obj(
"name" analyzer.name,
"version" analyzer.version,
"description" analyzer.description,
"dataTypeList" analyzer.dataTypeList,
"id" analyzer.id))
}
Loading

0 comments on commit 584d6d4

Please sign in to comment.