fix: add module

pubgo · Jan 26, 2024 · cf1b3ef · cf1b3ef
1 parent 01ab84f
commit cf1b3ef
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git a/cmd/cmd.go b/cmd/cmd.go
@@ -244,9 +244,22 @@ func Main() *cli.App {
 									return false
 								}
 
-								if !hasPath() {
+								var hasModule = func() bool {
+									for _, opt := range opts {
+										if strings.HasPrefix(opt, "module=") {
+											return true
+										}
+									}
+									return false
+								}
+
+								if !hasPath() && basePlugin.Paths != "" {
 									opts = append(opts, fmt.Sprintf("paths=%s", basePlugin.Paths))
 								}
+
+								if !hasModule() && basePlugin.Module != "" {
+									opts = append(opts, fmt.Sprintf("module=%s", basePlugin.Module))
+								}
 							}
 
 							if plg.Shell != "" || plg.Docker != "" {

diff --git a/cmd/config.go b/cmd/config.go
@@ -47,8 +47,9 @@ type Cfg struct {
 }
 
 type basePluginCfg struct {
-	Out   string `yaml:"out,omitempty"`
-	Paths string `yaml:"paths,omitempty"`
+	Out    string `yaml:"out,omitempty"`
+	Paths  string `yaml:"paths,omitempty"`
+	Module string `yaml:"module,omitempty"`
 }
 
 type plugin struct {