fix: fix typo (#382)

Signed-off-by: zongz <[email protected]>
kcl-lang · Jul 15, 2024 · 09233a5 · 09233a5
1 parent 81bcdec
commit 09233a5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/pkg/utils/utils.go b/pkg/utils/utils.go
@@ -306,7 +306,7 @@ func DirExists(path string) bool {
 	return err == nil
 }
 
-const ModRelativePathPattern = `^\$\{[^}]+:KCL_MOD\}/main\.k$`
+const ModRelativePathPattern = `\$\{([a-zA-Z0-9_-]+:)?KCL_MOD\}/`
 
 // If the path preffix is `${KCL_MOD}` or `${KCL_MOD:xxx}`
 func IsModRelativePath(s string) bool {

diff --git a/pkg/utils/utils_test.go b/pkg/utils/utils_test.go
@@ -296,3 +296,9 @@ func TestIsSymlinkExist(t *testing.T) {
 		assert.Equal(t, err, nil)
 	}()
 }
+
+func TestIsModRelativePath(t *testing.T) {
+	assert.Equal(t, IsModRelativePath("${KCL_MOD}/aaa"), true)
+	assert.Equal(t, IsModRelativePath("${helloworld:KCL_MOD}/aaa"), true)
+	assert.Equal(t, IsModRelativePath("xxx/xxx/xxx"), false)
+}