mirror of https://github.com/mitchell/shortnr.git
55 lines
1.9 KiB
Elixir
55 lines
1.9 KiB
Elixir
defmodule Shortnr.Transport.HTTP do
|
|
@moduledoc """
|
|
This module contains functions that can be used to handle HTTP requests and send responses, by
|
|
manipulating Plug.Conn.
|
|
"""
|
|
|
|
import Plug.Conn
|
|
|
|
@type error :: {:error, {atom(), String.t()}, Plug.Conn.t()}
|
|
@type ok_error :: {:ok, term(), Plug.Conn.t()} | error()
|
|
|
|
@spec handle(ok_error(), (... -> ok_error())) :: ok_error()
|
|
def handle(error = {:error, _sub_error, _conn}, _func), do: error
|
|
|
|
def handle({:ok, request, conn}, func) when is_function(func, 1) do
|
|
case func.(request) do
|
|
{:ok, response} -> {:ok, response, conn}
|
|
{:error, error_value} -> convert_error(error_value, conn)
|
|
end
|
|
end
|
|
|
|
def handle({:ok, _request, conn}, func) when is_function(func, 0) do
|
|
case func.() do
|
|
{:ok, response} -> {:ok, response, conn}
|
|
{:error, error_value} -> convert_error(error_value, conn)
|
|
end
|
|
end
|
|
|
|
defp convert_error({:invalid_argument, message}, conn),
|
|
do: {:error, {:bad_request, message}, conn}
|
|
|
|
defp convert_error(_, conn), do: {:error, {:internal_server_error, "unknown error"}, conn}
|
|
|
|
@spec send(ok_error(), atom(), Plug.Conn.t()) :: Plug.Conn.t()
|
|
def send({:error, {status, body}, _conn}, _success_status, original_conn) do
|
|
send_resp(original_conn, status_to_code(status), body)
|
|
end
|
|
|
|
def send({:ok, body, conn}, :found, _original_conn) do
|
|
conn = put_resp_header(conn, "Location", URI.to_string(body))
|
|
send_resp(conn, status_to_code(:found), URI.to_string(body))
|
|
end
|
|
|
|
def send({:ok, body, conn}, success_status, _original_conn) do
|
|
send_resp(conn, status_to_code(success_status), body)
|
|
end
|
|
|
|
defp status_to_code(:ok), do: 200
|
|
defp status_to_code(:created), do: 201
|
|
defp status_to_code(:found), do: 302
|
|
defp status_to_code(:bad_request), do: 400
|
|
defp status_to_code(:not_found), do: 404
|
|
defp status_to_code(:internal_server_error), do: 500
|
|
end
|