Monetization (#124)

* Integrate LemonSqueezy for payment processing (#113)

* handle lemonsqueezy subscriptions and create entitlements from them

* update existing entitlements

* add premium page and start implementing endpoints

* add pricing list to premium page

* fix build errors

* move plan info to backend and more refactoring

* implement subscription management

* store plan id in entitlements instead of lemonsqueezy ids

* [WIP] Entitlements (#123)

* implement mvp entitlement feature computation

* centralize feature computation

* enforce monthly credit usage limit

* enforce guild count limit

* minor

* minor

* Collaborators (#121)

* collaborators MVP

* fix rebase

* enforce max collaborators from entitlements

* minor

* minor

* Hand out roles to subscribers (#126)

* update json example

* hand out roles to premium subscribers

* run plan manager

* sqlc generate

Author: merlinfuchs

go.work.sum

@@ -139,6 +139,7 @@ github.com/jarcoal/httpmock v1.0.4/go.mod h1:ATjnClrvW/3tijVmpL/va5Z3aAyGvqU3gCT
 github.com/jedib0t/go-pretty v4.3.0+incompatible/go.mod h1:XemHduiw8R651AF9Pt4FwCTKeG3oo7hrHJAoznj9nag=
 github.com/jmespath/go-jmespath v0.4.0/go.mod h1:T8mJZnbsbmF+m6zOOFylbeCJqk5+pHWvzYPziyZiYoo=
 github.com/jmoiron/sqlx v1.3.5/go.mod h1:nRVWtLre0KfCLJvgxzCsLVMogSvQ1zNJtpYr2Ccp0mQ=
+github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
 github.com/josharian/intern v1.0.0/go.mod h1:5DoeVV0s6jJacbCEi61lwdGj/aVlrQvzHFFd8Hwg//Y=
 github.com/jpillora/backoff v1.0.0/go.mod h1:J/6gKK9jxlEcS3zixgDgUAsiuZ7yrSoa/FX5e0EB2j4=
 github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=

kite-service/cmd/server.go

@@ -11,10 +11,12 @@ import (
 	"github.com/kitecloud/kite/kite-service/internal/core/engine"
 	"github.com/kitecloud/kite/kite-service/internal/core/event"
 	"github.com/kitecloud/kite/kite-service/internal/core/gateway"
+	"github.com/kitecloud/kite/kite-service/internal/core/plan"
 	"github.com/kitecloud/kite/kite-service/internal/core/usage"
 	"github.com/kitecloud/kite/kite-service/internal/db/postgres"
 	"github.com/kitecloud/kite/kite-service/internal/db/s3"
 	"github.com/kitecloud/kite/kite-service/internal/logging"
+	"github.com/kitecloud/kite/kite-service/internal/model"
 	"github.com/sashabaranov/go-openai"
 	"github.com/urfave/cli/v2"
 )
@@ -88,10 +90,21 @@ func serverStartCMD(c *cli.Context) error {
 
 	handler := event.NewEventHandlerWrapper(engine, pg)
 
-	gateway := gateway.NewGatewayManager(pg, pg, handler)
+	billingPlans := make([]model.Plan, len(cfg.Billing.Plans))
+	for i, plan := range cfg.Billing.Plans {
+		billingPlans[i] = model.Plan(plan)
+	}
+
+	planManager := plan.NewPlanManager(pg, pg, billingPlans, plan.PlanManagerConfig{
+		DiscordBotToken: cfg.Discord.BotToken,
+		DiscordGuildID:  cfg.Discord.GuildID,
+	})
+	planManager.Run(ctx)
+
+	gateway := gateway.NewGatewayManager(pg, pg, planManager, handler)
 	gateway.Run(ctx)
 
-	usage := usage.NewUsageManager(pg, pg, cfg.UserLimits.CreditsPerMonth)
+	usage := usage.NewUsageManager(pg, pg, planManager)
 	usage.Run(ctx)
 
 	apiServer := api.NewAPIServer(api.APIServerConfig{
@@ -108,9 +121,15 @@ func serverStartCMD(c *cli.Context) error {
 			MaxMessagesPerApp:       cfg.UserLimits.MaxMessagesPerApp,
 			MaxEventListenersPerApp: cfg.UserLimits.MaxEventListenersPerApp,
 			MaxAssetSize:            cfg.UserLimits.MaxAssetSize,
-			CreditsPerMonth:         cfg.UserLimits.CreditsPerMonth,
 		},
-	}, pg, pg, pg, pg, pg, pg, pg, pg, pg, pg, pg, assetStore, gateway)
+		Billing: api.BillingConfig{
+			LemonSqueezyAPIKey:        cfg.Billing.LemonSqueezyAPIKey,
+			LemonSqueezySigningSecret: cfg.Billing.LemonSqueezySigningSecret,
+			LemonSqueezyStoreID:       cfg.Billing.LemonSqueezyStoreID,
+			TestMode:                  cfg.Billing.TestMode,
+			Plans:                     cfg.Billing.Plans,
+		},
+	}, pg, pg, pg, pg, pg, pg, pg, pg, pg, pg, pg, pg, pg, assetStore, gateway, planManager)
 	address := fmt.Sprintf("%s:%d", cfg.API.Host, cfg.API.Port)
 	if err := apiServer.Serve(ctx, address); err != nil {
 		slog.With("error", err).Error("Failed to start API server")

kite-service/go.mod

@@ -37,6 +37,7 @@ require (
 )
 
 require (
+	github.com/NdoleStudio/lemonsqueezy-go v1.2.4 // indirect
 	github.com/asaskevich/govalidator v0.0.0-20230301143203-a9d515a09cc2 // indirect
 	github.com/beorn7/perks v1.0.1 // indirect
 	github.com/cespare/xxhash/v2 v2.3.0 // indirect

kite-service/go.sum

@@ -2,6 +2,8 @@ github.com/Azure/go-ansiterm v0.0.0-20230124172434-306776ec8161 h1:L/gRVlceqvL25
 github.com/Azure/go-ansiterm v0.0.0-20230124172434-306776ec8161/go.mod h1:xomTg63KZ2rFqZQzSB4Vz2SUXa1BpHTVz9L5PTmPC4E=
 github.com/Microsoft/go-winio v0.6.1 h1:9/kr64B9VUZrLm5YYwbGtUJnMgqWVOdUAXu6Migciow=
 github.com/Microsoft/go-winio v0.6.1/go.mod h1:LRdKpFKfdobln8UmuiYcKPot9D2v6svN5+sAH+4kjUM=
+github.com/NdoleStudio/lemonsqueezy-go v1.2.4 h1:BhWlCUH+DIPfSn4g/V7f2nFkMCQuzno9DXKZ7YDrXXA=
+github.com/NdoleStudio/lemonsqueezy-go v1.2.4/go.mod h1:2uZlWgn9sbNxOx3JQWLlPrDOC6NT/wmSTOgL3U/fMMw=
 github.com/asaskevich/govalidator v0.0.0-20200108200545-475eaeb16496/go.mod h1:oGkLhpf+kjZl6xBf758TQhh5XrAeiJv/7FRz/2spLIg=
 github.com/asaskevich/govalidator v0.0.0-20230301143203-a9d515a09cc2 h1:DklsrG3dyBCFEj5IhUbnKptjxatkF07cF2ak3yi77so=
 github.com/asaskevich/govalidator v0.0.0-20230301143203-a9d515a09cc2/go.mod h1:WaHUgvxTVq04UNunO+XhnAqY/wQc+bxr74GqbsZ/Jqw=

kite-service/internal/api/access/middleware.go

@@ -4,6 +4,7 @@ import (
 	"errors"
 
 	"github.com/kitecloud/kite/kite-service/internal/api/handler"
+	"github.com/kitecloud/kite/kite-service/internal/model"
 	"github.com/kitecloud/kite/kite-service/internal/store"
 )
 
@@ -20,7 +21,17 @@ func (m *AccessManager) AppAccess(next handler.HandlerFunc) handler.HandlerFunc
 		}
 
 		if app.OwnerUserID != c.Session.UserID {
-			return handler.ErrForbidden("missing_access", "Access to app missing")
+			collaborator, err := m.appStore.Collaborator(c.Context(), appID, c.Session.UserID)
+			if err != nil {
+				if errors.Is(err, store.ErrNotFound) {
+					return handler.ErrForbidden("missing_access", "Access to app missing")
+				}
+				return err
+			}
+
+			c.UserAppRole = collaborator.Role
+		} else {
+			c.UserAppRole = model.AppCollaboratorRoleOwner
 		}
 
 		c.App = app

kite-service/internal/api/handler/app/collaborators.go

@@ -0,0 +1,97 @@
+package app
+
+import (
+	"errors"
+	"fmt"
+	"time"
+
+	"github.com/kitecloud/kite/kite-service/internal/api/handler"
+	"github.com/kitecloud/kite/kite-service/internal/api/wire"
+	"github.com/kitecloud/kite/kite-service/internal/model"
+	"github.com/kitecloud/kite/kite-service/internal/store"
+)
+
+func (h *AppHandler) HandleAppCollaboratorsList(c *handler.Context) (*wire.AppCollaboratorListResponse, error) {
+	collaborators, err := h.appStore.CollaboratorsByApp(c.Context(), c.App.ID)
+	if err != nil {
+		return nil, err
+	}
+
+	ownerUser, err := h.userStore.User(c.Context(), c.App.OwnerUserID)
+	if err != nil {
+		return nil, err
+	}
+
+	res := make(wire.AppCollaboratorListResponse, len(collaborators)+1)
+	res[0] = &wire.AppCollaborator{
+		User:      *wire.UserToWire(ownerUser, true),
+		Role:      string(model.AppCollaboratorRoleOwner),
+		CreatedAt: c.App.CreatedAt,
+		UpdatedAt: c.App.CreatedAt,
+	}
+
+	for i, collaborator := range collaborators {
+		res[i+1] = wire.CollaboratorToWire(collaborator)
+	}
+
+	return &res, nil
+}
+
+func (h *AppHandler) HandleAppCollaboratorCreate(c *handler.Context, req wire.AppCollaboratorCreateRequest) (*wire.AppCollaboratorCreateResponse, error) {
+	if !c.UserAppRole.CanManageCollaborators() {
+		return nil, handler.ErrForbidden("missing_permissions", "You don't have permissions to add collaborators to this app")
+	}
+
+	features := h.planManager.AppFeatures(c.Context(), c.App.ID)
+	if features.MaxCollaborators != 0 {
+		collaboratorCount, err := h.appStore.CountCollaboratorsByApp(c.Context(), c.App.ID)
+		if err != nil {
+			return nil, fmt.Errorf("failed to count collaborators: %w", err)
+		}
+
+		// We count the owner as a collaborator
+		if (collaboratorCount + 1) >= features.MaxCollaborators {
+			return nil, handler.ErrBadRequest("resource_limit", fmt.Sprintf("maximum number of collaborators (%d) reached", features.MaxCollaborators))
+		}
+	}
+
+	user, err := h.userStore.UserByDiscordID(c.Context(), req.DiscordUserID)
+	if err != nil {
+		if errors.Is(err, store.ErrNotFound) {
+			return nil, handler.ErrNotFound("unknown_user", "User not found")
+		}
+		return nil, err
+	}
+
+	if user.ID == c.App.OwnerUserID {
+		return nil, handler.ErrBadRequest("cannot_add_owner", "Cannot add owner as collaborator")
+	}
+
+	collaborator, err := h.appStore.CreateCollaborator(c.Context(), &model.AppCollaborator{
+		AppID:     c.App.ID,
+		UserID:    user.ID,
+		Role:      model.AppCollaboratorRole(req.Role),
+		CreatedAt: time.Now().UTC(),
+		UpdatedAt: time.Now().UTC(),
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	return wire.CollaboratorToWire(collaborator), nil
+}
+
+func (h *AppHandler) HandleAppCollaboratorDelete(c *handler.Context) (*wire.AppCollaboratorDeleteResponse, error) {
+	if !c.UserAppRole.CanManageCollaborators() {
+		return nil, handler.ErrForbidden("missing_permissions", "You don't have permissions to delete collaborators from this app")
+	}
+
+	userID := c.Param("userID")
+
+	err := h.appStore.DeleteCollaborator(c.Context(), c.App.ID, userID)
+	if err != nil {
+		return nil, err
+	}
+
+	return &wire.AppCollaboratorDeleteResponse{}, nil
+}

kite-service/internal/api/handler/app/handler.go

@@ -7,6 +7,7 @@ import (
 
 	"github.com/kitecloud/kite/kite-service/internal/api/handler"
 	"github.com/kitecloud/kite/kite-service/internal/api/wire"
+	"github.com/kitecloud/kite/kite-service/internal/core/plan"
 	"github.com/kitecloud/kite/kite-service/internal/model"
 	"github.com/kitecloud/kite/kite-service/internal/store"
 	"github.com/kitecloud/kite/kite-service/internal/util"
@@ -15,12 +16,21 @@ import (
 
 type AppHandler struct {
 	appStore       store.AppStore
+	userStore      store.UserStore
+	planManager    *plan.PlanManager
 	maxAppsPerUser int
 }
 
-func NewAppHandler(appStore store.AppStore, maxAppsPerUser int) *AppHandler {
+func NewAppHandler(
+	appStore store.AppStore,
+	userStore store.UserStore,
+	planManager *plan.PlanManager,
+	maxAppsPerUser int,
+) *AppHandler {
 	return &AppHandler{
 		appStore:       appStore,
+		userStore:      userStore,
+		planManager:    planManager,
 		maxAppsPerUser: maxAppsPerUser,
 	}
 }
@@ -196,6 +206,10 @@ func (h *AppHandler) HandleAppTokenUpdate(c *handler.Context, req wire.AppTokenU
 }
 
 func (h *AppHandler) HandleAppDelete(c *handler.Context) (*wire.AppDeleteResponse, error) {
+	if !c.UserAppRole.CanDeleteApp() {
+		return nil, handler.ErrForbidden("missing_permissions", "You don't have permissions to delete this app")
+	}
+
 	if err := h.appStore.DeleteApp(c.Context(), c.App.ID); err != nil {
 		return nil, fmt.Errorf("failed to delete app: %w", err)
 	}

kite-service/internal/api/handler/billing/checkout.go

@@ -0,0 +1,65 @@
+package billing
+
+import (
+	"fmt"
+	"strconv"
+	"time"
+
+	"github.com/NdoleStudio/lemonsqueezy-go"
+	"github.com/kitecloud/kite/kite-service/internal/api/handler"
+	"github.com/kitecloud/kite/kite-service/internal/api/wire"
+)
+
+func (h *BillingHandler) HandleAppCheckout(c *handler.Context, req wire.BillingCheckoutRequest) (*wire.BillingCheckoutResponse, error) {
+	user, err := h.userStore.User(c.Context(), c.Session.UserID)
+	if err != nil {
+		return nil, fmt.Errorf("failed to get user: %w", err)
+	}
+
+	redirectURL := fmt.Sprintf("%s/apps/%s/premium", h.config.AppPublicBaseURL, c.App.ID)
+
+	variantID, err := strconv.Atoi(req.LemonSqueezyVariantID)
+	if err != nil {
+		return nil, fmt.Errorf("failed to convert variant ID to int: %w", err)
+	}
+
+	storeID, err := strconv.Atoi(h.config.LemonSqueezyStoreID)
+	if err != nil {
+		return nil, fmt.Errorf("failed to convert store ID to int: %w", err)
+	}
+
+	res, _, err := h.client.Checkouts.Create(
+		c.Context(),
+		storeID,
+		variantID,
+		&lemonsqueezy.CheckoutCreateAttributes{
+			TestMode: ptr(h.config.TestMode),
+			CheckoutOptions: lemonsqueezy.CheckoutCreateOptions{
+				Embed: ptr(true),
+			},
+			CheckoutData: lemonsqueezy.CheckoutCreateData{
+				Name:  user.DisplayName,
+				Email: user.Email,
+				Custom: map[string]any{
+					"user_id": c.Session.UserID,
+					"app_id":  c.App.ID,
+				},
+			},
+			ProductOptions: lemonsqueezy.CheckoutCreateProductOptions{
+				RedirectURL: redirectURL,
+			},
+			ExpiresAt: ptr(time.Now().UTC().Add(time.Hour).Format(time.RFC3339)),
+		},
+	)
+	if err != nil {
+		return nil, fmt.Errorf("failed to create checkout: %w", err)
+	}
+
+	return &wire.BillingCheckoutResponse{
+		URL: res.Data.Attributes.URL,
+	}, nil
+}
+
+func ptr[T any](v T) *T {
+	return &v
+}

kite-service/internal/api/handler/billing/feature.go

@@ -0,0 +1,20 @@
+package billing
+
+import (
+	"github.com/kitecloud/kite/kite-service/internal/api/handler"
+	"github.com/kitecloud/kite/kite-service/internal/api/wire"
+)
+
+func (h *BillingHandler) HandleFeaturesGet(c *handler.Context) (*wire.FeaturesGetResponse, error) {
+	features := h.planManager.AppFeatures(c.Context(), c.App.ID)
+
+	res := wire.Features(features)
+	return &res, nil
+}
+
+func max(a, b int) int {
+	if a > b {
+		return a
+	}
+	return b
+}

kite-service/internal/api/handler/billing/handler.go

@@ -0,0 +1,48 @@
+package billing
+
+import (
+	"github.com/NdoleStudio/lemonsqueezy-go"
+	"github.com/kitecloud/kite/kite-service/internal/core/plan"
+	"github.com/kitecloud/kite/kite-service/internal/store"
+)
+
+type BillingHandlerConfig struct {
+	LemonSqueezyAPIKey        string
+	LemonSqueezySigningSecret string
+	LemonSqueezyStoreID       string
+	TestMode                  bool
+	AppPublicBaseURL          string
+}
+
+type BillingHandler struct {
+	config            BillingHandlerConfig
+	userStore         store.UserStore
+	subscriptionStore store.SubscriptionStore
+	entitlementStore  store.EntitlementStore
+	planManager       *plan.PlanManager
+
+	client *lemonsqueezy.Client
+}
+
+func NewBillingHandler(
+	config BillingHandlerConfig,
+	userStore store.UserStore,
+	subscriptionStore store.SubscriptionStore,
+	entitlementStore store.EntitlementStore,
+	planManager *plan.PlanManager,
+) *BillingHandler {
+	client := lemonsqueezy.New(
+		lemonsqueezy.WithAPIKey(config.LemonSqueezyAPIKey),
+		lemonsqueezy.WithSigningSecret(config.LemonSqueezySigningSecret),
+	)
+
+	return &BillingHandler{
+		config:            config,
+		userStore:         userStore,
+		subscriptionStore: subscriptionStore,
+		entitlementStore:  entitlementStore,
+		planManager:       planManager,
+
+		client: client,
+	}
+}