+ with {:ok, %{body: %{"errors" => false}}} <-
+ Elastix.Bulk.post(
+ url(),
+ d,
+ index: "activities",
+ type: "activity"
+ ) do
+ :ok
+ else
+ {:error, %{reason: err}} ->
+ Logger.error("Could not bulk put activity: #{err}")
+ :skipped
+
+ {:ok, %{body: body}} ->
+ IO.inspect(body)
+ :skipped
+ end
+ end
+
+ def bulk_post(data, :users) do
+ d =
+ data
+ |> Enum.filter(fn x -> x.actor_type == "Person" end)
+ |> Enum.map(fn d ->
+ [
+ %{index: %{_id: DocumentMappings.User.id(d)}},
+ DocumentMappings.User.encode(d)
+ ]
+ end)
+ |> List.flatten()
+
+ with {:ok, %{body: %{"errors" => false}}} <-
+ Elastix.Bulk.post(
+ url(),
+ d,
+ index: "users",
+ type: "user"
+ ) do
+ :ok
+ else
+ {:error, %{reason: err}} ->
+ Logger.error("Could not bulk put users: #{err}")
+ :skipped
+
+ {:ok, %{body: body}} ->
+ IO.inspect(body)
+ :skipped
+ end
+ end
+
+ def bulk_post(data, :hashtags) when is_list(data) do
+ d =
+ data
+ |> Enum.map(fn d ->
+ [
+ %{index: %{_id: DocumentMappings.Hashtag.id(d)}},
+ DocumentMappings.Hashtag.encode(d)
+ ]
+ end)
+ |> List.flatten()
+
+ with {:ok, %{body: %{"errors" => false}}} <-
+ Elastix.Bulk.post(
+ url(),
+ d,
+ index: "hashtags",
+ type: "hashtag"
+ ) do
+ :ok
+ else
+ {:error, %{reason: err}} ->
+ Logger.error("Could not bulk put hashtags: #{err}")
+ :skipped
+
+ {:ok, %{body: body}} ->
+ IO.inspect(body)
+ :skipped
+ end
+ end
+
+ def bulk_post(_, :hashtags), do: {:ok, nil}
+
+ def search(_, _, _, :skip), do: []
+
+ def search(:raw, index, type, q) do
+ with {:ok, raw_results} <- Elastix.Search.search(url(), index, [type], q) do
+ results =
+ raw_results
+ |> Map.get(:body, %{})
+ |> Map.get("hits", %{})
+ |> Map.get("hits", [])
+
+ {:ok, results}
+ else
+ {:error, e} ->
+ Logger.error(e)
+ {:error, e}
+ end
+ end
+
+ def search(:activities, q) do
+ with {:ok, results} <- search(:raw, "activities", "activity", q) do
+ results
+ |> Enum.map(fn result -> result["_id"] end)
+ |> Pleroma.Activity.all_by_ids_with_object()
+ |> Enum.sort(&(&1.inserted_at >= &2.inserted_at))
+ else
+ e ->
+ Logger.error(e)
+ []
+ end
+ end
+
+ def search(:users, q) do
+ with {:ok, results} <- search(:raw, "users", "user", q) do
+ results
+ |> Enum.map(fn result -> result["_id"] end)
+ |> Pleroma.User.get_all_by_ids()
+ else
+ e ->
+ Logger.error(e)
+ []
+ end
+ end
+
+ def search(:hashtags, q) do
+ with {:ok, results} <- search(:raw, "hashtags", "hashtag", q) do
+ results
+ |> Enum.map(fn result -> result["_source"]["hashtag"] end)
+ else
+ e ->
+ Logger.error(e)
+ []
+ end