dennii/dbgate
1
0
Fork 0
mirror of https://github.com/DeNNiiInc/dbgate.git synced 2026-04-19 03:45:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

open SQL file - try to detech encoding

Browse Source
This commit is contained in:
Jan Prochazka
2021-11-25 11:15:24 +01:00
parent 4f9566e630
commit 5de231dc1c
1 changed files with 18 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
packages/web/src/utility/openElectronFile.ts
View File

@@ -34,6 +34,22 @@ export async function openSqliteFile(filePath) {
});
}
function getFileEncoding(filePath, fs) {
var buf = Buffer.alloc(5);
var fd = fs.openSync(filePath, 'r');
fs.readSync(fd, buf, 0, 5, 0);
fs.closeSync(fd);
// https://en.wikipedia.org/wiki/Byte_order_mark
let e = null;
if (!e && buf[0] === 0xef && buf[1] === 0xbb && buf[2] === 0xbf) e = 'utf8';
if (!e && buf[0] === 0xfe && buf[1] === 0xff) e = 'utf16be';
if (!e && buf[0] === 0xff && buf[1] === 0xfe) e = 'utf16le';
if (!e) e = 'ascii';
return e;
}
export function openElectronFileCore(filePath, extensions) {
const nameLower = filePath.toLowerCase();
const path = window.require('path');
@@ -42,7 +58,8 @@ export function openElectronFileCore(filePath, extensions) {
const uploadListener = getUploadListener();
if (nameLower.endsWith('.sql')) {
const data = fs.readFileSync(filePath, { encoding: 'utf-8' });
const encoding = getFileEncoding(filePath, fs);
const data = fs.readFileSync(filePath, { encoding });
newQuery({
title: parsed.name,