Split Identifier struct into seperate types

This commit is contained in:
Pim Kunis 2021-08-19 16:31:03 +02:00
parent 24f0657710
commit 4b60a145ca
14 changed files with 283 additions and 196 deletions

View file

@ -2,6 +2,9 @@ defmodule MatrixServer do
alias MatrixServer.EncodableMap
@random_string_alphabet Enum.into(?a..?z, []) ++ Enum.into(?A..?Z, [])
@ipv6_regex ~r/^\[(?<ip>[^\]]+)\](?<port>:\d{1,5})?$/
@ipv4_regex ~r/^(?<hostname>[^:]+)(?<port>:\d{1,5})?$/
@dns_regex ~r/^[[:alnum:]-.]{1,255}$/
@spec get_mxid(String.t()) :: String.t()
def get_mxid(localpart) when is_binary(localpart) do
@ -110,7 +113,8 @@ defmodule MatrixServer do
%{object | signatures: new_sigs}
end
@spec validate_change_simple(Ecto.Changeset.t(), atom(), (term() -> boolean())) :: Ecto.Changeset.t()
@spec validate_change_simple(Ecto.Changeset.t(), atom(), (term() -> boolean())) ::
Ecto.Changeset.t()
def validate_change_simple(changeset, field, func) do
augmented_func = fn _, val ->
if func.(val), do: [], else: [{field, "invalid"}]
@ -146,4 +150,35 @@ defmodule MatrixServer do
|> String.replace("+", "-")
|> String.replace("/", "_")
end
@spec valid_domain?(String.t()) :: boolean()
def valid_domain?(domain) do
if String.starts_with?(domain, "[") do
# Parse as ipv6.
with %{"ip" => ip} <-
Regex.named_captures(@ipv6_regex, domain),
{:ok, _} <- :inet.parse_address(String.to_charlist(ip)) do
true
else
_ -> false
end
else
# Parse as ipv4 or dns name.
case Regex.named_captures(@ipv4_regex, domain) do
nil ->
false
%{"hostname" => hostname} ->
# Try to parse as ipv4.
case String.to_charlist(hostname) |> :inet.parse_address() do
{:ok, _} ->
true
{:error, _} ->
# Try to parse as dns name.
Regex.match?(@dns_regex, hostname)
end
end
end
end
end

View file

@ -1,173 +0,0 @@
defmodule MatrixServer.Identifier do
use Ecto.Type
alias MatrixServer.Identifier
defstruct [:type, :localpart, :domain]
def type(), do: :string
def cast(id) when is_binary(id) do
with %Identifier{} = identifier <- parse(id) do
{:ok, identifier}
end
end
def cast(_), do: :error
defp parse(id) do
{sigil, id} = String.split_at(id, 1)
case get_type(sigil) do
{:ok, :event} ->
parse_event(id)
{:ok, type} when type in [:user, :room, :group, :alias] ->
case String.split(id, ":", parts: 2) do
[localpart, domain] ->
case type do
:user -> parse_user(localpart, domain)
:room -> parse_room(localpart, domain)
:group -> parse_group(localpart, domain)
:alias -> parse_alias(localpart, domain)
end
_ ->
:error
end
:error ->
:error
end
end
defp get_type("@"), do: {:ok, :user}
defp get_type("!"), do: {:ok, :room}
defp get_type("$"), do: {:ok, :event}
defp get_type("+"), do: {:ok, :group}
defp get_type("#"), do: {:ok, :alias}
defp get_type(_), do: :error
@user_regex ~r/^[a-z0-9._=\-\/]+$/
defp parse_user(localpart, domain) do
if String.length(localpart) + String.length(domain) + 2 <= 255 and
Regex.match?(@user_regex, localpart) and
valid_domain?(domain) do
%Identifier{type: :user, localpart: localpart, domain: domain}
else
:error
end
end
defp parse_room(localpart, domain) do
if valid_domain?(domain) do
%Identifier{type: :room, localpart: localpart, domain: domain}
else
:error
end
end
@event_regex ~r/^[[:alnum:]-_]+$/
defp parse_event(localpart) do
if Regex.match?(@event_regex, localpart) do
%Identifier{type: :event, localpart: localpart}
else
:error
end
end
@group_regex ~r/^[[:lower:][:digit:]._=\-\/]+$/
defp parse_group(localpart, domain) do
if String.length(localpart) + String.length(domain) + 2 <= 255 and
Regex.match?(@group_regex, localpart) and
valid_domain?(domain) do
%Identifier{type: :group, localpart: localpart, domain: domain}
else
:error
end
end
defp parse_alias(localpart, domain) do
if String.length(localpart) + String.length(domain) + 2 <= 255 and valid_domain?(domain) do
%Identifier{type: :alias, localpart: localpart, domain: domain}
else
:error
end
end
def load(id) when is_binary(id) do
{:ok, parse_unvalidated(id)}
end
def load(_), do: :error
defp parse_unvalidated(id) do
{sigil, id} = String.split_at(id, 1)
case get_type(sigil) do
{:ok, :event} ->
%Identifier{type: :event, localpart: id}
{:ok, type} ->
[localpart, domain] = String.split(id, ":", parts: 2)
identifier = %Identifier{localpart: localpart, domain: domain}
case type do
:user -> %Identifier{identifier | type: :user}
:room -> %Identifier{identifier | type: :room}
:group -> %Identifier{identifier | type: :group}
:alias -> %Identifier{identifier | type: :alias}
end
end
end
def dump(%Identifier{type: :event, localpart: localpart}) do
{:ok, "$" <> localpart}
end
def dump(%Identifier{type: type, localpart: localpart, domain: domain}) do
{:ok, type_to_sigil(type) <> localpart <> ":" <> domain}
end
def dump(_), do: :error
defp type_to_sigil(:user), do: "@"
defp type_to_sigil(:room), do: "!"
defp type_to_sigil(:event), do: "$"
defp type_to_sigil(:group), do: "+"
defp type_to_sigil(:alias), do: "#"
@ipv6_regex ~r/^\[(?<ip>[^\]]+)\](?<port>:\d{1,5})?$/
@ipv4_regex ~r/^(?<hostname>[^:]+)(?<port>:\d{1,5})?$/
@dns_regex ~r/^[[:alnum:]-.]{1,255}$/
defp valid_domain?(domain) do
if String.starts_with?(domain, "[") do
# Parse as ipv6.
with %{"ip" => ip} <-
Regex.named_captures(@ipv6_regex, domain),
{:ok, _} <- :inet.parse_address(String.to_charlist(ip)) do
true
else
_ -> false
end
else
# Parse as ipv4 or dns name.
case Regex.named_captures(@ipv4_regex, domain) do
nil ->
false
%{"hostname" => hostname} ->
# Try to parse as ipv4.
case String.to_charlist(hostname) |> :inet.parse_address() do
{:ok, _} ->
true
{:error, _} ->
# Try to parse as dns name.
Regex.match?(@dns_regex, hostname)
end
end
end
end
end

