82 lines
2.1 KiB
Go
82 lines
2.1 KiB
Go
package controllers
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"git.myrkvi.com/myrkvi/bin/config"
|
|
"git.myrkvi.com/myrkvi/bin/global"
|
|
"git.myrkvi.com/myrkvi/bin/models"
|
|
"git.myrkvi.com/myrkvi/bin/utils"
|
|
"github.com/go-mail/mail"
|
|
"github.com/labstack/echo/v4"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/spf13/viper"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func PostEmailHandler(c echo.Context) error {
|
|
code := c.Param("id")
|
|
adminCode := c.FormValue("adminKey")
|
|
email := c.FormValue("email")
|
|
|
|
if !viper.GetBool("smtp.enabled") {
|
|
return utils.RenderErrorToast(c, "function not available")
|
|
}
|
|
|
|
file := models.File{}
|
|
|
|
result := global.DB.Where("page_key = ?", code).First(&file)
|
|
if result.Error == gorm.ErrRecordNotFound {
|
|
return utils.RenderErrorToast(c, "the file does not exist")
|
|
} else if result.Error != nil {
|
|
return utils.RenderErrorToast(c, "server error occurred")
|
|
}
|
|
|
|
if adminCode != file.AdminKey {
|
|
return utils.RenderErrorToast(c, "incorrect deletion key")
|
|
}
|
|
|
|
if file.Email != "" {
|
|
return utils.RenderErrorToast(c, "this file is already associated with an email")
|
|
}
|
|
|
|
file.Email = email
|
|
|
|
result = global.DB.Save(&file)
|
|
if result.Error != nil {
|
|
logrus.WithError(result.Error).
|
|
WithFields(logrus.Fields{
|
|
"file code": file.PageKey,
|
|
"email": email,
|
|
}).Errorln("failed to save email for file")
|
|
|
|
return utils.RenderErrorToast(c, "server error occurred")
|
|
}
|
|
|
|
smtp := config.SMTP
|
|
|
|
dial := mail.NewDialer(smtp.Host, int(smtp.Port), smtp.Username, smtp.Password)
|
|
|
|
message := mail.NewMessage()
|
|
message.SetHeader("From", smtp.From)
|
|
message.SetHeader("To", email)
|
|
message.SetHeader("Subject", fmt.Sprintf("[bin] Deletion key for file '%s'", file.Filename))
|
|
message.SetBody(
|
|
"text/plain",
|
|
fmt.Sprintf(
|
|
"Deletion key for the file located at %s is %s.\r\nYou can also use this link: %s",
|
|
fmt.Sprintf("%s/b/%s", config.BaseURL, file.PageKey),
|
|
file.AdminKey,
|
|
fmt.Sprintf("%s/b/%s?delcode=%s", config.BaseURL, file.PageKey, file.AdminKey),
|
|
),
|
|
)
|
|
|
|
err := dial.DialAndSend(message)
|
|
if err != nil {
|
|
return utils.RenderErrorToast(c, "could not send email")
|
|
}
|
|
|
|
return c.HTML(200, "<p>email sent!</p>")
|
|
|
|
}
|