fix(svelte): fix new import alias resolution, support custom registry

Author: WarningImHack3r

CHANGELOG.md

@@ -12,6 +12,11 @@
 
 ## [Unreleased]
 
+### Fixed
+
+- [Svelte] Fix invalid imports for added or updated components
+- [Svelte] Add support for custom registry
+
 ## [0.9.6] - 2025-08-23
 
 ### Fixed

src/main/kotlin/com/github/warningimhack3r/intellijshadcnplugin/backend/sources/Source.kt

@@ -39,7 +39,7 @@ abstract class Source<C : Config>(val project: Project, private val serializer:
     private val log = logger<Source<C>>()
     private var config: C? = null
 
-    protected val domain: String
+    protected open val domain: String
         get() = URI(getLocalConfig().`$schema`).let { uri ->
             "${uri.scheme}://${uri.host}".also {
                 log.debug("Parsed domain: $it")

src/main/kotlin/com/github/warningimhack3r/intellijshadcnplugin/backend/sources/config/SvelteConfig.kt

@@ -21,7 +21,7 @@ sealed class SvelteConfig : Config() {
     /**
      * The schema URL for the file.
      */
-    override val `$schema`: String = "https://shadcn-svelte.com/schema.json"
+    override val `$schema` = "https://shadcn-svelte.com/schema.json"
 
     /**
      * The Tailwind configuration.
@@ -50,8 +50,38 @@ sealed class SvelteConfig : Config() {
     @Serializable
     data class Aliases(
         override val components: String,
-        override val utils: String
-    ) : Config.Aliases()
+        override val utils: String,
+        val ui: String = $$"$lib/components/ui",
+        val hooks: String = $$"$lib/hooks",
+        val lib: String = $$"$lib"
+    ) : Config.Aliases() {
+
+        fun fromRaw(raw: String) = when (raw.uppercase()) {
+            Alias.COMPONENTS.name -> components
+            Alias.UTILS.name -> utils
+            Alias.UI.name -> ui
+            Alias.HOOKS.name -> hooks
+            Alias.LIB.name -> lib
+            else -> null
+        }
+
+        enum class Alias {
+            COMPONENTS,
+            UI,
+            HOOKS,
+            UTILS,
+            LIB;
+
+            val slotName = name.lowercase()
+
+            val placeholder = "$$name$"
+        }
+    }
+
+    /**
+     * The registry to use for API calls
+     */
+    val registry = "https://shadcn-svelte.com/registry"
 }
 
 /**

src/main/kotlin/com/github/warningimhack3r/intellijshadcnplugin/backend/sources/impl/SvelteSource.kt

@@ -29,12 +29,14 @@ open class SvelteSource(project: Project) : Source<SvelteConfig>(project, Svelte
         private var isJsUnsupportedNotified = false
     }
 
+    override val domain: String
+        get() = getLocalConfig().registry
+
     override var framework = "Svelte"
 
-    override fun getURLPathForRoot() = "registry/index.json"
+    override fun getURLPathForRoot() = "/index.json"
 
-    override fun getURLPathForComponent(componentName: String) =
-        "registry/$componentName.json"
+    override fun getURLPathForComponent(componentName: String) = "/$componentName.json"
 
     override fun usesDirectoriesForComponents() = true
 
@@ -115,15 +117,20 @@ open class SvelteSource(project: Project) : Source<SvelteConfig>(project, Svelte
         val importsPackagesReplacementVisitor = ImportsPackagesReplacementVisitor(project)
         runReadAction { file.accept(importsPackagesReplacementVisitor) }
         importsPackagesReplacementVisitor.replaceImports { `package` ->
-            `package`
-                .replace(Regex("^\\\$lib/registry/[^/]+"), escapeRegexValue(config.aliases.components))
-                .replace($$"$lib/utils", config.aliases.utils)
+            var renamedPackage = `package`
+            SvelteConfig.Aliases.Alias.entries.forEach { alias ->
+                renamedPackage = renamedPackage.replace(
+                    alias.placeholder,
+                    config.aliases.fromRaw(alias.slotName).orEmpty()
+                )
+            }
+            renamedPackage
         }
     }
 
     override fun fetchColors(): JsonElement {
-        val baseColor = getLocalConfig().tailwind.baseColor
-        return RequestSender.sendRequest("$domain/registry/colors/$baseColor.json").ok {
+        val config = getLocalConfig()
+        return RequestSender.sendRequest("${config.registry}/colors/${config.tailwind.baseColor}.json").ok {
             Json.parseToJsonElement(it.body)
         } ?: throw Exception("Colors not found")
     }

src/test/kotlin/com/github/warningimhack3r/intellijshadcnplugin/backend/sources/impl/SvelteReplacementsTests.kt

@@ -14,8 +14,11 @@ class SvelteReplacementsTests : ReplacementsTests() {
                 baseColor = "slate"
             ),
             aliases = SvelteConfig.Aliases(
-                components = $$"$lib/components",
-                utils = $$"$lib/replacedUtils"
+                components = $$"$lib/replacedComponents",
+                utils = $$"$lib/replacedUtils",
+                ui = $$"$lib/components/replacedUi",
+                hooks = $$"$lib/replacedHooks",
+                lib = $$"$replacedLib",
             )
         )
 
@@ -36,24 +39,20 @@ class SvelteReplacementsTests : ReplacementsTests() {
         )
     }
 
-    fun testImportMatchingComponentsAlias() {
-        compareImports($$"$lib/components/foo", $$"$lib/registry/default/foo")
+    fun testImportMatchingAlias() {
+        compareImports($$"$lib/replacedComponents", $$"$COMPONENTS$")
     }
 
-    fun testImportNotMatchingComponentAlias1() {
-        compareImports($$"$lib/components", $$"$lib/registry/foo")
+    fun testImportContainingAlias() {
+        compareImports($$"$lib/replacedHooks/myHook", $$"$HOOKS$/myHook")
     }
 
-    fun testImportNotMatchingComponentAlias2() {
-        compareImports($$"$lib/notregistry/something/foo", $$"$lib/notregistry/something/foo")
+    fun testImportMatchingAnywhere() {
+        compareImports($$"something/$replacedLib", $$"something/$LIB$")
     }
 
-    fun testImportMatchingUtilsAlias() {
-        compareImports($$"$lib/replacedUtils", $$"$lib/utils")
-    }
-
-    fun testImportMatchingUtilsAlias2() {
-        compareImports($$"$lib/replacedUtils.js", $$"$lib/utils.js")
+    fun testImportNotMatchingWrongCase() {
+        compareImports($$"$ui$", $$"$ui$")
     }
 
     fun testImportNotMatchingUtilsAlias() {