fix: added targeting to init also

mach-composer · Dec 10, 2024 · 6a1af59 · 6a1af59
1 parent b255a53
commit 6a1af59
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/docs/src/reference/cli/mach-composer_init.md b/docs/src/reference/cli/mach-composer_init.md
@@ -13,6 +13,7 @@ mach-composer init [flags]
   -h, --help                 help for init
       --ignore-version       Skip MACH composer version check
   -o, --output-path string   Outputs path to store the generated files. (default "deployments")
+  -s, --site string          Site to parse. If not set parse all sites.
       --var-file string      Use a variable file to parse the configuration with.
   -w, --workers int          The number of workers to use (default 1)
 ```

diff --git a/internal/cmd/init.go b/internal/cmd/init.go
@@ -11,6 +11,10 @@ import (
 	"github.com/mach-composer/mach-composer-cli/internal/runner"
 )
 
+var initFlags struct {
+	site string
+}
+
 var initCmd = &cobra.Command{
 	Use:   "init",
 	Short: "Initialize site directories Terraform files.",
@@ -28,14 +32,15 @@ var initCmd = &cobra.Command{
 
 func init() {
 	registerCommonFlags(initCmd)
+	initCmd.Flags().StringVarP(&initFlags.site, "site", "s", "", "Site to parse. If not set parse all sites.")
 }
 
 func initFunc(cmd *cobra.Command, _ []string) error {
 	cfg := loadConfig(cmd, true)
 	defer cfg.Close()
 	ctx := cmd.Context()
 
-	dg, err := graph.ToDeploymentGraph(cfg, commonFlags.outputPath)
+	dg, err := graph.ToDeploymentGraph(cfg, commonFlags.outputPath, graph.WithTargetSiteName(initFlags.site))
 	if err != nil {
 		return err
 	}