public class importDataFromCSVController {
private String clientId = '93701334664-
soqqm4cojb4rfn915hakjhjal8k697rd.apps.googleusercontent.com';
private String clientSecret = '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';
private String redirectUrl =
'https://drive.google.com/drive/u/0/folders/155Spx4tNEnRfziX6F2Whx11EzAwA-gjX';
public String accessToken { get; set; }
public Blob csvFileBody { get; set; }
public String csvAsString { get; set; }
public String[] csvFileLines { get; set; }
public List<Account> accountList { get; set; }
@future(callout=true)
public static void fetchAndProcessFilesFromGoogleDrive() {
try {
importDataFromCSVController controllerInstance = new
importDataFromCSVController();
controllerInstance.getFilesFromGoogleDrive();
} catch (Exception e) {
System.debug('Exception in fetchAndProcessFilesFromGoogleDrive: ' +
e.getMessage());
}
}
public importDataFromCSVController() {
csvFileLines = new String[]{};
accountList = new List<Account>();
refreshToken(); // Fetch the access token dynamically
}
private void refreshToken() {
accessToken = GoogleAuthService.refreshAccessToken(); // Fetch access token
dynamically
}
public void importCSVFile() {
try {
csvAsString = csvFileBody.toString();
csvFileLines = csvAsString.split('\n');
for (Integer i = 1; i < csvFileLines.size(); i++) {
String[] csvRecordData = csvFileLines[i].split(',');
if (csvRecordData.size() >= 5) {
Account accObj = new Account();
accObj.Name = csvRecordData[0].trim();
accObj.AccountNumber = csvRecordData[1].trim();
accObj.Type = csvRecordData[2].trim();
accObj.AccountSource = csvRecordData[3].trim();
accObj.Industry = csvRecordData[4].trim();
accountList.add(accObj);
} else {
System.debug('Skipping CSV line due to incomplete data: ' +
csvFileLines[i]);
}
}
if (!accountList.isEmpty()) {
insert accountList;
}
} catch (Exception e) {
ApexPages.Message errorMessage = new
ApexPages.Message(ApexPages.Severity.ERROR, 'An error occurred while importing
data. Please check the CSV file and try again.');
ApexPages.addMessage(errorMessage);
System.debug('Error importing CSV file: ' + e.getMessage());
}
}
// Define the set outside of the method, possibly in your class
Set<String> processedFiles = new Set<String>();
public void getFilesFromGoogleDrive() {
try {
String accessToken = GoogleAuthService.refreshAccessToken();
HttpRequest req = new HttpRequest();
req.setMethod('GET');
req.setEndpoint('https://www.googleapis.com/drive/v3/files');
req.setHeader('Authorization', 'Bearer ' + accessToken); // Use the
dynamically fetched access token
Http http = new Http();
HttpResponse res = http.send(req);
if (res.getStatusCode() == 200) {
String responseBody = res.getBody();
Map<String, Object> responseMap = (Map<String, Object>)
JSON.deserializeUntyped(responseBody);
List<Object> files = (List<Object>) responseMap.get('files');
if (files != null && !files.isEmpty()) {
for (Object fileObj : files) {
Map<String, Object> file = (Map<String, Object>) fileObj;
String fileId = (String) file.get('id');
String fileName = (String) file.get('name');
String mimeType = (String) file.get('mimeType');
// Now you can use the fileId to fetch the content of the file
if needed
System.debug('File ID: ' + fileId + ', File Name: ' + fileName
+ ', Mime Type: ' + mimeType);
if (mimeType == 'text/csv') {
System.debug('### File ID: ' + fileId);
System.debug('### File Name: ' + fileName);
if (!processedFiles.contains(fileId)) {
processCSVFile(fileId);
processedFiles.add(fileId);
} else {
System.debug('File ID ' + fileId + ' has already been
processed.');
}
}
}
if (processedFiles.isEmpty()) {
// throw new CustomException('No unprocessed CSV files found.');
}
} else {
System.debug('No files found in Google Drive.');
}
} else {
System.debug('Error fetching files from Google Drive');
}
} catch (Exception e) {
System.debug('Exception in getFilesFromGoogleDrive: ' + e.getMessage());
}
}
// Method to process a CSV file
private void processCSVFile(String fileId) {
CSVImportController controller = new CSVImportController();
controller.googleDriveFileId = fileId;
controller.importCSVFromGoogleDrive();
}
public void getFiles(string fileId) {
CSVImportController controller = new CSVImportController();
controller.googleDriveFileId = fileId;
controller.importCSVFromGoogleDrive();
}
}
