migrate from github

2026-07-15 22:42:31 +08:00
commit 3ad559f05e
262 changed files with 21637 additions and 0 deletions
@@ -0,0 +1,204 @@
+package grpcserver
+
+import (
+	"time"
+
+	"github.com/XingfenD/yoresee_doc/internal/dto"
+	"github.com/XingfenD/yoresee_doc/internal/mapper/doc_type_mapper"
+	"github.com/XingfenD/yoresee_doc/internal/media"
+	"github.com/XingfenD/yoresee_doc/internal/status"
+	"github.com/XingfenD/yoresee_doc/internal/utils"
+	pb "github.com/XingfenD/yoresee_doc/pkg/gen/yoresee_doc/v1"
+)
+
+func baseResponseFromErr(err error) *pb.BaseResponse {
+	if err == nil {
+		return baseResponseFromStatus(status.StatusSuccess)
+	}
+	return baseResponseFromStatus(err)
+}
+
+func baseResponseFromStatus(err error) *pb.BaseResponse {
+	st, ok := err.(*status.Status)
+	if !ok {
+		st = status.StatusServiceInternalError.(*status.Status)
+	}
+	return &pb.BaseResponse{
+		Code:    int32(st.Code),
+		Message: st.Message,
+	}
+}
+
+func timeToString(t time.Time) string {
+	if t.IsZero() {
+		return ""
+	}
+	return t.Format(time.RFC3339)
+}
+
+func toDocumentResponse(doc *dto.DocumentMetaResponse) *pb.DocumentResponse {
+	if doc == nil {
+		return nil
+	}
+	resp := &pb.DocumentResponse{
+		ExternalId:  doc.ExternalID,
+		Title:       doc.Title,
+		Type:        doc_type_mapper.ToProtoType(doc.Type),
+		Summary:     doc.Summary,
+		Tags:        doc.Tags,
+		ViewCount:   int32(doc.ViewCount),
+		EditCount:   int32(doc.EditCount),
+		CreatedAt:   timeToString(doc.CreatedAt),
+		UpdatedAt:   timeToString(doc.UpdatedAt),
+		HasChildren: doc.HasChildren,
+		IsPublic:    doc.IsPublic,
+	}
+	if len(doc.Children) > 0 {
+		resp.Children = make([]*pb.DocumentResponse, 0, len(doc.Children))
+		for _, child := range doc.Children {
+			resp.Children = append(resp.Children, toDocumentResponse(child))
+		}
+	}
+	return resp
+}
+
+func toKnowledgeBaseResponse(kb *dto.KnowledgeBaseResponse) *pb.KnowledgeBaseResponse {
+	if kb == nil {
+		return nil
+	}
+	return &pb.KnowledgeBaseResponse{
+		ExternalId:            kb.ExternalID,
+		Name:                  kb.Name,
+		Description:           kb.Description,
+		Cover:                 kb.Cover,
+		IsPublic:              kb.IsPublic,
+		CreatedAt:             timeToString(kb.CreatedAt),
+		UpdatedAt:             timeToString(kb.UpdatedAt),
+		DeletedAt:             timeToString(kb.DeletedAt),
+		CreatorUserExternalId: kb.CreatorUserExternalID,
+		CreatorName:           kb.CreatorName,
+		DocumentsCount:        kb.DocumentsCount,
+	}
+}
+
+func toUserResponse(user *dto.UserResponse) *pb.UserResponse {
+	if user == nil {
+		return nil
+	}
+	resp := &pb.UserResponse{
+		ExternalId: user.ExternalID,
+		Username:   user.Username,
+		Email:      user.Email,
+		Nickname:   user.Nickname,
+		Avatar:     media.BuildAvatarURL(user.ExternalID, user.AvatarObjectKey, user.AvatarVersion),
+		Status:     int32(user.Status),
+		CreatedAt:  timeToString(user.CreatedAt),
+		UpdatedAt:  timeToString(user.UpdatedAt),
+	}
+	if user.InvitationCode != nil {
+		resp.InvitationCode = user.InvitationCode
+	}
+	return resp
+}
+
+func toTemplateResponse(tpl *dto.TemplateResponse) *pb.TemplateResponse {
+	if tpl == nil {
+		return nil
+	}
+	return &pb.TemplateResponse{
+		Id:                      tpl.ID,
+		Name:                    tpl.Name,
+		Description:             tpl.Description,
+		Content:                 tpl.Content,
+		Type:                    doc_type_mapper.ToProtoType(tpl.Type),
+		Scope:                   tpl.Scope,
+		KnowledgeBaseExternalId: tpl.KnowledgeBaseExternalID,
+		Tags:                    tpl.Tags,
+		CreatedAt:               timeToString(tpl.CreatedAt),
+		UpdatedAt:               timeToString(tpl.UpdatedAt),
+	}
+}
+
+func toAttachmentResponse(attachment *dto.AttachmentResponse) *pb.AttachmentResponse {
+	if attachment == nil {
+		return nil
+	}
+	return &pb.AttachmentResponse{
+		ExternalId:         attachment.ExternalID,
+		DocumentExternalId: attachment.DocumentExternalID,
+		Name:               attachment.Name,
+		Size:               attachment.Size,
+		MimeType:           attachment.MimeType,
+		Path:               attachment.Path,
+		Url:                attachment.URL,
+		CreatedAt:          timeToString(attachment.CreatedAt),
+		UpdatedAt:          timeToString(attachment.UpdatedAt),
+	}
+}
+
+func toDocumentVersionResponse(item *dto.DocumentVersionResponse) *pb.DocumentVersionResponse {
+	if item == nil {
+		return nil
+	}
+	return &pb.DocumentVersionResponse{
+		Version:       int32(item.Version),
+		Title:         item.Title,
+		Content:       item.Content,
+		ChangeSummary: item.ChangeSummary,
+		CreatedAt:     timeToString(item.CreatedAt),
+	}
+}
+
+func toUserGroupResponse(group *dto.UserGroupResponse) *pb.UserGroupResponse {
+	if group == nil {
+		return nil
+	}
+	resp := &pb.UserGroupResponse{
+		ExternalId:            group.ExternalID,
+		Name:                  group.Name,
+		Description:           group.Description,
+		CreatorUserExternalId: group.CreatorUserExternalID,
+		MemberCount:           int32(group.MemberCount),
+	}
+	return resp
+}
+
+func toInvitationResponse(inv *dto.InvitationResponse) *pb.InvitationResponse {
+	if inv == nil {
+		return nil
+	}
+	resp := &pb.InvitationResponse{
+		Code:                inv.Code,
+		CreatedByExternalId: inv.CreatedByExternalID,
+		CreatedByName:       inv.CreatedByName,
+		UsedCnt:             inv.UsedCnt,
+		Disabled:            inv.Disabled,
+		CreatedAt:           timeToString(inv.CreatedAt),
+	}
+	if inv.MaxUsedCnt != nil {
+		resp.MaxUsedCnt = inv.MaxUsedCnt
+	}
+	if inv.ExpiresAt != nil {
+		resp.ExpiresAt = utils.Of(timeToString(*inv.ExpiresAt))
+	}
+	if inv.Note != nil {
+		resp.Note = inv.Note
+	}
+	return resp
+}
+
+func toInvitationRecordResponse(rec *dto.InvitationRecordResponse) *pb.InvitationRecordResponse {
+	if rec == nil {
+		return nil
+	}
+	resp := &pb.InvitationRecordResponse{
+		Code:   rec.Code,
+		UsedBy: rec.UsedBy,
+		UsedAt: timeToString(rec.UsedAt),
+		Status: rec.Status,
+	}
+	if rec.UsedByExternalID != "" {
+		resp.UsedByExternalId = utils.Of(rec.UsedByExternalID)
+	}
+	return resp
+}