import { ContainerDefinition, IndexKind, DataType, Database } from '@azure/cosmos';
import { AZURE_COSMOS_DB_ENTITY } from './cosmos-db.decorators';
import { getConnectionToken, getModelToken, pluralize } from './cosmos-db.utils';
import {Scope} from '@nestjs/common';

export interface PartitionKeyValues {
  PartitionKey: string;
}

export function createAzureCosmosDbProviders(
  connectionName?: string,
  models: { dto: any; collection?: string }[] = [],
) {
  const providers = (models || []).map(model => ({
    provide: getModelToken(model.dto.name),
    useFactory: async (database: Database) => {
      const entityDescriptor = Reflect.getMetadata(AZURE_COSMOS_DB_ENTITY, model.dto) as PartitionKeyValues;
      const partitionKey = entityDescriptor ? entityDescriptor.PartitionKey : null;
      const containerName = model.collection ?? pluralize(model.dto.name);
      const containerOptions: ContainerDefinition = {
        id: containerName,
      };

      // If the container has a DateTime field we add a Range Index
      // ref: https://docs.microsoft.com/en-us/azure/cosmos-db/working-with-dates#indexing-datetimes-for-range-queries
      if (Object.values(entityDescriptor).indexOf('DateTime') > -1) {
        containerOptions.indexingPolicy = {
          includedPaths: [
            {
              path: `/*`,
              indexes: [
                {
                  kind: IndexKind.Range,
                  dataType: DataType.String,
                  precision: -1,
                },
              ],
            },
          ],
        };
      }

      for (const key in entityDescriptor) {
        if (entityDescriptor.hasOwnProperty(key)) {
          const element = entityDescriptor[key];
          if (element === 'UniqueKey') {
            containerOptions.uniqueKeyPolicy.uniqueKeys.push({ paths: [`/${element}`] });
          }
        }
      }

      if (partitionKey != null) {
        containerOptions.partitionKey = {
          paths: [`/${partitionKey}`],
        };
      }
      // console.log('get container.', containerOptions.id)
      const coResponse = await database.containers.createIfNotExists(containerOptions);

      return coResponse.container;
    },
    inject: [getConnectionToken(connectionName)],
    scope: Scope.DEFAULT
  }));
  return providers;
}