feat/billing gate (#2021)

* billing functinoalities

Author: motatoes

backend/bootstrap/main.go

@@ -229,6 +229,9 @@ func Bootstrap(templates embed.FS, diggerController controllers.DiggerController
 		orgsApiGroup.GET("/settings/", controllers.GetOrgSettingsApi)
 		orgsApiGroup.PUT("/settings/", controllers.UpdateOrgSettingsApi)
 
+		billingApiGroup := apiGroup.Group("/billing")
+		billingApiGroup.GET("/", controllers.BillingStatusApi)
+		
 		reposApiGroup := apiGroup.Group("/repos")
 		reposApiGroup.GET("/", controllers.ListReposApi)
 		reposApiGroup.GET("/:repo_id/jobs", controllers.GetJobsForRepoApi)

backend/controllers/billing.go

@@ -0,0 +1,45 @@
+package controllers
+
+import (
+	"errors"
+	"github.com/diggerhq/digger/backend/middleware"
+	"github.com/diggerhq/digger/backend/models"
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+	"log/slog"
+	"net/http"
+)
+
+func BillingStatusApi(c *gin.Context) {
+	organisationId := c.GetString(middleware.ORGANISATION_ID_KEY)
+	organisationSource := c.GetString(middleware.ORGANISATION_SOURCE_KEY)
+
+	var org models.Organisation
+	err := models.DB.GormDB.Where("external_id = ? AND external_source = ?", organisationId, organisationSource).First(&org).Error
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			slog.Info("Organisation not found", "organisationId", organisationId, "source", organisationSource)
+			c.String(http.StatusNotFound, "Could not find organisation: "+organisationId)
+		} else {
+			slog.Error("Error fetching organisation", "organisationId", organisationId, "source", organisationSource, "error", err)
+			c.String(http.StatusInternalServerError, "Error fetching organisation")
+		}
+		return
+	}
+
+	monitoredProjectsCount, remainingFreeProjects, billableProjectsCount, err := models.DB.GetProjectsRemainingInFreePLan(org.ID)
+	if err != nil {
+		slog.Error("Error fetching remaining free projects", "error", err)
+		c.String(http.StatusInternalServerError, "Error fetching remaining free projects")
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{
+		"billing_plan":                   org.BillingPlan,
+		"billing_stripe_subscription_id": org.BillingStripeSubscriptionId,
+		"remaining_free_projects":        remainingFreeProjects,
+		"monitored_projects_count":       monitoredProjectsCount,
+		"billable_projects_count":        billableProjectsCount,
+		"monitored_projects_limit":       models.MaxFreePlanProjectsPerOrg,
+	})
+}

backend/controllers/orgs.go

@@ -42,6 +42,7 @@ func GetOrgSettingsApi(c *gin.Context) {
 		"drift_enabled":     org.DriftEnabled,
 		"drift_cron_tab":    org.DriftCronTab,
 		"drift_webhook_url": org.DriftWebhookUrl,
+		"billing_plan":      org.BillingPlan,
 	})
 }
 
