Packages

lmug

0.1.1
0.2.8 0.2.7 0.2.6 0.2.5 0.2.4 0.2.3 0.2.2 0.2.1 0.2.0 0.1.2 0.1.1 0.1.0 0.0.902 0.0.901 0.0.900

LFE HTTP Server Abstraction and Web Application Middleware Infrastructure

Current section

Files

Jump to
Readme
15 Versions
3 Dependencies
Dependants
Files
Activity
lmug src lmug-mw-content-type.lfe
Raw

src/lmug-mw-content-type.lfe 

(defmodule lmug-mw-content-type

  (doc "Set content type based upon file extension in URL.")

  (behaviour lmug-middleware)

  (export

   (wrap 1) (wrap 2)))



(defun filename-ext

  "Returns the file extension of a filename or filepath."

  ((ext) (when (is_binary ext))

   (filename-ext (binary_to_list ext)))

  ((ext) (when (is_list ext))

   (case (string:split (filename:extension url) ".")

     ('("" "") (lmug:default-content-type))

     (`("" ,ext) (lmug:ext->content-type ext))

     (_ (lmug:default-content-type)))))



(defun wrap (handler)

  "The same as #'wrap/2 but with an empty list for options."

  (wrap handler '()))



(defun wrap (handler _opts)

  "Middleware that adds a content type based upon the file extension in the URL."

  (lambda (req)

    (let ((resp (funcall handler req)))

      (http.response:add-header resp

                                #"Content-Type"

                                (filename-ext (mref req 'url))))))