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,159 @@
+package grpcserver
+
+import (
+	"context"
+	"strings"
+
+	"github.com/XingfenD/yoresee_doc/internal/domain_event"
+	"github.com/XingfenD/yoresee_doc/internal/dto"
+	"github.com/XingfenD/yoresee_doc/internal/model"
+	"github.com/XingfenD/yoresee_doc/internal/service/auth_service"
+	"github.com/XingfenD/yoresee_doc/internal/service/notification_service"
+	"github.com/XingfenD/yoresee_doc/internal/status"
+	pb "github.com/XingfenD/yoresee_doc/pkg/gen/yoresee_doc/v1"
+)
+
+type NotificationServiceServer struct {
+	pb.UnimplementedNotificationServiceServer
+}
+
+func NewNotificationServiceServer() *NotificationServiceServer {
+	return &NotificationServiceServer{}
+}
+
+func (s *NotificationServiceServer) CreateNotification(ctx context.Context, req *pb.CreateNotificationRequest) (*pb.CreateNotificationResponse, error) {
+	if req == nil {
+		return &pb.CreateNotificationResponse{Base: baseResponseFromStatus(status.StatusParamError)}, nil
+	}
+
+	userExternalID, ok := ctx.Value("user_external_id").(string)
+	if !ok || strings.TrimSpace(userExternalID) == "" {
+		return &pb.CreateNotificationResponse{Base: baseResponseFromStatus(status.StatusTokenInvalid)}, nil
+	}
+
+	receiverExternalIDs := req.ReceiverExternalIds
+	if len(receiverExternalIDs) == 0 {
+		receiverExternalIDs = []string{userExternalID}
+	}
+
+	if !onlySelf(receiverExternalIDs, userExternalID) {
+		isAdmin, err := auth_service.AuthSvc.IsAdmin(userExternalID)
+		if err != nil {
+			return &pb.CreateNotificationResponse{Base: baseResponseFromErr(err)}, nil
+		}
+		if !isAdmin {
+			return &pb.CreateNotificationResponse{Base: baseResponseFromStatus(status.StatusPermissionDenied)}, nil
+		}
+	}
+
+	evt := domain_event.NotificationCreateEvent{
+		ReceiverExternalIDs: receiverExternalIDs,
+		Type:                req.Type,
+		Title:               req.Title,
+		Content:             req.Content,
+		PayloadJSON:         req.PayloadJson,
+	}
+	if err := domain_event.PublishNotificationCreateEvent(ctx, evt); err != nil {
+		return &pb.CreateNotificationResponse{Base: baseResponseFromErr(status.StatusMQNotInitialized)}, nil
+	}
+
+	return &pb.CreateNotificationResponse{Base: baseResponseFromErr(nil)}, nil
+}
+
+func (s *NotificationServiceServer) ListNotifications(ctx context.Context, req *pb.ListNotificationsRequest) (*pb.ListNotificationsResponse, error) {
+	if req == nil {
+		return &pb.ListNotificationsResponse{Base: baseResponseFromStatus(status.StatusParamError)}, nil
+	}
+	userExternalID, ok := ctx.Value("user_external_id").(string)
+	if !ok || strings.TrimSpace(userExternalID) == "" {
+		return &pb.ListNotificationsResponse{Base: baseResponseFromStatus(status.StatusTokenInvalid)}, nil
+	}
+
+	var statusFilter *string
+	if req.Status != nil && strings.TrimSpace(req.GetStatus()) != "" {
+		statusFilter = req.Status
+	}
+
+	list, total, err := notification_service.NotificationSvc.ListNotifications(&dto.ListNotificationsRequest{
+		UserExternalID: userExternalID,
+		Status:         statusFilter,
+		Pagination: dto.Pagination{
+			Page:     int(req.Page),
+			PageSize: int(req.PageSize),
+		},
+	})
+	if err != nil {
+		return &pb.ListNotificationsResponse{Base: baseResponseFromErr(err)}, nil
+	}
+
+	respItems := make([]*pb.Notification, 0, len(list))
+	for _, item := range list {
+		respItems = append(respItems, toNotificationResponse(&item))
+	}
+
+	return &pb.ListNotificationsResponse{
+		Base:          baseResponseFromErr(nil),
+		Notifications: respItems,
+		Total:         total,
+	}, nil
+}
+
+func (s *NotificationServiceServer) MarkNotificationsRead(ctx context.Context, req *pb.MarkNotificationsReadRequest) (*pb.MarkNotificationsReadResponse, error) {
+	if req == nil {
+		return &pb.MarkNotificationsReadResponse{Base: baseResponseFromStatus(status.StatusParamError)}, nil
+	}
+	userExternalID, ok := ctx.Value("user_external_id").(string)
+	if !ok || strings.TrimSpace(userExternalID) == "" {
+		return &pb.MarkNotificationsReadResponse{Base: baseResponseFromStatus(status.StatusTokenInvalid)}, nil
+	}
+
+	err := notification_service.NotificationSvc.MarkRead(&dto.MarkNotificationsReadRequest{
+		UserExternalID: userExternalID,
+		ExternalIDs:    req.ExternalIds,
+	})
+	if err != nil {
+		return &pb.MarkNotificationsReadResponse{Base: baseResponseFromErr(err)}, nil
+	}
+	return &pb.MarkNotificationsReadResponse{Base: baseResponseFromErr(nil)}, nil
+}
+
+func (s *NotificationServiceServer) MarkAllNotificationsRead(ctx context.Context, req *pb.MarkAllNotificationsReadRequest) (*pb.MarkAllNotificationsReadResponse, error) {
+	userExternalID, ok := ctx.Value("user_external_id").(string)
+	if !ok || strings.TrimSpace(userExternalID) == "" {
+		return &pb.MarkAllNotificationsReadResponse{Base: baseResponseFromStatus(status.StatusTokenInvalid)}, nil
+	}
+	if err := notification_service.NotificationSvc.MarkAllRead(userExternalID); err != nil {
+		return &pb.MarkAllNotificationsReadResponse{Base: baseResponseFromErr(err)}, nil
+	}
+	return &pb.MarkAllNotificationsReadResponse{Base: baseResponseFromErr(nil)}, nil
+}
+
+func toNotificationResponse(item *model.Notification) *pb.Notification {
+	if item == nil {
+		return nil
+	}
+	return &pb.Notification{
+		ExternalId: item.ExternalID,
+		Type:       item.Type,
+		Status:     item.Status,
+		Title:      item.Title,
+		Content:    item.Content,
+		Payload:    item.Payload,
+		CreatedAt:  timeToString(item.CreatedAt),
+	}
+}
+
+func onlySelf(list []string, self string) bool {
+	if len(list) == 0 {
+		return true
+	}
+	for _, id := range list {
+		if strings.TrimSpace(id) == "" {
+			continue
+		}
+		if id != self {
+			return false
+		}
+	}
+	return true
+}