Reusable websocket component

nuxt\server\services\email\parser.ts
import fs from 'fs'
import { simpleParser } from 'mailparser'
....
import { sendWebSocetMessage } from 'websocketHelper'

// some mail parsing things

// after mail parsing send message
sendWebSocetMessage('newEmailSaved')

nuxt\server\services\email\parser.ts
import fs from 'fs'
import { simpleParser } from 'mailparser'
....
import { sendWebSocetMessage } from 'websocketHelper'

// some mail parsing things

// after mail parsing send message
sendWebSocetMessage('newEmailSaved')

let mqSubscription = null;

export default defineWebSocketHandler({
  open(peer) {
    mqSubscription = mq.subscribe('global-notifications', () => peer.publish(...send to client if needed...))
  },
  message(peer, message) {
    // (optional) reacting to messages from client
  },
  close(peer) {
    mqSubscription.destroy()
  },
})

let mqSubscription = null;

export default defineWebSocketHandler({
  open(peer) {
    mqSubscription = mq.subscribe('global-notifications', () => peer.publish(...send to client if needed...))
  },
  message(peer, message) {
    // (optional) reacting to messages from client
  },
  close(peer) {
    mqSubscription.destroy()
  },
})

let ws
const reconnectInterval = 5000 // Time in ms to wait before attempting to reconnect
let isManuallyClosed = false // Flag to prevent reconnection when connection is manually closed

// Function to handle WebSocket connection
export const wsConnect = async () => {
  const isSecure = process.env.SITE_PROTOCOL === 'https'
  const url = (isSecure ? 'wss://' : 'ws://') + process.env.SITE_NAME + '/api/websocket'

  // Close existing connection if present
  if (ws) {
    console.log('ws', 'Closing previous connection before reconnecting...')
    wsClose()
  }

  console.log('ws', 'Connecting to', url, '...')
  ws = new WebSocket(url)

  // Wait for the WebSocket to open
  await new Promise((resolve) => ws.addEventListener('open', resolve))

  console.log('ws', 'Connected!')

  // Handle connection close
  ws.addEventListener('close', handleWsClose)

  // Handle connection errors
  ws.addEventListener('error', handleWsError)
}

// Function to handle WebSocket closure
const handleWsClose = () => {
  console.log('ws', 'Connection closed.')
  if (!isManuallyClosed) {
    console.log('ws', `Attempting to reconnect in ${reconnectInterval / 1000} seconds...`)
    setTimeout(wsConnect, reconnectInterval) // Attempt to reconnect after delay
  }
}

// Function to handle WebSocket errors
const handleWsError = (error: any) => {
  console.error('ws', 'Connection error:', error)
  ws.close() // Close and trigger the reconnection logic
}

// Function to manually close the WebSocket connection
export const wsClose = () => {
  if (ws && ws.readyState === WebSocket.OPEN) {
    console.log('ws', 'Manually closing the connection...')
    isManuallyClosed = true
    ws.close()
  } else {
    console.log('ws', 'Connection is already closed.')
  }
}

let ws
const reconnectInterval = 5000 // Time in ms to wait before attempting to reconnect
let isManuallyClosed = false // Flag to prevent reconnection when connection is manually closed

// Function to handle WebSocket connection
export const wsConnect = async () => {
  const isSecure = process.env.SITE_PROTOCOL === 'https'
  const url = (isSecure ? 'wss://' : 'ws://') + process.env.SITE_NAME + '/api/websocket'

  // Close existing connection if present
  if (ws) {
    console.log('ws', 'Closing previous connection before reconnecting...')
    wsClose()
  }

  console.log('ws', 'Connecting to', url, '...')
  ws = new WebSocket(url)

  // Wait for the WebSocket to open
  await new Promise((resolve) => ws.addEventListener('open', resolve))

  console.log('ws', 'Connected!')

  // Handle connection close
  ws.addEventListener('close', handleWsClose)

  // Handle connection errors
  ws.addEventListener('error', handleWsError)
}

// Function to handle WebSocket closure
const handleWsClose = () => {
  console.log('ws', 'Connection closed.')
  if (!isManuallyClosed) {
    console.log('ws', `Attempting to reconnect in ${reconnectInterval / 1000} seconds...`)
    setTimeout(wsConnect, reconnectInterval) // Attempt to reconnect after delay
  }
}

// Function to handle WebSocket errors
const handleWsError = (error: any) => {
  console.error('ws', 'Connection error:', error)
  ws.close() // Close and trigger the reconnection logic
}

// Function to manually close the WebSocket connection
export const wsClose = () => {
  if (ws && ws.readyState === WebSocket.OPEN) {
    console.log('ws', 'Manually closing the connection...')
    isManuallyClosed = true
    ws.close()
  } else {
    console.log('ws', 'Connection is already closed.')
  }
}

// Function to send a message
export const wsSend = (message: string) => {
  if (ws && ws.readyState === WebSocket.OPEN) {
    console.log('sending message...', message)
    ws.send(message)
  } else {
    console.error('WebSocket is not open. Cannot send message:', message)
  }
}

// Function to send a ping message
export const ping = () => {
  if (ws && ws.readyState === WebSocket.OPEN) {
    console.log('ws', 'Sending ping')
    ws.send('ping')
  } else {
    console.error('WebSocket is not open. Cannot send ping.')
  }
}

// Function to connect, send a message, and disconnect
export const wsSendOnce = async (message: string) => {
  if (ws && ws.readyState === WebSocket.OPEN) {
    wsSend(message)
  } else {
    await wsConnect()
    wsSend(message)
  }
}

// Function to send a message
export const wsSend = (message: string) => {
  if (ws && ws.readyState === WebSocket.OPEN) {
    console.log('sending message...', message)
    ws.send(message)
  } else {
    console.error('WebSocket is not open. Cannot send message:', message)
  }
}

// Function to send a ping message
export const ping = () => {
  if (ws && ws.readyState === WebSocket.OPEN) {
    console.log('ws', 'Sending ping')
    ws.send('ping')
  } else {
    console.error('WebSocket is not open. Cannot send ping.')
  }
}

// Function to connect, send a message, and disconnect
export const wsSendOnce = async (message: string) => {
  if (ws && ws.readyState === WebSocket.OPEN) {
    wsSend(message)
  } else {
    await wsConnect()
    wsSend(message)
  }
}