Merge remote-tracking branch 'origin/main' into stevenmasley/404_vs_403

Emyrk · Emyrk · commit 9a4083804f1a · 2022-06-13T12:11:28.000-05:00
diff --git a/coderd/devtunnel/tunnel.go b/coderd/devtunnel/tunnel.go
@@ -51,7 +51,7 @@ type configExt struct {
 
 // NewWithConfig calls New with the given config. For documentation, see New.
 func NewWithConfig(ctx context.Context, logger slog.Logger, cfg Config) (*Tunnel, <-chan error, error) {
-	err := startUpdateRoutine(ctx, logger, cfg)
+	routineEnd, err := startUpdateRoutine(ctx, logger, cfg)
 	if err != nil {
 		return nil, nil, xerrors.Errorf("start update routine: %w", err)
 	}
@@ -101,6 +101,7 @@ allowed_ip=%s/128`,
 		case <-ctx.Done():
 			_ = wgListen.Close()
 			dev.Close()
+			<-routineEnd
 			close(ch)
 
 		case <-dev.Wait():
@@ -128,21 +129,24 @@ func New(ctx context.Context, logger slog.Logger) (*Tunnel, <-chan error, error)
 	return NewWithConfig(ctx, logger, cfg)
 }
 
-func startUpdateRoutine(ctx context.Context, logger slog.Logger, cfg Config) error {
+func startUpdateRoutine(ctx context.Context, logger slog.Logger, cfg Config) (<-chan struct{}, error) {
 	// Ensure we send the first config before spawning in the background.
 	_, err := sendConfigToServer(ctx, cfg)
 	if err != nil {
-		return xerrors.Errorf("send config to server: %w", err)
+		return nil, xerrors.Errorf("send config to server: %w", err)
 	}
 
+	endCh := make(chan struct{})
 	go func() {
+		defer close(endCh)
 		ticker := time.NewTicker(30 * time.Second)
 		defer ticker.Stop()
 
 		for {
 			select {
 			case <-ctx.Done():
-				break
+				return
+
 			case <-ticker.C:
 			}
 
@@ -152,19 +156,25 @@ func startUpdateRoutine(ctx context.Context, logger slog.Logger, cfg Config) err
 			}
 		}
 	}()
-	return nil
+	return endCh, nil
 }
 
-func sendConfigToServer(_ context.Context, cfg Config) (created bool, err error) {
+func sendConfigToServer(ctx context.Context, cfg Config) (created bool, err error) {
 	raw, err := json.Marshal(configExt(cfg))
 	if err != nil {
 		return false, xerrors.Errorf("marshal config: %w", err)
 	}
 
-	res, err := http.Post("https://"+EndpointHTTPS+"/tun", "application/json", bytes.NewReader(raw))
+	req, err := http.NewRequestWithContext(ctx, "POST", "https://"+EndpointHTTPS+"/tun", bytes.NewReader(raw))
+	if err != nil {
+		return false, xerrors.Errorf("new request: %w", err)
+	}
+
+	res, err := http.DefaultClient.Do(req)
 	if err != nil {
-		return false, xerrors.Errorf("send request: %w", err)
+		return false, xerrors.Errorf("do request: %w", err)
 	}
+
 	_, _ = io.Copy(io.Discard, res.Body)
 	_ = res.Body.Close()
 
@@ -177,7 +187,13 @@ func cfgPath() (string, error) {
 		return "", xerrors.Errorf("get user config dir: %w", err)
 	}
 
-	return filepath.Join(cfgDir, "coderv2", "devtunnel"), nil
+	cfgDir = filepath.Join(cfgDir, "coderv2")
+	err = os.MkdirAll(cfgDir, 0750)
+	if err != nil {
+		return "", xerrors.Errorf("mkdirall config dir %q: %w", cfgDir, err)
+	}
+
+	return filepath.Join(cfgDir, "devtunnel"), nil
 }
 
 func readOrGenerateConfig() (Config, error) {
diff --git a/coderd/devtunnel/tunnel_test.go b/coderd/devtunnel/tunnel_test.go
@@ -2,11 +2,13 @@ package devtunnel_test
 
 import (
 	"context"
+	"io"
 	"net"
 	"net/http"
 	"testing"
 	"time"
 
+	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 
 	"cdr.dev/slog/sloggers/slogtest"
@@ -47,17 +49,26 @@ func TestTunnel(t *testing.T) {
 	go server.Serve(tun.Listener)
 	defer tun.Listener.Close()
 
+	httpClient := &http.Client{
+		Timeout: 10 * time.Second,
+	}
+
 	require.Eventually(t, func() bool {
 		req, err := http.NewRequestWithContext(ctx, "GET", tun.URL, nil)
 		require.NoError(t, err)
 
-		res, err := http.DefaultClient.Do(req)
+		res, err := httpClient.Do(req)
 		require.NoError(t, err)
 		defer res.Body.Close()
+		_, _ = io.Copy(io.Discard, res.Body)
+
 		return res.StatusCode == http.StatusOK
 	}, time.Minute, time.Second)
 
+	httpClient.CloseIdleConnections()
+	assert.NoError(t, server.Close())
 	cancelTun()
+
 	select {
 	case <-errCh:
 	case <-time.After(10 * time.Second):
diff --git a/docs/README.md b/docs/README.md
@@ -1,5 +1,9 @@
 # Coder Documentation
 
+<p align="center">
+  <img src="images/hero-image.png">
+</p>
+
 ## Table of Contents
 
 - [About Coder](./about.md#about-coder)
diff --git a/provisioner/terraform/resources.go b/provisioner/terraform/resources.go
@@ -127,6 +127,10 @@ func ConvertResources(module *tfjson.StateModule, rawGraph string) ([]*proto.Res
 			}
 		}
 
+		if agentResource == nil {
+			continue
+		}
+
 		agents, exists := resourceAgents[agentResource.Label]
 		if !exists {
 			agents = make([]*proto.Agent, 0)

Original file line number	Diff line number	Diff line change
`@@ -127,6 +127,10 @@ func ConvertResources(module tfjson.StateModule, rawGraph string) ([]proto.Res`
`127`	`127`	`}`
`128`	`128`	`}`
`129`	`129`
	`130`	`+ if agentResource == nil {`
	`131`	`+ continue`
	`132`	`+ }`
	`133`	`+`
`130`	`134`	`agents, exists := resourceAgents[agentResource.Label]`
`131`	`135`	`if !exists {`
`132`	`136`	`agents = make([]*proto.Agent, 0)`