minor refactoring

stv0g · Mar 24, 2022 · 00876ae · 00876ae
1 parent 51fd47f
commit 00876ae
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/pkg/config/config.go b/pkg/config/config.go
@@ -119,7 +119,7 @@ func (c *S3Config) GetObjectURL(key string) *url.URL {
 	return u
 }
 
-func (c *expiration) Supported(cls string) bool {
+func (c *expiration) HasClass(cls string) bool {
 	for _, c := range c.Classes {
 		if c.Tag == cls {
 			return true

diff --git a/pkg/handlers/config.go b/pkg/handlers/config.go
@@ -5,8 +5,14 @@ import (
 	"github.com/stv0g/gose/pkg/config"
 )
 
+type featureResponse struct {
+	Shortener  bool `json:"shortener"`
+	NotifyMail bool `json:"notify_mail"`
+}
+
 type configResponse struct {
 	ExpirationClasses []config.ExpirationClass `json:"expiration_classes"`
+	Features          featureResponse          `json:"features"`
 }
 
 // HandleConfig returns runtime configuration to the frontend
@@ -15,5 +21,9 @@ func HandleConfig(c *gin.Context) {
 
 	c.JSON(200, &configResponse{
 		ExpirationClasses: cfg.S3.Expiration.Classes,
+		Features: featureResponse{
+			Shortener:  cfg.Shortener != nil,
+			NotifyMail: false,
+		},
 	})
 }
diff --git a/pkg/handlers/initiate.go b/pkg/handlers/initiate.go
@@ -90,7 +90,7 @@ func HandleInitiate(c *gin.Context) {
 	if req.Expiration == nil {
 		expiration = cfg.S3.Expiration.Default
 	} else {
-		if !cfg.S3.Expiration.Supported(*req.Expiration) {
+		if !cfg.S3.Expiration.HasClass(*req.Expiration) {
 			c.JSON(http.StatusInternalServerError, gin.H{"error": "invalid expiration class"})
 			return
 		}