rogee/atomctl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: issues

Browse Source
This commit is contained in:
Rogee
2025-02-10 15:10:39 +08:00
parent eb13231e9e
commit 266c2c4767
8 changed files with 91 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
pkg/ast/provider/provider.go
View File

@@ -59,6 +59,14 @@ type Provider struct {
ProviderFile string
}
func atomPackage(suffix string) string {
root := gomod.GetModuleName() + "/pkg/atom"
if suffix != "" {
return fmt.Sprintf("%s/%s", root, suffix)
}
return root
}
func Parse(source string) []Provider {
if strings.HasSuffix(source, "_test.go") {
return []Provider{}
@@ -256,8 +264,8 @@ func Parse(source string) []Provider {
modePkg := gomod.GetModuleName() + "/providers/grpc"
provider.Imports["git.ipao.vip/rogeecn/atom"] = ""
provider.Imports["git.ipao.vip/rogeecn/atom/contracts"] = ""
provider.Imports[atomPackage("")] = ""
provider.Imports[atomPackage("contracts")] = ""
provider.Imports[modePkg] = ""
provider.ProviderGroup = "atom.GroupInitial"
@@ -277,8 +285,8 @@ func Parse(source string) []Provider {
modePkg := gomod.GetModuleName() + "/providers/event"
provider.Imports["git.ipao.vip/rogeecn/atom"] = ""
provider.Imports["git.ipao.vip/rogeecn/atom/contracts"] = ""
provider.Imports[atomPackage("")] = ""
provider.Imports[atomPackage("contracts")] = ""
provider.Imports[modePkg] = ""
provider.ProviderGroup = "atom.GroupInitial"
@@ -297,8 +305,8 @@ func Parse(source string) []Provider {
modePkg := gomod.GetModuleName() + "/providers/job"
provider.Imports["git.ipao.vip/rogeecn/atom"] = ""
provider.Imports["git.ipao.vip/rogeecn/atom/contracts"] = ""
provider.Imports[atomPackage("")] = ""
provider.Imports[atomPackage("contracts")] = ""
provider.Imports["github.com/riverqueue/river"] = ""
provider.Imports[modePkg] = ""

4
pkg/ast/provider/render.go
View File

@@ -23,8 +23,8 @@ func Render(filename string, conf []Provider) error {
}()
imports := map[string]string{
"git.ipao.vip/rogeecn/atom/container": "",
"git.ipao.vip/rogeecn/atom/utils/opt": "",
atomPackage("container"): "",
atomPackage("opt"): "",
}
lo.ForEach(conf, func(item Provider, _ int) {
for k, v := range item.Imports {