spc-pleroma/lib/pleroma/helpers/inet_helper.ex

# Pleroma: A lightweight social networking server
# Copyright © 2017-2022 Pleroma Authors <https://pleroma.social/>
# SPDX-License-Identifier: AGPL-3.0-only

defmodule Pleroma.Helpers.InetHelper do
  def parse_address(ip) when is_tuple(ip) do
    {:ok, ip}
  end

  def parse_address(ip) when is_binary(ip) do
    ip
    |> String.to_charlist()
    |> parse_address()
  end

  def parse_address(ip) do
    :inet.parse_address(ip)
  end

  def parse_cidr(proxy) when is_binary(proxy) do
    proxy =
      cond do
        "/" in String.codepoints(proxy) -> proxy
        InetCidr.v4?(InetCidr.parse_address!(proxy)) -> proxy <> "/32"
        InetCidr.v6?(InetCidr.parse_address!(proxy)) -> proxy <> "/128"
      end

    InetCidr.parse_cidr!(proxy, true)
  end
end