516 lines
16 KiB
TypeScript
516 lines
16 KiB
TypeScript
import { Injectable, InternalServerErrorException } from '@nestjs/common';
|
|
import { OracleDBService } from 'src/db/db.service';
|
|
import { AuthLoginDTO } from './auth.dto';
|
|
import * as oracledb from 'oracledb';
|
|
import { ConfigService } from '@nestjs/config';
|
|
import axios, { AxiosRequestConfig, AxiosResponse } from 'axios';
|
|
import { Request } from 'express';
|
|
import * as jwkToPem from "jwk-to-pem"
|
|
import * as jwt from "jsonwebtoken"
|
|
import { UnauthorizedException } from 'src/exceptions/unauthorized.exception';
|
|
import { ConflictException } from 'src/exceptions/conflict.exception';
|
|
import { BadRequestException } from 'src/exceptions/badRequest.exception';
|
|
|
|
@Injectable()
|
|
export class AuthService {
|
|
private keyCache: Record<string, string> = {};
|
|
|
|
private readonly KEYCLOAK_URL: string;
|
|
private readonly KEYCLOAK_REALM: string;
|
|
private readonly KEYCLOAK_BASE_URL: string;
|
|
private readonly JWKS_URL: string;
|
|
private readonly TOKENS_INTROSPECT_URL: string;
|
|
private readonly CLIENT_ID: string;
|
|
private readonly CLIENT_SECRET: string;
|
|
private readonly TOKEN_URL: string;
|
|
private readonly USERS_URL: string;
|
|
|
|
constructor(
|
|
private readonly oracleDBService: OracleDBService,
|
|
private readonly configService: ConfigService,
|
|
) {
|
|
const KEYCLOAK_URL = this.configService.get<string>('KEYCLOAK_URL');
|
|
if (!KEYCLOAK_URL) throw new Error('Environment variable KEYCLOAK_URL is not set');
|
|
this.KEYCLOAK_URL = KEYCLOAK_URL;
|
|
|
|
const KEYCLOAK_REALM = this.configService.get<string>('KEYCLOAK_REALM');
|
|
if (!KEYCLOAK_REALM) throw new Error('Environment variable KEYCLOAK_REALM is not set');
|
|
this.KEYCLOAK_REALM = KEYCLOAK_REALM;
|
|
|
|
const CLIENT_ID = this.configService.get<string>('CLIENT_ID');
|
|
if (!CLIENT_ID) throw new Error('Environment variable KEYCLOAK CLIENT_ID is not set');
|
|
this.CLIENT_ID = CLIENT_ID;
|
|
|
|
const CLIENT_SECRET = this.configService.get<string>('CLIENT_SECRET');
|
|
if (!CLIENT_SECRET) throw new Error('Environment variable KEYCLOAK CLIENT_SECRET is not set');
|
|
this.CLIENT_SECRET = CLIENT_SECRET;
|
|
|
|
this.KEYCLOAK_BASE_URL = `${this.KEYCLOAK_URL}/realms/${this.KEYCLOAK_REALM}/protocol/openid-connect`;
|
|
|
|
this.JWKS_URL = `${this.KEYCLOAK_BASE_URL}/certs`
|
|
|
|
this.TOKENS_INTROSPECT_URL = `${this.KEYCLOAK_BASE_URL}/token/introspect`
|
|
|
|
this.TOKEN_URL = `${this.KEYCLOAK_BASE_URL}/token`
|
|
|
|
this.USERS_URL = `${this.KEYCLOAK_URL}/realms/${this.KEYCLOAK_REALM}/users`;
|
|
}
|
|
|
|
async login(body: AuthLoginDTO) {
|
|
let connection;
|
|
let rows = [];
|
|
try {
|
|
connection = await this.oracleDBService.getConnection();
|
|
if (!connection) {
|
|
throw new Error('No DB Connected');
|
|
}
|
|
|
|
const result = await connection.execute(
|
|
`BEGIN
|
|
USERLOGIN_PKG.ValidateUser(:P_EMAILADDR,:P_PASSWORD,:p_login_cursor);
|
|
END;`,
|
|
{
|
|
P_EMAILADDR: {
|
|
val: body.P_EMAILADDR,
|
|
type: oracledb.DB_TYPE_NVARCHAR,
|
|
},
|
|
P_PASSWORD: {
|
|
val: body.P_PASSWORD,
|
|
type: oracledb.DB_TYPE_NVARCHAR,
|
|
},
|
|
p_login_cursor: {
|
|
type: oracledb.CURSOR,
|
|
dir: oracledb.BIND_OUT,
|
|
},
|
|
},
|
|
{
|
|
outFormat: oracledb.OUT_FORMAT_OBJECT,
|
|
},
|
|
);
|
|
|
|
if (result.outBinds && result.outBinds.p_login_cursor) {
|
|
const cursor = result.outBinds.p_login_cursor;
|
|
let rowsBatch;
|
|
|
|
do {
|
|
rowsBatch = await cursor.getRows(100);
|
|
rows = rows.concat(rowsBatch);
|
|
} while (rowsBatch.length > 0);
|
|
|
|
await cursor.close();
|
|
} else {
|
|
throw new BadRequestException('Error executing request try after some time!');
|
|
}
|
|
|
|
if (rows[0]['ERRORMESG']) {
|
|
throw new BadRequestException('Invalid username or password!');
|
|
}
|
|
return { msg: 'Logged in successfully' };
|
|
} catch (err) {
|
|
throw new BadRequestException('Invalid username or password');
|
|
}
|
|
finally {
|
|
if (connection) {
|
|
try {
|
|
await connection.close();
|
|
} catch (closeErr) {
|
|
console.error('Failed to close connection:', closeErr);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private async getPublicKey(kid: string): Promise<string> {
|
|
if (this.keyCache[kid]) return this.keyCache[kid];
|
|
const { data } = await axios.get(this.JWKS_URL);
|
|
const key = data.keys.find((k) => k.kid === kid);
|
|
if (!key) throw new InternalServerErrorException('Key not found');
|
|
const pem = jwkToPem(key);
|
|
this.keyCache[kid] = pem;
|
|
return pem;
|
|
}
|
|
|
|
async introspectToken(token: string): Promise<any> {
|
|
|
|
const params = new URLSearchParams();
|
|
params.append('token', token);
|
|
params.append('client_id', this.CLIENT_ID);
|
|
params.append('client_secret', this.CLIENT_SECRET);
|
|
|
|
|
|
const response = await axios.post(this.TOKENS_INTROSPECT_URL, params, {
|
|
headers: { 'Content-Type': 'application/x-www-form-urlencoded' },
|
|
});
|
|
|
|
return response.data;
|
|
}
|
|
|
|
async decodeToken(token: string): Promise<any> {
|
|
const decodedHeader = jwt.decode(token, { complete: true });
|
|
if (!decodedHeader || typeof decodedHeader !== 'object') {
|
|
throw new Error('Invalid token format');
|
|
}
|
|
|
|
const kid: any = decodedHeader.header.kid;
|
|
const publicKey = await this.getPublicKey(kid);
|
|
|
|
const introspection = await this.introspectToken(token);
|
|
|
|
console.log("introspection : ", introspection.active, " : so Unauthorized!");
|
|
|
|
if (!introspection.active) {
|
|
throw new Error('Unauthorized');
|
|
}
|
|
|
|
const verified = jwt.verify(token, publicKey, { algorithms: ['RS256'] });
|
|
return verified;
|
|
}
|
|
|
|
async getAdminAccessTokenx(): Promise<string> {
|
|
const data = new URLSearchParams();
|
|
data.append('grant_type', 'client_credentials');
|
|
data.append('client_id', this.CLIENT_ID);
|
|
data.append('client_secret', this.CLIENT_SECRET);
|
|
try {
|
|
const response: AxiosResponse = await axios.post(this.KEYCLOAK_URL, data, {
|
|
headers: {
|
|
'Content-Type': 'application/x-www-form-urlencoded',
|
|
},
|
|
});
|
|
return response.data.access_token;
|
|
} catch (error) {
|
|
console.error('Error getting Keycloak access token:', error.response ? error.response.data : error.message);
|
|
throw new Error('Failed to obtain Keycloak access token');
|
|
}
|
|
}
|
|
|
|
async loginUser(username: string, password: string, req: Request): Promise<any> {
|
|
|
|
const accessToken = req.cookies['access_token'];
|
|
const refreshToken = req.cookies['refresh_token'];
|
|
|
|
// if (!accessToken && !refreshToken) {
|
|
|
|
const params = new URLSearchParams();
|
|
params.append('grant_type', 'password');
|
|
params.append('client_id', this.CLIENT_ID);
|
|
params.append('client_secret', this.CLIENT_SECRET);
|
|
params.append('username', username);
|
|
params.append('password', password);
|
|
|
|
try {
|
|
const response = await axios.post(this.TOKEN_URL, params, {
|
|
headers: { 'Content-Type': 'application/x-www-form-urlencoded' },
|
|
});
|
|
console.log('first time data ...');
|
|
|
|
let k = { ...response.data, email: username };
|
|
|
|
return k;
|
|
} catch (error) {
|
|
console.log(error.message);
|
|
|
|
throw new BadRequestException('Invalid username or password');
|
|
}
|
|
// }
|
|
|
|
// else if (accessToken && refreshToken) {
|
|
// return { access_token: accessToken, refresh_token: refreshToken, email: username }
|
|
// }
|
|
}
|
|
|
|
async getUserIdByEmail(email: string) {
|
|
|
|
let url = `${this.USERS_URL}?email=${email}`;
|
|
|
|
let adminAccessToken = await this.getAdminAccessToken();
|
|
|
|
const options: AxiosRequestConfig = {
|
|
method: 'GET',
|
|
url,
|
|
headers: {
|
|
Authorization: `Bearer ${adminAccessToken}`,
|
|
"Content-Type": "application/json"
|
|
}
|
|
};
|
|
|
|
try {
|
|
const { data } = await axios.request(options);
|
|
return data[0]?.id;
|
|
} catch (error) {
|
|
// console.error(error);
|
|
console.log(error.message);
|
|
}
|
|
}
|
|
|
|
async forgotPassword() {
|
|
|
|
const validatePasswordURLSearchParams = new URLSearchParams({
|
|
grant_type: 'password',
|
|
client_id: `${this.CLIENT_ID}`,
|
|
client_secret: `${this.CLIENT_SECRET}`,
|
|
username: `${'a@gmail.com'}`,
|
|
password: `${'A1!bcdef'}`,
|
|
});
|
|
|
|
let userID = await this.getUserIdByEmail("");
|
|
|
|
let resetPasswordURL = `${this.USERS_URL}/${userID}/reset-password`;
|
|
|
|
let adminAccessToken = await this.getAdminAccessToken();
|
|
|
|
const options1: AxiosRequestConfig = {
|
|
method: 'PUT',
|
|
url: resetPasswordURL,
|
|
headers: {
|
|
Authorization: `Bearer ${adminAccessToken}`
|
|
},
|
|
data: {
|
|
type: 'password',
|
|
value: 'A1!bcdef',
|
|
temporary: false,
|
|
}
|
|
};
|
|
|
|
const options2: AxiosRequestConfig = {
|
|
method: 'POST',
|
|
url: this.TOKEN_URL,
|
|
headers: {
|
|
'Content-Type': 'application/x-www-form-urlencoded',
|
|
},
|
|
data: validatePasswordURLSearchParams
|
|
};
|
|
|
|
try {
|
|
const { data } = await axios.request(options2);
|
|
return { statusCode: 200, message: 'password reset successfull' }
|
|
} catch (error) {
|
|
// console.error(error);
|
|
console.log(error.message);
|
|
|
|
throw new InternalServerErrorException()
|
|
|
|
}
|
|
}
|
|
|
|
async getAdminAccessToken() {
|
|
try {
|
|
const adminParams = new URLSearchParams();
|
|
adminParams.append('grant_type', 'client_credentials');
|
|
adminParams.append('client_id', this.CLIENT_ID);
|
|
adminParams.append('client_secret', this.CLIENT_SECRET);
|
|
|
|
const adminTokenResponse = await axios.post(this.TOKEN_URL,
|
|
adminParams,
|
|
{
|
|
headers: { 'Content-Type': 'application/x-www-form-urlencoded' },
|
|
});
|
|
console.log(adminTokenResponse.data);
|
|
|
|
const adminAccessToken = adminTokenResponse.data.access_token;
|
|
|
|
return adminAccessToken;
|
|
} catch (error) {
|
|
console.log(error.message);
|
|
throw error
|
|
}
|
|
}
|
|
|
|
async registerUser(body: AuthLoginDTO, req: Request) {
|
|
let det = await req['user'];
|
|
if (det?.email !== body.P_EMAILADDR) {
|
|
throw new BadRequestException();
|
|
}
|
|
const userPayload = {
|
|
username: body.P_EMAILADDR,
|
|
// firstName:"A",
|
|
// lastName:"B",
|
|
email: body.P_EMAILADDR,
|
|
emailVerified: true,
|
|
enabled: true,
|
|
credentials: [
|
|
{
|
|
type: 'password',
|
|
value: body.P_PASSWORD,
|
|
temporary: false,
|
|
},
|
|
],
|
|
};
|
|
|
|
console.log(userPayload);
|
|
|
|
const adminAccessToken = await this.getAdminAccessToken();
|
|
|
|
try {
|
|
const response = await axios.post(this.USERS_URL, userPayload, {
|
|
headers: {
|
|
Authorization: `Bearer ${adminAccessToken}`,
|
|
'Content-Type': 'application/json',
|
|
},
|
|
});
|
|
|
|
if (response.status === 201) {
|
|
|
|
const uid = await this.getUserIdByEmail(body.P_EMAILADDR);
|
|
|
|
const res = await this.assignRoleToUser(uid, "ca");
|
|
|
|
console.log(res);
|
|
|
|
return { message: 'User created successfully' };
|
|
}
|
|
|
|
} catch (error) {
|
|
// handle errors like user exists, validation errors, etc.
|
|
console.log(error.message);
|
|
|
|
if (error.message === "Request failed with status code 409") {
|
|
throw new ConflictException("User already exist");
|
|
}
|
|
|
|
throw new InternalServerErrorException('Failed to create user');
|
|
}
|
|
}
|
|
|
|
async logoutUser(refreshToken: string): Promise<any> {
|
|
const url = `${this.KEYCLOAK_BASE_URL}/logout`;
|
|
|
|
const params = new URLSearchParams();
|
|
params.append('client_id', this.CLIENT_ID);
|
|
params.append('client_secret', this.CLIENT_SECRET);
|
|
params.append('refresh_token', refreshToken);
|
|
|
|
try {
|
|
const response = await axios.post(url, params, {
|
|
headers: { 'Content-Type': 'application/x-www-form-urlencoded' },
|
|
});
|
|
return { statusCode: 200, message: 'Logged-Out successfully' };
|
|
} catch (error) {
|
|
throw new InternalServerErrorException('Logout failed');
|
|
}
|
|
}
|
|
|
|
async getClientUUID(): Promise<string> {
|
|
try {
|
|
const token = await this.getAdminAccessToken();
|
|
|
|
const realm = this.KEYCLOAK_REALM;
|
|
const clientId = this.CLIENT_ID;
|
|
const keycloakUrl = this.KEYCLOAK_URL;
|
|
|
|
const response = await axios.get(
|
|
`${keycloakUrl}/admin/realms/${realm}/clients?clientId=${encodeURIComponent(clientId)}`,
|
|
{
|
|
headers: {
|
|
Authorization: `Bearer ${token}`,
|
|
},
|
|
validateStatus: () => true, // manually handle status codes
|
|
}
|
|
);
|
|
|
|
if (response.status !== 200) {
|
|
throw new Error(`Failed to fetch client list. Status: ${response.status}. Message: ${JSON.stringify(response.data)}`);
|
|
}
|
|
|
|
const clients = response.data;
|
|
|
|
if (!Array.isArray(clients) || clients.length === 0) {
|
|
throw new Error(`Client with ID '${clientId}' not found in realm '${realm}'.`);
|
|
}
|
|
|
|
const clientUUID = clients[0].id;
|
|
|
|
if (!clientUUID) {
|
|
throw new Error(`Client UUID is missing in the response for clientId '${clientId}'.`);
|
|
}
|
|
|
|
return clientUUID;
|
|
} catch (error) {
|
|
console.error('❌ Error in getClientUUID:', error.message || error);
|
|
throw error; // rethrow for upstream handling
|
|
}
|
|
}
|
|
|
|
async getClientRole(roleName: string): Promise<any> {
|
|
const token = await this.getAdminAccessToken();
|
|
|
|
const clientUUID = await this.getClientUUID();
|
|
|
|
try {
|
|
const response = await axios.get(
|
|
`${this.KEYCLOAK_URL}/admin/realms/${this.KEYCLOAK_REALM}/clients/${clientUUID}/roles/${roleName}`,
|
|
{
|
|
headers: {
|
|
Authorization: `Bearer ${token}`,
|
|
},
|
|
},
|
|
);
|
|
|
|
return response.data;
|
|
} catch (error) {
|
|
console.error(`Error getting client role '${roleName}'`);
|
|
console.error(error.response?.data || error.message);
|
|
throw new InternalServerErrorException(`Could not find client role '${roleName}'`);
|
|
}
|
|
}
|
|
|
|
async assignRoleToUser(userId: string, roleName: string): Promise<any> {
|
|
try {
|
|
const token = await this.getAdminAccessToken();
|
|
|
|
const clientId = await this.getClientUUID();
|
|
|
|
const role = await this.getClientRole(roleName);
|
|
|
|
const url = `${this.KEYCLOAK_URL}/admin/realms/${this.KEYCLOAK_REALM}/users/${userId}/role-mappings/clients/${clientId}`;
|
|
|
|
const response = await axios.post(
|
|
url,
|
|
[role],
|
|
{
|
|
headers: {
|
|
Authorization: `Bearer ${token}`,
|
|
},
|
|
validateStatus: () => true, // To manually check status if needed
|
|
}
|
|
);
|
|
|
|
if (response.status >= 200 && response.status < 300) {
|
|
console.log(`✅ Role "${roleName}" successfully assigned to user ${userId}.`);
|
|
|
|
return { message: "Successfully assigned to user" }
|
|
} else {
|
|
throw new Error(`❌ Failed to assign role. Status: ${response.status}. Data: ${JSON.stringify(response.data)}`);
|
|
}
|
|
} catch (error) {
|
|
console.error(`🔥 Error assigning role "${roleName}" to user ${userId}:`, error.message || error);
|
|
throw error; // Rethrow to allow upstream handling
|
|
}
|
|
}
|
|
|
|
async getTokenFromRefreshToken(req: Request) {
|
|
const refreshToken = req.cookies['refresh_token'];
|
|
|
|
if (!refreshToken) {
|
|
throw new UnauthorizedException('Authentication failed');
|
|
}
|
|
|
|
const params = new URLSearchParams();
|
|
params.append('grant_type', 'refresh_token');
|
|
params.append('client_id', this.CLIENT_ID);
|
|
params.append('client_secret', this.CLIENT_SECRET);
|
|
params.append('refresh_token', refreshToken);
|
|
|
|
try {
|
|
const response = await axios.post(this.TOKEN_URL, params, {
|
|
headers: { 'Content-Type': 'application/x-www-form-urlencoded' },
|
|
});
|
|
|
|
return response.data;
|
|
} catch (error) {
|
|
console.log(error.message);
|
|
throw new UnauthorizedException('Authentication failed');
|
|
}
|
|
}
|
|
}
|