feat: 更新限时消息

Sioncovy · May 26, 2024 · 645c34b · 645c34b
1 parent 3c46e44
commit 645c34b
Show file tree

Hide file tree

Showing 3 changed files with 44 additions and 3 deletions.
diff --git a/src/components/Chat/InputArea/index.tsx b/src/components/Chat/InputArea/index.tsx
@@ -7,6 +7,7 @@ import { ChatMessageType, MessageSpecialType } from '@/typings'
 import { createObjectURL } from '@/utils'
 import {
  AudioOutlined,
+ ClockCircleOutlined,
  FileImageOutlined,
  FileOutlined,
  FireOutlined,
@@ -193,7 +194,33 @@ function InputArea({ userId, friendId }: InputAreaProps) {
  />
  }
  onClick={() => {
- setSpecial(MessageSpecialType.BurnAfterReading)
+ setSpecial(
+ special === MessageSpecialType.BurnAfterReading
+ ? MessageSpecialType.Normal
+ : MessageSpecialType.BurnAfterReading
+ )
+ }}
+ />
+ </Tooltip>
+ <Tooltip title="限时消息：发送五秒后自动销毁">
+ <Button
+ type="text"
+ icon={
+ <ClockCircleOutlined
+ style={{
+ color:
+ special === MessageSpecialType.BurnAfterTime
+ ? token.colorPrimary
+ : undefined
+ }}
+ />
+ }
+ onClick={() => {
+ setSpecial(
+ special === MessageSpecialType.BurnAfterTime
+ ? MessageSpecialType.Normal
+ : MessageSpecialType.BurnAfterTime
+ )
  }}
  />
  </Tooltip>

diff --git a/src/components/Chat/Message/index.tsx b/src/components/Chat/Message/index.tsx
@@ -48,12 +48,15 @@ function Message({
  if (special === MessageSpecialType.BurnAfterReading) {
  if (read) {
  const destroyTime = 5000 - (dayjs().valueOf() - updatedAt.valueOf())
-
- console.log('✨ ~ useEffect ~ destroyTime:', destroyTime)
  setTimeout(() => {
  mailpenDatabase.messages.findOne({ selector: { _id } }).remove()
  }, destroyTime)
  }
+ } else if (special === MessageSpecialType.BurnAfterTime) {
+ const destroyTime = 5000 - (dayjs().valueOf() - createdAt.valueOf())
+ setTimeout(() => {
+ mailpenDatabase.messages.findOne({ selector: { _id } }).remove()
+ }, destroyTime)
  }
  }, [read])
 

diff --git a/src/components/Sider/ChatSider/ChatItem/index.tsx b/src/components/Sider/ChatSider/ChatItem/index.tsx
@@ -40,6 +40,17 @@ function ChatItem({ chat }: ChatItemProps) {
  })
  return '[阅后即焚]'
  }
+ case MessageSpecialType.BurnAfterTime: {
+ if (content === '[限时消息]') return content
+ mailpenDatabase.chats.findOne({ selector: { _id: chat._id } }).update({
+ $set: {
+ message: {
+ ...message,
+ content: '[限时消息]'
+ }
+ }
+ })
+ }
  }
  switch (Number(type)) {
  case ChatMessageType.Text: {