Add scaffolding

kylecarbs · kylecarbs · commit 3dc1f0088033 · 2022-10-17T17:53:41.000Z
diff --git a/cli/gitaskpass/gitaskpass.go b/cli/gitaskpass/gitaskpass.go
@@ -0,0 +1,70 @@
+package gitaskpass
+
+import (
+	"net/url"
+	"regexp"
+	"strings"
+
+	"golang.org/x/xerrors"
+)
+
+// https://github.com/microsoft/vscode/blob/328646ebc2f5016a1c67e0b23a0734bd598ec5a8/extensions/git/src/askpass-main.ts#L46
+var hostReplace = regexp.MustCompile(`^["']+|["':]+$`)
+
+// CheckCommand returns true if the command arguments and environment
+// match those when the GIT_ASKPASS command is invoked by git.
+func CheckCommand(args, env []string) bool {
+	if len(args) != 1 || (!strings.HasPrefix(args[0], "Username ") && !strings.HasPrefix(args[0], "Password ")) {
+		return false
+	}
+	for _, e := range env {
+		if strings.HasPrefix(e, "GIT_PREFIX=") {
+			return true
+		}
+	}
+	return false
+}
+
+// Parse returns the user and host from a git askpass prompt. For
+// example: "user1" and "https://github.com". Note that for HTTP
+// protocols, the URL will never contain a path.
+//
+// For details on how the prompt is formatted, see `credential_ask_one`:
+// https://github.com/git/git/blob/bbe21b64a08f89475d8a3818e20c111378daa621/credential.c#L173-L191
+func Parse(prompt string) (user string, host string, err error) {
+	parts := strings.Split(prompt, " ")
+	if len(parts) < 3 {
+		return "", "", xerrors.Errorf("askpass prompt must contain 3 words; got %d: %q", len(parts), prompt)
+	}
+
+	switch parts[0] {
+	case "Username", "Password":
+	default:
+		return "", "", xerrors.Errorf("unknown prompt type: %q", prompt)
+	}
+
+	host = parts[2]
+	host = hostReplace.ReplaceAllString(host, "")
+
+	// Validate the input URL to ensure it's in an expected format.
+	u, err := url.Parse(host)
+	if err != nil {
+		return "", "", xerrors.Errorf("parse host failed: %w", err)
+	}
+
+	switch u.Scheme {
+	case "http", "https":
+	default:
+		return "", "", xerrors.Errorf("unsupported scheme: %q", u.Scheme)
+	}
+
+	if u.Host == "" {
+		return "", "", xerrors.Errorf("host is empty")
+	}
+
+	user = u.User.Username()
+	u.User = nil
+	host = u.String()
+
+	return user, host, nil
+}
diff --git a/cli/gitaskpass/gitaskpass_test.go b/cli/gitaskpass/gitaskpass_test.go
@@ -0,0 +1,72 @@
+package gitaskpass_test
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/coder/coder/cli/gitaskpass"
+)
+
+func TestCheckCommand(t *testing.T) {
+	t.Parallel()
+	t.Run("Success", func(t *testing.T) {
+		t.Parallel()
+		valid := gitaskpass.CheckCommand([]string{"Username "}, []string{"GIT_PREFIX=/example"})
+		require.True(t, valid)
+	})
+	t.Run("Failure", func(t *testing.T) {
+		t.Parallel()
+		valid := gitaskpass.CheckCommand([]string{}, []string{})
+		require.False(t, valid)
+	})
+}
+
+func TestParse(t *testing.T) {
+	t.Parallel()
+	for _, tc := range []struct {
+		in       string
+		wantUser string
+		wantHost string
+	}{
+		{
+			in:       "Username for 'https://github.com': ",
+			wantUser: "",
+			wantHost: "https://github.com",
+		},
+		{
+			in:       "Username for 'https://enterprise.github.com': ",
+			wantUser: "",
+			wantHost: "https://enterprise.github.com",
+		},
+		{
+			in:       "Username for 'http://wow.io': ",
+			wantUser: "",
+			wantHost: "http://wow.io",
+		},
+		{
+			in:       "Password for 'https://myuser@github.com': ",
+			wantUser: "myuser",
+			wantHost: "https://github.com",
+		},
+		{
+			in:       "Password for 'https://myuser@enterprise.github.com': ",
+			wantUser: "myuser",
+			wantHost: "https://enterprise.github.com",
+		},
+		{
+			in:       "Password for 'http://myuser@wow.io': ",
+			wantUser: "myuser",
+			wantHost: "http://wow.io",
+		},
+	} {
+		tc := tc
+		t.Run(tc.in, func(t *testing.T) {
+			t.Parallel()
+			user, host, err := gitaskpass.Parse(tc.in)
+			require.NoError(t, err)
+			require.Equal(t, tc.wantUser, user)
+			require.Equal(t, tc.wantHost, host)
+		})
+	}
+}
diff --git a/coderd/database/dump.sql b/coderd/database/dump.sql
diff --git a/coderd/database/migrations/000062_gitprovider.down.sql b/coderd/database/migrations/000062_gitprovider.down.sql
@@ -0,0 +1 @@
+DROP TABLE git_provider_links;
diff --git a/coderd/database/migrations/000062_gitprovider.up.sql b/coderd/database/migrations/000062_gitprovider.up.sql
@@ -0,0 +1,9 @@
+CREATE TABLE IF NOT EXISTS git_provider_links (
+  user_id uuid NOT NULL,
+  url text NOT NULL,
+  created_at timestamptz NOT NULL,
+  updated_at timestamptz NOT NULL,
+  oauth_access_token text NOT NULL,
+  oauth_refresh_token text NOT NULL,
+  oauth_expiry text NOT NULL
+);
diff --git a/coderd/database/models.go b/coderd/database/models.go
