Collect Scopes about @import

Author: thecrypticace

packages/tailwindcss-language-service/src/scopes/analyze.ts

@@ -5,6 +5,7 @@ import type {
   ScopeAtRule,
   ScopeClassList,
   ScopeClassName,
+  ScopeAtImport,
 } from './scope'
 import { ScopeTree } from './tree'
 import { getDocumentLanguages, LanguageBoundary } from '../util/getLanguageBoundaries'
@@ -235,5 +236,87 @@ function* analyzeAtRule(
     })
   }
 
+  // `@import` statements are special
+  else if (name === '@import') {
+    let importScope: ScopeAtImport = {
+      kind: 'css.at-rule.import',
+      children: [],
+      source: {
+        scope: overallSpan,
+        url: null,
+        sourceUrl: null,
+      },
+    }
+
+    scope.children.push(importScope)
+
+    let start = 0
+
+    for (let part of parts) {
+      let offset = start
+      let length = part.length + 1
+      let type: string | null = null
+      let quotable = true
+
+      if (part.startsWith('url(')) {
+        type = 'url'
+        quotable = true
+        part = part.slice(4)
+        offset += 4
+
+        if (part.endsWith(')')) {
+          part = part.slice(0, -1)
+        }
+      }
+
+      //
+      else if (part.startsWith('source(')) {
+        type = 'source-url'
+        quotable = true
+        part = part.slice(7)
+        offset += 7
+
+        if (part.endsWith(')')) {
+          part = part.slice(0, -1)
+        }
+      }
+
+      if (quotable && part.startsWith('"')) {
+        type ??= 'url'
+        part = part.slice(1)
+        offset += 1
+
+        if (part.endsWith('"')) {
+          part = part.slice(0, -1)
+        }
+      }
+
+      //
+      else if (quotable && part.startsWith("'")) {
+        type ??= 'url'
+        part = part.slice(1)
+        offset += 1
+
+        if (part.endsWith("'")) {
+          part = part.slice(0, -1)
+        }
+      }
+
+      if (type === 'url') {
+        importScope.source.url = [paramsSpan[0] + offset, paramsSpan[0] + offset + part.length]
+      }
+
+      //
+      else if (type === 'source-url') {
+        importScope.source.sourceUrl = [
+          paramsSpan[0] + offset,
+          paramsSpan[0] + offset + part.length,
+        ]
+      }
+
+      start += length
+    }
+  }
+
   yield scope
 }

packages/tailwindcss-language-service/src/scopes/scope.ts

@@ -233,6 +233,39 @@ export interface ScopeAtUtility {
   }
 }
 
+/**
+ * Text that represents a single class
+ *
+ * ```
+ * @import "./some-file.css";
+ * ^^^^^^^^^^^^^^^^^^^^^^^^^^
+ * ````
+ */
+export interface ScopeAtImport {
+  kind: 'css.at-rule.import'
+  children: AnyScope[]
+
+  source: {
+    scope: Span
+
+    // Marks the url of an import statement
+    // @import "./some-file.css";
+    //          ^^^^^^^^^^^^^^^
+    // @reference "./some-file.css";
+    //             ^^^^^^^^^^^^^^^
+    // @import url("./some-file.css");
+    //             ^^^^^^^^^^^^^^^^^
+    url: Span | null
+
+    // Marks an import statement's source url
+    // @import "./some-file.css" source("./foo");
+    //                                   ^^^^^
+    // @import "./some-file.css" source(none);
+    //                                  ^^^^
+    sourceUrl: Span | null
+  }
+}
+
 export type ScopeKind = keyof ScopeMap
 export type Scope<K extends ScopeKind> = ScopeMap[K]
 export type AnyScope = ScopeMap[ScopeKind]
@@ -245,4 +278,5 @@ type ScopeMap = {
   'class.name': ScopeClassName
   'css.at-rule': ScopeAtRule
   'css.at-rule.utility': ScopeAtUtility
+  'css.at-rule.import': ScopeAtImport
 }