Feat/BRIDGE-119 - Feature flag support

client_features.go

@@ -0,0 +1,19 @@
+package proton
+
+import (
+	"context"
+
+	"github.com/go-resty/resty/v2"
+)
+
+func (c *Client) GetFeatures(ctx context.Context) (FeatureFlagResult, error) {
+	res := FeatureFlagResult{}
+
+	if err := c.do(ctx, func(r *resty.Request) (*resty.Response, error) {
+		return r.SetResult(&res).Get("/feature/v2/frontend")
+	}); err != nil {
+		return FeatureFlagResult{}, err
+	}
+
+	return res, nil
+}

manager_features.go

@@ -0,0 +1,26 @@
+package proton
+
+import (
+	"context"
+)
+
+type FeatureFlagResult struct {
+	Code    int             `json:"Code"`
+	Toggles []FeatureToggle `json:"toggles"`
+}
+
+type FeatureToggle struct {
+	Name    string `json:"name"`
+	Enabled bool   `json:"enabled"`
+}
+
+func (m *Manager) GetFeatures(ctx context.Context) (FeatureFlagResult, error) {
+	responseData := FeatureFlagResult{}
+
+	_, err := m.r(ctx).SetResult(&responseData).Get("/feature/v2/frontend")
+	if err != nil {
+		return FeatureFlagResult{}, err
+	}
+
+	return responseData, nil
+}

server/backend/api.go

@@ -1294,6 +1294,27 @@ func (b *Backend) GenerateContactID(userID string) (string, error) {
 	})
 }
 
+func (b *Backend) GetFeatureFlags() []proton.FeatureToggle {
+	return readBackendRet(b, func(b *unsafeBackend) []proton.FeatureToggle {
+		return b.featureFlags
+	})
+}
+
+func (b *Backend) PushFeatureFlag(flagName string) {
+	writeBackend(b, func(b *unsafeBackend) {
+		b.featureFlags = append(b.featureFlags, proton.FeatureToggle{
+			Name:    flagName,
+			Enabled: true,
+		})
+	})
+}
+
+func (b *Backend) DeleteFeatureFlags() {
+	writeBackend(b, func(b *unsafeBackend) {
+		b.featureFlags = []proton.FeatureToggle{}
+	})
+}
+
 func buildEvent(
 	updates []update,
 	addresses map[string]*address,

server/backend/backend.go

@@ -48,6 +48,8 @@ type unsafeBackend struct {
 	enableDedup bool
 
 	csTicket []string
+
+	featureFlags []proton.FeatureToggle
 }
 
 func readBackendRet[T any](b *Backend, f func(b *unsafeBackend) T) T {

server/features.go

@@ -0,0 +1,17 @@
+package server
+
+import (
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+)
+
+func (s *Server) handleGetFeatures() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		ff := s.b.GetFeatureFlags()
+		c.JSON(http.StatusOK, gin.H{
+			"Code":    1000,
+			"Toggles": ff,
+		})
+	}
+}

server/router.go

@@ -26,6 +26,9 @@ func initRouter(s *Server) {
 		s.applyRateLimit(),
 	)
 
+	// Feature flag route. Needs to be updated when user specific feature flags are implemented
+	s.r.GET("/feature/v2/frontend", s.handleGetFeatures())
+
 	if core := s.r.Group("/core/v4"); core != nil {
 		// Domains routes don't need authentication.
 		if domains := core.Group("/domains"); domains != nil {

server/server.go

@@ -252,3 +252,11 @@ func (s *Server) Close() {
 	s.proxyTransport.CloseIdleConnections()
 	s.s.Close()
 }
+
+func (s *Server) PushFeatureFlag(flagName string) {
+	s.b.PushFeatureFlag(flagName)
+}
+
+func (s *Server) DeleteFeatureFlags() {
+	s.b.DeleteFeatureFlags()
+}