Backup_FME_personal/force-app/main/default/classes/AWSS3Feature.cls

608 lines
25 KiB
OpenEdge ABL

public with sharing class AWSS3Feature extends CloudStorageBase {
private static String SERVER_URL = System.Label.Ngrok;
public override OperationResult listFiles(Map<String, Object> params) {
List<Map<String, String>> resultFiles = new List<Map<String, String>>();
try {
String endpoint = SERVER_URL + '/api/v1/s3/all';
System.debug('Endpoint: ' + endpoint);
Boolean isServiceMode = (Boolean) params.get('isServiceMode');
Map<String, Object> payload = (Map<String, Object>) params.get('payload');
String key = (String) params.get('key');
// String requestBody = '{"aws": ' + JSON.serialize(payload) + ', "key": '+ key + '}';
Map<String, Object> body = new Map<String, Object>();
body.put('aws', payload);
body.put('key', key);
String requestBody = JSON.serialize(body);
System.debug('requestBody: ' + requestBody);
Map<String, String> headers = new Map<String, String>{
'Content-Type' => 'application/json'
};
HttpResponse res = CloudStorageUtlilty.makeCallout(requestBody, endpoint, headers, 'POST');
System.debug('Status Code: ' + res.getStatusCode());
System.debug('Response Body: ' + res.getBody());
if(res.getStatusCode() == 200) {
System.debug('response body: '+ res.getBody());
List<Object> fileList = (List<Object>) JSON.deserializeUntyped(res.getBody());
processItems(fileList, resultFiles);
}
else {
System.debug('response body after fail: '+ res.getBody());
System.debug('S3 call failed!');
return OperationResult.error('Someting went wrong');
}
} catch (Exception e) {
throw new AuraHandledException('Exception: ' + e.getMessage());
}
System.debug('Final resultFiles: ' + JSON.serializePretty(resultFiles));
return OperationResult.success(resultFiles);
}
@AuraEnabled
public static OperationResult storeAwsS3Credentials(Boolean isAdminCred, Map<String, Object> payload){
try {
System.debug('method in AWS S3 for token store');
String userId = SalesforceEnvironmentUtil.getUserId();
String accessKey = (String) payload.get('accessKey');
String secretAccessKey = (String) payload.get('secretAccessKey');
String region = (String) payload.get('region');
String bucketName = (String) payload.get('bucketName');
Map<String, Object> tokenMap = new Map<String, Object>();
tokenMap.put('HIC_FilesME__Auth_Provider_Name__c', 'AwsS3');
// tokenMap.put('HIC_FilesME__External_Storage_Type__c', 'AwsS3');
//added
tokenMap.put('HIC_FilesME__Provider__c', 'AwsS3');
tokenMap.put('HIC_FilesME__Access_Key__c', accessKey);
tokenMap.put('HIC_FilesME__Secret_Access_Key__c', secretAccessKey);
tokenMap.put('HIC_FilesME__Region__c', region);
tokenMap.put('HIC_FilesME__Bucket__c', bucketName);
tokenMap.put('HIC_FilesME__Is_Admin_level_Credentials__c', isAdminCred);
tokenMap.put('HIC_FilesME__User__c', userId);
Map<String, Object> body = new Map<String, Object>();
body.put('aws', payload);
body.put('key', '');
String requestBody = JSON.serialize(body);
System.debug('requestBody: ' + requestBody);
Map<String, String> headers = new Map<String, String>{
'Content-Type' => 'application/json'
};
String endpoint = SERVER_URL + '/api/v1/s3/all';
HttpResponse res = CloudStorageUtlilty.makeCallout(requestBody, endpoint, headers, 'POST');
if(res.getStatusCode() == 200){
System.debug('Storing token');
OAuthTokenManager.storeToken(tokenMap);
List<sObject> record = [
SELECT id FROM HIC_FilesME__External_Token__c
WHERE HIC_FilesME__Is_Admin_level_Credentials__c = :isAdminCred
AND HIC_FilesME__Provider__c = 'AwsS3'
];
System.debug('record size-> '+ record.size());
if(record != null && record.size() > 0) {
return OperationResult.success('Credentials stored');
}
else {
return OperationResult.error('Credentials not stored');
}
}
else {
System.debug('Invalid credentials');
return OperationResult.error('Invalid credentials');
}
}
catch(Exception e){
throw new AuraHandledException('Exception occoured: '+ e.getMessage());
}
}
private static void processItems(List<Object> items, List<Map<String, String>> resultFiles) {
for (Object item : items) {
Map<String, Object> fileMap = (Map<String, Object>) item;
String id = (String) fileMap.get('name');
String name = (String) fileMap.get('name');
name = name.removeEnd('/');
List<String> nameSplit = name.split('/');
name = nameSplit[nameSplit.size()-1];
String type = (String) fileMap.get('type');
Boolean isFolder = type != null && type.toLowerCase() == 'folder';
String iconName = (String) fileMap.get('value');
if(iconName == null && isFolder){
iconName = 'doctype:folder';
}
Map<String, String> fileData = new Map<String, String>{
'id' => id,
'path' => id,
'name' => name,
'type' => type,
'iconName' => iconName,
'mimeType' => iconName,
'isFolder' => String.valueOf(isFolder)
};
resultFiles.add(fileData);
if (isFolder && fileMap.containsKey('children')) {
List<Object> children = (List<Object>) fileMap.get('children');
if (children != null && !children.isEmpty()) {
processItems(children, resultFiles);
}
}
}
}
//to delete
public override OperationResult deleteFile(Map<String, Object> params) {
try {
System.debug('>>>entered deleteFile SFTP');
String key = (String) params.get('fileId');
Boolean isServiceMode = (Boolean) params.get('isServiceMode');
Map<String, Object> payload = (Map<String, Object>) params.get('payload');
Map<String, Object> body = new Map<String, Object>();
body.put('aws', payload);
body.put('key', key);
String requestBody = JSON.serialize(body);
String endpoint = SERVER_URL + '/api/v1/s3';
System.debug('Create Request Body: ' + requestBody);
Map<String, String> headers = new Map<String, String>{
'Content-Type' => 'application/json',
'accept' => '*/*'
};
HttpResponse res = CloudStorageUtlilty.makeCallout(requestBody, endpoint, headers, 'DELETE');
System.debug('>>>res=' + res.getBody());
System.debug('>>>res.getStatusCode()=' + res.getStatusCode());
if (res.getStatusCode() == 200 || res.getStatusCode() == 204) {
return OperationResult.success(res.getBody());
} else {
return OperationResult.error(res.getBody());
}
} catch (Exception e) {
System.debug('>>>error=' + e.getMessage());
throw new AuraHandledException(e.getMessage());
}
}
//to create
public override OperationResult createFolder(Map<String, Object> params) {
Map<String, Object> resultMap = new Map<String, Object>();
System.debug('Inside createFolder');
try {
String parentFolderId = (String) params.get('parentFolderId');
Boolean isServiceMode = (Boolean) params.get('isServiceMode');
String folderName = (String) params.get('folderName');
String key = (String) params.get('key');
Map<String, Object> payload = (Map<String, Object>) params.get('payload');
Map<String, Object> body = new Map<String, Object>();
body.put('aws', payload);
body.put('key', key);
String encodedFolderName = EncodingUtil.urlEncode(folderName, 'UTF-8');
String endpoint = SERVER_URL + '/api/v1/s3/' + folderName;
System.debug('Create Request Body: ' + JSON.serializePretty(body));
String requestBody = JSON.Serialize(body);
Map<String, String> headers = new Map<String, String>{
'Content-Type' => 'application/json',
'accept' => '*/*'
};
HttpResponse res = CloudStorageUtlilty.makeCallout(requestBody, endpoint, headers, 'POST');
System.debug('>>>res=' + res.getBody());
System.debug('>>>res.getStatusCode()=' + res.getStatusCode());
if (res.getStatusCode() != 200 && res.getStatusCode() != 201) {
return OperationResult.error('Error: ' + res.getStatusCode() + ' - ' + res.getBody());
}
String responseBody = res.getBody();
// if(res.getBody() != null){
// resultMap = (Map<String, Object>) JSON.deserializeUntyped(responseBody);
// }
resultMap.put('code', '200');
System.debug('>>>resultMap=' + resultMap);
return OperationResult.success(resultMap);
} catch (Exception e) {
throw new AuraHandledException('AWS S3 Folder Creation Failed: ' + e.getMessage());
}
}
//rename
public override OperationResult renameFile(Map<String, Object> params) {
Map<String, Object> resultMap = new Map<String, Object>();
System.debug('Inside renameFolder');
try {
String fileId = (String) params.get('fileId');
Boolean isServiceMode = (Boolean) params.get('isServiceMode');
String newFileName = (String) params.get('newFileName');
Map<String, Object> payload = (Map<String, Object>) params.get('payload');
String endpoint = SERVER_URL + '/api/v1/s3/rename';
if(fileId.endsWith('/')){
fileId = fileId.substring(0, fileId.length() - 1);
}
if(newFileName.endsWith('/')){
newFileName = newFileName.substring(0, newFileName.length() - 1);
}
Map<String, Object> body = new Map<String, Object>();
body.put('aws', payload);
body.put('oldName', fileId);
body.put('newName', newFileName);
String requestBody = JSON.serialize(body);
System.debug('Rename Request Body: ' + requestBody);
Map<String, String> headers = new Map<String, String>{
'Content-Type' => 'application/json',
'accept' => '*/*'
};
HttpResponse res = CloudStorageUtlilty.makeCallout(requestBody, endpoint, headers, 'POST');
if (res.getStatusCode() != 200) {
return OperationResult.error('Error: ' + res.getStatusCode() + ' - ' + res.getBody());
}
String responseBody = res.getBody();
resultMap = (Map<String, Object>) JSON.deserializeUntyped(responseBody);
resultMap.put('code', '200');
return OperationResult.success(resultMap);
} catch (Exception e) {
throw new AuraHandledException('SFTP Folder Rename Failed: ' + e.getMessage());
}
}
public override OperationResult moveFileAndFolder(Map<String, Object> params) {
Map<String, Object> resultMap = new Map<String, Object>();
System.debug('Inside moveFileAndFolder');
try {
Boolean isServiceMode = (Boolean) params.get('isServiceMode');
String sourceKey = (String) params.get('oldFolderId');
String destinationKey = (String) params.get('newFolderId');
Map<String, Object> payload = (Map<String, Object>) params.get('payload');
String endpoint = SERVER_URL + '/api/v1/s3/move';
Map<String, Object> body = new Map<String, Object>();
body.put('aws', payload);
body.put('sourceKey', sourceKey);
body.put('destinationKey', destinationKey);
body.put('overwrite', true);
String requestBody = JSON.serialize(body);
System.debug('Move Request Body: ' + requestBody);
Map<String, String> headers = new Map<String, String>{
'Content-Type' => 'application/json',
'accept' => '*/*'
};
HttpResponse res = CloudStorageUtlilty.makeCallout(requestBody, endpoint, headers, 'POST');
if (res.getStatusCode() != 200) {
return OperationResult.error('Error: ' + res.getStatusCode() + ' - ' + res.getBody());
}
String responseBody = res.getBody();
resultMap = (Map<String, Object>) JSON.deserializeUntyped(responseBody);
resultMap.put('code', '200');
return OperationResult.success(resultMap);
} catch (Exception e) {
throw new AuraHandledException('SFTP Folder Move Failed: ' + e.getMessage());
}
}
// download
public override OperationResult downloadFile(Map<String, Object> params) {
System.debug('AWSS3Feature.downloadFile');
OperationResult result;
try {
Blob fileBlob = fetchFileAndConvertToBase64(params);
if (fileBlob != null) {
result = OperationResult.success(EncodingUtil.base64Encode(fileBlob));
System.debug('File downloaded successfully.');
} else {
result = OperationResult.error('Failed to download file');
}
} catch (Exception e) {
result = OperationResult.error('Error downloading file: ' + e.getMessage());
FME_Logger.error(e.getMessage(), 'AWSS3Feature' + e.getLineNumber(), 'downloadFile');
}
return result;
}
public static Blob fetchFileAndConvertToBase64(Map<String, Object> params) {
Blob fileBlob;
try {
Boolean isServiceMode = (Boolean) params.get('isServiceMode');
String fileId = (String) params.get('fileId');
String accessToken = (String) params.get('accessToken');
String userId = (String) SalesforceEnvironmentUtil.getUserId();
Map<String, Object> payload = (Map<String, Object>) params.get('payload');
// 2. AWS S3 download endpoint
String downloadUrl = SERVER_URL + '/api/v1/s3/download';
Map<String, String> headers = new Map<String, String>{
'Content-Type' => 'application/json',
'accept' => '*/*'
};
Map<String, Object> body = new Map<String, Object> {
'aws'=> payload,
'key'=> fileId
};
String requestBody = JSON.serialize(body);
// 3. Download file
HttpResponse res = CloudStorageUtlilty.makeCallout(requestBody, downloadUrl, headers, 'POST');
// 5. Success
if (res.getStatusCode() == 200) {
fileBlob = res.getBodyAsBlob();
} else {
System.debug('AWS S3 download failed: ' + res.getStatusCode() + ' | ' + res.getBody());
}
} catch (Exception e) {
System.debug(
'Error in fetchFileAndConvertToBase64: ' +
e.getMessage() + ' | Line: ' + e.getLineNumber()
);
FME_Logger.error(
e.getMessage(),
'AWSS3Feature' + e.getLineNumber(),
'fetchFileAndConvertToBase64'
);
}
return fileBlob;
}
public override OperationResult linkFiles(String recordId) {
List<Map<String, String>> resultFolders = new List<Map<String, String>>();
try {
List<HIC_FilesME__Linked_Folder__c> linkedFolders = [
SELECT Id,Name,CreatedDate,CreatedBy.Name,HIC_FilesME__External_Folder_ID__c,
HIC_FilesME__Folder_Name__c,
HIC_FilesME__Folder_URL__c,
HIC_FilesME__Storage_Provider__c,
HIC_FilesME__Access_Level__c,
HIC_FilesME__Connected_Username__c
FROM HIC_FilesME__Linked_Folder__c
WHERE HIC_FilesME__Linked_Record__c = :recordId
AND HIC_FilesME__Is_Active__c = true
];
for (HIC_FilesME__Linked_Folder__c folder : linkedFolders) {
Map<String, String> folderData = new Map<String, String>();
folderData.put('id', folder.HIC_FilesME__External_Folder_ID__c);
folderData.put('recordId', folder.Id);
folderData.put('name', folder.HIC_FilesME__Folder_Name__c);
folderData.put('externalStorageType', folder.HIC_FilesME__Storage_Provider__c);
folderData.put('accessLevel', folder.HIC_FilesME__Access_Level__c);
folderData.put('createdTime', String.valueOf(folder.CreatedDate));
folderData.put('createdBy', folder.CreatedBy != null ? folder.CreatedBy.Name : '-');
folderData.put('linkedBy', folder.HIC_FilesME__Connected_Username__c);
folderData.put('url', folder.HIC_FilesME__Folder_URL__c);
resultFolders.add(folderData);
}
System.debug('resultFolders >>>> ' + resultFolders);
return OperationResult.success(resultFolders);
} catch (Exception e) {FME_Logger.error(e.getMessage(),'LinkedFolderFeature' + e.getLineNumber(),'linkFiles');
return OperationResult.error('Failed to fetch linked folders. ' + e.getMessage());
}
}
public override OperationResult toLinkFiles(String recordId,String folderDetails,String providerType) {
try {
System.debug('providerType >>>> ' + providerType);
System.debug('recordId >>>> ' + recordId);
System.debug('JSON.deserializeUntyped folderDetails >>>> ' + JSON.deserializeUntyped(folderDetails));
Map<String, Object> folderMap =
(Map<String, Object>) JSON.deserializeUntyped(folderDetails);
HIC_FilesME__Linked_Folder__c linkedFolder = new HIC_FilesME__Linked_Folder__c();
linkedFolder.HIC_FilesME__Linked_Record__c = recordId;
linkedFolder.HIC_FilesME__External_Folder_ID__c = (String) folderMap.get('id');
linkedFolder.HIC_FilesME__Folder_Name__c = (String) folderMap.get('name');
//add
linkedFolder.HIC_FilesME__Created__c = UserInfo.getUserId();
linkedFolder.HIC_FilesME__Created_Date__c = System.now();
linkedFolder.HIC_FilesME__Connected_Username__c = (String) folderMap.get('linkedBy');
linkedFolder.HIC_FilesME__Folder_URL__c = (String) folderMap.get('url');
linkedFolder.HIC_FilesME__Storage_Provider__c = providerType;
// linkedFolder.HIC_FilesME__Access_Level__c = (String) folderMap.get('accessLevel');
linkedFolder.HIC_FilesME__Is_Active__c = true;
String key = providerType +' '+ recordId +' '+ (String) folderMap.get('id');
linkedFolder.HIC_FilesME__Key__c = key;
upsert linkedFolder HIC_FilesME__Key__c;
// insert linkedFolder;
System.debug('linkedFolder >>>> ' + linkedFolder);
return OperationResult.success(linkedFolder);
} catch (Exception e) {
FME_Logger.error(e.getMessage(),'LinkedFolderFeature' + e.getLineNumber(),'toLinkFiles' );
return OperationResult.error('Error linking folder: ' + e.getMessage());
}
}
public override OperationResult unlinkFiles(String recordId, String filesId, Boolean deleteFromDrive) {
System.debug('recordId >>>> ' + recordId);
System.debug('filesId >>>> ' + filesId);
System.debug('deleteFromDrive >>>> ' + deleteFromDrive);
OperationResult result;
try {
List<HIC_FilesME__Linked_Folder__c> fileLinks = [SELECT Id FROM HIC_FilesME__Linked_Folder__c WHERE HIC_FilesME__Linked_Record__c = :recordId AND HIC_FilesME__External_Folder_ID__c = :filesId];
if (fileLinks.size() > 0) {
delete fileLinks;
// if (deleteFromDrive) {
// deleteFile(filesId, true, null);
// }
result = OperationResult.success(null);
} else {
result = OperationResult.error('No record found');
}
} catch (Exception e) {
result = OperationResult.error('Error unlinking file: ' + e.getMessage());
FME_Logger.error(e.getMessage(), 'AWSS3Feature' + e.getLineNumber(), 'unlinkFiles');
}
return result;
}
//send email
public override OperationResult emailFile(Map<String,Object> params) {
OperationResult result;
try {
//maildetails
Map<String, Object> mailDetails =(Map<String, Object>) JSON.deserializeUntyped(JSON.serialize(params.get('mailDetails')));
String fileId = (String) params.get('fileId');
String fileName = (String) params.get('fileName');
String mimeType = (String) params.get('toSendFileNameMimeType');
String senderId = (String) mailDetails.get('fromUser');
String toUser = (String) mailDetails.get('toUser');
String ccUser = (String) mailDetails.get('ccUser');
String bccUser = (String) mailDetails.get('bccUser');
String subject = (String) mailDetails.get('subject');
String body = (String) mailDetails.get('body');
Messaging.SingleEmailMessage mail = new Messaging.SingleEmailMessage();
// Required
mail.setToAddresses(new List<String>{ toUser });
// Optional
if (String.isNotBlank(ccUser))
mail.setCcAddresses(new List<String>{ ccUser });
if (String.isNotBlank(bccUser))
mail.setBccAddresses(new List<String>{ bccUser });
mail.setSubject(subject);
// Remove HTML tags safely
mail.setPlainTextBody(body != null ? body.replaceAll('<[^>]+>', '') : '');
if (String.isNotBlank(senderId) && senderId.startsWith('0D2')) {
mail.setOrgWideEmailAddressId(senderId);
}
Map<String, Object> downloadParams = new Map<String, Object>(params);
downloadParams.put('fileId', fileId);
downloadParams.put('fileName', fileName);
OperationResult op = downloadFile(downloadParams);
if (op != null && op.data != null) {
Blob fileBlob = EncodingUtil.base64Decode((String) op.data);
Messaging.EmailFileAttachment attachment = new Messaging.EmailFileAttachment();
attachment.setFileName(fileName);
attachment.setBody(fileBlob);
attachment.setContentType(String.isNotBlank(mimeType) ? mimeType: 'application/octet-stream');
mail.setFileAttachments(new List<Messaging.EmailFileAttachment>{ attachment });
}
Messaging.SendEmailResult[] results = Messaging.sendEmail( new List<Messaging.SingleEmailMessage>{ mail });
if (!results.isEmpty() && results[0].isSuccess()) {
result = OperationResult.success(null);
System.debug(' Email sent successfully');
} else {
String errMsg = (!results.isEmpty() && results[0].getErrors().size() > 0)? results[0].getErrors()[0].getMessage(): 'Unknown send failure';
result = OperationResult.error(errMsg);
}
} catch (Exception e) {
FME_Logger.error(e.getMessage(),'EmailFeature ' + e.getLineNumber(),'emailFile');
result = OperationResult.error('Error sending email: ' + e.getMessage());
}
System.debug('result:' + result);
return result;
}
}