import { ConfigurationService } from '@ghostfolio/api/services/configuration/configuration.service';
import { decodeDataSource } from '@ghostfolio/common/helper';

import {
  CallHandler,
  ExecutionContext,
  Injectable,
  NestInterceptor
} from '@nestjs/common';
import { DataSource } from '@prisma/client';
import { Observable } from 'rxjs';

@Injectable()
export class TransformDataSourceInRequestInterceptor<T>
  implements NestInterceptor<T, any>
{
  public constructor(
    private readonly configurationService: ConfigurationService
  ) {}

  public intercept(
    context: ExecutionContext,
    next: CallHandler<T>
  ): Observable<any> {
    const http = context.switchToHttp();
    const request = http.getRequest();

    if (this.configurationService.get('ENABLE_FEATURE_SUBSCRIPTION')) {
      if (request.body.activities) {
        request.body.activities = request.body.activities.map((activity) => {
          if (DataSource[activity.dataSource]) {
            return activity;
          } else {
            return {
              ...activity,
              dataSource: decodeDataSource(activity.dataSource)
            };
          }
        });
      }

      if (request.body.dataSource && !DataSource[request.body.dataSource]) {
        request.body.dataSource = decodeDataSource(request.body.dataSource);
      }

      if (request.params.dataSource && !DataSource[request.params.dataSource]) {
        request.params.dataSource = decodeDataSource(request.params.dataSource);
      }
    }

    return next.handle();
  }
}