diff --git a/cmd/tf-preview-gh/main.go b/cmd/tf-preview-gh/main.go new file mode 100644 index 0000000..901e410 --- /dev/null +++ b/cmd/tf-preview-gh/main.go @@ -0,0 +1,260 @@ +package main + +import ( + "bufio" + "context" + "fmt" + "io" + "net/http" + "net/url" + "os" + "os/exec" + "os/signal" + "slices" + "strings" + "syscall" + "time" + + "github.com/cenkalti/backoff" + "github.com/google/go-github/v57/github" + "github.com/hashicorp/go-slug" + "golang.ngrok.com/ngrok" + "golang.ngrok.com/ngrok/config" +) + +type LocalContent struct { + dir string +} + +func (c *LocalContent) ServeHTTP(rw http.ResponseWriter, req *http.Request) { + rw.Header().Set("Content-Type", "application/octet-stream") + + _, err := slug.Pack(c.dir, rw, true) + if err != nil { + fmt.Printf("failed to pack contents: %+v\n", err) + return + } + + fmt.Println("workspace was downloaded") +} + +func startServer(ctx context.Context) (string, error) { + listenerCtx, cancelListener := context.WithCancel(context.Background()) + + connected := make(chan struct{}) + go func() { + select { + case <-connected: + case <-time.After(10 * time.Second): + cancelListener() + } + }() + + listener, err := ngrok.Listen(listenerCtx, config.HTTPEndpoint(), ngrok.WithAuthtokenFromEnv()) + if err != nil { + return "", err + } + close(connected) + + cwd, err := os.Getwd() + if err != nil { + listener.Close() + return "", err + } + handler := &LocalContent{ + dir: cwd, + } + + server := &http.Server{ + Handler: handler, + } + + go func() { + <-ctx.Done() + shutdownCtx, cancel := context.WithTimeout(context.Background(), 10*time.Second) + defer cancel() + if err := server.Shutdown(shutdownCtx); err != nil { + fmt.Printf("failed to shutdown server: %+v\n", err) + } + }() + + go func() { + if err := server.Serve(listener); err != http.ErrServerClosed { + fmt.Printf("server failed: %+v\n", err) + } + }() + + return listener.URL(), nil +} + +type countingReader struct { + io.Reader + readBytes int +} + +func (c *countingReader) Read(dst []byte) (int, error) { + n, err := c.Reader.Read(dst) + c.readBytes += n + return n, err +} + +var ignoredGroupNames = []string{ + "Operating System", + "Runner Image", + "Runner Image Provisioner", + "GITHUB_TOKEN Permissions", +} + +func streamLogs(logsURL *url.URL, skip int64) (int64, error) { + logs, err := http.Get(logsURL.String()) + if err != nil { + return 0, err + } + if logs.StatusCode != http.StatusOK { + return 0, fmt.Errorf("invalid status for logs: %d", logs.StatusCode) + } + defer logs.Body.Close() + + if _, err := io.Copy(io.Discard, io.LimitReader(logs.Body, skip)); err != nil { + return 0, err + } + + r := &countingReader{Reader: logs.Body} + scanner := bufio.NewScanner(r) + groupDepth := 0 + for scanner.Scan() { + line := scanner.Text() + ts, rest, ok := strings.Cut(line, " ") + if !ok { + rest = ts + } + if groupName, ok := strings.CutPrefix(rest, "##[group]"); ok { + groupDepth++ + if !slices.Contains(ignoredGroupNames, groupName) { + fmt.Printf("\n# %s\n", groupName) + } + } + if groupDepth == 0 { + fmt.Println(rest) + } + if strings.HasPrefix(rest, "##[endgroup]") { + groupDepth-- + } + } + if err := scanner.Err(); err != nil { + return int64(r.readBytes), err + } + + return int64(r.readBytes), err +} + +const ( + owner = "ffddorf" + repo = "terraform-playground" + workflowFilename = "preview.yaml" +) + +func main() { + ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM) + defer cancel() + + serverURL, err := startServer(ctx) + if err != nil { + panic(err) + } + + // steal token from GH CLI + cmd := exec.CommandContext(ctx, "gh", "auth", "token") + out, err := cmd.Output() + if err != nil { + panic(err) + } + + token := strings.TrimSpace(string(out)) + gh := github.NewClient(nil).WithAuthToken(token) + + startedAt := time.Now().UTC() + + // start workflow + _, err = gh.Actions.CreateWorkflowDispatchEventByFileName(ctx, + owner, repo, workflowFilename, + github.CreateWorkflowDispatchEventRequest{ + Ref: "main", + Inputs: map[string]interface{}{ + "workspace_transfer_url": serverURL, + }, + }, + ) + if err != nil { + panic(err) + } + + fmt.Println("Waiting for run to start...") + + // find workflow run + var run *github.WorkflowRun + err = backoff.Retry(func() error { + workflows, _, err := gh.Actions.ListWorkflowRunsByFileName( + ctx, owner, repo, workflowFilename, + &github.ListWorkflowRunsOptions{ + Created: fmt.Sprintf(">=%s", startedAt.Format("2006-01-02T15:04")), + }, + ) + if err != nil { + return backoff.Permanent(err) + } + if len(workflows.WorkflowRuns) == 0 { + return fmt.Errorf("no workflow runs found") + } + + run = workflows.WorkflowRuns[0] + return nil + }, backoff.NewExponentialBackOff()) + if err != nil { + panic(err) + } + + var jobID int64 + err = backoff.Retry(func() error { + jobs, _, err := gh.Actions.ListWorkflowJobs(ctx, + owner, repo, *run.ID, + &github.ListWorkflowJobsOptions{}, + ) + if err != nil { + return backoff.Permanent(err) + } + if len(jobs.Jobs) == 0 { + return fmt.Errorf("no jobs found") + } + + jobID = *jobs.Jobs[0].ID + return nil + }, backoff.NewExponentialBackOff()) + if err != nil { + panic(err) + } + + logsURL, _, err := gh.Actions.GetWorkflowJobLogs(ctx, owner, repo, jobID, 2) + if err != nil { + panic(err) + } + + var readBytes int64 + for { + n, err := streamLogs(logsURL, readBytes) + if err != nil { + panic(err) + } + readBytes += n + + // check if job is done + job, _, err := gh.Actions.GetWorkflowJobByID(ctx, owner, repo, jobID) + if err != nil { + panic(err) + } + if job.CompletedAt != nil { + fmt.Println("Job complete.") + break + } + } +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..65d442b --- /dev/null +++ b/go.mod @@ -0,0 +1,31 @@ +module github.com/ffddorf/tf-preview-github + +go 1.21.5 + +require ( + github.com/cenkalti/backoff v2.2.1+incompatible + github.com/google/go-github/v57 v57.0.0 + github.com/hashicorp/go-slug v0.13.3 + github.com/stretchr/testify v1.8.4 + golang.ngrok.com/ngrok v1.7.0 +) + +require ( + github.com/davecgh/go-spew v1.1.1 // indirect + github.com/go-stack/stack v1.8.1 // indirect + github.com/google/go-querystring v1.1.0 // indirect + github.com/inconshreveable/log15 v3.0.0-testing.5+incompatible // indirect + github.com/inconshreveable/log15/v3 v3.0.0-testing.5 // indirect + github.com/jpillora/backoff v1.0.0 // indirect + github.com/mattn/go-colorable v0.1.13 // indirect + github.com/mattn/go-isatty v0.0.20 // indirect + github.com/pmezard/go-difflib v1.0.0 // indirect + go.uber.org/multierr v1.11.0 // indirect + golang.ngrok.com/muxado/v2 v2.0.0 // indirect + golang.org/x/net v0.19.0 // indirect + golang.org/x/sync v0.6.0 // indirect + golang.org/x/sys v0.16.0 // indirect + golang.org/x/term v0.16.0 // indirect + google.golang.org/protobuf v1.32.0 // indirect + gopkg.in/yaml.v3 v3.0.1 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..7b0232a --- /dev/null +++ b/go.sum @@ -0,0 +1,57 @@ +github.com/cenkalti/backoff v2.2.1+incompatible h1:tNowT99t7UNflLxfYYSlKYsBpXdEet03Pg2g16Swow4= +github.com/cenkalti/backoff v2.2.1+incompatible/go.mod h1:90ReRw6GdpyfrHakVjL/QHaoyV4aDUVVkXQJJJ3NXXM= +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/go-stack/stack v1.8.1 h1:ntEHSVwIt7PNXNpgPmVfMrNhLtgjlmnZha2kOpuRiDw= +github.com/go-stack/stack v1.8.1/go.mod h1:dcoOX6HbPZSZptuspn9bctJ+N/CnF5gGygcUP3XYfe4= +github.com/google/go-cmp v0.5.2/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI= +github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY= +github.com/google/go-github/v57 v57.0.0 h1:L+Y3UPTY8ALM8x+TV0lg+IEBI+upibemtBD8Q9u7zHs= +github.com/google/go-github/v57 v57.0.0/go.mod h1:s0omdnye0hvK/ecLvpsGfJMiRt85PimQh4oygmLIxHw= +github.com/google/go-querystring v1.1.0 h1:AnCroh3fv4ZBgVIf1Iwtovgjaw/GiKJo8M8yD/fhyJ8= +github.com/google/go-querystring v1.1.0/go.mod h1:Kcdr2DB4koayq7X8pmAG4sNG59So17icRSOU623lUBU= +github.com/hashicorp/go-slug v0.13.3 h1:JiYNpOkD0HmMWw/lNYiBAUD6+WIBIV7UftKiqIbpNqM= +github.com/hashicorp/go-slug v0.13.3/go.mod h1:RA4C+ezyC2nDsiPM5+1djqagveBBJdSN/fM2QCUziYQ= +github.com/hashicorp/yamux v0.1.1 h1:yrQxtgseBDrq9Y652vSRDvsKCJKOUD+GzTS4Y0Y8pvE= +github.com/hashicorp/yamux v0.1.1/go.mod h1:CtWFDAQgb7dxtzFs4tWbplKIe2jSi3+5vKbgIO0SLnQ= +github.com/inconshreveable/log15 v3.0.0-testing.5+incompatible h1:VryeOTiaZfAzwx8xBcID1KlJCeoWSIpsNbSk+/D2LNk= +github.com/inconshreveable/log15 v3.0.0-testing.5+incompatible/go.mod h1:cOaXtrgN4ScfRrD9Bre7U1thNq5RtJ8ZoP4iXVGRj6o= +github.com/inconshreveable/log15/v3 v3.0.0-testing.5 h1:h4e0f3kjgg+RJBlKOabrohjHe47D3bbAB9BgMrc3DYA= +github.com/inconshreveable/log15/v3 v3.0.0-testing.5/go.mod h1:3GQg1SVrLoWGfRv/kAZMsdyU5cp8eFc1P3cw+Wwku94= +github.com/jpillora/backoff v1.0.0 h1:uvFg412JmmHBHw7iwprIxkPMI+sGQ4kzOWsMeHnm2EA= +github.com/jpillora/backoff v1.0.0/go.mod h1:J/6gKK9jxlEcS3zixgDgUAsiuZ7yrSoa/FX5e0EB2j4= +github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA= +github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg= +github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM= +github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY= +github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= +github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/stretchr/testify v1.8.4 h1:CcVxjf3Q8PM0mHUKJCdn+eZZtm5yQwehR5yeSVQQcUk= +github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo= +go.uber.org/multierr v1.11.0 h1:blXXJkSxSSfBVBlC76pxqeO+LN3aDfLQo+309xJstO0= +go.uber.org/multierr v1.11.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y= +golang.ngrok.com/muxado/v2 v2.0.0 h1:bu9eIDhRdYNtIXNnqat/HyMeHYOAbUH55ebD7gTvW6c= +golang.ngrok.com/muxado/v2 v2.0.0/go.mod h1:wzxJYX4xiAtmwumzL+QsukVwFRXmPNv86vB8RPpOxyM= +golang.ngrok.com/ngrok v1.7.0 h1:xwcr8QWue+ehgn54hdQwTya4B6A1qXg6+IRim6WINmA= +golang.ngrok.com/ngrok v1.7.0/go.mod h1:ruVcXZ7Rre5O9oeqqa8uZCB3Xtkt2PoyjF3eW9b7t6A= +golang.org/x/crypto v0.16.0 h1:mMMrFzRSCF0GvB7Ne27XVtVAaXLrPmgPC7/v0tkwHaY= +golang.org/x/crypto v0.16.0/go.mod h1:gCAAfMLgwOJRpTjQ2zCCt2OcSfYMTeZVSRtQlPC7Nq4= +golang.org/x/net v0.19.0 h1:zTwKpTd2XuCqf8huc7Fo2iSy+4RHPd10s4KzeTnVr1c= +golang.org/x/net v0.19.0/go.mod h1:CfAk/cbD4CthTvqiEl8NpboMuiuOYsAr/7NOjZJtv1U= +golang.org/x/sync v0.6.0 h1:5BMeUDZ7vkXGfEr1x9B4bRcTH4lpkTkpdh0T/J+qjbQ= +golang.org/x/sync v0.6.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk= +golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.16.0 h1:xWw16ngr6ZMtmxDyKyIgsE93KNKz5HKmMa3b8ALHidU= +golang.org/x/sys v0.16.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= +golang.org/x/term v0.16.0 h1:m+B6fahuftsE9qjo0VWp2FW0mB3MTJvR0BaMQrq0pmE= +golang.org/x/term v0.16.0/go.mod h1:yn7UURbUtPyrVJPGPq404EukNFxcm/foM+bV/bfcDsY= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +google.golang.org/protobuf v1.32.0 h1:pPC6BG5ex8PDFnkbrGU3EixyhKcQ2aDuBS36lqK/C7I= +google.golang.org/protobuf v1.32.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= +gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=