import { DatabaseTypeCode } from '@/constants';
import sqlServer from '@/service/sql';
import { format } from 'sql-formatter';

/**
 * 格式化sql
 */
export function formatSql(sql: string, dbType: DatabaseTypeCode) {
  const arr = [
    'bigquery',
    'db2',
    'hive',
    'mariadb',
    'mysql',
    'n1ql',
    'plsql',
    'postgresql',
    'redshift',
    'spark',
    'sqlite',
    'sql',
    'trino',
    'transactsql',
    'singlestoredb',
    'snowflake',
  ];
  const language = arr.includes(dbType.toLowerCase()) ? dbType.toLowerCase() : 'sql';
  return new Promise((r: (sql: string) => void) => {
    let formatRes = '';
    // debugger;
    try {
      formatRes = format(sql || '', { language });
      // formatRes = '';
    } catch (e) {
      console.log('Frontend format sql error', e);
    }
    // // 如果格式化失败，直接返回原始sql
    if (!formatRes) {
      sqlServer
        .sqlFormat({
          sql,
          dbType,
        })
        .then((res) => {
          formatRes = res;
          r(formatRes);
        });
    } else {
      r(formatRes);
    }
  });
}