Cloudreve/service/admin/tools.go

package admin

import (
	"encoding/hex"
	"net/http"
	"strconv"

	"github.com/cloudreve/Cloudreve/v4/inventory/types"

	"github.com/cloudreve/Cloudreve/v4/application/dependency"
	"github.com/cloudreve/Cloudreve/v4/pkg/boolset"
	"github.com/cloudreve/Cloudreve/v4/pkg/filemanager/manager"
	request2 "github.com/cloudreve/Cloudreve/v4/pkg/request"
	"github.com/cloudreve/Cloudreve/v4/pkg/serializer"
	"github.com/cloudreve/Cloudreve/v4/pkg/setting"
	"github.com/cloudreve/Cloudreve/v4/pkg/wopi"
	"github.com/gin-gonic/gin"
	"github.com/wneessen/go-mail"
)

type (
	HashIDService struct {
		ID     int    `json:"id"`
		Type   int    `json:"type"`
		HashID string `json:"hash_id"`
	}
	HashIDParamCtx struct{}
)

func (service *HashIDService) Encode(c *gin.Context) (string, error) {
	dep := dependency.FromContext(c)
	res, err := dep.HashIDEncoder().Encode([]int{service.ID, service.Type})
	if err != nil {
		return "", err
	}
	return res, nil
}

func (service *HashIDService) Decode(c *gin.Context) (int, error) {
	dep := dependency.FromContext(c)
	res, err := dep.HashIDEncoder().Decode(service.HashID, service.Type)
	if err != nil {
		return 0, err
	}

	return res, nil
}

type (
	BsEncodeService struct {
		Bool []int `json:"bool"`
	}
	BsEncodeParamCtx struct{}
	BsEncodeRes      struct {
		Hex string
		B64 []byte
	}
)

func (service *BsEncodeService) Encode(c *gin.Context) (*BsEncodeRes, error) {
	bs := &boolset.BooleanSet{}
	for _, v := range service.Bool {
		boolset.Set(v, true, bs)
	}

	res, err := bs.MarshalBinary()
	if err != nil {
		return nil, err
	}

	return &BsEncodeRes{
		Hex: hex.EncodeToString(res),
		B64: res,
	}, nil
}

type (
	BsDecodeService struct {
		Code string `json:"code"`
	}
	BsDecodeParamCtx struct{}
	BsDecodeRes      struct {
		Bool []int `json:"bool"`
	}
)

func (service *BsDecodeService) Decode(c *gin.Context) (*BsDecodeRes, error) {
	bs, err := boolset.FromString(service.Code)
	if err != nil {
		return nil, err
	}

	res := []int{}
	for i := 0; i < len(*bs)*8; i++ {
		if bs.Enabled(i) {
			res = append(res, i)
		}
	}

	return &BsDecodeRes{
		Bool: res,
	}, nil
}

type (
	FetchWOPIDiscoveryService struct {
		Endpoint string `form:"endpoint" binding:"required"`
	}
	FetchWOPIDiscoveryParamCtx struct{}
)

func (s *FetchWOPIDiscoveryService) Fetch(c *gin.Context) (*types.ViewerGroup, error) {
	dep := dependency.FromContext(c)
	requestClient := dep.RequestClient(request2.WithContext(c), request2.WithLogger(dep.Logger()))
	content, err := requestClient.Request("GET", s.Endpoint, nil).CheckHTTPResponse(http.StatusOK).GetResponse()
	if err != nil {
		return nil, serializer.NewError(serializer.CodeInternalSetting, "WOPI endpoint id unavailable", err)
	}

	vg, err := wopi.DiscoveryXmlToViewerGroup(content)
	if err != nil {
		return nil, serializer.NewError(serializer.CodeParamErr, "Failed to parse WOPI response", err)
	}

	return vg, nil
}

type (
	TestSMTPService struct {
		Settings map[string]string `json:"settings" binding:"required"`
		To       string            `json:"to" binding:"required,email"`
	}
	TestSMTPParamCtx struct{}
)

func (s *TestSMTPService) Test(c *gin.Context) error {
	port, err := strconv.Atoi(s.Settings["smtpPort"])
	if err != nil {
		return serializer.NewError(serializer.CodeParamErr, "Invalid SMTP port", err)
	}

	tlsPolicy := mail.TLSOpportunistic
	if setting.IsTrueValue(s.Settings["smtpEncryption"]) {
		tlsPolicy = mail.TLSMandatory
	}
	d, diaErr := mail.NewClient(s.Settings["smtpHost"],
		mail.WithPort(port),
		mail.WithSMTPAuth(mail.SMTPAuthAutoDiscover), mail.WithTLSPortPolicy(tlsPolicy),
		mail.WithUsername(s.Settings["smtpUser"]), mail.WithPassword(s.Settings["smtpPass"]),
	)
	if diaErr != nil {
		return serializer.NewError(serializer.CodeInternalSetting, "Failed to create SMTP client: "+diaErr.Error(), diaErr)
	}

	m := mail.NewMsg()
	if err := m.FromFormat(s.Settings["fromName"], s.Settings["fromAdress"]); err != nil {
		return serializer.NewError(serializer.CodeInternalSetting, "Failed to set FROM address: "+err.Error(), err)
	}
	m.ReplyToFormat(s.Settings["fromName"], s.Settings["replyTo"])
	m.To(s.To)
	m.Subject("Cloudreve SMTP Test")
	m.SetMessageID()
	m.SetBodyString(mail.TypeTextHTML, "This is a test email from Cloudreve.")

	err = d.DialAndSendWithContext(c, m)
	if err != nil {
		return serializer.NewError(serializer.CodeInternalSetting, "Failed to send test email: "+err.Error(), err)
	}

	return nil
}

func ClearEntityUrlCache(c *gin.Context) {
	dep := dependency.FromContext(c)
	dep.KV().Delete(manager.EntityUrlCacheKeyPrefix)
}