Finally Its done quick quick quick
This commit is contained in:
vedang29
+278
-109
@@ -1,128 +1,297 @@
|
||||
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.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.stereotype.Component;
|
||||
import org.springframework.http.HttpStatus;
|
||||
import org.springframework.http.ResponseEntity;
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
import org.springframework.web.multipart.MultipartFile;
|
||||
|
||||
import java.util.Scanner;
|
||||
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;
|
||||
|
||||
@Component
|
||||
public class HDFScontroller implements Runnable {
|
||||
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 UserManager userManager;
|
||||
private final HDFSOperations hdfsOperations;
|
||||
|
||||
@Autowired
|
||||
public HDFScontroller(UserManager userManager, HDFSOperations hdfsOperations) {
|
||||
this.userManager = userManager;
|
||||
public HDFScontroller(HDFSOperations hdfsOperations) {
|
||||
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();
|
||||
}
|
||||
}
|
||||
@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;
|
||||
|
||||
Reference in New Issue
Block a user