@qe-mcp/ena 0.0.0-stage → 0.1.17

package/xlsx.js

@@ -0,0 +1,130 @@
+/**
+ * xlsx.js — read the first worksheet of an .xlsx file as CSV text, as the ENA
+ * Web Tool does for an uploaded .xlsx (rENA-webtool js/XlsxReader.js, ported
+ * to Node: zlib instead of JSZip, jsdom's DOMParser instead of the browser's).
+ *
+ * Parity with the web tool, and so with R's openxlsx::read.xlsx(sheet = 1):
+ *   - first sheet in tab order only; its first row is the header row
+ *   - values as stored: dates stay Excel serial numbers (no date detection)
+ *   - booleans as TRUE / FALSE; shared and inline strings resolved
+ *   - rows are taken in file order and padded to the widest row
+ */
+
+import { inflateRawSync } from 'zlib';
+import { JSDOM } from 'jsdom';
+
+// ── ZIP: the entries of an .xlsx (a ZIP of XML parts) ────────────────────────
+function unzip(buf) {
+    // End of central directory: scan back over a possible archive comment.
+    let eocd = -1;
+    for (let i = buf.length - 22; i >= Math.max(0, buf.length - 65557); i--)
+        if (buf.readUInt32LE(i) === 0x06054b50) { eocd = i; break; }
+    if (eocd < 0) throw new Error('not an .xlsx file (no ZIP directory found)');
+    const count = buf.readUInt16LE(eocd + 10);
+    let p = buf.readUInt32LE(eocd + 16);
+    const files = new Map();
+    for (let n = 0; n < count; n++) {
+        if (buf.readUInt32LE(p) !== 0x02014b50) throw new Error('corrupt .xlsx (bad ZIP directory)');
+        const method = buf.readUInt16LE(p + 10);
+        const size = buf.readUInt32LE(p + 20);
+        const nameLen = buf.readUInt16LE(p + 28), extraLen = buf.readUInt16LE(p + 30), commentLen = buf.readUInt16LE(p + 32);
+        const local = buf.readUInt32LE(p + 42);
+        const name = buf.toString('utf8', p + 46, p + 46 + nameLen);
+        p += 46 + nameLen + extraLen + commentLen;
+        files.set(name, () => {
+            const start = local + 30 + buf.readUInt16LE(local + 26) + buf.readUInt16LE(local + 28);
+            const data = buf.subarray(start, start + size);
+            if (method === 0) return data;
+            if (method === 8) return inflateRawSync(data);
+            throw new Error(`unsupported compression in .xlsx entry ${name}`);
+        });
+    }
+    return { file: (name) => files.get(name)?.() ?? null };
+}
+
+let _parser;
+function parseXml(buf) {
+    _parser ??= new (new JSDOM('').window.DOMParser)();
+    return _parser.parseFromString(buf.toString('utf8'), 'application/xml');
+}
+
+// "A1" / "BC12" → 0-based column. Column letters are base-26 (bijective).
+function refCol(ref) {
+    let col = 0;
+    for (let i = 0; i < ref.length; i++) {
+        const ch = ref.charCodeAt(i);
+        if (ch < 65 || ch > 90) break;
+        col = col * 26 + (ch - 64);
+    }
+    return col - 1;
+}
+
+// Text of a shared-string <si> or inline <is> (rich-text runs concatenated).
+function textOf(elem) {
+    if (!elem) return '';
+    let out = '';
+    for (const t of elem.getElementsByTagName('t')) out += t.textContent;
+    return out;
+}
+
+function firstSheetPath(zip) {
+    const wb = zip.file('xl/workbook.xml');
+    if (!wb) throw new Error('not an .xlsx workbook (xl/workbook.xml missing)');
+    const sheets = parseXml(wb).getElementsByTagName('sheet');
+    if (!sheets.length) throw new Error('the workbook has no sheets');
+    const rid = sheets[0].getAttribute('r:id')
+        || sheets[0].getAttributeNS('http://schemas.openxmlformats.org/officeDocument/2006/relationships', 'id');
+    const rels = zip.file('xl/_rels/workbook.xml.rels');
+    if (rels) {
+        for (const rel of parseXml(rels).getElementsByTagName('Relationship')) {
+            if (rel.getAttribute('Id') !== rid) continue;
+            const target = rel.getAttribute('Target');
+            if (target.charAt(0) === '/') return target.slice(1);
+            return target.indexOf('xl/') === 0 ? target : 'xl/' + target;
+        }
+    }
+    return 'xl/worksheets/sheet1.xml';
+}
+
+function csvField(v) {
+    if (v == null) return '';
+    const s = String(v);
+    return /[",\r\n]/.test(s) ? '"' + s.replace(/"/g, '""') + '"' : s;
+}
+
+function cellValue(cell, shared) {
+    const t = cell.getAttribute('t');           // s | inlineStr | str | b | e | n | (none)
+    if (t === 'inlineStr') return textOf(cell.getElementsByTagName('is')[0]);
+    const v = cell.getElementsByTagName('v')[0];
+    const raw = v ? v.textContent : '';
+    if (t === 's') return shared[parseInt(raw, 10)] || '';
+    if (t === 'b') return raw === '1' ? 'TRUE' : 'FALSE';
+    return raw;
+}
+
+/** CSV text of the first worksheet of the .xlsx in buf. */
+export function xlsxToCSV(buf) {
+    const zip = unzip(buf);
+    const sst = zip.file('xl/sharedStrings.xml');
+    const shared = sst ? Array.from(parseXml(sst).getElementsByTagName('si'), textOf) : [];
+    const path = firstSheetPath(zip);
+    const sheet = zip.file(path);
+    if (!sheet) throw new Error(`worksheet not found: ${path}`);
+
+    const grid = [];
+    let maxCol = -1;
+    for (const rowEl of parseXml(sheet).getElementsByTagName('row')) {
+        const row = [];
+        Array.from(rowEl.getElementsByTagName('c')).forEach((cell, c) => {
+            const ref = cell.getAttribute('r');
+            const col = ref ? refCol(ref) : c;
+            row[col] = cellValue(cell, shared);
+            if (col > maxCol) maxCol = col;
+        });
+        grid.push(row);
+    }
+    if (maxCol < 0) return '';
+    return grid.map(row => Array.from({ length: maxCol + 1 }, (_, c) => csvField(row[c])).join(',')).join('\r\n');
+}
+
+export const isXlsx = (path) => /\.xlsx$/i.test(String(path ?? ''));