barechat/index.js

Anonymous chat anywhere with commandline

/* global Bare */
import readline from 'readline'  // Module for reading user input in terminal
import tty from 'tty'            // Module to control terminal behavior
import process from 'process'    // Process control for Bare

import { getBackend } from './lib/chat-core'
import { createFileStore, parseArgs } from './lib/helper'

// Parse command line arguments
const args = parseArgs(Bare.argv.slice(2))
// console.log('args:', JSON.stringify(args))

// Initialize file store with provided path (or null if storage disabled)
const fileStore = createFileStore(args.store)

const {
  swarm,
  getMemberId,
  joinRoom,
  sendMessage
} = getBackend(args)

Bare
  .on('suspend', () => console.log('suspended'))
  .on('resume', () => console.log('resumed'))
  .on('exit', () => {
    sendMessage('leaved')
    console.log('exited')
    swarm.destroy()
  })

const rl = readline.createInterface({
  input: new tty.ReadStream(0),
  output: new tty.WriteStream(1)
})

// When there's a new connection, listen for new messages, and output them to the terminal
swarm.on('connection', peer => {
  const memberId = getMemberId(peer)
  console.log(`[info] New peer ${memberId} joined`)
  peer.on('data', event => appendMessage({ memberId, event: JSON.parse(event) }))
  peer.on('error', e => console.log(`Connection error: ${e}`))
})

// When there's updates to the swarm, update the peers count
swarm.on('update', () => {
  console.log(`[info] Number of connections is now ${swarm.connections.size}`)
})

// create or join with a topic
await joinChatRoom(args.topic || '')

rl.input.setMode(tty.constants.MODE_RAW) // Enable raw input mode for efficient key reading
rl.on('data', line => {
  sendMessage(line)
  // Store own messages too
  appendMessage({ memberId: 'me', event: { message: line } })
  rl.prompt()
})
rl.prompt()

rl.on('close', () => {
  process.kill(process.pid, 'SIGINT')
})

async function joinChatRoom (topicStr) {
  const { done, topic } = await joinRoom(topicStr)
  if (done) {
    console.log(`[info] Joined chat room ${topic}`)
    fileStore.setupLogFile(topic)
  } else {
    console.log(`[info] Joined fail`)
  }
}

function appendMessage ({ memberId, event }) {
  // Output chat msgs to terminal
  console.log(`[${memberId}] ${event?.message}`)

  // Save message to log file using the fileStore
  if (fileStore.isEnabled()) {
    fileStore.saveMessage(memberId, event?.message)
  }
}