Skip to content

Commit

Permalink
#1731 Use implicit parameter for FieldsParser in initial query
Browse files Browse the repository at this point in the history
  • Loading branch information
To-om committed Jan 28, 2021
1 parent 68dc6cd commit 656b645
Show file tree
Hide file tree
Showing 35 changed files with 5 additions and 39 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,6 @@ class PublicAction @Inject() (actionSrv: ActionSrv, organisationSrv: Organisatio
Query.init[Traversal.V[Action]]("listAction", (graph, authContext) => actionSrv.startTraversal(graph).visible(organisationSrv)(authContext))
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Action]](
"getAction",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => actionSrv.get(idOrName)(graph).visible(organisationSrv)(authContext)
)
override val pageQuery: ParamQuery[OutputParam] = Query.withParam[OutputParam, Traversal.V[Action], IteratorOutput](
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,6 @@ class PublicAnalyzerTemplate @Inject() (analyzerTemplateSrv: AnalyzerTemplateSrv
Query.init[Traversal.V[AnalyzerTemplate]]("listAnalyzerTemplate", (graph, _) => analyzerTemplateSrv.startTraversal(graph))
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[AnalyzerTemplate]](
"getReportTemplate",
FieldsParser[EntityIdOrName],
(idOrName, graph, _) => analyzerTemplateSrv.get(idOrName)(graph)
)
override val pageQuery: ParamQuery[OutputParam] =
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,6 @@ class PublicJob @Inject() (jobSrv: JobSrv) extends PublicData with JobRenderer {
Query.init[Traversal.V[Job]]("listJob", (graph, authContext) => jobSrv.startTraversal(graph).visible(authContext))
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Job]](
"getJob",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => jobSrv.get(idOrName)(graph).visible(authContext)
)
override val pageQuery: ParamQuery[OutputParam] =
Expand Down
1 change: 0 additions & 1 deletion thehive/app/org/thp/thehive/controllers/v0/AlertCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -357,7 +357,6 @@ class PublicAlert @Inject() (
)
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Alert]](
"getAlert",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => alertSrv.get(idOrName)(graph).visible(organisationSrv)(authContext)
)
override val pageQuery: ParamQuery[OutputParam] =
Expand Down
5 changes: 2 additions & 3 deletions thehive/app/org/thp/thehive/controllers/v0/AuditCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -3,9 +3,8 @@ package org.thp.thehive.controllers.v0
import akka.actor.ActorRef
import akka.pattern.ask
import akka.util.Timeout
import javax.inject.{Inject, Named, Singleton}
import org.thp.scalligraph.EntityIdOrName
import org.thp.scalligraph.controllers.{Entrypoint, FieldsParser}
import org.thp.scalligraph.controllers.Entrypoint
import org.thp.scalligraph.models.{Database, UMapping}
import org.thp.scalligraph.query._
import org.thp.scalligraph.traversal.TraversalOps._
Expand All @@ -17,6 +16,7 @@ import org.thp.thehive.services._
import play.api.libs.json.{JsArray, JsObject, Json}
import play.api.mvc.{Action, AnyContent, Results}

import javax.inject.{Inject, Named, Singleton}
import scala.concurrent.ExecutionContext
import scala.concurrent.duration.DurationInt

Expand Down Expand Up @@ -67,7 +67,6 @@ class AuditCtrl @Inject() (
class PublicAudit @Inject() (auditSrv: AuditSrv, organisationSrv: OrganisationSrv, db: Database) extends PublicData {
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Audit]](
"getAudit",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => auditSrv.get(idOrName)(graph).visible(organisationSrv)(authContext)
)

