ash_prefixed_id

if Code.ensure_loaded?(AshPostgres.CustomExtension) do
  defmodule AshPrefixedId.PostgresExtension do
    @moduledoc """
    AshPostgres custom extension that provides `uuid_generate_v7()` and
    `timestamp_from_uuid_v7()` PostgreSQL functions.

    ## Usage

    Add to your Repo's `installed_extensions`:

        defmodule MyApp.Repo do
          use AshPostgres.Repo, otp_app: :my_app

          def installed_extensions do
            ["ash-prefixed-id", ...]
          end
        end

    Then generate and run the migration:

        mix ash.codegen install_prefixed_id_extension
        mix ash.migrate

    ## A note on `LEAKPROOF`

    `timestamp_from_uuid_v7/1` is intentionally NOT marked `LEAKPROOF`.
    LEAKPROOF only matters when a function appears in a query that
    crosses a security barrier (Row Level Security, `security_barrier`
    views) and you want the planner to push the predicate through that
    barrier. Ash's authorization is application-layer (Ash policies) and
    AshPostgres does not enable RLS or security-barrier views by default,
    so the attribute provides no practical benefit for typical users.

    Marking a function `LEAKPROOF` requires PostgreSQL superuser, which
    blocks every managed Postgres provider (RDS, Cloud SQL, Heroku,
    Supabase, Neon, Aiven, etc.). Defaulting to portable migrations is
    worth the lost optimizer hint.

    Users who actually rely on RLS / security-barrier views can mark the
    function leakproof themselves as superuser, after the migration
    runs:

        ALTER FUNCTION timestamp_from_uuid_v7(uuid) LEAKPROOF;
    """

    use AshPostgres.CustomExtension, name: "ash-prefixed-id", latest_version: 1

    @impl true
    def install(_version) do
      """
      execute(\"\"\"
      CREATE OR REPLACE FUNCTION uuid_generate_v7()
      RETURNS UUID
      AS $$
      DECLARE
        timestamp    TIMESTAMPTZ;
        microseconds INT;
      BEGIN
        timestamp    = clock_timestamp();
        microseconds = (cast(extract(microseconds FROM timestamp)::INT - (floor(extract(milliseconds FROM timestamp))::INT * 1000) AS DOUBLE PRECISION) * 4.096)::INT;

        RETURN encode(
          set_byte(
            set_byte(
              overlay(uuid_send(gen_random_uuid()) placing substring(int8send(floor(extract(epoch FROM timestamp) * 1000)::BIGINT) FROM 3) FROM 1 FOR 6
            ),
            6, (b'0111' || (microseconds >> 8)::bit(4))::bit(8)::int
          ),
          7, microseconds::bit(8)::int
        ),
        'hex')::UUID;
      END
      $$
      LANGUAGE PLPGSQL
      VOLATILE;
      \"\"\")

      execute(\"\"\"
      CREATE OR REPLACE FUNCTION timestamp_from_uuid_v7(_uuid uuid)
      RETURNS TIMESTAMP WITHOUT TIME ZONE
      AS $$
        SELECT to_timestamp(('x0000' || substr(_uuid::TEXT, 1, 8) || substr(_uuid::TEXT, 10, 4))::BIT(64)::BIGINT::NUMERIC / 1000);
      $$
      LANGUAGE SQL
      IMMUTABLE PARALLEL SAFE STRICT;
      \"\"\")
      """
    end

    @impl true
    def uninstall(_version) do
      """
      execute(\"DROP FUNCTION IF EXISTS uuid_generate_v7()\")
      execute(\"DROP FUNCTION IF EXISTS timestamp_from_uuid_v7(uuid)\")
      """
    end
  end
end