@@ -62,9 +63,11 @@ func UpdateOrgSettingsApi(c *gin.Context) {
 		return
 	}
 	var reqBody struct {
-		DriftEnabled    bool   `json:"drift_enabled"`
-		DriftCronTab    string `json:"drift_cron_tab"`
-		DriftWebhookUrl string `json:"drift_webhook_url"`
+		DriftEnabled                *bool   `json:"drift_enabled,omitempty"`
+		DriftCronTab                *string `json:"drift_cron_tab,omitempty"`
+		DriftWebhookUrl             *string `json:"drift_webhook_url,omitempty"`
+		BillingPlan                 *string `json:"billing_plan,omitempty"`
+		BillingStripeSubscriptionId *string `json:"billing_stripe_subscription_id,omitempty"`
 	}
 	err = json.NewDecoder(c.Request.Body).Decode(&reqBody)
 	if err != nil {
@@ -73,9 +76,26 @@ func UpdateOrgSettingsApi(c *gin.Context) {
 		return
 	}
 
-	org.DriftEnabled = reqBody.DriftEnabled
-	org.DriftCronTab = reqBody.DriftCronTab
-	org.DriftWebhookUrl = reqBody.DriftWebhookUrl
+	if reqBody.DriftEnabled != nil {
+		org.DriftEnabled = *reqBody.DriftEnabled
+	}
+
+	if reqBody.DriftCronTab != nil {
+		org.DriftCronTab = *reqBody.DriftCronTab
+	}
+
+	if reqBody.DriftWebhookUrl != nil {
+		org.DriftWebhookUrl = *reqBody.DriftWebhookUrl
+	}
+
+	if reqBody.BillingPlan != nil {
+		org.BillingPlan = models.BillingPlan(*reqBody.BillingPlan)
+	}
+
+	if reqBody.BillingStripeSubscriptionId != nil {
+		org.BillingStripeSubscriptionId = *reqBody.BillingStripeSubscriptionId
+	}
+
 	err = models.DB.GormDB.Save(&org).Error
 	if err != nil {
 		slog.Error("Error saving organisation", "organisationId", organisationId, "source", organisationSource, "error", err)

backend/migrations/20250712014920.sql

@@ -0,0 +1,2 @@
+-- Modify "organisations" table
+ALTER TABLE "public"."organisations" ADD COLUMN "billing_plan" text NULL DEFAULT 'free', ADD COLUMN "billing_stripe_subscription_id" text NULL;

backend/migrations/atlas.sum

@@ -1,4 +1,4 @@
-h1:94F268ZEPf2Y9xBgyPcV2kVk0fb4+6flqUIRqoaO4cs=
+h1:lTHG1XO1wASaCOl6gEp635MxFxAzxBuzxTe8pOPqKQc=
 20231227132525.sql h1:43xn7XC0GoJsCnXIMczGXWis9d504FAWi4F1gViTIcw=
 20240115170600.sql h1:IW8fF/8vc40+eWqP/xDK+R4K9jHJ9QBSGO6rN9LtfSA=
 20240116123649.sql h1:R1JlUIgxxF6Cyob9HdtMqiKmx/BfnsctTl5rvOqssQw=
@@ -56,3 +56,4 @@ h1:94F268ZEPf2Y9xBgyPcV2kVk0fb4+6flqUIRqoaO4cs=
 20250711030148.sql h1:hN972A3rU9TN4bCUo1muJWwEqNQQyIPlmYgsePNH00w=
 20250711030248.sql h1:oKzrMdfE75UyMSx++OMrXOAi1AOLN6kuv8iBLq8+srs=
 20250711030323.sql h1:vN9g0H99CItCw9aG6tRo73py5qQ4iD6qew2Y66WcoBk=
+20250712014920.sql h1:tu6MmFyXwCEvXjhjMYze79Vs3+OPqv6te5IiJsjnd5k=

backend/models/orgs.go

@@ -6,14 +6,26 @@ import (
 	"gorm.io/gorm"
 )
 
+type BillingPlan string
+
+const (
+	BillingPlanFree    BillingPlan = "free"
+	BillingPlanPro     BillingPlan = "pro"
+	BillingPlanPremium BillingPlan = "unlimited" // custom plan with unlimited resources
+)
+
+const MaxFreePlanProjectsPerOrg uint = 3
+
 type Organisation struct {
 	gorm.Model
-	Name            string `gorm:"Index:idx_organisation"`
-	ExternalSource  string `gorm:"uniqueIndex:idx_external_source"`
-	ExternalId      string `gorm:"uniqueIndex:idx_external_source"`
-	DriftEnabled    bool   `gorm:"default:false"`
-	DriftWebhookUrl string
-	DriftCronTab    string `gorm:"default:'0 0 * * *'"`
+	Name                        string `gorm:"Index:idx_organisation"`
+	ExternalSource              string `gorm:"uniqueIndex:idx_external_source"`
+	ExternalId                  string `gorm:"uniqueIndex:idx_external_source"`
+	DriftEnabled                bool   `gorm:"default:false"`
+	DriftWebhookUrl             string
+	DriftCronTab                string      `gorm:"default:'0 0 * * *'"`
+	BillingPlan                 BillingPlan `gorm:"default:'free'"`
+	BillingStripeSubscriptionId string
 }
 
 type Repo struct {

backend/models/storage.go

@@ -5,6 +5,7 @@ import (
 	"errors"
 	"fmt"
 	"log/slog"
+	"math"
 	"net/http"
 	"time"
 
@@ -44,6 +45,19 @@ func (db *Database) GetProjectsFromContext(c *gin.Context, orgIdKey string) ([]P
 	return projects, true
 }
 
+func (db *Database) GetProjectsRemainingInFreePLan(orgId uint) (uint, uint, uint, error) {
+
+	var countOfMonitoredProjects int64
+	err := db.GormDB.Model(&Project{}).Where("organisation_id = ? AND drift_enabled = ?", orgId, true).Count(&countOfMonitoredProjects).Error
+	if err != nil {
+		slog.Error("Error fetching project count", "error", err)
+		return 0, 0, 0, err
+	}
+	remainingFreeProjects := uint(math.Max(0, float64(MaxFreePlanProjectsPerOrg)-float64(countOfMonitoredProjects)))
+	billableProjectsCount := uint(math.Max(0, float64(countOfMonitoredProjects)-float64(MaxFreePlanProjectsPerOrg)))
+	return uint(countOfMonitoredProjects), remainingFreeProjects, billableProjectsCount, nil
+}
+
 func (db *Database) GetReposFromContext(c *gin.Context, orgIdKey string) ([]Repo, bool) {
 	loggedInOrganisationId, exists := c.Get(orgIdKey)