33 lines
804 B
JavaScript
33 lines
804 B
JavaScript
'use strict';
|
|
|
|
const COPY_FORMATS = ['TEXT', 'CSV', 'BINARY'];
|
|
|
|
const regex = /\bFORMAT\s+(\w+)/;
|
|
|
|
module.exports = {
|
|
getFormatFromCopyQuery (copyQuery) {
|
|
let format = 'TEXT'; // Postgres default format
|
|
|
|
copyQuery = copyQuery.toUpperCase();
|
|
|
|
if (!copyQuery.startsWith('COPY ')) {
|
|
return false;
|
|
}
|
|
|
|
if (copyQuery.includes(' WITH') && copyQuery.includes('FORMAT ')) {
|
|
const result = regex.exec(copyQuery);
|
|
|
|
if (result && result.length === 2) {
|
|
if (COPY_FORMATS.includes(result[1])) {
|
|
format = result[1];
|
|
format = format.toUpperCase();
|
|
} else {
|
|
format = false;
|
|
}
|
|
}
|
|
}
|
|
|
|
return format;
|
|
}
|
|
};
|