feat: add api 'GetFullSchemaType' to support external package in kcl doc (#188)

* feat: add api 'GetFullSchemaType' to support external package in kcl doc

Signed-off-by: zongz <[email protected]>

* chore: bump kcl artifact version

Signed-off-by: zongz <[email protected]>

* fix: remove useless changes

Signed-off-by: zongz <[email protected]>

* fix: make fmt

Signed-off-by: zongz <[email protected]>

* fix: move add kfiles code out the method ParseArgs

Signed-off-by: zongz <[email protected]>

---------

Signed-off-by: zongz <[email protected]>

Author: zong-zhe

pkg/kcl/api.go

@@ -199,6 +199,26 @@ func GetSchemaType(file, code, schemaName string) ([]*gpyrpc.KclType, error) {
 	return resp.SchemaTypeList, nil
 }
 
+func GetFullSchemaType(pathList []string, schemaName string, opts ...Option) ([]*gpyrpc.KclType, error) {
+	opts = append(opts, *NewOption().Merge(WithKFilenames(pathList...)))
+	args, err := ParseArgs(pathList, opts...)
+	if err != nil {
+		return nil, err
+	}
+
+	client := service.NewKclvmServiceClient()
+	resp, err := client.GetFullSchemaType(&gpyrpc.GetFullSchemaType_Args{
+		ExecArgs:   args.ExecProgram_Args,
+		SchemaName: schemaName,
+	})
+
+	if err != nil {
+		return nil, err
+	}
+
+	return resp.SchemaTypeList, nil
+}
+
 func GetSchemaTypeMapping(file, code, schemaName string) (map[string]*gpyrpc.KclType, error) {
 	client := service.NewKclvmServiceClient()
 	resp, err := client.GetSchemaTypeMapping(&gpyrpc.GetSchemaTypeMapping_Args{

pkg/kcl/api_test.go

@@ -4,10 +4,12 @@ package kcl
 
 import (
 	"fmt"
+	"path/filepath"
 	"reflect"
 	"sort"
 	"testing"
 
+	"github.com/stretchr/testify/assert"
 	"kcl-lang.io/kcl-go/pkg/tools/list"
 )
 
@@ -102,3 +104,16 @@ func TestListUpstreamFiles(t *testing.T) {
 		t.Fatalf("\nexpect = %v\ngot    = %v", expect, deps)
 	}
 }
+
+func TestGetFullSchemaType(t *testing.T) {
+	testPath := filepath.Join(".", "testdata", "get_schema_ty")
+	tys, err := GetFullSchemaType(
+		[]string{filepath.Join(testPath, "aaa")},
+		"",
+		WithExternalPkgs(fmt.Sprintf("bbb=%s", filepath.Join(testPath, "bbb"))),
+	)
+	assert.Equal(t, err, nil)
+	assert.Equal(t, len(tys), 1)
+	assert.Equal(t, tys[0].Filename, filepath.Join("testdata", "get_schema_ty", "bbb", "main.k"))
+	assert.Equal(t, tys[0].SchemaName, "B")
+}

pkg/kcl/testdata/get_schema_ty/aaa/kcl.mod

@@ -0,0 +1,5 @@
+[package]
+name = "aaa"
+edition = "0.0.1"
+version = "0.0.1"
+

pkg/kcl/testdata/get_schema_ty/aaa/main.k

@@ -0,0 +1,5 @@
+import bbb as b
+
+a = b.B {
+    name: "b instance in a"
+}

pkg/kcl/testdata/get_schema_ty/bbb/kcl.mod

@@ -0,0 +1,5 @@
+[package]
+name = "bbb"
+edition = "0.0.1"
+version = "0.0.1"
+

pkg/kcl/testdata/get_schema_ty/bbb/main.k

@@ -0,0 +1,2 @@
+schema B:
+    name: str