msimerson/Haraka

'use strict';

const fs      = require('fs');

class FsyncWriteStream extends fs.WriteStream {
    constructor (path, options) {
        super(path, options);
    }

    close (cb) {
        const self = this;
        if (cb) this.once('close', cb);

        if (this.closed || 'number' !== typeof this.fd) {
            if ('number' !== typeof this.fd) {
                this.once('open', close);
                return;
            }
            return setImmediate(this.emit.bind(this, 'close'));
        }
        this.closed = true;
        close();

        function close (fd) {
            fs.fsync(fd || self.fd, er => {
                if (er) {
                    self.emit('error', er);
                    return;
                }

                fs.close(fd || self.fd, err => {
                    if (err) {
                        self.emit('error', err);
                    }
                    else {
                        self.emit('close');
                    }
                });
                self.fd = null;
            });
        }
    }
}

module.exports = FsyncWriteStream;