Merge pull request #135 from isimluk/intel_ruls

CrowdStrike Intelligence Report Downloads

Makefile

@@ -8,7 +8,7 @@ build:
 
 clean-generate: remove-generated generate
 
-generate: specs/swagger-download-patch.json
+generate: specs/swagger-pdf-download-patch.json
 	$(GO) run github.com/go-swagger/go-swagger/cmd/swagger generate client --skip-validation -f $^ -t falcon
 
 .PHONY: build generate remove-generated
@@ -28,6 +28,10 @@ specs/swagger-download-patch.json: specs/swagger-stripped-oauth.json
 	# We add missing binary response body spec to the swagger
 	jq '.definitions."domain.DownloadItem"."type"="string" | .definitions."domain.DownloadItem"."format"="binary"' $< > $@
 
+specs/swagger-pdf-download-patch.json: specs/swagger-download-patch.json
+	# We add missing binary response body spec to the swagger
+	jq '.paths."/intel/entities/report-files/v1"."get"."responses"."200"."schema"={"$$ref": "#/definitions/domain.DownloadItem"} | .paths."/intel/entities/rules-latest-files/v1"."get"."responses"."200"."schema"={"$$ref": "#/definitions/domain.DownloadItem"}' $< > $@
+
 specs/swagger.json:
 	@echo "Sorry swagger.json needs to be obtained manually at this moment"
 	@exit 1

falcon/api_client.go

@@ -6,6 +6,7 @@ import (
 	"time"
 
 	"github.com/crowdstrike/gofalcon/falcon/client"
+	httpruntime "github.com/go-openapi/runtime"
 	httptransport "github.com/go-openapi/runtime/client"
 	"github.com/go-openapi/strfmt"
 	log "github.com/sirupsen/logrus"
@@ -33,6 +34,7 @@ func NewClient(ac *ApiConfig) (*client.CrowdStrikeAPISpecification, error) {
 	customTransport := httptransport.NewWithClient(
 		ac.Host(), ac.BasePath(), []string{}, authenticatedClient)
 	customTransport.Debug = ac.Debug
+	customTransport.Consumers["application/pdf"] = httpruntime.ByteStreamConsumer()
 
 	return client.New(customTransport, strfmt.Default), nil
 }