git.squeep.com Git - akkoma/blob - lib/mix/tasks/pleroma/user.ex

   1 defmodule Mix.Tasks.Pleroma.User do
   2   use Mix.Task
   3   alias Pleroma.{Repo, User}
   4
   5   @shortdoc "Manages Pleroma users"
   6   @moduledoc """
   7   Manages Pleroma users.
   8
   9   ## Create a new user.
  10
  11       mix pleroma.user new NICKNAME EMAIL [OPTION...]
  12
  13   Options:
  14   - `--name NAME` - the user's name (i.e., "Lain Iwakura")
  15   - `--bio BIO` - the user's bio
  16   - `--password PASSWORD` - the user's password
  17   - `--moderator`/`--no-moderator` - whether the user is a moderator
  18   - `--admin`/`--no-admin` - whether the user is an admin
  19
  20   ## Delete the user's account.
  21
  22       mix pleroma.user rm NICKNAME
  23
  24   ## Deactivate or activate the user's account.
  25
  26       mix pleroma.user toggle_activated NICKNAME
  27
  28   ## Create a password reset link.
  29
  30       mix pleroma.user reset_password NICKNAME
  31
  32   ## Set the value of the given user's settings.
  33
  34       mix pleroma.user set NICKNAME [OPTION...]
  35
  36   Options:
  37   - `--locked`/`--no-locked` - whether the user's account is locked
  38   - `--moderator`/`--no-moderator` - whether the user is a moderator
  39   - `--admin`/`--no-admin` - whether the user is an admin
  40   """
  41
  42   def run(["new", nickname, email | rest]) do
  43     {options, [], []} =
  44       OptionParser.parse(
  45         rest,
  46         strict: [
  47           name: :string,
  48           bio: :string,
  49           password: :string,
  50           moderator: :boolean,
  51           admin: :boolean
  52         ]
  53       )
  54
  55     name = Keyword.get(options, :name, nickname)
  56     bio = Keyword.get(options, :bio, "")
  57
  58     {password, generated_password?} =
  59       case Keyword.get(options, :password) do
  60         nil ->
  61           {:crypto.strong_rand_bytes(16) |> Base.encode64(), true}
  62
  63         password ->
  64           {password, false}
  65       end
  66
  67     moderator? = Keyword.get(options, :moderator, false)
  68     admin? = Keyword.get(options, :admin, false)
  69
  70     Mix.shell().info("""
  71     A user will be created with the following information:
  72       - nickname: #{nickname}
  73       - email: #{email}
  74       - password: #{
  75       if(generated_password?, do: "[generated; a reset link will be created]", else: password)
  76     }
  77       - name: #{name}
  78       - bio: #{bio}
  79       - moderator: #{if(moderator?, do: "true", else: "false")}
  80       - admin: #{if(admin?, do: "true", else: "false")}
  81     """)
  82
  83     proceed? = Mix.shell().yes?("Continue?")
  84
  85     unless not proceed? do
  86       Mix.Task.run("app.start")
  87
  88       params =
  89         %{
  90           nickname: nickname,
  91           email: email,
  92           password: password,
  93           password_confirmation: password,
  94           name: name,
  95           bio: bio
  96         }
  97         |> IO.inspect()
  98
  99       user = User.register_changeset(%User{}, params)
 100       Repo.insert!(user)
 101
 102       Mix.shell().info("User #{nickname} created")
 103
 104       if moderator? do
 105         run(["set", nickname, "--moderator"])
 106       end
 107
 108       if admin? do
 109         run(["set", nickname, "--admin"])
 110       end
 111
 112       if generated_password? do
 113         run(["reset_password", nickname])
 114       end
 115
 116     else
 117       Mix.shell().info("User will not be created.")
 118     end
 119   end
 120
 121   def run(["rm", nickname]) do
 122     Mix.Task.run("app.start")
 123
 124     with %User{local: true} = user <- User.get_by_nickname(nickname) do
 125       User.delete(user)
 126       Mix.shell().info("User #{nickname} deleted.")
 127     else
 128       _ ->
 129         Mix.shell().error("No local user #{nickname}")
 130     end
 131   end
 132
 133   def run(["toggle_activated", nickname]) do
 134     Mix.Task.run("app.start")
 135
 136     with %User{local: true} = user <- User.get_by_nickname(nickname) do
 137       User.deactivate(user, !user.info["deactivated"])
 138       Mix.shell().info("Activation status of #{nickname}: #{user.info["deactivated"]}")
 139     else
 140       _ ->
 141         Mix.shell().error("No local user #{nickname}")
 142     end
 143   end
 144
 145   def run(["reset_password", nickname]) do
 146     Mix.Task.run("app.start")
 147
 148     with %User{local: true} = user <- User.get_by_nickname(nickname),
 149          {:ok, token} <- Pleroma.PasswordResetToken.create_token(user) do
 150       Mix.shell().info("Generated password reset token for #{user.nickname}")
 151
 152       IO.puts(
 153         "URL: #{
 154           Pleroma.Web.Router.Helpers.util_url(
 155             Pleroma.Web.Endpoint,
 156             :show_password_reset,
 157             token.token
 158           )
 159         }"
 160       )
 161     else
 162       _ ->
 163         Mix.shell().error("No local user #{nickname}")
 164     end
 165   end
 166
 167   def run(["set", nickname | rest]) do
 168     {options, [], []} =
 169       OptionParser.parse(
 170         rest,
 171         strict: [
 172           moderator: :boolean,
 173           admin: :boolean,
 174           locked: :boolean
 175         ]
 176       )
 177
 178     case Keyword.get(options, :moderator) do
 179       nil -> nil
 180       value -> set_moderator(nickname, value)
 181     end
 182
 183     case Keyword.get(options, :locked) do
 184       nil -> nil
 185       value -> set_locked(nickname, value)
 186     end
 187
 188     case Keyword.get(options, :admin) do
 189       nil -> nil
 190       value -> set_admin(nickname, value)
 191     end
 192   end
 193
 194   defp set_moderator(nickname, value) do
 195     Application.ensure_all_started(:pleroma)
 196
 197     with %User{local: true} = user <- User.get_by_nickname(nickname) do
 198       info =
 199         user.info
 200         |> Map.put("is_moderator", value)
 201
 202       cng = User.info_changeset(user, %{info: info})
 203       {:ok, user} = User.update_and_set_cache(cng)
 204
 205       Mix.shell().info("Moderator status of #{nickname}: #{user.info["is_moderator"]}")
 206     else
 207       _ ->
 208         Mix.shell().error("No local user #{nickname}")
 209     end
 210   end
 211
 212   defp set_admin(nickname, value) do
 213     Application.ensure_all_started(:pleroma)
 214
 215     with %User{local: true} = user <- User.get_by_nickname(nickname) do
 216       info =
 217         user.info
 218         |> Map.put("is_admin", value)
 219
 220       cng = User.info_changeset(user, %{info: info})
 221       {:ok, user} = User.update_and_set_cache(cng)
 222
 223       Mix.shell().info("Admin status of #{nickname}: #{user.info["is_admin"]}")
 224     else
 225       _ ->
 226         Mix.shell().error("No local user #{nickname}")
 227     end
 228   end
 229
 230   defp set_locked(nickname, value) do
 231     Mix.Ecto.ensure_started(Repo, [])
 232
 233     with %User{local: true} = user <- User.get_by_nickname(nickname) do
 234       info =
 235         user.info
 236         |> Map.put("locked", value)
 237
 238       cng = User.info_changeset(user, %{info: info})
 239       user = Repo.update!(cng)
 240
 241       IO.puts("Locked status of #{nickname}: #{user.info["locked"]}")
 242     else
 243       _ ->
 244         IO.puts("No local user #{nickname}")
 245     end
 246   end
 247 end