mirror of
https://github.com/cpvalente/ontime.git
synced 2026-08-05 15:33:59 +00:00
fix: prevent sheet name too long in excel export
This commit is contained in:
committed by
Carlos Valente
parent
1cee679397
commit
fd39cab845
@@ -0,0 +1,37 @@
|
||||
import xlsx from 'xlsx';
|
||||
|
||||
import { demoDb } from '../../../models/demoProject.js';
|
||||
import { generateExcelFile } from '../excel.service.js';
|
||||
|
||||
describe('generateExcelFile()', () => {
|
||||
it('sanitises long worksheet names to an Excel-compatible value', () => {
|
||||
const buffer = generateExcelFile(
|
||||
{
|
||||
...demoDb.rundowns.default,
|
||||
title: 'This is a very long name with many characters and weird things: like [Main]/?*',
|
||||
},
|
||||
demoDb.customFields,
|
||||
);
|
||||
|
||||
const workbook = xlsx.read(buffer, { type: 'buffer' });
|
||||
const worksheetName = workbook.SheetNames[0];
|
||||
|
||||
expect(worksheetName).toBeDefined();
|
||||
expect(worksheetName.length).toBeLessThanOrEqual(31);
|
||||
expect(worksheetName).not.toMatch(/[:\\/?*3[\]]/);
|
||||
});
|
||||
|
||||
it('falls back to default worksheet name when title is fully invalid', () => {
|
||||
const buffer = generateExcelFile(
|
||||
{
|
||||
...demoDb.rundowns.default,
|
||||
title: '[]:*?/\\',
|
||||
},
|
||||
demoDb.customFields,
|
||||
);
|
||||
|
||||
const workbook = xlsx.read(buffer, { type: 'buffer' });
|
||||
|
||||
expect(workbook.SheetNames[0]).toBe('Rundown');
|
||||
});
|
||||
});
|
||||
@@ -21,6 +21,18 @@ import { rundownToTabular } from './excel.utils.js';
|
||||
// we keep the excel data in memory to allow the flow upload -> preview
|
||||
let excelData: WorkBook = xlsx.utils.book_new();
|
||||
|
||||
const maxWorksheetNameLength = 31;
|
||||
const invalidWorksheetCharsRegex = /[:\\/?*[\]]/g;
|
||||
|
||||
function getValidWorksheetName(title: string): string {
|
||||
const sanitisedTitle = title.replaceAll(invalidWorksheetCharsRegex, ' ').trim().replace(/\s+/g, ' ');
|
||||
|
||||
const withFallback = sanitisedTitle.length > 0 ? sanitisedTitle : 'Rundown';
|
||||
const truncatedTitle = withFallback.slice(0, maxWorksheetNameLength).trim();
|
||||
|
||||
return truncatedTitle.length > 0 ? truncatedTitle : 'Rundown';
|
||||
}
|
||||
|
||||
/**
|
||||
* Receives and parses an excel file
|
||||
* The file is deleted after being read
|
||||
@@ -93,7 +105,8 @@ export function generateExcelFile(rundown: Rundown, customFields: CustomFields):
|
||||
|
||||
const workbook = xlsx.utils.book_new();
|
||||
const worksheet = xlsx.utils.aoa_to_sheet(rundownToTabular(rundown, customFields));
|
||||
xlsx.utils.book_append_sheet(workbook, worksheet, rundown.title || 'Rundown');
|
||||
const worksheetName = getValidWorksheetName(rundown.title || 'Rundown');
|
||||
xlsx.utils.book_append_sheet(workbook, worksheet, worksheetName);
|
||||
|
||||
return xlsx.write(workbook, { type: 'buffer', bookType: 'xlsx' });
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user