Expand Down
1 change: 0 additions & 1 deletion thehive/app/org/thp/thehive/controllers/v0/CaseCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -197,7 +197,6 @@ class PublicCase @Inject() (
override val getQuery: ParamQuery[EntityIdOrName] =
Query.initWithParam[EntityIdOrName, Traversal.V[Case]](
"getCase",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => caseSrv.get(idOrName)(graph).visible(organisationSrv)(authContext)
)
override val pageQuery: ParamQuery[OutputParam] =
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -102,7 +102,6 @@ class PublicCaseTemplate @Inject() (
.init[Traversal.V[CaseTemplate]]("listCaseTemplate", (graph, authContext) => organisationSrv.get(authContext.organisation)(graph).caseTemplates)
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[CaseTemplate]](
"getCaseTemplate",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => caseTemplateSrv.get(idOrName)(graph).visible(authContext)
)
override val pageQuery: ParamQuery[OutputParam] = Query.withParam[OutputParam, Traversal.V[CaseTemplate], IteratorOutput](
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,6 @@ class PublicCustomField @Inject() (customFieldSrv: CustomFieldSrv) extends Publi
override val outputQuery: Query = Query.output[CustomField with Entity]
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[CustomField]](
"getCustomField",
FieldsParser[EntityIdOrName],
(idOrName, graph, _) => customFieldSrv.get(idOrName)(graph)
)
override val publicProperties: PublicProperties =
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,6 @@ class PublicDashboard @Inject() (

override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Dashboard]](
"getDashboard",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => dashboardSrv.get(idOrName)(graph).visible(authContext)
)

Expand Down
1 change: 0 additions & 1 deletion thehive/app/org/thp/thehive/controllers/v0/LogCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,6 @@ class PublicLog @Inject() (logSrv: LogSrv, organisationSrv: OrganisationSrv) ext
Query.init[Traversal.V[Log]]("listLog", (graph, authContext) => logSrv.startTraversal(graph).visible(organisationSrv)(authContext))
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Log]](
"getLog",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => logSrv.get(idOrName)(graph).visible(organisationSrv)(authContext)
)
override val pageQuery: ParamQuery[OutputParam] = Query.withParam[OutputParam, Traversal.V[Log], IteratorOutput](
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -368,7 +368,6 @@ class PublicObservable @Inject() (
)
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Observable]](
"getObservable",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => observableSrv.get(idOrName)(graph).visible(organisationSrv)(authContext)
)
override val pageQuery: ParamQuery[OutputParam] =
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,6 @@ class PublicObservableType @Inject() (observableTypeSrv: ObservableTypeSrv) exte
override val outputQuery: Query = Query.output[ObservableType with Entity]
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[ObservableType]](
"getObservableType",
FieldsParser[EntityIdOrName],
(idOrName, graph, _) => observableTypeSrv.get(idOrName)(graph)
)
override val publicProperties: PublicProperties = PublicPropertyListBuilder[ObservableType]
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -138,7 +138,6 @@ class PublicOrganisation @Inject() (organisationSrv: OrganisationSrv) extends Pu
override val outputQuery: Query = Query.output[Organisation with Entity]
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Organisation]](
"getOrganisation",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => organisationSrv.get(idOrName)(graph).visible(authContext)
)
override val extraQueries: Seq[ParamQuery[_]] = Seq(
Expand Down
1 change: 0 additions & 1 deletion thehive/app/org/thp/thehive/controllers/v0/PageCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,6 @@ class PublicPage @Inject() (pageSrv: PageSrv, organisationSrv: OrganisationSrv)
Query.init[Traversal.V[Page]]("listPage", (graph, authContext) => organisationSrv.get(authContext.organisation)(graph).pages)
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Page]](
"getPage",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => pageSrv.get(idOrName)(graph).visible(authContext)
)
val pageQuery: ParamQuery[OutputParam] = Query.withParam[OutputParam, Traversal.V[Page], IteratorOutput](
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,6 @@ class PublicProfile @Inject() (profileSrv: ProfileSrv) extends PublicData {

override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Profile]](
"getProfile",
FieldsParser[EntityIdOrName],
(idOrName, graph, _) => profileSrv.get(idOrName)(graph)
)
val initialQuery: Query =
Expand Down
1 change: 0 additions & 1 deletion thehive/app/org/thp/thehive/controllers/v0/TagCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -113,7 +113,6 @@ class PublicTag @Inject() (tagSrv: TagSrv, organisationSrv: OrganisationSrv) ext
override val outputQuery: Query = Query.output[Tag with Entity]
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Tag]](
"getTag",
FieldsParser[EntityIdOrName],
(idOrName, graph, _) => tagSrv.get(idOrName)(graph)
)
implicit val stringRenderer: Renderer[String] = Renderer.toJson[String, String](identity)
Expand Down
1 change: 0 additions & 1 deletion thehive/app/org/thp/thehive/controllers/v0/TaskCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -117,7 +117,6 @@ class PublicTask @Inject() (taskSrv: TaskSrv, organisationSrv: OrganisationSrv,
)
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Task]](
"getTask",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => taskSrv.get(idOrName)(graph).inOrganisation(organisationSrv.currentId(graph, authContext))
)
override val outputQuery: Query =
Expand Down
1 change: 0 additions & 1 deletion thehive/app/org/thp/thehive/controllers/v0/UserCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -232,7 +232,6 @@ class PublicUser @Inject() (userSrv: UserSrv, organisationSrv: OrganisationSrv)
Query.init[Traversal.V[User]]("listUser", (graph, authContext) => organisationSrv.get(authContext.organisation)(graph).users)
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[User]](
"getUser",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => userSrv.get(idOrName)(graph).visible(authContext)
)
override val pageQuery: ParamQuery[OutputParam] = Query.withParam[OutputParam, Traversal.V[User], IteratorOutput](
Expand Down
1 change: 0 additions & 1 deletion thehive/app/org/thp/thehive/controllers/v1/AlertCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,6 @@ class AlertCtrl @Inject() (

override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Alert]](
"getAlert",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => alertSrv.get(idOrName)(graph).visible(organisationSrv)(authContext)
)
override val pageQuery: ParamQuery[OutputParam] = Query.withParam[OutputParam, Traversal.V[Alert], IteratorOutput](
Expand Down
3 changes: 1 addition & 2 deletions thehive/app/org/thp/thehive/controllers/v1/AuditCtrl.scala
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
package org.thp.thehive.controllers.v1

import org.thp.scalligraph.EntityIdOrName
import org.thp.scalligraph.controllers.{Entrypoint, FieldsParser}
import org.thp.scalligraph.controllers.Entrypoint
import org.thp.scalligraph.models.{Database, Schema}
import org.thp.scalligraph.query.{ParamQuery, PublicProperties, Query}
import org.thp.scalligraph.traversal.TraversalOps._
Expand Down Expand Up @@ -32,7 +32,6 @@ class AuditCtrl @Inject() (
val publicProperties: PublicProperties = properties.audit
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Audit]](
"getAudit",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => auditSrv.get(idOrName)(graph).visible(organisationSrv)(authContext)
)

