• Home
• com.github.workingDog
• taxii2lib_2.12
• 0.1
• source code
• Collection.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.kodekutters.taxii.Collection.scala Maven / Gradle / Ivy

package com.kodekutters.taxii

import com.kodekutters.stix.Bundle
import play.api.libs.json.Json

import scala.concurrent.ExecutionContext.Implicits.global
import scala.concurrent.Future

/**
  * This Endpoint provides general information about a Collection, which can be used to help
  * users and clients decide whether and how they want to interact with it.
  *
  * @param conn the connection to the com.kodekutters.taxii server
  */
case class Collection(taxiiCollection: TaxiiCollection, api_root: String,
                      conn: TaxiiConnection = TaxiiConnection("", 0, "", "", "")) {

  def this(url: String, id: String, api_root: String, user: String, password: String) =
    this(new TaxiiCollection(id, "temp", true, true), api_root, new TaxiiConnection(url, user, password, 5))

  def this(id: String, api_root: String, user: String, password: String) =
    this(new TaxiiCollection(id, "temp", true, true), api_root, new TaxiiConnection(api_root, user, password, 5))

  def this(id: String, api_root: String) = this(new TaxiiCollection(id, "temp", true, true), api_root)


  val basePath = api_root + "collections/" + taxiiCollection.id
  val thePath = basePath + "/objects/"

  def getObjects(filter: Option[Seq[(String, String)]] = None, range: String = ""): Future[Option[Bundle]] = {
    val theHeader = if (range == null || range.isEmpty) conn.stixHeaders
    else (conn.stixHeaders.toMap + ("Range" -> ("items=" + range))).toSeq
    conn.fetch[Bundle](thePath, theHeader, filter).map(_.toOption)
  }

  def getObject(obj_id: String, filter: Option[Seq[(String, String)]] = None): Future[Option[Bundle]] =
    conn.fetch[Bundle](thePath + obj_id + "/", conn.stixHeaders, filter).map(_.toOption)

  def addObjects(bundle: Bundle): Future[Option[TaxiiStatus]] =
    conn.post(thePath, Json.toJson[Bundle](bundle)).map(_.toOption)

  def getManifests(filter: Option[Seq[(String, String)]] = None, range: String = ""): Future[Option[TaxiiManifest]] = {
    val thePath = basePath + "/manifest/"
    val theHeader = if (range == null || range.isEmpty) conn.getHeaders
    else (conn.getHeaders.toMap + ("Range" -> ("items=" + range))).toSeq
    conn.fetch[TaxiiManifest](thePath, theHeader, filter).map(_.toOption)
  }

}