vedang29
409 lines
16 KiB
Java
409 lines
16 KiB
Java
package com.skycrate.backend.skycrateBackend.controller;
|
|
|
|
import com.skycrate.backend.skycrateBackend.dto.ResponseDTO;
|
|
import com.skycrate.backend.skycrateBackend.services.EncryptionUtil;
|
|
import com.skycrate.backend.skycrateBackend.services.HDFSOperations;
|
|
import com.skycrate.backend.skycrateBackend.utils.KeyUtil;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.http.HttpStatus;
|
|
import org.springframework.http.ResponseEntity;
|
|
import org.springframework.web.bind.annotation.*;
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
|
import java.io.IOException;
|
|
import java.nio.file.Files;
|
|
import java.nio.file.Path;
|
|
import java.nio.file.Paths;
|
|
import java.nio.file.StandardCopyOption;
|
|
import java.security.PrivateKey;
|
|
import java.security.PublicKey;
|
|
import java.util.List;
|
|
|
|
import static com.skycrate.backend.skycrateBackend.utils.KeyUtil.getPrivateKeyForUser;
|
|
import static com.skycrate.backend.skycrateBackend.utils.KeyUtil.getPublicKeyForUser;
|
|
|
|
@RestController
|
|
@RequestMapping("/api/hdfs")
|
|
public class HDFScontroller {
|
|
|
|
private final HDFSOperations hdfsOperations;
|
|
|
|
@Autowired
|
|
public HDFScontroller(HDFSOperations hdfsOperations) {
|
|
this.hdfsOperations = hdfsOperations;
|
|
}
|
|
|
|
@PostMapping("/createFolder")
|
|
public ResponseDTO createFolder(@RequestParam String hdfsPath) {
|
|
System.out.println("==> Received createFolder call for: " + hdfsPath);
|
|
try {
|
|
hdfsOperations.createFolder(hdfsPath);
|
|
return new ResponseDTO("Folder created successfully", true);
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
return new ResponseDTO("Failed to create folder: " + e.getMessage(), false);
|
|
}
|
|
}
|
|
|
|
|
|
@PostMapping("/uploadFile")
|
|
public ResponseDTO uploadFile(
|
|
@RequestParam("file") MultipartFile file,
|
|
@RequestParam String hdfsPath,
|
|
@RequestParam String uploadedFileName,
|
|
@RequestParam String username) {
|
|
try {
|
|
// Save file locally first
|
|
String localPath = saveFileLocally(file);
|
|
System.out.println("File saved locally at: " + localPath);
|
|
|
|
// Upload file to HDFS
|
|
hdfsOperations.uploadFile(localPath, hdfsPath, uploadedFileName, username);
|
|
return new ResponseDTO("File uploaded successfully", true);
|
|
} catch (IOException e) {
|
|
e.printStackTrace();
|
|
return new ResponseDTO("Failed to upload file locally: " + e.getMessage(), false);
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
return new ResponseDTO("Failed to upload file to HDFS: " + e.getMessage(), false);
|
|
}
|
|
}
|
|
|
|
|
|
private String saveFileLocally(MultipartFile file) throws IOException {
|
|
// Create a temporary directory if it doesn't exist
|
|
Path tmpDir = Paths.get("tmp");
|
|
if (!Files.exists(tmpDir)) {
|
|
Files.createDirectories(tmpDir); // Create the directory if it doesn't exist
|
|
}
|
|
|
|
Path path = tmpDir.resolve(file.getOriginalFilename());
|
|
|
|
// Copy the file to the local directory
|
|
Files.copy(file.getInputStream(), path, StandardCopyOption.REPLACE_EXISTING);
|
|
|
|
return path.toString(); // Return the local path for further processing
|
|
}
|
|
|
|
|
|
@PostMapping("/downloadFile")
|
|
public ResponseDTO downloadFile(
|
|
@RequestParam String hdfsPath,
|
|
@RequestParam String localPath,
|
|
@RequestParam String username) {
|
|
try {
|
|
hdfsOperations.downloadFile(hdfsPath, localPath, username);
|
|
return new ResponseDTO("File downloaded successfully", true);
|
|
} catch (Exception e) {
|
|
return new ResponseDTO("Failed to download file: " + e.getMessage(), false);
|
|
}
|
|
}
|
|
|
|
@DeleteMapping("/deleteFile")
|
|
public ResponseDTO deleteFile(@RequestParam String hdfsPath) {
|
|
try {
|
|
hdfsOperations.deleteFile(hdfsPath);
|
|
return new ResponseDTO("File deleted successfully", true);
|
|
} catch (Exception e) {
|
|
return new ResponseDTO("Failed to delete file: " + e.getMessage(), false);
|
|
}
|
|
}
|
|
|
|
@DeleteMapping("/deleteFolder")
|
|
public ResponseDTO deleteFolder(@RequestParam String hdfsPath) {
|
|
try {
|
|
hdfsOperations.deleteFolder(hdfsPath);
|
|
return new ResponseDTO("Folder deleted successfully", true);
|
|
} catch (Exception e) {
|
|
return new ResponseDTO("Failed to delete folder: " + e.getMessage(), false);
|
|
}
|
|
}
|
|
|
|
@GetMapping("/listFiles")
|
|
public ResponseEntity<?> listFiles(@RequestParam String hdfsPath) {
|
|
try {
|
|
List<String> files = hdfsOperations.listFilesAndFolders(hdfsPath);
|
|
return ResponseEntity.ok(files); // Returns the list as JSON array
|
|
} catch (Exception e) {
|
|
return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
|
|
.body("Failed to list files: " + e.getMessage());
|
|
}
|
|
}
|
|
|
|
// @GetMapping("/listFiles")
|
|
// public ResponseDTO listFiles(@RequestParam String hdfsPath) {
|
|
// try {
|
|
// hdfsOperations.listFilesAndFolders(hdfsPath);
|
|
// return new ResponseDTO("Listed files successfully", true);
|
|
// } catch (Exception e) {
|
|
// return new ResponseDTO("Failed to list files: " + e.getMessage(), false);
|
|
// }
|
|
// }
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
//package com.skycrate.backend.skycrateBackend.controller;
|
|
//
|
|
//import com.skycrate.backend.skycrateBackend.dto.User;
|
|
//import com.skycrate.backend.skycrateBackend.repository.UserManager;
|
|
//import com.skycrate.backend.skycrateBackend.services.HDFSOperations;
|
|
//import org.springframework.beans.factory.annotation.Autowired;
|
|
//import org.springframework.stereotype.Component;
|
|
//
|
|
//import java.util.Scanner;
|
|
//
|
|
//@Component
|
|
//public class HDFScontroller implements Runnable {
|
|
//
|
|
// private final UserManager userManager;
|
|
// private final HDFSOperations hdfsOperations;
|
|
//
|
|
// @Autowired
|
|
// public HDFScontroller(UserManager userManager, HDFSOperations hdfsOperations) {
|
|
// this.userManager = userManager;
|
|
// this.hdfsOperations = hdfsOperations;
|
|
// }
|
|
//
|
|
// @Override
|
|
// public void run() {
|
|
// Scanner scanner = new Scanner(System.in);
|
|
//
|
|
// while (true) {
|
|
// System.out.print("Enter your username: ");
|
|
// String username = scanner.nextLine();
|
|
//
|
|
// System.out.print("Enter your password: ");
|
|
// String password = scanner.nextLine();
|
|
//
|
|
// if (!userManager.authenticate(username, password)) {
|
|
// System.out.println("❌ Authentication failed. Exiting...");
|
|
// return;
|
|
// }
|
|
//
|
|
// User user;
|
|
// try {
|
|
// user = userManager.getUser(username);
|
|
// } catch (Exception e) {
|
|
// System.out.println("❌ Error creating user: " + e.getMessage());
|
|
// return;
|
|
// }
|
|
//
|
|
// while (true) {
|
|
// System.out.println("\nChoose an option:");
|
|
// System.out.println("1. Create a folder");
|
|
// System.out.println("2. Upload a file (encrypted)");
|
|
// System.out.println("3. Upload a file to a specific folder (encrypted)");
|
|
// System.out.println("4. Download a file");
|
|
// System.out.println("5. Delete a file");
|
|
// System.out.println("6. Delete a folder");
|
|
// System.out.println("7. List files and folders");
|
|
// System.out.println("8. Exit");
|
|
//
|
|
// int choice = scanner.nextInt();
|
|
// scanner.nextLine(); // Consume newline
|
|
//
|
|
// try {
|
|
// switch (choice) {
|
|
// case 1:
|
|
// System.out.print("Enter folder name: ");
|
|
// hdfsOperations.createFolder(scanner.nextLine());
|
|
// break;
|
|
//
|
|
// case 2:
|
|
// System.out.print("Enter file path to upload: ");
|
|
// String filePath = scanner.nextLine();
|
|
// System.out.print("Enter HDFS destination path: ");
|
|
// String hdfsDestinationFolder = scanner.nextLine();
|
|
// System.out.print("Enter the name for the uploaded file: ");
|
|
// String uploadedFileName = scanner.nextLine();
|
|
// hdfsOperations.uploadFile(filePath, hdfsDestinationFolder, uploadedFileName, user);
|
|
// break;
|
|
//
|
|
// case 3:
|
|
// System.out.print("Enter folder name to upload to: ");
|
|
// String targetFolder = scanner.nextLine();
|
|
// System.out.print("Enter file path to upload: ");
|
|
// String fileToUpload = scanner.nextLine();
|
|
// System.out.print("Enter the name for the uploaded file: ");
|
|
// String uploadedFileNameSpecific = scanner.nextLine();
|
|
// hdfsOperations.uploadFile(fileToUpload, targetFolder, uploadedFileNameSpecific, user);
|
|
// break;
|
|
//
|
|
// case 4:
|
|
// System.out.print("Enter HDFS file path to download: ");
|
|
// String hdfsFilePath = scanner.nextLine();
|
|
// System.out.print("Enter local path to save the downloaded file: ");
|
|
// String localDownloadPath = scanner.nextLine();
|
|
// hdfsOperations.downloadFile(hdfsFilePath, localDownloadPath, user);
|
|
// break;
|
|
//
|
|
// case 5:
|
|
// System.out.print("Enter HDFS file path to delete: ");
|
|
// hdfsOperations.deleteFile(scanner.nextLine());
|
|
// break;
|
|
//
|
|
// case 6:
|
|
// System.out.print("Enter HDFS folder path to delete: ");
|
|
// hdfsOperations.deleteFolder(scanner.nextLine());
|
|
// break;
|
|
//
|
|
// case 7:
|
|
// System.out.print("Enter HDFS path to list files and folders: ");
|
|
// hdfsOperations.listFilesAndFolders(scanner.nextLine());
|
|
// break;
|
|
//
|
|
// case 8:
|
|
// System.out.println("✅ Exiting...");
|
|
// return;
|
|
//
|
|
// default:
|
|
// System.out.println("⚠️ Invalid option. Please try again.");
|
|
// }
|
|
// } catch (Exception e) {
|
|
// System.out.println("❌ Operation failed: " + e.getMessage());
|
|
// e.printStackTrace();
|
|
// }
|
|
// }
|
|
// }
|
|
// }
|
|
//}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
//package com.skycrate.backend.skycrateBackend.controller;
|
|
//
|
|
//import com.skycrate.backend.skycrateBackend.dto.User;
|
|
//import com.skycrate.backend.skycrateBackend.repository.UserManager;
|
|
//import com.skycrate.backend.skycrateBackend.services.HDFSOperations;
|
|
//import org.springframework.beans.factory.annotation.Autowired;
|
|
//import org.springframework.stereotype.Component;
|
|
//
|
|
//import java.util.Scanner;
|
|
//
|
|
//@Component
|
|
//public class HDFScontroller implements Runnable {
|
|
//
|
|
// private final UserManager userManager;
|
|
// private final HDFSOperations hdfsOperations;
|
|
//
|
|
// @Autowired
|
|
// public HDFScontroller(UserManager userManager, HDFSOperations hdfsOperations) {
|
|
// this.userManager = userManager;
|
|
// this.hdfsOperations = hdfsOperations;
|
|
// }
|
|
//
|
|
// @Override
|
|
// public void run() {
|
|
// Scanner scanner = new Scanner(System.in);
|
|
//
|
|
// while (true) {
|
|
// System.out.print("Enter your username: ");
|
|
// String username = scanner.nextLine();
|
|
//
|
|
// System.out.print("Enter your password: ");
|
|
// String password = scanner.nextLine();
|
|
//
|
|
// if (!userManager.authenticate(username, password)) {
|
|
// System.out.println("X Authentication failed. Exiting...");
|
|
// return;
|
|
// }
|
|
//
|
|
// User user;
|
|
// try {
|
|
// user = userManager.getUser(username);
|
|
// } catch (Exception e) {
|
|
// System.out.println("X Error creating user: " + e.getMessage());
|
|
// return;
|
|
// }
|
|
//
|
|
// while (true) {
|
|
// System.out.println("\nChoose an option:");
|
|
// System.out.println("1. Create a folder");
|
|
// System.out.println("2. Upload a file (encrypted)");
|
|
// System.out.println("3. Upload a file to a specific folder (encrypted)");
|
|
// System.out.println("4. Download a file");
|
|
// System.out.println("5. Delete a file");
|
|
// System.out.println("6. Delete a folder");
|
|
// System.out.println("7. List files and folders");
|
|
// System.out.println("8. Exit");
|
|
//
|
|
// int choice = scanner.nextInt();
|
|
// scanner.nextLine(); // Consume newline
|
|
//
|
|
// switch (choice) {
|
|
// case 1:
|
|
// System.out.print("Enter folder name: ");
|
|
// hdfsOperations.createFolder(scanner.nextLine());
|
|
// break;
|
|
// case 2:
|
|
// System.out.print("Enter file path to upload: ");
|
|
// String filePath = scanner.nextLine();
|
|
// System.out.print("Enter HDFS destination path: ");
|
|
// String hdfsDestinationFolder = scanner.nextLine();
|
|
// System.out.print("Enter the name for the uploaded file: ");
|
|
// String uploadedFileName = scanner.nextLine();
|
|
// hdfsOperations.uploadFile(filePath, hdfsDestinationFolder, uploadedFileName, user);
|
|
// break;
|
|
// case 3:
|
|
// System.out.print("Enter folder name to upload to: ");
|
|
// String targetFolder = scanner.nextLine();
|
|
// System.out.print("Enter file path to upload: ");
|
|
// String fileToUpload = scanner.nextLine();
|
|
// System.out.print("Enter the name for the uploaded file: ");
|
|
// String uploadedFileNameSpecific = scanner.nextLine();
|
|
// hdfsOperations.uploadFile(fileToUpload, targetFolder, uploadedFileNameSpecific, user);
|
|
// break;
|
|
// case 4:
|
|
// System.out.print("Enter HDFS file path to download: ");
|
|
// String hdfsFilePath = scanner.nextLine();
|
|
// System.out.print("Enter local path to save the downloaded file: ");
|
|
// String localDownloadPath = scanner.nextLine();
|
|
// hdfsOperations.downloadFile(hdfsFilePath, localDownloadPath, user);
|
|
// break;
|
|
// case 5:
|
|
// System.out.print("Enter HDFS file path to delete: ");
|
|
// hdfsOperations.deleteFile(scanner.nextLine());
|
|
// break;
|
|
// case 6:
|
|
// System.out.print("Enter HDFS folder path to delete: ");
|
|
// hdfsOperations.deleteFolder(scanner.nextLine());
|
|
// break;
|
|
// case 7:
|
|
// System.out.print("Enter HDFS path to list files and folders: ");
|
|
// hdfsOperations.listFilesAndFolders(scanner.nextLine());
|
|
// break;
|
|
// case 8:
|
|
// System.out.println("Exiting...");
|
|
// return;
|
|
// default:
|
|
// System.out.println("Invalid option. Please try again.");
|
|
// }
|
|
// }
|
|
// }
|
|
// }
|
|
//}
|