line_socket.js

'use strict'
// A subclass of Socket which reads data by line

const net = require('node:net')
const utils = require('haraka-utils')

const tls_socket = require('./tls_socket')

class Socket extends net.Socket {
    constructor(options) {
        super(options)
        setup_line_processor(this)
    }
}

function setup_line_processor(socket) {
    let current_data = ''

    socket.on('data', function on_socket_data(data) {
        current_data += data
        let results
        while ((results = utils.line_regexp.exec(current_data))) {
            const this_line = results[1]
            current_data = current_data.slice(this_line.length)
            socket.emit('line', this_line)
        }
    })

    socket.on('end', function on_socket_end() {
        if (current_data.length) {
            socket.emit('line', current_data)
        }
        current_data = ''
    })
}

exports.Socket = Socket

// New interface - uses TLS
exports.connect = (port, host) => {
    let options = {}
    if (typeof port === 'object') {
        options = port
    } else {
        options.port = port
        options.host = host
    }
    const sock = tls_socket.connect(options)
    setup_line_processor(sock)
    return sock
}