You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
114 lines
2.6 KiB
114 lines
2.6 KiB
package controllers
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/cloudreve/Cloudreve/v4/pkg/hashid"
|
|
"github.com/cloudreve/Cloudreve/v4/pkg/serializer"
|
|
"github.com/cloudreve/Cloudreve/v4/service/share"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// CreateShare 创建分享
|
|
func CreateShare(c *gin.Context) {
|
|
service := ParametersFromContext[*share.ShareCreateService](c, share.ShareCreateParamCtx{})
|
|
uri, err := service.Upsert(c, 0)
|
|
if err != nil {
|
|
c.JSON(200, serializer.Err(c, err))
|
|
return
|
|
}
|
|
|
|
c.JSON(200, serializer.Response{Data: uri})
|
|
}
|
|
|
|
// EditShare 编辑分享
|
|
func EditShare(c *gin.Context) {
|
|
service := ParametersFromContext[*share.ShareCreateService](c, share.ShareCreateParamCtx{})
|
|
uri, err := service.Upsert(c, hashid.FromContext(c))
|
|
if err != nil {
|
|
c.JSON(200, serializer.Err(c, err))
|
|
return
|
|
}
|
|
|
|
c.JSON(200, serializer.Response{Data: uri})
|
|
}
|
|
|
|
// GetShare 查看分享
|
|
func GetShare(c *gin.Context) {
|
|
service := ParametersFromContext[*share.ShareInfoService](c, share.ShareInfoParamCtx{})
|
|
info, err := service.Get(c)
|
|
if err != nil {
|
|
c.JSON(200, serializer.Err(c, err))
|
|
return
|
|
}
|
|
|
|
c.JSON(200, serializer.Response{Data: info})
|
|
}
|
|
|
|
// ListShare 列出分享
|
|
func ListShare(c *gin.Context) {
|
|
service := ParametersFromContext[*share.ListShareService](c, share.ListShareParamCtx{})
|
|
resp, err := service.List(c)
|
|
if err != nil {
|
|
c.JSON(200, serializer.Err(c, err))
|
|
c.Abort()
|
|
return
|
|
}
|
|
|
|
if resp != nil {
|
|
c.JSON(200, serializer.Response{
|
|
Data: resp,
|
|
})
|
|
}
|
|
}
|
|
|
|
// DeleteShare 删除分享
|
|
func DeleteShare(c *gin.Context) {
|
|
err := share.DeleteShare(c, hashid.FromContext(c))
|
|
if err != nil {
|
|
c.JSON(200, serializer.Err(c, err))
|
|
return
|
|
}
|
|
|
|
c.JSON(200, serializer.Response{})
|
|
}
|
|
|
|
func ShareRedirect(c *gin.Context) {
|
|
service := ParametersFromContext[*share.ShortLinkRedirectService](c, share.ShortLinkRedirectParamCtx{})
|
|
|
|
// Check if the request is from a social media bot
|
|
if isSocialMediaBot(c.GetHeader("User-Agent")) {
|
|
html, err := service.RenderOGPage(c)
|
|
if err == nil {
|
|
c.Header("Content-Type", "text/html; charset=utf-8")
|
|
c.String(http.StatusOK, html)
|
|
return
|
|
}
|
|
// Fall back to redirect on error
|
|
}
|
|
|
|
c.Redirect(http.StatusFound, service.RedirectTo(c))
|
|
}
|
|
|
|
// isSocialMediaBot checks if the User-Agent belongs to a social media crawler
|
|
func isSocialMediaBot(ua string) bool {
|
|
bots := []string{
|
|
"facebookexternalhit",
|
|
"twitterbot",
|
|
"linkedinbot",
|
|
"whatsapp",
|
|
"slackbot",
|
|
"discordbot",
|
|
"telegrambot",
|
|
"facebot",
|
|
}
|
|
ua = strings.ToLower(ua)
|
|
for _, bot := range bots {
|
|
if strings.Contains(ua, bot) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|