# Pleroma: A lightweight social networking server
-# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
+# Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
# SPDX-License-Identifier: AGPL-3.0-only
defmodule Mix.Tasks.Pleroma.Database do
alias Pleroma.Object
alias Pleroma.Repo
alias Pleroma.User
+
require Logger
require Pleroma.Constants
+
import Ecto.Query
import Mix.Pleroma
+
use Mix.Task
@shortdoc "A collection of database related tasks"
|> Stream.run()
end
- def run(["fill_old_hashtags", month_limit]) do
- import Ecto.Query
-
- start_pleroma()
-
- month_limit = String.to_integer(month_limit)
-
- if month_limit < 1 do
- shell_error("Invalid `month_limit` argument, needs to be greater than 1")
- else
- time_limit = DateTime.utc_now() |> Timex.shift(months: -month_limit)
-
- from(
- o in Object,
- where: fragment("(?)->>'hashtags' is null", o.data),
- where: fragment("(?)->>'tag' != '[]'", o.data),
- where: o.inserted_at < ^time_limit,
- select: %{id: o.id, tag: fragment("(?)->>'tag'", o.data)}
- )
- |> Pleroma.Repo.chunk_stream(200, :batches)
- |> Stream.each(fn objects ->
- Repo.transaction(fn ->
- objects_first = objects |> List.first()
- objects_last = objects |> List.last()
-
- Logger.info(
- "fill_old_hashtags: #{objects_first.id} (#{objects_first.inserted_at}) -- #{
- objects_last.id
- } (#{objects_last.inserted_at})"
- )
-
- objects
- |> Enum.map(fn object ->
- tags =
- object.tag
- |> Jason.decode!()
- |> Enum.filter(&is_bitstring(&1))
-
- Object
- |> where([o], o.id == ^object.id)
- |> update([o],
- set: [data: fragment("safe_jsonb_set(?, '{hashtags}', ?, true)", o.data, ^tags)]
- )
- |> Repo.update_all([], timeout: :infinity)
- end)
- end)
- end)
- |> Stream.run()
- end
- end
-
def run(["vacuum", args]) do
start_pleroma()