View file

@ -43,7 +43,11 @@ defmodule MatrixServer.RoomServer do
end
end
@spec create_room(pid(), MatrixServer.Account.t(), MatrixServerWeb.Client.Request.CreateRoom.t()) :: {:ok, String.t()} | {:error, atom()}
@spec create_room(
pid(),
MatrixServer.Account.t(),
MatrixServerWeb.Client.Request.CreateRoom.t()
) :: {:ok, String.t()} | {:error, atom()}
def create_room(pid, account, input) do
GenServer.call(pid, {:create_room, account, input})
end

View file

@ -204,8 +204,8 @@ defmodule MatrixServer.StateResolution.Authorization do
defp get_action_power_level(:invite, _), do: 50
defp get_action_power_level(:ban, %{"ban" => pl}), do: pl
defp get_action_power_level(:ban, _), do: 50
defp get_action_power_level(:redact, %{"redact" => pl}), do: pl
defp get_action_power_level(:redact, _), do: 50
# defp get_action_power_level(:redact, %{"redact" => pl}), do: pl
# defp get_action_power_level(:redact, _), do: 50
defp get_action_power_level(:kick, %{"kick" => pl}), do: pl
defp get_action_power_level(:kick, _), do: 50

View file

@ -0,0 +1,43 @@
defmodule MatrixServer.Types.AliasId do
use Ecto.Type
alias MatrixServer.Types.AliasId
defstruct [:localpart, :domain]
defimpl String.Chars, for: AliasId do
def to_string(%AliasId{localpart: localpart, domain: domain}) do
"#" <> localpart <> ":" <> domain
end
end
def type(), do: :string
def cast(s) when is_binary(s) do
with "#" <> rest <- s,
[localpart, domain] <- String.split(rest, ":", parts: 2) do
if String.length(localpart) + String.length(domain) + 2 <= 255 and
MatrixServer.valid_domain?(domain) do
%AliasId{localpart: localpart, domain: domain}
else
:error
end
else
_ -> :error
end
end
def cast(_), do: :error
def load(s) when is_binary(s) do
"@" <> rest = s
[localpart, domain] = String.split(rest, ":", parts: 2)
%AliasId{localpart: localpart, domain: domain}
end
def load(_), do: :error
def dump(%AliasId{} = alias_id), do: to_string(alias_id)
def dump(_), do: :error
end

View file

@ -0,0 +1,42 @@
defmodule MatrixServer.Types.EventId do
use Ecto.Type
alias MatrixServer.Types.EventId
defstruct [:id]
@id_regex ~r/^[[:alnum:]-_]+$/
defimpl String.Chars, for: EventId do
def to_string(%EventId{id: id}) do
"$" <> id
end
end
def type(), do: :string
def cast(s) when is_binary(s) do
with "$" <> id <- s do
if Regex.match?(@id_regex, id) do
%EventId{id: id}
else
:error
end
else
_ -> :error
end
end
def cast(_), do: :error
def load(s) when is_binary(s) do
"$" <> rest = s
%EventId{id: rest}
end
def load(_), do: :error
def dump(%EventId{} = event_id), do: to_string(event_id)
def dump(_), do: :error
end