Expand Down
1 change: 0 additions & 1 deletion thehive/app/org/thp/thehive/controllers/v1/CaseCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,6 @@ class CaseCtrl @Inject() (
Query.init[Traversal.V[Case]]("listCase", (graph, authContext) => caseSrv.startTraversal(graph).visible(organisationSrv)(authContext))
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Case]](
"getCase",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => caseSrv.get(idOrName)(graph).visible(organisationSrv)(authContext)
)
override val pageQuery: ParamQuery[OutputParam] = Query.withParam[OutputParam, Traversal.V[Case], IteratorOutput](
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,6 @@ class CaseTemplateCtrl @Inject() (
.init[Traversal.V[CaseTemplate]]("listCaseTemplate", (graph, authContext) => organisationSrv.get(authContext.organisation)(graph).caseTemplates)
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[CaseTemplate]](
"getCaseTemplate",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => caseTemplateSrv.get(idOrName)(graph).visible(authContext)
)
override val pageQuery: ParamQuery[OutputParam] = Query.withParam[OutputParam, Traversal.V[CaseTemplate], IteratorOutput](
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,6 @@ class CustomFieldCtrl @Inject() (entrypoint: Entrypoint, db: Database, customFie
override val outputQuery: Query = Query.output[CustomField with Entity]
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[CustomField]](
"getCustomField",
FieldsParser[EntityIdOrName],
(idOrName, graph, _) => customFieldSrv.get(idOrName)(graph)
)
override val publicProperties: PublicProperties = PublicPropertyListBuilder[CustomField]
Expand Down
1 change: 0 additions & 1 deletion thehive/app/org/thp/thehive/controllers/v1/LogCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,6 @@ class LogCtrl @Inject() (
Query.init[Traversal.V[Log]]("listLog", (graph, authContext) => logSrv.startTraversal(graph).visible(organisationSrv)(authContext))
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Log]](
"getLog",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => logSrv.get(idOrName)(graph).visible(organisationSrv)(authContext)
)
override val pageQuery: ParamQuery[OutputParam] = Query.withParam[OutputParam, Traversal.V[Log], IteratorOutput](
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,6 @@ class ObservableCtrl @Inject() (
)
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Observable]](
"getObservable",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => observableSrv.get(idOrName)(graph).visible(organisationSrv)(authContext)
)
override val pageQuery: ParamQuery[OutputParam] = Query.withParam[OutputParam, Traversal.V[Observable], IteratorOutput](
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,6 @@ class ObservableTypeCtrl @Inject() (
override val outputQuery: Query = Query.output[ObservableType with Entity]
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[ObservableType]](
"getObservableType",
FieldsParser[EntityIdOrName],
(idOrName, graph, _) => observableTypeSrv.get(idOrName)(graph)
)
override val publicProperties: PublicProperties = PublicPropertyListBuilder[ObservableType]
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,6 @@ class OrganisationCtrl @Inject() (
override val outputQuery: Query = Query.output[RichOrganisation, Traversal.V[Organisation]](_.richOrganisation)
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Organisation]](
"getOrganisation",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => organisationSrv.get(idOrName)(graph).visible(authContext)
)
override val extraQueries: Seq[ParamQuery[_]] = Seq(
Expand Down
3 changes: 1 addition & 2 deletions thehive/app/org/thp/thehive/controllers/v1/PatternCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,6 @@ class PatternCtrl @Inject() (
override val outputQuery: Query = Query.output[RichPattern, Traversal.V[Pattern]](_.richPattern)
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Pattern]](
"getPattern",
FieldsParser[EntityIdOrName],
(idOrName, graph, _) => patternSrv.get(idOrName)(graph)
)

