Integrate tenant authentication

containers/docker-compose.yaml

@@ -36,6 +36,7 @@ services:
     init: true
     depends_on:
       - trtl
+      - quarterdeck
     ports:
       - 8080:8080
     environment:
@@ -45,6 +46,10 @@ services:
       - TENANT_LOG_LEVEL=info
       - TENANT_CONSOLE_LOG=true
       - TENANT_ALLOW_ORIGINS=http://localhost:3000
+      - TENANT_AUTH_KEYS_URL=http://quarterdeck:8088/.well-known/jwks.json
+      - TENANT_AUTH_AUDIENCE=http://localhost:3000
+      - TENANT_AUTH_ISSUER=http://localhost:8088
+      - TENANT_AUTH_COOKIE_DOMAIN=localhost
       - TENANT_DATABASE_URL=trtl://trtl:4436
       - TENANT_DATABASE_INSECURE=true
       - TENANT_SENDGRID_API_KEY
@@ -142,4 +147,4 @@ services:
     ports:
       - 9080:3000
     volumes:
-      - ./monitor/grafana:/var/lib/grafana
+      - ./monitor/grafana:/var/lib/grafana

pkg/tenant/api/v1/client.go

@@ -50,6 +50,7 @@ func New(endpoint string, opts ...ClientOption) (_ TenantClient, err error) {
 type APIv1 struct {
 	endpoint *url.URL
 	client   *http.Client
+	creds    string
 }
 
 // Ensures the APIv1 implements the TenantClient interface
@@ -795,7 +796,10 @@ func (s *APIv1) NewRequest(ctx context.Context, method, path string, data interf
 	req.Header.Add("Accept-Encoding", acceptEncode)
 	req.Header.Add("Content-Type", contentType)
 
-	// TODO: add authentication if it is available
+	// Add authentication if it is available
+	if s.creds != "" {
+		req.Header.Add("Authorization", "Bearer "+s.creds)
+	}
 
 	// Adds CSRF protection if it is available
 	if s.client.Jar != nil {
@@ -846,3 +850,54 @@ func (s *APIv1) Do(req *http.Request, data interface{}, checkStatus bool) (rep *
 
 	return rep, nil
 }
+
+// SetCredentials is a helper function for external users to override credentials at
+// runtime by directly passing in the token, which is useful for testing.
+// TODO: Pass in a credentials interface instead of the token string.
+func (c *APIv1) SetCredentials(token string) {
+	c.creds = token
+}
+
+// SetCSRFProtect is a helper function to set CSRF cookies on the client. This is not
+// possible in a browser because of the HttpOnly flag. This method should only be used
+// for testing purposes and an error is returned if the URL is not localhost. For live
+// clients - the server should set these cookies. If protect is false, then the cookies
+// are removed from the client by setting the cookies to an empty slice.
+func (c *APIv1) SetCSRFProtect(protect bool) error {
+	if c.client.Jar == nil {
+		return errors.New("client does not have a cookie jar, cannot set cookies")
+	}
+
+	if c.endpoint.Hostname() != "127.0.0.1" && c.endpoint.Hostname() != "localhost" {
+		return fmt.Errorf("csrf protect is for local testing only, cannot set cookies for %s", c.endpoint.Hostname())
+	}
+
+	// The URL for the cookies
+	u := c.endpoint.ResolveReference(&url.URL{Path: "/"})
+
+	var cookies []*http.Cookie
+	if protect {
+		cookies = []*http.Cookie{
+			{
+				Name:     "csrf_token",
+				Value:    "testingcsrftoken",
+				Expires:  time.Now().Add(10 * time.Minute),
+				HttpOnly: false,
+			},
+			{
+				Name:     "csrf_reference_token",
+				Value:    "testingcsrftoken",
+				Expires:  time.Now().Add(10 * time.Minute),
+				HttpOnly: true,
+			},
+		}
+	} else {
+		cookies = c.client.Jar.Cookies(u)
+		for _, cookie := range cookies {
+			cookie.MaxAge = -1
+		}
+	}
+
+	c.client.Jar.SetCookies(u, cookies)
+	return nil
+}

pkg/tenant/config/config.go

@@ -24,12 +24,21 @@ type Config struct {
 	LogLevel     logger.LevelDecoder `split_words:"true" default:"info"`                  // $TENANT_LOG_LEVEL
 	ConsoleLog   bool                `split_words:"true" default:"false"`                 // $TENANT_CONSOLE_LOG
 	AllowOrigins []string            `split_words:"true" default:"http://localhost:3000"` // $TENANT_ALLOW_ORIGINS
+	Auth         AuthConfig          `split_words:"true"`
 	Database     DatabaseConfig      `split_words:"true"`
 	SendGrid     SendGridConfig      `split_words:"false"`
 	Sentry       sentry.Config
 	processed    bool // set when the config is properly procesesed from the environment
 }
 
+// Configures the authentication and authorization for the Tenant API.
+type AuthConfig struct {
+	KeysURL      string `split_words:"true"`
+	Audience     string `split_words:"true"`
+	Issuer       string `split_words:"true"`
+	CookieDomain string `split_words:"true"`
+}
+
 // Configures the connection to trtl for replicated data storage.
 type DatabaseConfig struct {
 	URL      string `default:"trtl://localhost:4436"`
@@ -102,6 +111,10 @@ func (c Config) Validate() (err error) {
 		return err
 	}
 
+	if err = c.Auth.Validate(); err != nil {
+		return err
+	}
+
 	return nil
 }
 
@@ -117,6 +130,11 @@ func (c Config) AllowAllOrigins() bool {
 	return false
 }
 
+func (c AuthConfig) Validate() error {
+	// TODO: Validate the keys URL if provided
+	return nil
+}
+
 // If not insecure, the cert and pool paths are required.
 func (c DatabaseConfig) Validate() (err error) {
 	// If in testing mode, configuration is valid

pkg/tenant/config/config_test.go

@@ -20,6 +20,10 @@ var testEnv = map[string]string{
 	"TENANT_LOG_LEVEL":                "error",
 	"TENANT_CONSOLE_LOG":              "true",
 	"TENANT_ALLOW_ORIGINS":            "http://localhost:8888,http://localhost:8080",
+	"TENANT_AUTH_KEYS_URL":            "http://localhost:8080/.well-known/jwks.json",
+	"TENANT_AUTH_AUDIENCE":            "audience",
+	"TENANT_AUTH_ISSUER":              "issuer",
+	"TENANT_AUTH_COOKIE_DOMAIN":       "localhost",
 	"TENANT_DATABASE_URL":             "trtl://localhost:4436",
 	"TENANT_DATABASE_INSECURE":        "true",
 	"TENANT_DATABASE_CERT_PATH":       "path/to/certs.pem",
@@ -64,6 +68,10 @@ func TestConfig(t *testing.T) {
 	require.Equal(t, zerolog.ErrorLevel, conf.GetLogLevel())
 	require.True(t, conf.ConsoleLog)
 	require.Len(t, conf.AllowOrigins, 2)
+	require.Equal(t, testEnv["TENANT_AUTH_KEYS_URL"], conf.Auth.KeysURL)
+	require.Equal(t, testEnv["TENANT_AUTH_AUDIENCE"], conf.Auth.Audience)
+	require.Equal(t, testEnv["TENANT_AUTH_ISSUER"], conf.Auth.Issuer)
+	require.Equal(t, testEnv["TENANT_AUTH_COOKIE_DOMAIN"], conf.Auth.CookieDomain)
 	require.Equal(t, testEnv["TENANT_DATABASE_URL"], conf.Database.URL)
 	require.True(t, conf.Database.Insecure)
 	require.Equal(t, testEnv["TENANT_DATABASE_CERT_PATH"], conf.Database.CertPath)
@@ -148,6 +156,10 @@ func TestAllowAllOrigins(t *testing.T) {
 	require.True(t, conf.AllowAllOrigins(), "expected allow all origins to be true when * is set")
 }
 
+func TestAuth(t *testing.T) {
+	// TODO: test AuthConfig validation
+}
+
 func TestDatabase(t *testing.T) {
 	// TODO: test DatabaseConfig validation
 }