🎉 Init project from scratch

pkg/services/auth.go

@@ -0,0 +1,90 @@
+package services
+
+import (
+	"code.smartsheep.studio/hydrogen/interactive/pkg/database"
+	"code.smartsheep.studio/hydrogen/interactive/pkg/models"
+	"code.smartsheep.studio/hydrogen/interactive/pkg/security"
+	"errors"
+	"fmt"
+	"github.com/google/uuid"
+	"gorm.io/gorm"
+	"strconv"
+	"time"
+)
+
+type PassportUserinfo struct {
+	Sub                uint   `json:"sub"`
+	Email              string `json:"email"`
+	Picture            string `json:"picture"`
+	PreferredUsernames string `json:"preferred_usernames"`
+}
+
+func LinkAccount(userinfo PassportUserinfo) (models.Account, error) {
+	var account models.Account
+	if err := database.C.Where(&models.Account{
+		ExternalID: userinfo.Sub,
+	}).First(&account).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			account = models.Account{
+				Name:         userinfo.PreferredUsernames,
+				Avatar:       userinfo.Picture,
+				EmailAddress: userinfo.Email,
+				PowerLevel:   0,
+				ExternalID:   userinfo.Sub,
+			}
+			return account, database.C.Save(&account).Error
+		}
+		return account, err
+	}
+
+	return account, nil
+}
+
+func GetToken(account models.Account) (string, string, error) {
+	var err error
+	var refresh, access string
+
+	sub := strconv.Itoa(int(account.ID))
+	access, err = security.EncodeJwt(
+		uuid.NewString(),
+		security.JwtAccessType,
+		sub,
+		[]string{"interactive"},
+		time.Now().Add(30*time.Minute),
+	)
+	if err != nil {
+		return refresh, access, err
+	}
+	refresh, err = security.EncodeJwt(
+		uuid.NewString(),
+		security.JwtRefreshType,
+		sub,
+		[]string{"interactive"},
+		time.Now().Add(30*24*time.Hour),
+	)
+	if err != nil {
+		return refresh, access, err
+	}
+
+	return access, refresh, nil
+}
+
+func RefreshToken(token string) (string, string, error) {
+	parseInt := func(str string) int {
+		val, _ := strconv.Atoi(str)
+		return val
+	}
+
+	var account models.Account
+	if claims, err := security.DecodeJwt(token); err != nil {
+		return "404", "403", err
+	} else if claims.Type != security.JwtRefreshType {
+		return "404", "403", fmt.Errorf("invalid token type, expected refresh token")
+	} else if err := database.C.Where(models.Account{
+		BaseModel: models.BaseModel{ID: uint(parseInt(claims.Subject))},
+	}).First(&account).Error; err != nil {
+		return "404", "403", err
+	}
+
+	return GetToken(account)
+}

pkg/services/mailer.go

@@ -0,0 +1,51 @@
+package services
+
+import (
+	"crypto/tls"
+	"fmt"
+	"net/smtp"
+	"net/textproto"
+
+	"github.com/jordan-wright/email"
+	"github.com/spf13/viper"
+)
+
+func SendMail(target string, subject string, content string) error {
+	mail := &email.Email{
+		To:      []string{target},
+		From:    viper.GetString("mailer.name"),
+		Subject: subject,
+		Text:    []byte(content),
+		Headers: textproto.MIMEHeader{},
+	}
+	return mail.SendWithTLS(
+		fmt.Sprintf("%s:%d", viper.GetString("mailer.smtp_host"), viper.GetInt("mailer.smtp_port")),
+		smtp.PlainAuth(
+			"",
+			viper.GetString("mailer.username"),
+			viper.GetString("mailer.password"),
+			viper.GetString("mailer.smtp_host"),
+		),
+		&tls.Config{ServerName: viper.GetString("mailer.smtp_host")},
+	)
+}
+
+func SendMailHTML(target string, subject string, content string) error {
+	mail := &email.Email{
+		To:      []string{target},
+		From:    viper.GetString("mailer.name"),
+		Subject: subject,
+		HTML:    []byte(content),
+		Headers: textproto.MIMEHeader{},
+	}
+	return mail.SendWithTLS(
+		fmt.Sprintf("%s:%d", viper.GetString("mailer.smtp_host"), viper.GetInt("mailer.smtp_port")),
+		smtp.PlainAuth(
+			"",
+			viper.GetString("mailer.username"),
+			viper.GetString("mailer.password"),
+			viper.GetString("mailer.smtp_host"),
+		),
+		&tls.Config{ServerName: viper.GetString("mailer.smtp_host")},
+	)
+}