forked from nimbolus/terraform-backend
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
4 changed files
with
216 additions
and
31 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,175 @@ | ||
| package scaffold | ||
|
|
||
| import ( | ||
| "bufio" | ||
| "context" | ||
| "errors" | ||
| "fmt" | ||
| "io" | ||
| "net/url" | ||
| "os" | ||
| "path/filepath" | ||
| "slices" | ||
| "strings" | ||
|
|
||
| "github.com/hashicorp/hcl/v2" | ||
| "github.com/hashicorp/hcl/v2/hclwrite" | ||
| "github.com/nimbolus/terraform-backend/pkg/git" | ||
| "github.com/nimbolus/terraform-backend/pkg/tfcontext" | ||
| "github.com/spf13/cobra" | ||
| "github.com/zclconf/go-cty/cty" | ||
| ) | ||
|
|
||
| var ( | ||
| backendAddress string | ||
| ) | ||
|
|
||
| func NewCommand() *cobra.Command { | ||
| cmd := &cobra.Command{ | ||
| Use: "scaffold", | ||
| Short: "scaffold the necessary config to use the GitHub Actions Terraform workflow", | ||
| RunE: func(cmd *cobra.Command, args []string) error { | ||
| return run(cmd.Context()) | ||
| }, | ||
| } | ||
|
|
||
| cmd.Flags().StringVar(&backendAddress, "backend-url", "https://ffddorf-terraform-backend.fly.dev/", "URL to use as the backend address") | ||
|
|
||
| return cmd | ||
| } | ||
|
|
||
| func run(ctx context.Context) error { | ||
| cwd, err := os.Getwd() | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| if err := writeBackendConfig(cwd); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| // todo: create github actions workflows | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func prompt(text string) (string, error) { | ||
| stdout := os.Stderr | ||
| fmt.Fprint(stdout, text) | ||
|
|
||
| rdr := bufio.NewReader(os.Stdin) | ||
| answer, err := rdr.ReadBytes('\n') | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
| return string(answer[:len(answer)-1]), nil | ||
| } | ||
|
|
||
| func writeBackendConfig(dir string) (reterr error) { | ||
| var file *hclwrite.File | ||
| var outFile io.WriteCloser | ||
| var backendBlock *hclwrite.Block | ||
|
|
||
| _, filename, err := tfcontext.FindBackendBlock(dir) | ||
| if err == nil { | ||
| relPath, _ := filepath.Rel(dir, filename) | ||
| answer, err := prompt(fmt.Sprintf("There is an existing backend config at %s. Do you want to replace it? [y/N] ", relPath)) | ||
| if err != nil { | ||
| return | ||
| } | ||
| if !strings.EqualFold(answer, "y") { | ||
| return errors.New("aborting") | ||
| } | ||
|
|
||
| b, err := os.ReadFile(filename) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| var diags hcl.Diagnostics | ||
| file, diags = hclwrite.ParseConfig(b, filename, hcl.Pos{}) | ||
| if len(diags) > 0 { | ||
| return errors.Join(diags) | ||
| } | ||
| var tfBlock *hclwrite.Block | ||
| for _, block := range file.Body().Blocks() { | ||
| if block.Type() != "terraform" { | ||
| continue | ||
| } | ||
| tfBlock = block | ||
| for _, innerBlock := range block.Body().Blocks() { | ||
| if innerBlock.Type() == "backend" { | ||
| backendBlock = innerBlock | ||
| } | ||
| } | ||
| } | ||
| if backendBlock == nil { | ||
| return errors.New("backend block not found anymore") | ||
| } | ||
| if backendBlock.Labels()[0] != "http" { | ||
| tfBlock.Body().RemoveBlock(backendBlock) | ||
| backendBlock = tfBlock.Body().AppendNewBlock("backend", nil) | ||
| } | ||
|
|
||
| outFile, err = os.Create(filename) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| defer func() { | ||
| if reterr != nil { | ||
| // restore original content | ||
| _, _ = outFile.Write(b) | ||
| } | ||
| _ = outFile.Close() | ||
| }() | ||
| } else { | ||
| file = hclwrite.NewEmptyFile() | ||
| tfBlock := file.Body().AppendNewBlock("terraform", nil) | ||
| backendBlock = tfBlock.Body().AppendNewBlock("backend", nil) | ||
| filename = filepath.Join(dir, "backend.tf") | ||
| outFile, err = os.Create(filename) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| defer outFile.Close() | ||
| } | ||
|
|
||
| origin, err := git.RepoOrigin() | ||
| if err != nil { | ||
| return err | ||
| } | ||
| segments := strings.Split(origin.Path, "/") | ||
| if len(segments) < 2 { | ||
| return fmt.Errorf("invalid repo path: %s", origin.Path) | ||
| } | ||
| repo := segments[1] | ||
|
|
||
| backendURL, err := url.Parse(backendAddress) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| backendURL.Path = filepath.Join(backendURL.Path, "state", repo, "default") | ||
| address := backendURL.String() | ||
|
|
||
| backendBlock.SetLabels([]string{"http"}) | ||
| backendBody := backendBlock.Body() | ||
| backendAttributes := []string{"address", "lock_address", "unlock_address", "username"} | ||
| for name := range backendBody.Attributes() { | ||
| if slices.Contains(backendAttributes, name) { | ||
| continue | ||
| } | ||
| backendBody.RemoveAttribute(name) | ||
| } | ||
| backendBody.SetAttributeValue("address", cty.StringVal(address)) | ||
| backendBody.SetAttributeValue("lock_address", cty.StringVal(address)) | ||
| backendBody.SetAttributeValue("unlock_address", cty.StringVal(address)) | ||
| backendBody.SetAttributeValue("username", cty.StringVal("github_pat")) | ||
|
|
||
| if _, err := file.WriteTo(outFile); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| relPath, _ := filepath.Rel(dir, filename) | ||
| fmt.Printf("Wrote backend config to: %s\n", relPath) | ||
| return nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters