From fec2b18870f6806e602009632b52b9fe89ebfb83 Mon Sep 17 00:00:00 2001
From: Dany Castillo <31006608+dcastil@users.noreply.github.com>
Date: Fri, 10 Dec 2021 09:12:08 +0100
Subject: [PATCH] add support for columns and isTshirtSize validator

---
 src/default-config.ts   | 43 +++++++++++++++++++----------------------
 src/validators.ts       |  9 +++++++--
 tests/class-map.test.ts |  1 +
 3 files changed, 28 insertions(+), 25 deletions(-)

diff --git a/src/default-config.ts b/src/default-config.ts
index f62da723..4f57f5f5 100644
--- a/src/default-config.ts
+++ b/src/default-config.ts
@@ -1,5 +1,12 @@
 import { fromTheme } from './from-theme'
-import { isAny, isCustomLength, isCustomValue, isInteger, isLength } from './validators'
+import {
+    isAny,
+    isCustomLength,
+    isCustomValue,
+    isInteger,
+    isLength,
+    isTshirtSize,
+} from './validators'
 
 export function getDefaultConfig() {
     const colors = fromTheme('colors')
@@ -26,8 +33,6 @@ export function getDefaultConfig() {
     const space = fromTheme('space')
     const translate = fromTheme('translate')
 
-    const getSizesSimple = () => ['sm', 'md', 'lg', 'xl', '2xl'] as const
-    const getSizesExtended = () => ['3xl', '4xl', '5xl', '6xl', '7xl'] as const
     const getOverscroll = () => ['auto', 'contain', 'none'] as const
     const getOverflow = () => ['auto', 'hidden', 'visible', 'scroll'] as const
     const getSpacingWithAuto = () => ['auto', spacing] as const
@@ -77,10 +82,10 @@ export function getDefaultConfig() {
         theme: {
             colors: [isAny],
             spacing: [isLength],
-            blur: ['none', '', ...getSizesSimple(), '3xl', isCustomLength],
+            blur: ['none', '', isTshirtSize, isCustomLength],
             brightness: [isInteger],
             borderColor: [colors],
-            borderRadius: ['none', '', ...getSizesSimple(), '3xl', 'full', isCustomLength],
+            borderRadius: ['none', '', 'full', isTshirtSize, isCustomLength],
             borderWidth: getLengthWithEmpty(),
             contrast: [isInteger],
             grayscale: getZeroAndEmpty(),
@@ -255,6 +260,11 @@ export function getDefaultConfig() {
              * @see https://tailwindcss.com/docs/container
              */
             container: ['container'],
+            /**
+             * Columns
+             * @see https://tailwindcss.com/docs/columns
+             */
+            columns: [{ columns: [isTshirtSize] }],
             /**
              * Box Decoration Break
              * @see https://tailwindcss.com/docs/box-decoration-break
@@ -654,13 +664,12 @@ export function getDefaultConfig() {
                     'max-w': [
                         '0',
                         'none',
-                        ...getSizesSimple(),
-                        ...getSizesExtended(),
                         'full',
                         'min',
                         'max',
                         'prose',
-                        { screen: getSizesSimple() },
+                        { screen: [isTshirtSize] },
+                        isTshirtSize,
                         isCustomLength,
                     ],
                 },
@@ -690,19 +699,7 @@ export function getDefaultConfig() {
              * Font Size
              * @see https://tailwindcss.com/docs/font-size
              */
-            'font-size': [
-                {
-                    text: [
-                        'xs',
-                        ...getSizesSimple(),
-                        'base',
-                        ...getSizesExtended(),
-                        '8xl',
-                        '9xl',
-                        isCustomLength,
-                    ],
-                },
-            ],
+            'font-size': [{ text: ['base', isTshirtSize, isCustomLength] }],
             /**
              * Font Smoothing
              * @see https://tailwindcss.com/docs/font-smoothing
@@ -1097,7 +1094,7 @@ export function getDefaultConfig() {
              * Box Shadow
              * @see https://tailwindcss.com/docs/box-shadow
              */
-            shadow: [{ shadow: ['', ...getSizesSimple(), 'inner', 'none'] }],
+            shadow: [{ shadow: ['', 'inner', 'none', isTshirtSize] }],
             /**
              * Box Shadow Color
              * @see https://tailwindcss.com/docs/box-shadow-color
@@ -1138,7 +1135,7 @@ export function getDefaultConfig() {
              * Drop Shadow
              * @see https://tailwindcss.com/docs/drop-shadow
              */
-            'drop-shadow': [{ 'drop-shadow': ['', ...getSizesSimple(), 'none'] }],
+            'drop-shadow': [{ 'drop-shadow': ['', 'none', isTshirtSize] }],
             /**
              * Grayscale
              * @see https://tailwindcss.com/docs/grayscale
diff --git a/src/validators.ts b/src/validators.ts
index 17a84e03..33cf5241 100644
--- a/src/validators.ts
+++ b/src/validators.ts
@@ -1,14 +1,15 @@
 const customValueRegex = /^\[(.+)\]$/
 const fractionRegex = /^\d+\/\d+$/
 const stringLengths = new Set(['px', 'full', 'screen'])
+const tshirtUnitRegex = /^(\d+)?(xs|sm|md|lg|xl)$/
 const lengthUnitRegex = /\d+(%|px|em|rem|vh|vw|pt|pc|in|cm|mm|cap|ch|ex|lh|rlh|vi|vb|vmin|vmax)/
 
 export function isLength(classPart: string) {
     return (
-        isCustomLength(classPart) ||
         !Number.isNaN(Number(classPart)) ||
         stringLengths.has(classPart) ||
-        fractionRegex.test(classPart)
+        fractionRegex.test(classPart) ||
+        isCustomLength(classPart)
     )
 }
 
@@ -39,3 +40,7 @@ export function isCustomValue(classPart: string) {
 export function isAny() {
     return true
 }
+
+export function isTshirtSize(classPart: string) {
+    return tshirtUnitRegex.test(classPart)
+}
diff --git a/tests/class-map.test.ts b/tests/class-map.test.ts
index efcffe7e..3c4afc04 100644
--- a/tests/class-map.test.ts
+++ b/tests/class-map.test.ts
@@ -72,6 +72,7 @@ test('class map has correct class groups at first part', () => {
         caret: ['caret-color'],
         clear: ['clear'],
         col: ['col-end', 'col-start', 'col-start-end'],
+        columns: ['columns'],
         container: ['container'],
         content: ['align-content', 'content'],
         contents: ['display'],