OneAuth/libs/fs/file.go

package fs

import (
	"context"
	"encoding/json"
	"fmt"
	"github.com/olivere/elastic/v7"
	"github.com/veypi/OneAuth/cfg"
	"github.com/veypi/OneAuth/libs/webdav"
	"github.com/veypi/OneAuth/models"
	"github.com/veypi/OneAuth/models/index"
	"github.com/veypi/utils/log"
	"gorm.io/gorm"
	"net/http"
	"strconv"
	"time"
)

func updateUserSize(uid uint, delta int64) error {
	if uid == 0 {
		return nil
	}
	err := cfg.DB().Model(&models.User{}).Where("ID = ?", uid).
		Update("Used", gorm.Expr("Used + ?", delta)).Error
	return err
}

func removeFile(r *http.Request, prefix string, h *webdav.Handler) error {
	f := &models.File{}
	f.OwnerID = h.OwnerID
	f.Path = prefix
	res, err := index.File.Get().Id(f.ID()).Do(r.Context())
	if err != nil {
		if elastic.IsNotFound(err) {
			return nil
		}
		return err
	}
	_, err = index.File.Delete().Id(f.ID()).Do(r.Context())
	if err == nil {
		err = json.Unmarshal(res.Source, f)
		if err != nil {
			return err
		}
		uid, _ := strconv.Atoi(h.OwnerID)
		if f.Size > 0 && uid > 0 {
			err = updateUserSize(uint(uid), -int64(f.Size))
		}
		_ = addHistory(r, h, models.ActDelete, res.Id, f.Path)
	}
	return err
}

func updateFile(ctx context.Context, fileID string, props interface{}) error {
	_, err := index.File.Update().Id(fileID).Doc(props).Do(ctx)
	return err
}

func getFile(ctx context.Context, path string, h *webdav.Handler) (*models.File, error) {
	f := &models.File{}
	f.OwnerID = h.OwnerID
	f.Path = path
	//body, err := cfg.ES().Update().Index(models.IndexFile).Id(f.ID()).Upsert(f).Do(ctx)
	body, err := index.File.Get().Id(f.ID()).Do(ctx)
	if elastic.IsNotFound(err) {
		f.CreatedAt = time.Now()
		_, err = index.File.Index().Id(f.ID()).BodyJson(f).Do(ctx)
		return f, err
	} else if err != nil {
		return f, err
	}
	err = json.Unmarshal(body.Source, f)
	return f, err
}

func addHistory(r *http.Request, h *webdav.Handler, act models.Action, fileID string, tag string) error {
	his := &models.History{
		Action:  act,
		OwnerID: h.OwnerID,
		ActorID: h.ActorID,
		FileID:  fileID,
		Tag:     tag,
		IP:      r.RemoteAddr,
	}
	his.CreatedAt = time.Now()
	_, err := index.History.Index().BodyJson(his).Do(r.Context())
	if err != nil {
		return err
	}
	if act == models.ActGet && fileID != "" {
		_, err = index.File.Update().Id(fileID).
			Script(elastic.NewScript("ctx._source.Count ++")).Do(r.Context())
	}
	return err
}

func CleanHistory(days uint) error {
	res, err := index.History.DeleteByQuery().
		Query(elastic.NewRangeQuery("CreatedAt").Lt(fmt.Sprintf("now-%dd/d", days))).
		Do(cfg.Ctx)
	if err == nil && res.Deleted > 0 {
		log.Info().Msgf("clean %d records of history.", res.Deleted)
	}
	return err
}