git.squeep.com Git - akkoma/blob - lib/pleroma/instances/instance.ex

   1 defmodule Pleroma.Instances.Instance do
   2   @moduledoc "Instance."
   3
   4   alias Pleroma.Instances
   5   alias Pleroma.Instances.Instance
   6
   7   use Ecto.Schema
   8
   9   import Ecto.{Query, Changeset}
  10
  11   alias Pleroma.Repo
  12
  13   schema "instances" do
  14     field(:host, :string)
  15     field(:unreachable_since, :naive_datetime)
  16     field(:reachability_checked_at, :naive_datetime)
  17
  18     timestamps()
  19   end
  20
  21   def update_changeset(struct, params \\ %{}) do
  22     struct
  23     |> cast(params, [:host, :unreachable_since, :reachability_checked_at])
  24     |> unique_constraint(:host)
  25   end
  26
  27   def reachable?(url) do
  28     !Repo.one(
  29       from(i in Instance,
  30         where:
  31           i.host == ^host(url) and i.unreachable_since <= ^Instances.reachability_time_threshold(),
  32         select: true
  33       )
  34     )
  35   end
  36
  37   def set_reachable(url) do
  38     Repo.update_all(
  39       from(i in Instance, where: i.host == ^host(url)),
  40       set: [
  41         unreachable_since: nil,
  42         reachability_checked_at: DateTime.utc_now()
  43       ]
  44     )
  45   end
  46
  47   def set_unreachable(url, unreachable_since \\ nil) do
  48     unreachable_since = unreachable_since || DateTime.utc_now()
  49     host = host(url)
  50     existing_record = Repo.get_by(Instance, %{host: host})
  51
  52     changes = %{
  53       unreachable_since: unreachable_since,
  54       reachability_checked_at: NaiveDateTime.utc_now()
  55     }
  56
  57     if existing_record do
  58       update_changes =
  59         if existing_record.unreachable_since &&
  60              NaiveDateTime.compare(existing_record.unreachable_since, unreachable_since) != :gt,
  61            do: Map.delete(changes, :unreachable_since),
  62            else: changes
  63
  64       {:ok, _instance} = Repo.update(update_changeset(existing_record, update_changes))
  65     else
  66       {:ok, _instance} = Repo.insert(update_changeset(%Instance{}, Map.put(changes, :host, host)))
  67     end
  68   end
  69
  70   defp host(url_or_host) do
  71     if url_or_host =~ ~r/^http/i do
  72       URI.parse(url_or_host).host
  73     else
  74       url_or_host
  75     end
  76   end
  77 end