feat: subscription settings

2 files changed, 126 insertions, 24 deletions

diff --git a/internal/router/api-v1-general-page-body.go b/internal/router/api-v1-general-page-body.go
index 6f856d3..66f1234 100644
--- a/internal/router/api-v1-general-page-body.go
+++ b/internal/router/api-v1-general-page-body.go
@@ -95,33 +95,12 @@ func Api_V1_GeneralPage_Body(l map[string]*locale.LocaleConfig, langs []config.A
 				queryTags = append(queryTags, string(match[1]))
 			}
 
-			pages, err := b2Client.Scan(lang + "/")
+			tagsArray, err := getTags(b2Client, lang)
 			if err != nil {
-				slog.Warn("failed to scan pages via b2", slog.String("error", err.Error()))
-				return c.Status(fiber.ErrInternalServerError.Code).SendString(fmt.Sprintf("failed to scan pages via b2: %s", slog.String("error", err.Error())))
+				slog.Warn("failed to get the available tags", slog.String("error", err.Error()))
+				return c.Status(fiber.ErrInternalServerError.Code).SendString("failed to gather available tags")
 			}
 
-			tagsMap := make(map[string]int)
-			for _, page := range pages {
-				for _, tag := range page.Metadata.Tags {
-					tagsMap[tag]++
-				}
-			}
-			slog.Debug("enlist pages for catalogue", slog.Int("tag_count", len(tagsMap)), slog.Int("page_count", len(pages)), slog.String("path", c.Path()))
-
-			type Tag struct {
-				Name  string `json:"Name" yaml:"name"`
-				Count int    `json:"Count" yaml:"count"`
-			}
-
-			tagsArray := make([]Tag, 0, len(tagsMap))
-			for tag, count := range tagsMap {
-				tagsArray = append(tagsArray, Tag{tag, count})
-			}
-			slices.SortFunc(tagsArray, func(a Tag, b Tag) int {
-				return strings.Compare(a.Name, b.Name)
-			})
-
 			values["Tags"] = tagsArray
 			values["QuerySort"] = querySort
 			values["QueryTags"] = strings.Join(queryTags, ",")
