diff --git a/src/app.module.ts b/src/app.module.ts index ad2215c..dcb27a8 100644 --- a/src/app.module.ts +++ b/src/app.module.ts @@ -1,12 +1,19 @@ -import { Module } from '@nestjs/common'; +import { MiddlewareConsumer, Module } from '@nestjs/common'; import { DbModule } from './db/db.module'; import { OracleModule } from './oracle/oracle.module'; import { AuthModule } from './auth/auth.module'; import { MssqlModule } from './mssql/mssql.module'; +import { OriginCheckMiddleware } from './middleware/OriginCheck.middleware'; @Module({ imports: [MssqlModule, AuthModule, DbModule, OracleModule], controllers: [], providers: [], }) -export class AppModule {} +export class AppModule { + configure(consumer: MiddlewareConsumer) { + consumer + .apply(OriginCheckMiddleware) + .forRoutes('*'); // Apply to all routes + } +} diff --git a/src/middleware/OriginCheck.middleware.ts b/src/middleware/OriginCheck.middleware.ts new file mode 100644 index 0000000..2478dda --- /dev/null +++ b/src/middleware/OriginCheck.middleware.ts @@ -0,0 +1,21 @@ +import { Injectable, Logger, NestMiddleware } from '@nestjs/common'; +import { Request, Response, NextFunction } from 'express'; + +@Injectable() +export class OriginCheckMiddleware implements NestMiddleware { + use(req: Request, res: Response, next: NextFunction) { + const allowedOrigin = 'http://your-specified-domain.com'; // Replace with your domain + const origin = req.headers.origin; + + const logger = new Logger('OriginCheckMiddleware'); + logger.log('user-agent', req.headers['user-agent']); + // console.log(req.headers['user-agent']); + + + // if (origin === allowedOrigin) { + next(); + // } else { + // res.status(403).send('Access denied'); + // } + } +} \ No newline at end of file diff --git a/src/mssql/uscib-managed-sp/carnet-sequence/carnet-sequence.service.ts b/src/mssql/uscib-managed-sp/carnet-sequence/carnet-sequence.service.ts index 5f499b1..a235df7 100644 --- a/src/mssql/uscib-managed-sp/carnet-sequence/carnet-sequence.service.ts +++ b/src/mssql/uscib-managed-sp/carnet-sequence/carnet-sequence.service.ts @@ -17,7 +17,7 @@ export class CarnetSequenceService { const request = new Request(connection); request.input('P_SPID', mssql.Int, body.p_spid); const result = await request.execute('carnetsys.GETCARNETSEQUENCE'); - return { data: result.recordsets }; + return { data: result.recordset }; } catch (error) { console.error('Error executing stored procedure:', error); diff --git a/src/mssql/uscib-managed-sp/region/region.service.ts b/src/mssql/uscib-managed-sp/region/region.service.ts index fa51a69..4d5aee7 100644 --- a/src/mssql/uscib-managed-sp/region/region.service.ts +++ b/src/mssql/uscib-managed-sp/region/region.service.ts @@ -15,7 +15,7 @@ export class RegionService { connection = await this.mssqlDBService.getConnection(); const request = new Request(connection); const result = await request.execute('carnetsys.GETREGIONS'); - return { data: result.recordsets }; + return { data: result.recordset }; } catch (error) { console.error('Error executing stored procedure:', error); @@ -35,7 +35,7 @@ export class RegionService { request.input('P_REGION', mssql.VarChar(30), body.p_region); request.input('P_NAME', mssql.VarChar(30), body.p_name); const result = await request.execute('carnetsys.INSERTNEWREGION'); - return { data: result.recordsets }; + return { data: result.recordset }; } catch (error) { console.error('Error executing stored procedure:', error); diff --git a/src/mssql/uscib-managed-sp/sp-contacts/sp-contacts.service.ts b/src/mssql/uscib-managed-sp/sp-contacts/sp-contacts.service.ts index bb04654..a0a0137 100644 --- a/src/mssql/uscib-managed-sp/sp-contacts/sp-contacts.service.ts +++ b/src/mssql/uscib-managed-sp/sp-contacts/sp-contacts.service.ts @@ -17,7 +17,7 @@ export class SpContactsService { const request = new Request(connection); request.input('P_SPID', mssql.Int, body.p_SPid); const result = await request.execute('carnetsys.GETSPALLCONTACTS'); - return { data: result.recordsets }; + return { data: result.recordset }; } catch (error) { console.error('Error executing stored procedure:', error); @@ -36,7 +36,7 @@ export class SpContactsService { const request = new Request(connection); request.input('P_SPID', mssql.Int, body.p_SPid); const result = await request.execute('carnetsys.GETSPDEFAULTCONTACTS'); - return { data: result.recordsets }; + return { data: result.recordset }; } catch (error) { console.error('Error executing stored procedure:', error); @@ -55,7 +55,7 @@ export class SpContactsService { const request = new Request(connection); request.input('P_SPCONTACTID', mssql.Int, body.p_spcontactid); const result = await request.execute('carnetsys.INACTIVATESPCONTACTS'); - return { data: result.recordsets }; + return { data: result.recordset }; } catch (error) { console.error('Error executing stored procedure:', error); diff --git a/src/mssql/uscib-managed-sp/sp/sp.service.ts b/src/mssql/uscib-managed-sp/sp/sp.service.ts index 59627fd..c512258 100644 --- a/src/mssql/uscib-managed-sp/sp/sp.service.ts +++ b/src/mssql/uscib-managed-sp/sp/sp.service.ts @@ -16,7 +16,7 @@ export class SpService { connection = await this.mssqlDBService.getConnection(); const request = new Request(connection); const result = await request.execute('carnetsys.GETALLSPS'); - return { data: result.recordsets }; + return { data: result.recordset }; } catch (error) { console.error('Error executing stored procedure:', error); @@ -35,7 +35,7 @@ export class SpService { const request = new Request(connection); request.input('P_SPID', mssql.Int, body.p_spid); const result = await request.execute('carnetsys.GETSPBYSPID'); - return { data: result.recordsets }; + return { data: result.recordset }; } catch (error) { console.error('Error executing stored procedure:', error);