"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.indexGff3 = void 0;
const tslib_1 = require("tslib");
const common_1 = require("../types/common");
const cli_progress_1 = require("cli-progress");
const zlib_1 = require("zlib");
const readline_1 = (0, tslib_1.__importDefault)(require("readline"));
const path_1 = (0, tslib_1.__importDefault)(require("path"));
const fs_1 = (0, tslib_1.__importDefault)(require("fs"));
function indexGff3(config, attributes, outLocation, typesToExclude, quiet) {
    return (0, tslib_1.__asyncGenerator)(this, arguments, function* indexGff3_1() {
        var e_1, _a;
        const { adapter, trackId } = config;
        const { gffGzLocation: { uri }, } = adapter;
        // progress bar code was aided by blog post at
        // https://webomnizz.com/download-a-file-with-progressbar-using-node-js/
        const progressBar = new cli_progress_1.SingleBar({
            format: '{bar} ' + trackId + ' {percentage}% | ETA: {eta}s',
            etaBuffer: 2000,
        }, cli_progress_1.Presets.shades_classic);
        let fileDataStream;
        let totalBytes = 0;
        let receivedBytes = 0;
        if ((0, common_1.isURL)(uri)) {
            fileDataStream = yield (0, tslib_1.__await)((0, common_1.createRemoteStream)(uri));
            totalBytes = +(fileDataStream.headers['content-length'] || 0);
        }
        else {
            const filename = path_1.default.isAbsolute(uri) ? uri : path_1.default.join(outLocation, uri);
            totalBytes = fs_1.default.statSync(filename).size;
            fileDataStream = fs_1.default.createReadStream(filename);
        }
        if (!quiet) {
            progressBar.start(totalBytes, 0);
        }
        fileDataStream.on('data', chunk => {
            receivedBytes += chunk.length;
            progressBar.update(receivedBytes);
        });
        const rl = readline_1.default.createInterface({
            input: uri.endsWith('.gz')
                ? fileDataStream.pipe((0, zlib_1.createGunzip)())
                : fileDataStream,
        });
        try {
            for (var rl_1 = (0, tslib_1.__asyncValues)(rl), rl_1_1; rl_1_1 = yield (0, tslib_1.__await)(rl_1.next()), !rl_1_1.done;) {
                const line = rl_1_1.value;
                if (line.startsWith('#')) {
                    continue;
                }
                else if (line.startsWith('>')) {
                    break;
                }
                const [seq_id, , type, start, end, , , , col9] = line.split('\t');
                const locStr = `${seq_id}:${start}..${end}`;
                if (!typesToExclude.includes(type)) {
                    // turns gff3 attrs into a map, and converts the arrays into space
                    // separated strings
                    const col9attrs = Object.fromEntries(col9
                        .split(';')
                        .map(f => f.trim())
                        .filter(f => !!f)
                        .map(f => f.split('='))
                        .map(([key, val]) => [
                        key.trim(),
                        decodeURIComponent(val).trim().split(',').join(' '),
                    ]));
                    const attrs = attributes
                        .map(attr => col9attrs[attr])
                        .filter((f) => !!f);
                    if (attrs.length) {
                        const record = JSON.stringify([
                            encodeURIComponent(locStr),
                            encodeURIComponent(trackId),
                            ...attrs.map(a => encodeURIComponent(a)),
                        ]).replace(/,/g, '|');
                        yield yield (0, tslib_1.__await)(`${record} ${[...new Set(attrs)].join(' ')}\n`);
                    }
                }
            }
        }
        catch (e_1_1) { e_1 = { error: e_1_1 }; }
        finally {
            try {
                if (rl_1_1 && !rl_1_1.done && (_a = rl_1.return)) yield (0, tslib_1.__await)(_a.call(rl_1));
            }
            finally { if (e_1) throw e_1.error; }
        }
        progressBar.stop();
    });
}
exports.indexGff3 = indexGff3;