@@ -135,8 +114,33 @@ func Api_V1_GeneralPage_Body(l map[string]*locale.LocaleConfig, langs []config.A
 			if err != nil {
 				slog.Error("get info from mailer about a client", slog.String("error", err.Error()))
 			}
+
+			tagsArray, err := getTags(b2Client, lang)
+			if err != nil {
+				slog.Warn("failed to get the available tags", slog.String("error", err.Error()))
+				return c.Status(fiber.ErrInternalServerError.Code).SendString("failed to gather available tags")
+			}
+
+			subscriptionType, tags, err := Mailer.GetSubscriptions(c.IP())
+			if err != nil {
+				slog.Warn("failed to get the user subscriptions", slog.String("error", err.Error()))
+				return c.Status(fiber.ErrInternalServerError.Code).SendString("failed to get the user subscriptions")
+			}
+
+			switch subscriptionType {
+			case mailer.None:
+				values["TagsPicked"] = "none"
+			case mailer.All:
+				values["TagsPicked"] = "all"
+			case mailer.Specific:
+				values["TagsPicked"] = "specific"
+			}
+
+			values["TagsPickedList"] = tags
+
 			values["Email"] = email
 			values["EmailCode"] = c.Query("email_code")
+			values["ExistingTags"] = tagsArray
 
 			additionalTemplates = append(additionalTemplates, "views/pages/user-page.html")
 		} else if len(pathParts) == 3 && pathParts[1] == "blog" {
@@ -204,3 +208,34 @@ func readBlogPost(md goldmark.Markdown, b2Client *b2.B2Client, sourceName string
 
 	return metadata, buf.String(), nil
 }
+
+type Tag struct {
+	Name  string `json:"Name" yaml:"name"`
+	Count int    `json:"Count" yaml:"count"`
+}
+
+func getTags(b2Client *b2.B2Client, lang string) (tags []Tag, err error) {
+	pages, err := b2Client.Scan(lang + "/")
+	if err != nil {
+		slog.Warn("failed to scan pages via b2", slog.String("error", err.Error()))
+		return nil, fmt.Errorf("failed to scan pages via b2: %w", err)
+	}
+
+	tagsMap := make(map[string]int)
+	for _, page := range pages {
+		for _, tag := range page.Metadata.Tags {
+			tagsMap[tag]++
+		}
+	}
+	slog.Debug("enlist pages for catalogue", slog.Int("tag_count", len(tagsMap)), slog.Int("page_count", len(pages)), slog.String("lang", lang))
+
+	tagsArray := make([]Tag, 0, len(tagsMap))
+	for tag, count := range tagsMap {
+		tagsArray = append(tagsArray, Tag{tag, count})
+	}
+	slices.SortFunc(tagsArray, func(a Tag, b Tag) int {
+		return strings.Compare(a.Name, b.Name)
+	})
+
+	return tagsArray, nil
+}
diff --git a/internal/router/api-v1-subs.go b/internal/router/api-v1-subs.go
new file mode 100644
index 0000000..8995ebe
--- /dev/null
+++ b/internal/router/api-v1-subs.go
@@ -0,0 +1,67 @@
+package router
+
+import (
+	"net/url"
+	"strings"
+
+	"github.com/SayaAndy/saya-today-web/internal/mailer"
+	"github.com/SayaAndy/saya-today-web/locale"
+	"github.com/gofiber/fiber/v2"
+)
+
+func init() {
+	assert(0, tm.Add("personal-page-status", "views/partials/personal-page-status.html"))
+}
+
+func Api_V1_Subs_Put(l map[string]*locale.LocaleConfig) func(c *fiber.Ctx) error {
+	return func(c *fiber.Ctx) error {
+		c.Set(fiber.HeaderContentType, fiber.MIMETextPlainCharsetUTF8)
+
+		id := c.IP()
+		lang := "en"
+
+		var referer, path string
+		var pathParts []string
+		var urlStruct *url.URL
+		var err error
+
+		referer = c.Get("Referer", "")
+		if referer == "" {
+			goto skipFetchingLang
+		}
+
+		urlStruct, err = url.ParseRequestURI(referer)
+		if err != nil {
+			goto skipFetchingLang
+		}
+
+		path = urlStruct.EscapedPath()
+		pathParts = strings.Split(strings.Trim(path, "/"), "/")
+		if len(pathParts) == 0 {
+			goto skipFetchingLang
+		}
+
+		lang = pathParts[0]
+
+	skipFetchingLang:
+		subscriptionType := c.FormValue("tags")
+		var subscriptionTypeEnum mailer.SubscriptionType
+		switch subscriptionType {
+		case "all":
+			subscriptionTypeEnum = mailer.All
+		case "none":
+			subscriptionTypeEnum = mailer.None
+		case "specific":
+			subscriptionTypeEnum = mailer.Specific
+		default:
+			return api_v1_email_sendStatusHtml(c, "subs-message", l, "Failed", fiber.StatusUnprocessableEntity, lang, l[lang].UserProfile.SubscribeInvalidType, map[string]string{})
+		}
+
+		specificTags := c.FormValue("tags_picked")
+		if err = Mailer.Subscribe(id, subscriptionTypeEnum, specificTags); err != nil {
+			return api_v1_email_sendStatusHtml(c, "subs-message", l, "Failed", fiber.StatusUnprocessableEntity, lang, l[lang].UserProfile.FailedToSubscribe, map[string]string{})
+		}
+
+		return api_v1_email_sendStatusHtml(c, "subs-message", l, "OK", fiber.StatusOK, lang, l[lang].UserProfile.SubscribedSuccessfully, map[string]string{})
+	}
+}