@or-sdk/pgsql 1.0.3-beta.1920.0 → 1.0.3-beta.1935.0

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';