mirror of
https://github.com/ctwj/urldb.git
synced 2025-11-25 03:15:04 +08:00
364 lines
11 KiB
Go
364 lines
11 KiB
Go
package converter
|
||
|
||
import (
|
||
"encoding/json"
|
||
"reflect"
|
||
"time"
|
||
|
||
"github.com/ctwj/urldb/db/dto"
|
||
"github.com/ctwj/urldb/db/entity"
|
||
)
|
||
|
||
// ToResourceResponse 将Resource实体转换为ResourceResponse
|
||
func ToResourceResponse(resource *entity.Resource) dto.ResourceResponse {
|
||
response := dto.ResourceResponse{
|
||
ID: resource.ID,
|
||
Key: resource.Key,
|
||
Title: resource.Title,
|
||
Description: resource.Description,
|
||
URL: resource.URL,
|
||
PanID: resource.PanID,
|
||
SaveURL: resource.SaveURL,
|
||
FileSize: resource.FileSize,
|
||
CategoryID: resource.CategoryID,
|
||
ViewCount: resource.ViewCount,
|
||
IsValid: resource.IsValid,
|
||
IsPublic: resource.IsPublic,
|
||
CreatedAt: resource.CreatedAt,
|
||
UpdatedAt: resource.UpdatedAt,
|
||
Cover: resource.Cover,
|
||
Author: resource.Author,
|
||
ErrorMsg: resource.ErrorMsg,
|
||
SyncedToMeilisearch: resource.SyncedToMeilisearch,
|
||
SyncedAt: resource.SyncedAt,
|
||
}
|
||
|
||
// 设置分类名称
|
||
if resource.Category.ID != 0 {
|
||
response.CategoryName = resource.Category.Name
|
||
}
|
||
|
||
// 设置平台信息
|
||
if resource.Pan.ID != 0 {
|
||
panResponse := dto.PanResponse{
|
||
ID: resource.Pan.ID,
|
||
Name: resource.Pan.Name,
|
||
Key: resource.Pan.Key,
|
||
Icon: resource.Pan.Icon,
|
||
Remark: resource.Pan.Remark,
|
||
}
|
||
response.Pan = &panResponse
|
||
}
|
||
|
||
// 转换标签
|
||
response.Tags = make([]dto.TagResponse, len(resource.Tags))
|
||
for i, tag := range resource.Tags {
|
||
response.Tags[i] = dto.TagResponse{
|
||
ID: tag.ID,
|
||
Name: tag.Name,
|
||
Description: tag.Description,
|
||
ResourceCount: 0, // 在资源上下文中，标签的资源数量不相关
|
||
}
|
||
}
|
||
|
||
return response
|
||
}
|
||
|
||
// ToResourceResponseFromMeilisearch 将MeilisearchDocument转换为ResourceResponse（包含高亮信息）
|
||
func ToResourceResponseFromMeilisearch(doc interface{}) dto.ResourceResponse {
|
||
// 使用反射来获取MeilisearchDocument的字段
|
||
docValue := reflect.ValueOf(doc)
|
||
if docValue.Kind() == reflect.Ptr {
|
||
docValue = docValue.Elem()
|
||
}
|
||
|
||
response := dto.ResourceResponse{}
|
||
|
||
// 获取基本字段
|
||
if idField := docValue.FieldByName("ID"); idField.IsValid() {
|
||
response.ID = uint(idField.Uint())
|
||
}
|
||
if titleField := docValue.FieldByName("Title"); titleField.IsValid() {
|
||
response.Title = titleField.String()
|
||
}
|
||
if descField := docValue.FieldByName("Description"); descField.IsValid() {
|
||
response.Description = descField.String()
|
||
}
|
||
if urlField := docValue.FieldByName("URL"); urlField.IsValid() {
|
||
response.URL = urlField.String()
|
||
}
|
||
if coverField := docValue.FieldByName("Cover"); coverField.IsValid() {
|
||
response.Cover = coverField.String()
|
||
}
|
||
if saveURLField := docValue.FieldByName("SaveURL"); saveURLField.IsValid() {
|
||
response.SaveURL = saveURLField.String()
|
||
}
|
||
if fileSizeField := docValue.FieldByName("FileSize"); fileSizeField.IsValid() {
|
||
response.FileSize = fileSizeField.String()
|
||
}
|
||
if keyField := docValue.FieldByName("Key"); keyField.IsValid() {
|
||
response.Key = keyField.String()
|
||
}
|
||
if categoryField := docValue.FieldByName("Category"); categoryField.IsValid() {
|
||
response.CategoryName = categoryField.String()
|
||
}
|
||
if authorField := docValue.FieldByName("Author"); authorField.IsValid() {
|
||
response.Author = authorField.String()
|
||
}
|
||
if createdAtField := docValue.FieldByName("CreatedAt"); createdAtField.IsValid() {
|
||
response.CreatedAt = createdAtField.Interface().(time.Time)
|
||
}
|
||
if updatedAtField := docValue.FieldByName("UpdatedAt"); updatedAtField.IsValid() {
|
||
response.UpdatedAt = updatedAtField.Interface().(time.Time)
|
||
}
|
||
|
||
// 处理PanID
|
||
if panIDField := docValue.FieldByName("PanID"); panIDField.IsValid() && !panIDField.IsNil() {
|
||
panIDPtr := panIDField.Interface().(*uint)
|
||
if panIDPtr != nil {
|
||
response.PanID = panIDPtr
|
||
}
|
||
}
|
||
|
||
// 处理Tags
|
||
if tagsField := docValue.FieldByName("Tags"); tagsField.IsValid() {
|
||
tags := tagsField.Interface().([]string)
|
||
response.Tags = make([]dto.TagResponse, len(tags))
|
||
for i, tagName := range tags {
|
||
response.Tags[i] = dto.TagResponse{
|
||
Name: tagName,
|
||
}
|
||
}
|
||
}
|
||
|
||
// 处理高亮字段
|
||
if titleHighlightField := docValue.FieldByName("TitleHighlight"); titleHighlightField.IsValid() {
|
||
response.TitleHighlight = titleHighlightField.String()
|
||
}
|
||
if descHighlightField := docValue.FieldByName("DescriptionHighlight"); descHighlightField.IsValid() {
|
||
response.DescriptionHighlight = descHighlightField.String()
|
||
}
|
||
if categoryHighlightField := docValue.FieldByName("CategoryHighlight"); categoryHighlightField.IsValid() {
|
||
response.CategoryHighlight = categoryHighlightField.String()
|
||
}
|
||
if tagsHighlightField := docValue.FieldByName("TagsHighlight"); tagsHighlightField.IsValid() {
|
||
tagsHighlight := tagsHighlightField.Interface().([]string)
|
||
response.TagsHighlight = make([]string, len(tagsHighlight))
|
||
copy(response.TagsHighlight, tagsHighlight)
|
||
}
|
||
|
||
return response
|
||
}
|
||
|
||
// ToResourceResponseList 将Resource实体列表转换为ResourceResponse列表
|
||
func ToResourceResponseList(resources []entity.Resource) []dto.ResourceResponse {
|
||
responses := make([]dto.ResourceResponse, len(resources))
|
||
for i, resource := range resources {
|
||
responses[i] = ToResourceResponse(&resource)
|
||
}
|
||
return responses
|
||
}
|
||
|
||
// ToCategoryResponse 将Category实体转换为CategoryResponse
|
||
func ToCategoryResponse(category *entity.Category, resourceCount int64, tagNames []string) dto.CategoryResponse {
|
||
return dto.CategoryResponse{
|
||
ID: category.ID,
|
||
Name: category.Name,
|
||
Description: category.Description,
|
||
ResourceCount: resourceCount,
|
||
TagNames: tagNames,
|
||
}
|
||
}
|
||
|
||
// ToCategoryResponseList 将Category实体列表转换为CategoryResponse列表
|
||
func ToCategoryResponseList(categories []entity.Category, resourceCounts map[uint]int64, tagNamesMap map[uint][]string) []dto.CategoryResponse {
|
||
responses := make([]dto.CategoryResponse, len(categories))
|
||
for i, category := range categories {
|
||
resourceCount := resourceCounts[category.ID]
|
||
tagNames := tagNamesMap[category.ID]
|
||
responses[i] = ToCategoryResponse(&category, resourceCount, tagNames)
|
||
}
|
||
return responses
|
||
}
|
||
|
||
// ToTagResponse 将Tag实体转换为TagResponse
|
||
func ToTagResponse(tag *entity.Tag, resourceCount int64) dto.TagResponse {
|
||
response := dto.TagResponse{
|
||
ID: tag.ID,
|
||
Name: tag.Name,
|
||
Description: tag.Description,
|
||
CategoryID: tag.CategoryID,
|
||
ResourceCount: resourceCount,
|
||
}
|
||
|
||
// 设置分类名称
|
||
if tag.CategoryID != nil && tag.Category.ID != 0 {
|
||
response.CategoryName = tag.Category.Name
|
||
} else if tag.CategoryID != nil {
|
||
// 如果CategoryID存在但Category没有预加载，设置为"未知分类"
|
||
response.CategoryName = "未知分类"
|
||
}
|
||
|
||
return response
|
||
}
|
||
|
||
// ToTagResponseList 将Tag实体列表转换为TagResponse列表
|
||
func ToTagResponseList(tags []entity.Tag, resourceCounts map[uint]int64) []dto.TagResponse {
|
||
responses := make([]dto.TagResponse, len(tags))
|
||
for i, tag := range tags {
|
||
resourceCount := resourceCounts[tag.ID]
|
||
responses[i] = ToTagResponse(&tag, resourceCount)
|
||
}
|
||
return responses
|
||
}
|
||
|
||
// ToPanResponse 将Pan实体转换为PanResponse
|
||
func ToPanResponse(pan *entity.Pan) dto.PanResponse {
|
||
return dto.PanResponse{
|
||
ID: pan.ID,
|
||
Name: pan.Name,
|
||
Key: pan.Key,
|
||
Icon: pan.Icon,
|
||
Remark: pan.Remark,
|
||
}
|
||
}
|
||
|
||
// ToPanResponseList 将Pan实体列表转换为PanResponse列表
|
||
func ToPanResponseList(pans []entity.Pan) []dto.PanResponse {
|
||
responses := make([]dto.PanResponse, len(pans))
|
||
for i, pan := range pans {
|
||
responses[i] = ToPanResponse(&pan)
|
||
}
|
||
return responses
|
||
}
|
||
|
||
// ToCksResponse 将Cks实体转换为CksResponse
|
||
func ToCksResponse(cks *entity.Cks) dto.CksResponse {
|
||
response := dto.CksResponse{
|
||
ID: cks.ID,
|
||
PanID: cks.PanID,
|
||
Idx: cks.Idx,
|
||
Ck: cks.Ck,
|
||
IsValid: cks.IsValid,
|
||
Space: cks.Space,
|
||
LeftSpace: cks.LeftSpace,
|
||
UsedSpace: cks.UsedSpace,
|
||
Username: cks.Username,
|
||
VipStatus: cks.VipStatus,
|
||
ServiceType: cks.ServiceType,
|
||
Remark: cks.Remark,
|
||
}
|
||
|
||
// 设置平台信息
|
||
if cks.Pan.ID != 0 {
|
||
response.Pan = &dto.PanResponse{
|
||
ID: cks.Pan.ID,
|
||
Name: cks.Pan.Name,
|
||
Key: cks.Pan.Key,
|
||
Icon: cks.Pan.Icon,
|
||
Remark: cks.Pan.Remark,
|
||
}
|
||
}
|
||
|
||
return response
|
||
}
|
||
|
||
// ToCksResponseList 将Cks实体列表转换为CksResponse列表
|
||
func ToCksResponseList(cksList []entity.Cks) []dto.CksResponse {
|
||
responses := make([]dto.CksResponse, len(cksList))
|
||
for i, cks := range cksList {
|
||
responses[i] = ToCksResponse(&cks)
|
||
}
|
||
return responses
|
||
}
|
||
|
||
// ToReadyResourceResponse 将ReadyResource实体转换为ReadyResourceResponse
|
||
func ToReadyResourceResponse(resource *entity.ReadyResource) dto.ReadyResourceResponse {
|
||
isDeleted := !resource.DeletedAt.Time.IsZero()
|
||
var deletedAt *time.Time
|
||
if isDeleted {
|
||
deletedAt = &resource.DeletedAt.Time
|
||
}
|
||
|
||
return dto.ReadyResourceResponse{
|
||
ID: resource.ID,
|
||
Title: resource.Title,
|
||
Description: resource.Description,
|
||
URL: resource.URL,
|
||
Category: resource.Category,
|
||
Tags: resource.Tags,
|
||
Img: resource.Img,
|
||
Source: resource.Source,
|
||
Extra: resource.Extra,
|
||
Key: resource.Key,
|
||
ErrorMsg: resource.ErrorMsg,
|
||
CreateTime: resource.CreateTime,
|
||
IP: resource.IP,
|
||
DeletedAt: deletedAt,
|
||
IsDeleted: isDeleted,
|
||
}
|
||
}
|
||
|
||
// ToReadyResourceResponseList 将ReadyResource实体列表转换为ReadyResourceResponse列表
|
||
func ToReadyResourceResponseList(resources []entity.ReadyResource) []dto.ReadyResourceResponse {
|
||
responses := make([]dto.ReadyResourceResponse, len(resources))
|
||
for i, resource := range resources {
|
||
responses[i] = ToReadyResourceResponse(&resource)
|
||
}
|
||
return responses
|
||
}
|
||
|
||
// RequestToReadyResource 将ReadyResourceRequest转换为ReadyResource实体
|
||
// func RequestToReadyResource(req *dto.ReadyResourceRequest) *entity.ReadyResource {
|
||
// if req == nil {
|
||
// return nil
|
||
// }
|
||
|
||
// return &entity.ReadyResource{
|
||
// Title: &req.Title,
|
||
// Description: req.Description,
|
||
// URL: req.Url,
|
||
// Category: req.Category,
|
||
// Tags: req.Tags,
|
||
// Img: req.Img,
|
||
// Source: req.Source,
|
||
// Extra: req.Extra,
|
||
// Key: req.Key,
|
||
// }
|
||
// }
|
||
|
||
// TaskToGoogleIndexTaskOutput 将Task实体转换为GoogleIndexTaskOutput
|
||
func TaskToGoogleIndexTaskOutput(task *entity.Task, stats map[string]int) dto.GoogleIndexTaskOutput {
|
||
output := dto.GoogleIndexTaskOutput{
|
||
ID: task.ID,
|
||
Name: task.Name,
|
||
Description: task.Description,
|
||
Type: string(task.Type),
|
||
Status: string(task.Status),
|
||
Progress: task.Progress,
|
||
TotalItems: stats["total"],
|
||
ProcessedItems: stats["completed"] + stats["failed"],
|
||
SuccessfulItems: stats["completed"],
|
||
FailedItems: stats["failed"],
|
||
PendingItems: stats["pending"],
|
||
ProcessingItems: stats["processing"],
|
||
IndexedURLs: task.IndexedURLs,
|
||
FailedURLs: task.FailedURLs,
|
||
ConfigID: task.ConfigID,
|
||
CreatedAt: task.CreatedAt,
|
||
UpdatedAt: task.UpdatedAt,
|
||
StartedAt: task.StartedAt,
|
||
CompletedAt: task.CompletedAt,
|
||
}
|
||
|
||
// 设置进度数据
|
||
if task.ProgressData != "" {
|
||
var progressData map[string]interface{}
|
||
if err := json.Unmarshal([]byte(task.ProgressData), &progressData); err == nil {
|
||
output.ProgressData = progressData
|
||
}
|
||
}
|
||
|
||
return output
|
||
}
|