multipart interceptor

This commit is contained in:
Tommi Reiman 2018-09-07 23:07:38 +03:00
parent ca02680e2d
commit 08349e61df
3 changed files with 63 additions and 9 deletions

View file

@ -0,0 +1,55 @@
(ns reitit.http.interceptors.multipart
(:require [reitit.coercion :as coercion]
[ring.middleware.multipart-params :as multipart-params]
[clojure.spec.alpha :as s]
[spec-tools.core :as st])
(:import (java.io File)))
(s/def ::filename string?)
(s/def ::content-type string?)
(s/def ::tempfile (partial instance? File))
(s/def ::bytes bytes?)
(s/def ::size int?)
(def temp-file-part
"Spec for file param created by ring.middleware.multipart-params.temp-file store."
(st/spec
{:spec (s/keys :req-un [::filename ::content-type ::tempfile ::size])
:swagger/type "file"}))
(def bytes-part
"Spec for file param created by ring.middleware.multipart-params.byte-array store."
(st/spec
{:spec (s/keys :req-un [::filename ::content-type ::bytes])
:swagger/type "file"}))
(defn- coerced-request [request coercers]
(if-let [coerced (if coercers (coercion/coerce-request coercers request))]
(update request :parameters merge coerced)
request))
;;
;; public api
;;
(defn multipart-interceptor
"Creates a Interceptor to handle the multipart params, based on
ring.middleware.multipart-params, taking same options. Mounts only
if endpoint has `[:parameters :multipart]` defined. Publishes coerced
parameters into `[:parameters :multipart]` under request."
([]
(multipart-interceptor nil))
([options]
{:name ::multipart
:compile (fn [{:keys [parameters coercion]} opts]
(if-let [multipart (:multipart parameters)]
(let [parameter-coercion {:multipart (coercion/->ParameterCoercion
:multipart-params :string true true)}
opts (assoc opts ::coercion/parameter-coercion parameter-coercion)
coercers (if multipart (coercion/request-coercers coercion parameters opts))]
{:data {:swagger {:consumes ^:replace #{"multipart/form-data"}}}
:enter (fn [ctx]
(let [request (-> (:request ctx)
(multipart-params/multipart-params-request options)
(coerced-request coercers))]
(assoc ctx :request request)))})))}))

View file

@ -1,4 +1,4 @@
(ns ^:no-doc reitit.ring.middleware.multipart
(ns reitit.ring.middleware.multipart
(:refer-clojure :exclude [compile])
(:require [reitit.coercion :as coercion]
[ring.middleware.multipart-params :as multipart-params]
@ -40,14 +40,10 @@
:wrap (fn [handler]
(fn
([request]
(try
(-> request
(multipart-params/multipart-params-request options)
(coerced-request coercers)
(handler))
(catch Exception e
(.printStackTrace e)
(throw e))))
(handler)))
([request respond raise]
(-> request
(multipart-params/multipart-params-request options)

View file

@ -0,0 +1,3 @@
(ns reitit.http.interceptors.multipart-test)
;; TODO