54 lines
1.4 KiB
TypeScript
54 lines
1.4 KiB
TypeScript
import { redactAttributes } from '@ghostfolio/api/helper/object.helper';
|
|
import { encodeDataSource } from '@ghostfolio/common/helper';
|
|
import {
|
|
CallHandler,
|
|
ExecutionContext,
|
|
Injectable,
|
|
NestInterceptor
|
|
} from '@nestjs/common';
|
|
import { DataSource } from '@prisma/client';
|
|
import { Observable } from 'rxjs';
|
|
import { map } from 'rxjs/operators';
|
|
|
|
import { ConfigurationService } from '../services/configuration.service';
|
|
|
|
@Injectable()
|
|
export class TransformDataSourceInResponseInterceptor<T>
|
|
implements NestInterceptor<T, any>
|
|
{
|
|
public constructor(
|
|
private readonly configurationService: ConfigurationService
|
|
) {}
|
|
|
|
public intercept(
|
|
context: ExecutionContext,
|
|
next: CallHandler<T>
|
|
): Observable<any> {
|
|
return next.handle().pipe(
|
|
map((data: any) => {
|
|
if (this.configurationService.get('ENABLE_FEATURE_SUBSCRIPTION')) {
|
|
data = redactAttributes({
|
|
options: [
|
|
{
|
|
attribute: 'dataSource',
|
|
valueMap: Object.keys(DataSource).reduce(
|
|
(valueMap, dataSource) => {
|
|
valueMap[dataSource] = encodeDataSource(
|
|
DataSource[dataSource]
|
|
);
|
|
return valueMap;
|
|
},
|
|
{}
|
|
)
|
|
}
|
|
],
|
|
object: data
|
|
});
|
|
}
|
|
|
|
return data;
|
|
})
|
|
);
|
|
}
|
|
}
|