cloudreve/pkg/serializer/share.go

package serializer

import (
	model "github.com/HFO4/cloudreve/models"
	"github.com/HFO4/cloudreve/pkg/hashid"
	"time"
)

// Share 分享信息序列化
type Share struct {
	Key        string        `json:"key"`
	Locked     bool          `json:"locked"`
	IsDir      bool          `json:"is_dir"`
	CreateDate string        `json:"create_date,omitempty"`
	Downloads  int           `json:"downloads"`
	Views      int           `json:"views"`
	Expire     int64         `json:"expire"`
	Preview    bool          `json:"preview"`
	Creator    *shareCreator `json:"creator,omitempty"`
	Source     *shareSource  `json:"source,omitempty"`
}

type shareCreator struct {
	Key       string `json:"key"`
	Nick      string `json:"nick"`
	GroupName string `json:"group_name"`
}

type shareSource struct {
	Name string `json:"name"`
	Size uint64 `json:"size"`
}

// myShareItem 我的分享列表条目
type myShareItem struct {
	Key             string       `json:"key"`
	IsDir           bool         `json:"is_dir"`
	Password        string       `json:"password"`
	CreateDate      string       `json:"create_date,omitempty"`
	Downloads       int          `json:"downloads"`
	RemainDownloads int          `json:"remain_downloads"`
	Views           int          `json:"views"`
	Expire          int64        `json:"expire"`
	Preview         bool         `json:"preview"`
	Source          *shareSource `json:"source,omitempty"`
}

// BuildShareList 构建我的分享列表响应
func BuildShareList(shares []model.Share, total int) Response {
	res := make([]myShareItem, 0, total)
	now := time.Now().Unix()
	for i := 0; i < len(shares); i++ {
		item := myShareItem{
			Key:             hashid.HashID(shares[i].ID, hashid.ShareID),
			IsDir:           shares[i].IsDir,
			Password:        shares[i].Password,
			CreateDate:      shares[i].CreatedAt.Format("2006-01-02 15:04:05"),
			Downloads:       shares[i].Downloads,
			Views:           shares[i].Views,
			Preview:         shares[i].PreviewEnabled,
			Expire:          -1,
			RemainDownloads: shares[i].RemainDownloads,
		}
		if shares[i].Expires != nil {
			item.Expire = shares[i].Expires.Unix() - now
			if item.Expire == 0 {
				item.Expire = 0
			}
		}
		if shares[i].File.ID != 0 {
			item.Source = &shareSource{
				Name: shares[i].File.Name,
				Size: shares[i].File.Size,
			}
		} else if shares[i].Folder.ID != 0 {
			item.Source = &shareSource{
				Name: shares[i].Folder.Name,
			}
		}

		res = append(res, item)
	}

	return Response{Data: map[string]interface{}{
		"total": total,
		"items": res,
	}}
}

// BuildShareResponse 构建获取分享信息响应
func BuildShareResponse(share *model.Share, unlocked bool) Share {
	creator := share.Creator()
	resp := Share{
		Key:    hashid.HashID(share.ID, hashid.ShareID),
		Locked: !unlocked,
		Creator: &shareCreator{
			Key:       hashid.HashID(creator.ID, hashid.UserID),
			Nick:      creator.Nick,
			GroupName: creator.Group.Name,
		},
		CreateDate: share.CreatedAt.Format("2006-01-02 15:04:05"),
	}

	// 未解锁时只返回基本信息
	if !unlocked {
		return resp
	}

	resp.IsDir = share.IsDir
	resp.Downloads = share.Downloads
	resp.Views = share.Views
	resp.Preview = share.PreviewEnabled

	if share.Expires != nil {
		resp.Expire = share.Expires.Unix() - time.Now().Unix()
	}

	if share.IsDir {
		source := share.SourceFolder()
		resp.Source = &shareSource{
			Name: source.Name,
			Size: 0,
		}
	} else {
		source := share.SourceFile()
		resp.Source = &shareSource{
			Name: source.Name,
			Size: source.Size,
		}
	}

	return resp

}