|
| 1 | +package chats |
| 2 | + |
| 3 | +import ( |
| 4 | + "context" |
| 5 | + "go.mongodb.org/mongo-driver/bson" |
| 6 | + "go.mongodb.org/mongo-driver/bson/primitive" |
| 7 | + database "messagewith-server/chats/database" |
| 8 | + errorConstants "messagewith-server/error-constants" |
| 9 | + "messagewith-server/graph/model" |
| 10 | + "messagewith-server/users" |
| 11 | + usersDatabase "messagewith-server/users/database" |
| 12 | +) |
| 13 | + |
| 14 | +var ( |
| 15 | + repository R |
| 16 | + cachedUsers = map[string]*model.User{} |
| 17 | +) |
| 18 | + |
| 19 | +type service struct{} |
| 20 | + |
| 21 | +func getService(repo R) *service { |
| 22 | + repository = repo |
| 23 | + return &service{} |
| 24 | +} |
| 25 | + |
| 26 | +func (service *service) CreateChat(ctx context.Context, user *usersDatabase.User, secondUserID string) (*model.Chat, error) { |
| 27 | + secondUserObjectID, err := primitive.ObjectIDFromHex(secondUserID) |
| 28 | + if err != nil { |
| 29 | + return nil, errorConstants.ErrInvalidID |
| 30 | + } |
| 31 | + |
| 32 | + _, err = repository.FindOne(ctx, bson.M{"users": bson.M{"$size": 2, "$in": []primitive.ObjectID{user.ID, secondUserObjectID}}}) |
| 33 | + if err == nil { |
| 34 | + return nil, errorConstants.ErrChatAlreadyCreated |
| 35 | + } |
| 36 | + |
| 37 | + secondUser, err := users.Service.GetPlainUser(ctx, &secondUserID, nil, nil) |
| 38 | + if err != nil { |
| 39 | + return nil, errorConstants.ErrInvalidID |
| 40 | + } |
| 41 | + |
| 42 | + chat := &database.Chat{ |
| 43 | + ID: primitive.NewObjectID(), |
| 44 | + DeletedBy: []primitive.ObjectID{}, |
| 45 | + LastViewedMessages: []database.LastViewedMessage{}, |
| 46 | + Messages: []*database.Message{}, |
| 47 | + MessagesCount: 0, |
| 48 | + Users: []primitive.ObjectID{user.ID, secondUserObjectID}, |
| 49 | + } |
| 50 | + |
| 51 | + err = repository.Create(ctx, chat) |
| 52 | + if err != nil { |
| 53 | + panic(err) |
| 54 | + } |
| 55 | + |
| 56 | + return FilterChat(ctx, chat, user.ID, secondUser), nil |
| 57 | +} |
| 58 | + |
| 59 | +func FilterChat(ctx context.Context, chat *database.Chat, userObjectID primitive.ObjectID, secondUser *usersDatabase.User) *model.Chat { |
| 60 | + modelChat := &model.Chat{} |
| 61 | + modelChat.MessagesCount = int(chat.MessagesCount) |
| 62 | + modelChat.User = users.FilterUser(secondUser) |
| 63 | + modelChat.ID = chat.ID.Hex() |
| 64 | + modelChat.Messages = FilterAllMessages(ctx, userObjectID, chat.Messages) |
| 65 | + modelChat.LastViewedMessage = FilterLastViewedMessage(ctx, chat.LastViewedMessages, chat.Messages) |
| 66 | + |
| 67 | + return modelChat |
| 68 | +} |
| 69 | + |
| 70 | +func GetChatUser(ctx context.Context, userID primitive.ObjectID) *model.User { |
| 71 | + var filteredUser *model.User |
| 72 | + itemIdHex := userID.Hex() |
| 73 | + |
| 74 | + if cachedUsers[itemIdHex] != nil { |
| 75 | + filteredUser = cachedUsers[itemIdHex] |
| 76 | + } else { |
| 77 | + user, _ := users.Service.GetUser(ctx, &itemIdHex, nil, nil) |
| 78 | + cachedUsers[itemIdHex] = user |
| 79 | + filteredUser = user |
| 80 | + } |
| 81 | + |
| 82 | + return filteredUser |
| 83 | +} |
| 84 | + |
| 85 | +func GetMessage(allMessages []*database.Message, id primitive.ObjectID) *database.Message { |
| 86 | + for _, item := range allMessages { |
| 87 | + if item.ID == id { |
| 88 | + return item |
| 89 | + } |
| 90 | + } |
| 91 | + |
| 92 | + return nil |
| 93 | +} |
| 94 | + |
| 95 | +func FilterLastViewedMessage(ctx context.Context, lastViewedMessages []database.LastViewedMessage, allMessages []*database.Message) *model.LastViewedMessage { |
| 96 | + if len(lastViewedMessages) == 1 { |
| 97 | + return &model.LastViewedMessage{ |
| 98 | + Message: FilterMessage(ctx, nil, GetMessage(allMessages, lastViewedMessages[0].Message)), |
| 99 | + User: GetChatUser(ctx, lastViewedMessages[0].User), |
| 100 | + Time: int(lastViewedMessages[0].Time.Time().Unix()), |
| 101 | + } |
| 102 | + } |
| 103 | + |
| 104 | + return nil |
| 105 | +} |
| 106 | + |
| 107 | +func FilterMessage(ctx context.Context, userObjectID *primitive.ObjectID, message *database.Message) *model.Message { |
| 108 | + for _, deletedBy := range message.DeletedBy { |
| 109 | + if deletedBy == *userObjectID { |
| 110 | + return nil |
| 111 | + } |
| 112 | + } |
| 113 | + |
| 114 | + var filteredUser *model.User |
| 115 | + messageIdHex := message.ID.Hex() |
| 116 | + |
| 117 | + return &model.Message{ |
| 118 | + ID: messageIdHex, |
| 119 | + User: filteredUser, |
| 120 | + MessageType: model.MessageType(message.Type), |
| 121 | + Content: message.Content, |
| 122 | + IsForwarded: message.IsForwarded, |
| 123 | + Reactions: FilterAllReactions(ctx, message.Reactions), |
| 124 | + SendTime: int(message.SendTime.Time().Unix()), |
| 125 | + } |
| 126 | +} |
| 127 | + |
| 128 | +func FilterAllMessages(ctx context.Context, userObjectID primitive.ObjectID, messages []*database.Message) []*model.Message { |
| 129 | + allModelMessages := make([]*model.Message, 0) |
| 130 | + |
| 131 | + for _, item := range messages { |
| 132 | + filteredMessage := FilterMessage(ctx, &userObjectID, item) |
| 133 | + if filteredMessage == nil { |
| 134 | + continue |
| 135 | + } |
| 136 | + |
| 137 | + allModelMessages = append(allModelMessages, filteredMessage) |
| 138 | + } |
| 139 | + |
| 140 | + return allModelMessages |
| 141 | +} |
| 142 | + |
| 143 | +func FilterAllReactions(ctx context.Context, reactions []*database.Reaction) []*model.Reaction { |
| 144 | + allModelReactions := make([]*model.Reaction, 0) |
| 145 | + |
| 146 | + for _, item := range reactions { |
| 147 | + filteredUser := GetChatUser(ctx, item.User) |
| 148 | + |
| 149 | + allModelReactions = append(allModelReactions, &model.Reaction{ |
| 150 | + User: filteredUser, |
| 151 | + Emoji: item.Emoji, |
| 152 | + }) |
| 153 | + } |
| 154 | + |
| 155 | + return allModelReactions |
| 156 | +} |
0 commit comments