Expand Down Expand Up @@ -72,7 +71,7 @@ class PatternCtrl @Inject() (

def get(patternId: String): Action[AnyContent] =
entrypoint("get pattern")
.authRoTransaction(db) { implicit request => implicit graph =>
.authRoTransaction(db) { _ => implicit graph =>
patternSrv
.get(EntityIdOrName(patternId))
.richPattern
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,6 @@ class ProcedureCtrl @Inject() (
override val outputQuery: Query = Query.output[RichProcedure, Traversal.V[Procedure]](_.richProcedure)
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Procedure]](
"getProcedure",
FieldsParser[EntityIdOrName],
(idOrName, graph, _) => procedureSrv.get(idOrName)(graph)
)

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,6 @@ class ProfileCtrl @Inject() (

override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Profile]](
"getProfile",
FieldsParser[EntityIdOrName],
(idOrName, graph, _) => profileSrv.get(idOrName)(graph)
)
val entityName: String = "profile"
Expand Down
1 change: 0 additions & 1 deletion thehive/app/org/thp/thehive/controllers/v1/ShareCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,6 @@ class ShareCtrl @Inject() (
override val outputQuery: Query = Query.outputWithContext[RichShare, Traversal.V[Share]]((shareSteps, _) => shareSteps.richShare)
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Share]](
"getShare",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => shareSrv.get(idOrName)(graph).visible(authContext)
)
override val extraQueries: Seq[ParamQuery[_]] = Seq(
Expand Down
1 change: 0 additions & 1 deletion thehive/app/org/thp/thehive/controllers/v1/TaskCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,6 @@ class TaskCtrl @Inject() (
)
override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[Task]](
"getTask",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => taskSrv.get(idOrName)(graph).visible(organisationSrv)(authContext)
)
override val outputQuery: Query =
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,6 @@ class TaxonomyCtrl @Inject() (
override val getQuery: ParamQuery[EntityIdOrName] =
Query.initWithParam[EntityIdOrName, Traversal.V[Taxonomy]](
"getTaxonomy",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => taxonomySrv.get(idOrName)(graph).visible(authContext)
)
override val pageQuery: ParamQuery[OutputParam] =
Expand Down
1 change: 0 additions & 1 deletion thehive/app/org/thp/thehive/controllers/v1/UserCtrl.scala
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,6 @@ class UserCtrl @Inject() (

override val getQuery: ParamQuery[EntityIdOrName] = Query.initWithParam[EntityIdOrName, Traversal.V[User]](
"getUser",
FieldsParser[EntityIdOrName],
(idOrName, graph, authContext) => userSrv.get(idOrName)(graph).visible(authContext)
)

Expand Down

0 comments on commit 656b645

Please sign in to comment.