Add support for sorting class attribute on Astro components (#129)

* Add support for sorting `class` attribute on Astro components

* Update changelog

* Simplify test

* Add support for custom elements

* Update changelog

Author: thecrypticace

CHANGELOG.md

@@ -7,7 +7,9 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
-- Nothing yet!
+### Fixed
+
+- Add support for sorting `class` attribute on custom elements and Astro components ([#129](https://github.com/tailwindlabs/prettier-plugin-tailwindcss/pull/129))
 
 ## [0.2.3] - 2023-02-15
 

src/index.js

@@ -573,7 +573,7 @@ export const parsers = {
 }
 
 function transformAstro(ast, { env, changes }) {
-  if (ast.type === "element") {
+  if (ast.type === "element" || ast.type === "custom-element" || ast.type === "component") {
     for (let attr of ast.attributes ?? []) {
       if (attr.name === "class" && attr.type === "attribute" && attr.kind === "quoted") {
         attr.value = sortClasses(attr.value, {

tests/test.js

@@ -265,6 +265,16 @@ let tests = {
   }
 </style>`,
     ],
+    t`---
+import Layout from '../layouts/Layout.astro'
+import Custom from '../components/Custom.astro'
+---
+
+<Layout>
+  <main class="${yes}"></main>
+  <my-element class="${yes}"></my-element>
+  <Custom class="${yes}" />
+</Layout>`,
   ],
 };