@or-sdk/pgsql 1.0.3-beta.1942.0 → 1.0.3-beta.1943.0

package/src/types.ts

@@ -25,6 +25,11 @@ export type PgsqlConfig = {
    * Api version
    */
   version?: string;
+
+  /**
+   * use external database
+   */
+  isExternal?: boolean;
 };
 
 export type QueryField = {
@@ -49,9 +54,18 @@ export type ExecuteQueryResponse<T> = {
 
 export type ExecuteQueryArgs = {
   query: string;
-  params?: any[];
+  params?: {
+    [key: string]: unknown;
+  };
   database: string;
-  isExternal: boolean;
+};
+
+export type CreateDatabaseResponse = {
+  ok: number;
+  result: [
+    unknown[],
+    unknown[],
+  ];
 };
 
 export type ListDatabasesResponseItem = {
@@ -73,7 +87,7 @@ export type GetDBSizeResult = {
 
 export type TableColumn = {
   name: string;
-  type: string;
+  type: number;
 };
 
 export type CreateTableArgs = {
@@ -82,7 +96,6 @@ export type CreateTableArgs = {
   table: string;
   columns: TableColumn[];
   primaryKey: string;
-  isExternal: boolean;
 };
 
 export type AddColumnsArgs = {
@@ -90,7 +103,6 @@ export type AddColumnsArgs = {
   schema: string;
   table: string;
   columns: TableColumn[];
-  isExternal: boolean;
 };
 
 export type Row = {
@@ -106,19 +118,9 @@ export type InsertArgs = {
   context?: {
     progress?: number;
   };
-  isExternal: boolean;
 };
 
-export type GetTableDataProps = {
-  database: string;
-  schema: string;
-  table: string;
-  isExternal: boolean;
-  chunkSize?: number;
-  offset?: number;
-  limit?: number;
-  params?: any[];
-};
+export type EncodedValue = string | number | boolean;
 
 export type SelectAllArgs = {
   database: string;
@@ -130,8 +132,6 @@ export type SelectAllArgs = {
   chunkSize?: number;
   offset?: number;
   limit?: number;
-  isExternal: boolean;
-  params?: any[];
 };
 
 export type EditRowArgs = {
@@ -142,7 +142,6 @@ export type EditRowArgs = {
   value: unknown;
   row: Row;
   primaryKeys: string;
-  isExternal: boolean;
 };
 
 export type DeleteRowsArgs = {
@@ -151,7 +150,6 @@ export type DeleteRowsArgs = {
   table: string;
   rows: Row[];
   primaryKeys: string;
-  isExternal: boolean;
 };
 
 export type DatabaseItem = {

package/src/utils/encodeValue.ts

@@ -0,0 +1,11 @@
+import { EncodedValue } from '../types';
+import _ from 'lodash';
+
+function encodeValue(value: unknown): EncodedValue {
+  if (_.isUndefined(value) || value === '' || _.isNull(value)) return 'NULL';
+  if (_.isString(value)) return `'${value.replaceAll('\'', '\'\'')}'`;
+  if (_.isNumber(value) || _.isBoolean(value)) return value;
+  return `'${JSON.stringify(value).replaceAll('\'', '\'\'')}'`;
+}
+
+export default encodeValue;

package/src/utils/extractDatabaseNames.ts

@@ -0,0 +1,7 @@
+import { ListDatabasesResponseItem } from '../types';
+
+function extractDatabaseNames(databaseNameArr: ListDatabasesResponseItem[]): string[] {
+  return databaseNameArr.map(({ name }) => name);
+}
+
+export default extractDatabaseNames;

package/src/utils/getDeleteRowsQuery.ts

@@ -0,0 +1,10 @@
+import { Row } from '../types';
+import { encodeValue, splitPrimaryKeys } from '../utils';
+
+function getDeleteRowQuery(schema: string, table: string, rows: Row[], primaryKeys: string): string {
+  const pk = splitPrimaryKeys(primaryKeys);
+  return `delete from ${schema}.${table}
+where (${rows.map(row => `${pk.map(primaryKey => `${primaryKey} = ${encodeValue(row[primaryKey])}`).join(' AND ')}`).join(') \nOR (')})`;
+}
+
+export default getDeleteRowQuery;

package/src/utils/getEditRowQuery.ts

@@ -0,0 +1,12 @@
+import { encodeValue, splitPrimaryKeys } from '../utils';
+import { Row } from '../types';
+
+function getEditRowQuery(schema: string, table: string, key: string, value: unknown, row: Row, primaryKeys: string): string {
+  value = encodeValue(value);
+  const pk = splitPrimaryKeys(primaryKeys);
+  return `update ${schema}.${table}
+set ${key} = ${value}
+where ${pk.map(primaryKey => `${primaryKey} = ${encodeValue(row[primaryKey])}`).join(' AND ')}`;
+}
+
+export default getEditRowQuery;

package/src/utils/getInsertKeys.ts

@@ -0,0 +1,9 @@
+import { Row } from '../types';
+
+function getInsertKeys(rows: Row[]): string[] {
+  const allKeys = {};
+  rows.forEach(x => Object.assign(allKeys, x));
+  return Object.keys(allKeys);
+}
+
+export default getInsertKeys;

package/src/utils/getInsertQueries.ts

@@ -0,0 +1,18 @@
+import { Row } from '../types';
+import { encodeValue, getInsertKeys } from '../utils';
+import _ from 'lodash';
+
+function getInsertQueries(schema: string, table: string, insertRows: Row[], chunkSize: number): string[] {
+  const rows = _.isArray(insertRows) ? insertRows : [insertRows];
+
+  const keys = getInsertKeys(rows);
+  const stringifyRow = (row: Row) => `${keys.map(key => encodeValue(row[key])).join(', ')}`;
+
+  return _.chain(rows)
+    .chunk(chunkSize)
+    .map((chunk) => `insert into ${schema}.${table} (${keys.join(', ')})
+values (${chunk.map(stringifyRow).join('), (')});`)
+    .value();
+}
+
+export default getInsertQueries;

package/src/utils/index.ts

@@ -2,6 +2,9 @@
  * @internal
  */
 
+export { default as extractDatabaseNames } from './extractDatabaseNames';
+export { default as getInsertKeys } from './getInsertKeys';
+export { default as encodeValue } from './encodeValue';
 export { default as splitPrimaryKeys } from './splitPrimaryKeys';
 export { default as getListTablesQuery } from './getListTablesQuery';
 export { default as createSchemaQuery } from './createSchemaQuery';
@@ -9,7 +12,10 @@ export { default as getDropSchemaQuery } from './getDropSchemaQuery';
 export { default as getCreateTableQuery } from './getCreateTableQuery';
 export { default as getDropTableQuery } from './getDropTableQuery';
 export { default as getAddColumnsQuery } from './getAddColumnsQuery';
+export { default as getInsertQueries } from './getInsertQueries';
 export { default as getSelectAllCountQuery } from './getSelectAllCountQuery';
 export { default as getSelectAllQuery } from './getSelectAllQuery';
+export { default as getEditRowQuery } from './getEditRowQuery';
+export { default as getDeleteRowsQuery } from './getDeleteRowsQuery';
 export { default as getGenerateTableSchemaQuery } from './getGenerateTableSchemaQuery';
 export { default as getGetPrimaryKeysQuery } from './getGetPrimaryKeysQuery';