Add OSS license

Author: noah

internal/server/api/shared/middleware.go

@@ -34,12 +34,16 @@ func (m *Middleware) IsLicenseExpired() gin.HandlerFunc {
 			return
 		}
 
+		if lic.IsOSS() {
+			return
+		}
+
 		if lic.IsOverLimit() {
 			gb.AbortWithErrorResponse(c, http.StatusPaymentRequired, "The member count is over the limit.")
 			return
 		}
 
-		if !lic.IsTrial() && lic.IsExpired() {
+		if lic.IsStandard() && lic.IsExpired() {
 			now := time.Now()
 			if lic.ExpiredAt.Add(extraDuration).Before(now) {
 				gb.AbortWithErrorResponse(c, http.StatusPaymentRequired, "The license is expired.")

internal/server/api/shared/middleware_test.go

@@ -15,6 +15,33 @@ import (
 func TestMiddleware_IsLicenseExpired(t *testing.T) {
 	month := 30 * 24 * time.Hour
 
+	t.Run("Return 200 when the license is OSS.", func(t *testing.T) {
+		ctrl := gomock.NewController(t)
+		m := mock.NewMockInteractor(ctrl)
+
+		m.
+			EXPECT().
+			GetLicense(gomock.Any()).
+			Return(vo.NewOSSLicense(), nil)
+
+		gin.SetMode(gin.ReleaseMode)
+		router := gin.New()
+
+		lm := NewMiddleware(m)
+		router.GET("/repos", lm.IsLicenseExpired(), func(c *gin.Context) {
+			c.Status(http.StatusOK)
+		})
+
+		req, _ := http.NewRequest("GET", "/repos", nil)
+		w := httptest.NewRecorder()
+
+		router.ServeHTTP(w, req)
+
+		if w.Code != http.StatusOK {
+			t.Fatalf("IsLicenseExpired = %v, wanted %v", w.Code, http.StatusOK)
+		}
+	})
+
 	t.Run("Return 402 error when the count of member is over the limit.", func(t *testing.T) {
 		ctrl := gomock.NewController(t)
 		m := mock.NewMockInteractor(ctrl)

vo/license.go

@@ -7,7 +7,11 @@ const (
 )
 
 const (
-	LicenseKindTrial    LicenseKind = "trial"
+	// LicenseKindOSS is a license for the community edition.
+	LicenseKindOSS LicenseKind = "oss"
+	// LicenseKindTrial is a trial license of the enterprise edition.
+	LicenseKindTrial LicenseKind = "trial"
+	// LicenseKindStandard is a license of the enterprise edition.
 	LicenseKindStandard LicenseKind = "standard"
 )
 
@@ -28,6 +32,13 @@ type (
 	}
 )
 
+func NewOSSLicense() *License {
+	return &License{
+		Kind:        LicenseKindOSS,
+		MemberCount: -1,
+	}
+}
+
 func NewTrialLicense(cnt int) *License {
 	return &License{
 		Kind:        LicenseKindTrial,
@@ -45,14 +56,24 @@ func NewStandardLicense(cnt int, d *SigningData) *License {
 	}
 }
 
+func (l *License) IsOSS() bool {
+	return l.Kind == LicenseKindOSS
+}
+
 func (l *License) IsTrial() bool {
 	return l.Kind == LicenseKindTrial
 }
 
+func (l *License) IsStandard() bool {
+	return l.Kind == LicenseKindStandard
+}
+
+// IsOverLimit verify it is over the limit of the license.
 func (l *License) IsOverLimit() bool {
 	return l.MemberCount > l.MemberLimit
 }
 
+// IsExpired verify that the license is expired or not.
 func (l *License) IsExpired() bool {
 	return l.ExpiredAt.Before(time.Now())
 }