defmodule FrenzyWeb.FeverController do use FrenzyWeb, :controller alias Frenzy.{Repo, Feed, Item} import Ecto.Query plug :api_check def api_check(conn, _) do if Map.has_key?(conn.params, "api") do conn else conn |> resp(400, "Invalid request") |> halt() end end def get(conn, params) do json(conn, %{api_version: 2, auth: 0}) end def post(conn, %{"api_key" => api_key} = params) do case validate_key(api_key) do :invalid -> resp(conn, 401, "Invalid API key") :ok -> json(conn, fever_response(params)) end end defp validate_key(api_key) do auth = Application.get_env(:frenzy, :auth) username = auth[:username] password = auth[:password] expected = :crypto.hash(:md5, "#{username}:#{password}") |> Base.encode16(case: :lower) case api_key |> String.downcase do ^expected -> :ok _ -> :invalid end end defp fever_response(params) do res = %{api_version: 2, auth: 1} |> mark(params) |> unread_recently_read(params) |> groups(params) |> feeds(params) |> favicons(params) |> links(params) |> unread(params) |> saved(params) |> items(params) end defp mark(res, %{"mark" => "item", "id" => id, "as" => as} = params) do item = Repo.get(Item, id) |> Repo.preload(:feed) diff = case as do "read" -> %{read: true, read_date: Timex.now} "unread" -> %{read: false, read_date: nil} _ -> %{} end changeset = Item.changeset(item, diff) Repo.update(changeset) res end defp mark(res, _), do: res defp unread_recently_read(res, %{"unread_recently_read" => 1}) do Repo.all(from i in Item, where: i.read, where: i.read_date >= from_now(-1, "hour")) |> Enum.map(fn i -> Item.changeset(i, %{read: false, read_date: nil}) end) |> Enum.map(&Repo.update/1) res end defp unread_recently_read(res, _), do: res defp groups(res, %{"groups" => _}) do res |> Map.put(:groups, []) |> Map.put(:feeds_groups, []) end defp groups(res, _), do: res defp feeds(res, %{"feeds" => _}) do feeds = Repo.all(Feed) |> Enum.map(&Feed.to_fever/1) res |> Map.put(:feeds, feeds) |> Map.put(:feeds_groups, []) end defp feeds(res, _), do: res defp favicons(res, %{"favicons" => _}) do res |> Map.put(:favicons, []) end defp favicons(res, _), do: res defp links(res, %{"links" => _}) do res |> Map.put(:links, []) end defp links(res, _), do: res defp unread(res, %{"unread_item_ids" => _}) do unread = Repo.all(from Item, where: [read: false]) |> Enum.map(fn item -> item.id end) |> Enum.join(",") res |> Map.put(:unread_item_ids, unread) end defp unread(res, _), do: res defp saved(res, %{"saved_item_ids" => _}) do res |> Map.put(:saved_item_ids, "") end defp saved(res, _), do: res defp items(res, %{"items" => _} = params) do items = cond do Map.has_key?(params, "with_ids") -> params["with_ids"] |> String.split(",") |> Enum.map(fn id -> {id, _} = id |> String.trim |> Integer.parse Repo.get(Item, id) end) Map.has_key?(params, "since_id") -> since = Repo.get(Item, params["since_id"]) {since, _} = Integer.parse(since) Repo.all(from i in Item, where: i.inserted_at > ^since.inserted_at, order_by: [asc: :id], limit: 50) Map.has_key?(params, "max_id") -> max = Repo.get(Item, params["max_id"]) {max, _} = Integer.parse(max) Repo.all(from i in Item, where: i.inserted_at < ^max.inserted_at, order_by: [desc: :id], limit: 50) true -> [] end items = items |> Enum.reject(&is_nil/1) |> Enum.map(&Item.to_fever/1) res |> Map.put(:items, items) |> Map.put(:total_items, Enum.count(items)) end defp items(res, _), do: res end