Merge pull request #209 from kcl-lang/entry-set-api

feat: add entry set APIs

Author: Peefy

pkg/opt/opt.go

@@ -52,6 +52,11 @@ func (opts *CompileOptions) ExtendEntries(entries []string) {
 	opts.entries = append(opts.entries, entries...)
 }
 
+// SetEntries will set the entries of the compiler.
+func (opts *CompileOptions) SetEntries(entries []string) {
+	opts.entries = entries
+}
+
 // SetHasSettingsYaml will set the 'hasSettingsYaml' flag.
 func (opts *CompileOptions) SetHasSettingsYaml(hasSettingsYaml bool) {
 	opts.hasSettingsYaml = hasSettingsYaml

pkg/opt/opt_test.go

@@ -14,4 +14,9 @@ func TestWorkDirAsPkgPath(t *testing.T) {
 	assert.Equal(t, opts.PkgPath(), "")
 	opts.Merge(kcl.WithWorkDir("test_work_dir"))
 	assert.Equal(t, opts.PkgPath(), "test_work_dir")
+	opts.ExtendEntries([]string{"file1.k", "file2.k"})
+	opts.ExtendEntries([]string{"file3.k", "file4.k"})
+	assert.Equal(t, opts.Entries(), []string{"file1.k", "file2.k", "file3.k", "file4.k"})
+	opts.SetEntries([]string{"override.k"})
+	assert.Equal(t, opts.Entries(), []string{"override.k"})
 }