View file

@ -0,0 +1,46 @@
defmodule MatrixServer.Types.GroupId do
use Ecto.Type
alias MatrixServer.Types.GroupId
defstruct [:localpart, :domain]
@localpart_regex ~r/^[[:lower:][:digit:]._=\-\/]+$/
defimpl String.Chars, for: GroupId do
def to_string(%GroupId{localpart: localpart, domain: domain}) do
"+" <> localpart <> ":" <> domain
end
end
def type(), do: :string
def cast(s) when is_binary(s) do
with "+" <> rest <- s,
[localpart, domain] <- String.split(rest, ":", parts: 2) do
if String.length(localpart) + String.length(domain) + 2 <= 255 and
Regex.match?(@localpart_regex, localpart) and
MatrixServer.valid_domain?(domain) do
%GroupId{localpart: localpart, domain: domain}
else
:error
end
else
_ -> :error
end
end
def cast(_), do: :error
def load(s) when is_binary(s) do
"@" <> rest = s
[localpart, domain] = String.split(rest, ":", parts: 2)
%GroupId{localpart: localpart, domain: domain}
end
def load(_), do: :error
def dump(%GroupId{} = group_id), do: to_string(group_id)
def dump(_), do: :error
end

View file

@ -0,0 +1,42 @@
defmodule MatrixServer.Types.RoomId do
use Ecto.Type
alias MatrixServer.Types.RoomId
defstruct [:localpart, :domain]
defimpl String.Chars, for: RoomId do
def to_string(%RoomId{localpart: localpart, domain: domain}) do
"!" <> localpart <> ":" <> domain
end
end
def type(), do: :string
def cast(s) when is_binary(s) do
with "!" <> rest <- s,
[localpart, domain] <- String.split(rest, ":", parts: 2) do
if MatrixServer.valid_domain?(domain) do
%RoomId{localpart: localpart, domain: domain}
else
:error
end
else
_ -> :error
end
end
def cast(_), do: :error
def load(s) when is_binary(s) do
"!" <> rest = s
[localpart, domain] = String.split(rest, ":", parts: 2)
%RoomId{localpart: localpart, domain: domain}
end
def load(_), do: :error
def dump(%RoomId{} = room_id), do: to_string(room_id)
def dump(_), do: :error
end

View file

@ -0,0 +1,45 @@
defmodule MatrixServer.Types.UserId do
use Ecto.Type
alias MatrixServer.Types.UserId
defstruct [:localpart, :domain]
@localpart_regex ~r/^[a-z0-9._=\-\/]+$/
defimpl String.Chars, for: UserId do
def to_string(%UserId{localpart: localpart, domain: domain}) do
"@" <> localpart <> ":" <> domain
end
end
def type(), do: :string
def cast(s) when is_binary(s) do
with "@" <> rest <- s,
[localpart, domain] <- String.split(rest, ":", parts: 2) do
if String.length(localpart) + String.length(domain) + 2 <= 255 and
Regex.match?(@localpart_regex, localpart) and MatrixServer.valid_domain?(domain) do
%UserId{localpart: localpart, domain: domain}
else
:error
end
else
_ -> :error
end
end
def cast(_), do: :error
def load(s) when is_binary(s) do
"@" <> rest = s
[localpart, domain] = String.split(rest, ":", parts: 2)
%UserId{localpart: localpart, domain: domain}
end
def load(_), do: :error
def dump(%UserId{} = user_id), do: to_string(user_id)
def dump(_), do: :error
end

View file

@ -5,7 +5,8 @@ defmodule MatrixServerWeb.Federation.QueryController do
import MatrixServerWeb.Error
import Ecto.Query
alias MatrixServer.{Repo, Account, Identifier}
alias MatrixServer.{Repo, Account}
alias MatrixServer.Types.UserId
defmodule ProfileRequest do
use Ecto.Schema
@ -14,7 +15,7 @@ defmodule MatrixServerWeb.Federation.QueryController do
@primary_key false
embedded_schema do
field :user_id, Identifier
field :user_id, UserId
field :field, :string
end
@ -31,8 +32,7 @@ defmodule MatrixServerWeb.Federation.QueryController do
end
def profile(conn, params) do
with {:ok,
%ProfileRequest{user_id: %Identifier{type: :user, localpart: localpart, domain: domain}}} <-
with {:ok, %ProfileRequest{user_id: %UserId{localpart: localpart, domain: domain}}} <-
ProfileRequest.validate(params) do
if domain == MatrixServer.server_name() do
case Repo.one(from a in Account, where: a.localpart == ^localpart) do

View file

@ -11,7 +11,10 @@ defmodule MatrixServer.MixProject do
start_permanent: Mix.env() == :prod,
aliases: aliases(),
deps: deps(),
dialyzer: [plt_add_deps: :app_tree]
dialyzer: [
plt_add_deps: :app_tree,
flags: [:error_handling, :race_conditions, :underspecs, :unknown, :unmatched_returns]
]
]
end