Skip to content

Commit

Permalink
#1483 Fix application load
Browse files Browse the repository at this point in the history
  • Loading branch information
To-om committed Sep 7, 2020
1 parent 409ed55 commit affc1a0
Show file tree
Hide file tree
Showing 20 changed files with 27 additions and 20 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ class ActionCtrl @Inject() (
logSrv: LogSrv,
alertSrv: AlertSrv,
implicit val executionContext: ExecutionContext,
override val queryExecutor: QueryExecutor,
@Named("v0") override val queryExecutor: QueryExecutor,
override val publicData: PublicAction
) extends AuditRenderer
with QueryCtrl {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ class AnalyzerTemplateCtrl @Inject() (
override val entrypoint: Entrypoint,
@Named("with-thehive-cortex-schema") override val db: Database,
analyzerTemplateSrv: AnalyzerTemplateSrv,
override val queryExecutor: QueryExecutor,
@Named("v0") override val queryExecutor: QueryExecutor,
override val publicData: PublicAnalyzerTemplate
) extends QueryCtrl {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ class JobCtrl @Inject() (
observableSrv: ObservableSrv,
errorHandler: ErrorHandler,
implicit val ec: ExecutionContext,
override val queryExecutor: QueryExecutor,
@Named("v0") override val queryExecutor: QueryExecutor,
override val publicData: PublicJob
) extends QueryCtrl {
def get(jobId: String): Action[AnyContent] =
Expand Down
2 changes: 2 additions & 0 deletions thehive/app/org/thp/thehive/TheHiveModule.scala
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ import org.thp.scalligraph.auth._
import org.thp.scalligraph.janus.JanusDatabase
import org.thp.scalligraph.models.{Database, Schema}
import org.thp.scalligraph.services.{GenIntegrityCheckOps, HadoopStorageSrv, S3StorageSrv}
import org.thp.thehive.controllers.v0.QueryExecutorVersion0Provider
import org.thp.thehive.models.{DatabaseProvider, TheHiveSchemaDefinition}
import org.thp.thehive.services.notification.notifiers._
import org.thp.thehive.services.notification.triggers._
Expand Down Expand Up @@ -79,6 +80,7 @@ class TheHiveModule(environment: Environment, configuration: Configuration) exte
val queryExecutorBindings = ScalaMultibinder.newSetBinder[QueryExecutor](binder)
queryExecutorBindings.addBinding.to[TheHiveQueryExecutorV0]
queryExecutorBindings.addBinding.to[TheHiveQueryExecutorV1]
bind[QueryExecutor].annotatedWithName("v0").toProvider[QueryExecutorVersion0Provider]
ScalaMultibinder.newSetBinder[Connector](binder)
val schemaBindings = ScalaMultibinder.newSetBinder[Schema](binder)
schemaBindings.addBinding.to[TheHiveSchemaDefinition]
Expand Down
2 changes: 1 addition & 1 deletion thehive/app/org/thp/thehive/controllers/v0/AlertCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@ class AlertCtrl @Inject() (
caseSrv: CaseSrv,
override val publicData: PublicAlert,
@Named("with-thehive-schema") implicit val db: Database,
override val queryExecutor: QueryExecutor
@Named("v0") override val queryExecutor: QueryExecutor
) extends QueryCtrl {
def create: Action[AnyContent] =
entrypoint("create alert")
Expand Down
2 changes: 1 addition & 1 deletion thehive/app/org/thp/thehive/controllers/v0/AuditCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ class AuditCtrl @Inject() (
override val publicData: PublicAudit,
@Named("with-thehive-schema") implicit override val db: Database,
implicit val ec: ExecutionContext,
val queryExecutor: QueryExecutor
@Named("v0") override val queryExecutor: QueryExecutor
) extends AuditRenderer
with QueryCtrl {
implicit val timeout: Timeout = Timeout(5.minutes)
Expand Down
2 changes: 1 addition & 1 deletion thehive/app/org/thp/thehive/controllers/v0/CaseCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ class CaseCtrl @Inject() (
tagSrv: TagSrv,
userSrv: UserSrv,
override val publicData: PublicCase,
override val queryExecutor: QueryExecutor,
@Named("v0") override val queryExecutor: QueryExecutor,
@Named("with-thehive-schema") implicit override val db: Database
) extends CaseRenderer
with QueryCtrl {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ class CaseTemplateCtrl @Inject() (
auditSrv: AuditSrv,
override val publicData: PublicCaseTemplate,
@Named("with-thehive-schema") implicit override val db: Database,
override val queryExecutor: QueryExecutor
@Named("v0") override val queryExecutor: QueryExecutor
) extends QueryCtrl {
def create: Action[AnyContent] =
entrypoint("create case template")
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ class CustomFieldCtrl @Inject() (
@Named("with-thehive-schema") override val db: Database,
customFieldSrv: CustomFieldSrv,
override val publicData: PublicCustomField,
override val queryExecutor: QueryExecutor
@Named("v0") override val queryExecutor: QueryExecutor
) extends QueryCtrl
with AuditRenderer {
def create: Action[AnyContent] =
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ class DashboardCtrl @Inject() (
userSrv: UserSrv,
@Named("with-thehive-schema") implicit val db: Database,
override val publicData: PublicDashboard,
override val queryExecutor: QueryExecutor
@Named("v0") override val queryExecutor: QueryExecutor
) extends QueryCtrl {
def create: Action[AnyContent] =
entrypoint("create dashboard")
Expand Down
2 changes: 1 addition & 1 deletion thehive/app/org/thp/thehive/controllers/v0/LogCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ class LogCtrl @Inject() (
@Named("with-thehive-schema") override val db: Database,
logSrv: LogSrv,
taskSrv: TaskSrv,
override val queryExecutor: QueryExecutor,
@Named("v0") override val queryExecutor: QueryExecutor,
override val publicData: PublicLog
) extends QueryCtrl {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ class ObservableCtrl @Inject() (
observableSrv: ObservableSrv,
observableTypeSrv: ObservableTypeSrv,
caseSrv: CaseSrv,
override val queryExecutor: QueryExecutor,
@Named("v0") override val queryExecutor: QueryExecutor,
override val publicData: PublicObservable
) extends ObservableRenderer
with QueryCtrl {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ class ObservableTypeCtrl @Inject() (
override val entrypoint: Entrypoint,
@Named("with-thehive-schema") override val db: Database,
observableTypeSrv: ObservableTypeSrv,
override val queryExecutor: QueryExecutor,
@Named("v0") override val queryExecutor: QueryExecutor,
override val publicData: PublicObservableType
) extends QueryCtrl {
def get(idOrName: String): Action[AnyContent] =
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ class OrganisationCtrl @Inject() (
organisationSrv: OrganisationSrv,
userSrv: UserSrv,
@Named("with-thehive-schema") implicit override val db: Database,
override val queryExecutor: QueryExecutor,
@Named("v0") override val queryExecutor: QueryExecutor,
override val publicData: PublicOrganisation
) extends QueryCtrl {
def create: Action[AnyContent] =
Expand Down
2 changes: 1 addition & 1 deletion thehive/app/org/thp/thehive/controllers/v0/PageCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ class PageCtrl @Inject() (
override val entrypoint: Entrypoint,
pageSrv: PageSrv,
@Named("with-thehive-schema") override val db: Database,
override val queryExecutor: QueryExecutor,
@Named("v0") override val queryExecutor: QueryExecutor,
override val publicData: PublicPage
) extends QueryCtrl {
def get(idOrTitle: String): Action[AnyContent] =
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ class ProfileCtrl @Inject() (
profileSrv: ProfileSrv,
override val publicData: PublicProfile,
@Named("with-thehive-schema") implicit val db: Database,
override val queryExecutor: QueryExecutor
@Named("v0") override val queryExecutor: QueryExecutor
) extends QueryCtrl {
def create: Action[AnyContent] =
entrypoint("create profile")
Expand Down
2 changes: 1 addition & 1 deletion thehive/app/org/thp/thehive/controllers/v0/TagCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ class TagCtrl @Inject() (
override val entrypoint: Entrypoint,
@Named("with-thehive-schema") override val db: Database,
tagSrv: TagSrv,
override val queryExecutor: QueryExecutor,
@Named("v0") override val queryExecutor: QueryExecutor,
override val publicData: PublicTag
) extends QueryCtrl {
def importTaxonomy: Action[AnyContent] =
Expand Down
2 changes: 1 addition & 1 deletion thehive/app/org/thp/thehive/controllers/v0/TaskCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ class TaskCtrl @Inject() (
userSrv: UserSrv,
organisationSrv: OrganisationSrv,
shareSrv: ShareSrv,
override val queryExecutor: QueryExecutor,
@Named("v0") override val queryExecutor: QueryExecutor,
override val publicData: PublicTask
) extends QueryCtrl {

Expand Down
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
package org.thp.thehive.controllers.v0

import javax.inject.{Inject, Named, Singleton}
import javax.inject.{Inject, Named, Provider, Singleton}
import org.scalactic.Good
import org.thp.scalligraph.BadRequestError
import org.thp.scalligraph.{BadRequestError, GlobalQueryExecutor}
import org.thp.scalligraph.auth.AuthContext
import org.thp.scalligraph.controllers.{FObject, Field, FieldsParser}
import org.thp.scalligraph.models._
Expand Down Expand Up @@ -206,3 +206,8 @@ class ChildQueryInputFilter(childType: String, childFilter: InputQuery[Traversal
.getOrElse(throw BadRequestError(s"$traversalType hasn't child $childType"))
}
}

@Singleton
class QueryExecutorVersion0Provider @Inject() (globalQueryExecutor: GlobalQueryExecutor) extends Provider[QueryExecutor] {
override def get(): QueryExecutor = globalQueryExecutor.get(0)
}
2 changes: 1 addition & 1 deletion thehive/app/org/thp/thehive/controllers/v0/UserCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ class UserCtrl @Inject() (
organisationSrv: OrganisationSrv,
auditSrv: AuditSrv,
@Named("with-thehive-schema") implicit override val db: Database,
override val queryExecutor: QueryExecutor,
@Named("v0") override val queryExecutor: QueryExecutor,
override val publicData: PublicUser
) extends QueryCtrl {
def current: Action[AnyContent] =
Expand Down

0 comments on commit affc1a0

Please sign in to comment.