-
-
Notifications
You must be signed in to change notification settings - Fork 6
/
utils.ts
36 lines (31 loc) · 1.09 KB
/
utils.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
import { workspace } from 'coc.nvim';
import fs from 'fs';
import { inspect } from 'util';
// util to debug values when developing
// tail -f /tmp/coc-sql-debug
export const log = (something: any) => {
fs.writeFileSync('/tmp/coc-sql-debug', inspect(something, { depth: null }) + '\n', { flag: 'a' });
};
const defaultDatabase = 'mysql';
const supportedDatabases = ['db2', 'bigquery', 'hive', 'mysql', 'mariadb', 'postgresql', 'transactsql'];
// Guess database from environment variable DATABASE_URL
function guessDatabase(): string {
if (process.env.DATABASE_URL) {
let dbType = process.env.DATABASE_URL.toLowerCase().split(':')[0];
if (dbType === 'postgres') {
dbType = 'postgresql';
}
if (supportedDatabases.includes(dbType)) {
return dbType;
}
}
return defaultDatabase;
}
// gets database from coc config or else, tries to guess value from DATABASE_URL env variable
export function getDatabase(): string {
let database = workspace.getConfiguration('sql').get('database') as string;
if (database === 'guess') {
database = guessDatabase();
}
return database;
}