refactor(extensions): reorganize operations extensions

src/index.ts

@@ -10,10 +10,12 @@ export type {
 } from './operations/domains';
 export type {
   CreateExtension,
+  CreateExtensionFn,
   CreateExtensionOptions,
   DropExtension,
   Extension,
-} from './operations/extensionsTypes';
+  StringExtension,
+} from './operations/extensions';
 export type {
   CreateFunction,
   DropFunction,

src/operations/extensions/createExtension.ts

@@ -0,0 +1,37 @@
+import type { MigrationOptions } from '../../types';
+import type { DropOptions, IfNotExistsOption } from '../generalTypes';
+import { dropExtension } from './dropExtension';
+import type { StringExtension } from './shared';
+
+export interface CreateExtensionOptions extends IfNotExistsOption {
+  schema?: string;
+}
+
+export type CreateExtensionFn = (
+  extension: StringExtension | StringExtension[],
+  options?: CreateExtensionOptions & DropOptions
+) => string | string[];
+
+export type CreateExtension = CreateExtensionFn & {
+  reverse: CreateExtensionFn;
+};
+
+export function createExtension(mOptions: MigrationOptions): CreateExtension {
+  const _create: CreateExtension = (_extensions, options = {}) => {
+    const { ifNotExists, schema } = options;
+
+    const extensions = Array.isArray(_extensions) ? _extensions : [_extensions];
+    const ifNotExistsStr = ifNotExists ? ' IF NOT EXISTS' : '';
+    const schemaStr = schema ? ` SCHEMA ${mOptions.literal(schema)}` : '';
+
+    return extensions.map((extension) => {
+      const extensionStr = mOptions.literal(extension);
+
+      return `CREATE EXTENSION${ifNotExistsStr} ${extensionStr}${schemaStr};`;
+    });
+  };
+
+  _create.reverse = dropExtension(mOptions);
+
+  return _create;
+}

src/operations/extensions/dropExtension.ts

@@ -0,0 +1,26 @@
+import type { MigrationOptions } from '../../types';
+import type { DropOptions } from '../generalTypes';
+import type { StringExtension } from './shared';
+
+export type DropExtension = (
+  extension: StringExtension | StringExtension[],
+  dropOptions?: DropOptions
+) => string | string[];
+
+export function dropExtension(mOptions: MigrationOptions): DropExtension {
+  const _drop: DropExtension = (_extensions, options = {}) => {
+    const { ifExists, cascade } = options;
+
+    const extensions = Array.isArray(_extensions) ? _extensions : [_extensions];
+    const ifExistsStr = ifExists ? ' IF EXISTS' : '';
+    const cascadeStr = cascade ? ' CASCADE' : '';
+
+    return extensions.map((extension) => {
+      const extensionStr = mOptions.literal(extension);
+
+      return `DROP EXTENSION${ifExistsStr} ${extensionStr}${cascadeStr};`;
+    });
+  };
+
+  return _drop;
+}

src/operations/extensions/index.ts

@@ -0,0 +1,3 @@
+export * from './createExtension';
+export * from './dropExtension';
+export * from './shared';

src/operations/extensionsTypes.ts → src/operations/extensions/shared.ts

@@ -1,8 +1,4 @@
-import type {
-  DropOptions,
-  IfNotExistsOption,
-  LiteralUnion,
-} from './generalTypes';
+import type { LiteralUnion } from '../generalTypes';
 
 export type Extension =
   | 'adminpack'
@@ -51,22 +47,4 @@ export type Extension =
   | 'uuid-ossp'
   | 'xml2';
 
-export interface CreateExtensionOptions extends IfNotExistsOption {
-  schema?: string;
-}
-
-type StringExtension = LiteralUnion<Extension>;
-
-type CreateExtensionFn = (
-  extension: StringExtension | StringExtension[],
-  options?: CreateExtensionOptions & DropOptions
-) => string | string[];
-
-export type CreateExtension = CreateExtensionFn & {
-  reverse: CreateExtensionFn;
-};
-
-export type DropExtension = (
-  extension: StringExtension | StringExtension[],
-  dropOptions?: DropOptions
-) => string | string[];
+export type StringExtension = LiteralUnion<Extension>;

src/types.ts

@@ -7,7 +7,7 @@ import type {
   QueryResult,
 } from 'pg';
 import type * as domains from './operations/domains';
-import type * as extensions from './operations/extensionsTypes';
+import type * as extensions from './operations/extensions';
 import type * as functions from './operations/functionsTypes';
 import type { Name } from './operations/generalTypes';
 import type * as indexes from './operations/indexesTypes';