feat: subscription settings

3 files changed, 202 insertions, 24 deletions

diff --git a/internal/mailer/mailer.go b/internal/mailer/mailer.go
index 450b818..cd5eda8 100644
--- a/internal/mailer/mailer.go
+++ b/internal/mailer/mailer.go
@@ -43,6 +43,14 @@ type Mailer struct {
 	l map[string]*locale.LocaleConfig
 }
 
+type SubscriptionType int
+
+const (
+	All SubscriptionType = iota
+	None
+	Specific
+)
+
 func NewMailer(db *sql.DB, clientHost string, mailHost string, publicName string, mailAddress string, username string, password string, salt []byte, localization map[string]*locale.LocaleConfig) (*Mailer, error) {
 	verificationCodes, err := ristretto.NewCache(&ristretto.Config[uint64, string]{
 		NumCounters:            10000,
@@ -279,3 +287,71 @@ func (m *Mailer) Verify(verificationCodeEncoded string, lang string) error {
 	delete(m.lostMailMap, verificationSegments[0])
 	return nil
 }
+
+func (m *Mailer) GetSubscriptions(userId string) (subscriptionType SubscriptionType, tags []string, err error) {
+	tx, err := m.db.Begin()
+	if err != nil {
+		return None, nil, fmt.Errorf("failed to initialize transaction with db: %s", err)
+	}
+
+	hash := m.GetHash(userId)
+
+	var rows *sql.Rows
+	if rows, err = tx.Query(`SELECT tags FROM subscription_user_to_tags_table WHERE user_id=? LIMIT 1;`, hash); err != nil {
+		tx.Rollback()
+		return None, nil, fmt.Errorf("failed to query user-to-tags table in db for the user: %s", err)
+	}
+	defer tx.Commit()
+	defer rows.Close()
+
+	if !rows.Next() {
+		return None, nil, nil
+	}
+
+	tagsString := ""
+	if err = rows.Scan(&tagsString); err != nil {
+		return None, nil, fmt.Errorf("failed to scan the result from user-to-tags query: %s", err)
+	}
+
+	switch tagsString {
+	case "":
+		return None, nil, nil
+	case "_all":
+		return All, nil, nil
+	default:
+		return Specific, strings.Split(tagsString, ","), nil
+	}
+}
+
+func (m *Mailer) Subscribe(userId string, subscriptionType SubscriptionType, tags ...string) error {
+	tx, err := m.db.Begin()
+	if err != nil {
+		return fmt.Errorf("failed to initialize transaction with db: %s", err)
+	}
+
+	tagsOutput := ""
+	switch subscriptionType {
+	case All:
+		tagsOutput = "_all"
+	case None:
+		tagsOutput = ""
+	case Specific:
+		tagsOutput = strings.Join(tags, ",")
+	}
+
+	hash := m.GetHash(userId)
+
+	if _, err = tx.Exec(`INSERT INTO subscription_user_to_tags_table(user_id, tags) VALUES(?, ?)
+  ON CONFLICT(user_id) DO UPDATE SET
+  	tags=excluded.tags;`, hash, tagsOutput); err != nil {
+		tx.Rollback()
+		return fmt.Errorf("failed to configure user-to-tags table in db for the user: %s", err)
+	}
+
+	if err = tx.Commit(); err != nil {
+		tx.Rollback()
+		return fmt.Errorf("failed to commit transaction to db: %s", err)
+	}
+
+	return nil
+}
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{})
+	}
+}