Generate PgHasArrayType impls for enums

sqlx-macros-core/src/derives/attributes.rs

@@ -236,12 +236,6 @@ pub fn check_enum_attributes(input: &DeriveInput) -> syn::Result<SqlxContainerAt
         input
     );
 
-    assert_attribute!(
-        !attributes.no_pg_array,
-        "unused #[sqlx(no_pg_array)]; derive does not emit `PgHasArrayType` impls for enums",
-        input
-    );
-
     Ok(attributes)
 }
 

sqlx-macros-core/src/derives/type.rs

@@ -184,6 +184,18 @@ fn expand_derive_has_sql_type_strong_enum(
                 }
             }
         ));
+
+        if !attributes.no_pg_array {
+            let array_ty_name = pg_array_type_name(ident, attributes.type_name.as_ref());
+            tts.extend(quote!(
+                    #[automatically_derived]
+                    impl ::sqlx::postgres::PgHasArrayType for #ident {
+                        fn array_type_info() -> ::sqlx::postgres::PgTypeInfo {
+                            ::sqlx::postgres::PgTypeInfo::with_name(#array_ty_name)
+                        }
+                    }
+            ));
+        }
     }
 
     if cfg!(feature = "sqlite") {
@@ -235,3 +247,21 @@ fn type_name(ident: &Ident, explicit_name: Option<&TypeName>) -> TokenStream {
         quote_spanned!(ident.span()=> #s)
     })
 }
+
+fn pg_array_type_name(ident: &Ident, explicit_name: Option<&TypeName>) -> TokenStream {
+    explicit_name
+        .map(|tn| {
+            let s = pg_array_type_name_from_type_name(&tn.val);
+            quote! { #s }
+        })
+        .unwrap_or_else(|| {
+            let s = pg_array_type_name_from_type_name(&ident.to_string());
+            quote_spanned!(ident.span()=> #s)
+        })
+}
+
+fn pg_array_type_name_from_type_name(name: &str) -> String {
+    name.split_once('.')
+        .map(|(schema, name)| format!("{}._{}", schema, name))
+        .unwrap_or_else(|| format!("_{}", name))
+}

tests/postgres/describe.rs

@@ -52,7 +52,7 @@ async fn it_describes_enum() -> anyhow::Result<()> {
 
     let ty = d.columns()[0].type_info();
 
-    assert_eq!(ty.name(), "status");
+    assert_eq!(ty.name(), "public.status");
 
     assert_eq!(
         format!("{:?}", ty.kind()),