backup + decriptare toate fisiere done
This commit is contained in:
andrei-mihnea-cerbu
@@ -0,0 +1,217 @@
|
||||
const fs = require('fs').promises; // Ensure you use the promise-based API
|
||||
const path = require('path');
|
||||
const { decryptFileInPlace, encryptFileInPlace, encryptFileWithKey, decryptFileWithKey} = require("../src/main/aes_encrypt");
|
||||
const {lockFile, unlockFile} = require("../helpers/lock_mechanism");
|
||||
|
||||
function extractFilePaths(directoryStructure) {
|
||||
let paths = [];
|
||||
const traverse = (dir, currentPath) => {
|
||||
Object.keys(dir).forEach(key => {
|
||||
if (key === 'files') {
|
||||
dir[key].forEach(file => paths.push(path.join(currentPath, file)));
|
||||
} else {
|
||||
traverse(dir[key], path.join(currentPath, key));
|
||||
}
|
||||
});
|
||||
};
|
||||
traverse(directoryStructure, '');
|
||||
return paths;
|
||||
}
|
||||
|
||||
async function fetchFiles(ip, filePaths) {
|
||||
let results = {};
|
||||
|
||||
for (const filePath of filePaths) {
|
||||
const url = `http://${ip}:3000/file_path`;
|
||||
try {
|
||||
const response = await fetch(url, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json'
|
||||
},
|
||||
body: JSON.stringify({ filePath })
|
||||
});
|
||||
|
||||
if (!response.ok) throw new Error(`HTTP status ${response.status}`);
|
||||
|
||||
const arrayBuffer = await response.arrayBuffer(); // Fetch the response as an ArrayBuffer
|
||||
results[filePath] = Buffer.from(arrayBuffer);
|
||||
} catch (error) {
|
||||
console.error(`Error fetching ${filePath}:`, error);
|
||||
results[filePath] = null;
|
||||
}
|
||||
}
|
||||
|
||||
return results;
|
||||
}
|
||||
|
||||
async function removeDirectory(directoryPath) {
|
||||
try {
|
||||
// Check if the directory exists
|
||||
const stats = await fs.stat(directoryPath);
|
||||
if (!stats.isDirectory()) {
|
||||
console.log('The specified path is not a directory.');
|
||||
return;
|
||||
}
|
||||
|
||||
// Read all the contents of the directory
|
||||
const files = await fs.readdir(directoryPath);
|
||||
|
||||
// Loop through each file/directory and delete them
|
||||
for (const file of files) {
|
||||
const currentPath = path.join(directoryPath, file);
|
||||
const currentStats = await fs.stat(currentPath);
|
||||
|
||||
if (currentStats.isDirectory()) {
|
||||
// Recursive call for directories
|
||||
await removeDirectory(currentPath);
|
||||
} else {
|
||||
// Delete file
|
||||
await fs.unlink(currentPath);
|
||||
}
|
||||
}
|
||||
|
||||
// Finally, delete the directory itself
|
||||
await fs.rmdir(directoryPath);
|
||||
//console.log(`Directory removed: ${directoryPath}`);
|
||||
} catch (error) {
|
||||
console.error(`Error removing directory: ${error.message}`);
|
||||
}
|
||||
}
|
||||
|
||||
async function decryptUserSystemConfig() {
|
||||
const configPath = path.join(__dirname, '..', 'usersInSystem.json');
|
||||
await lockFile(configPath);
|
||||
await decryptFileInPlace(configPath);
|
||||
}
|
||||
|
||||
async function encryptUserSystemConfig(){
|
||||
const configPath = path.join(__dirname, '..', 'usersInSystem.json');
|
||||
await unlockFile(configPath);
|
||||
await encryptFileInPlace(configPath);
|
||||
}
|
||||
|
||||
async function processBackup() {
|
||||
await decryptUserSystemConfig();
|
||||
const usersConfigPath = path.join(__dirname, '..', 'usersInSystem.json');
|
||||
const backupConfigPath = path.join(__dirname, '..', 'backupSchemes.json');
|
||||
const baseBackupDirPath = path.join(__dirname, '..', 'backup_directories');
|
||||
|
||||
try {
|
||||
// Read and parse the users configuration
|
||||
const usersData = await fs.readFile(usersConfigPath, 'utf8');
|
||||
const usersConfig = JSON.parse(usersData);
|
||||
|
||||
// Read and parse the backup configuration
|
||||
const backupData = await fs.readFile(backupConfigPath, 'utf8');
|
||||
const jsonData = JSON.parse(backupData);
|
||||
|
||||
for (const key in jsonData) {
|
||||
const userBackupDir = path.join(baseBackupDirPath, key);
|
||||
await removeDirectory(userBackupDir);
|
||||
|
||||
const node = jsonData[key];
|
||||
const directoryStructure = JSON.parse(node.directoryStructure);
|
||||
const filePaths = extractFilePaths(directoryStructure);
|
||||
const files = await fetchFiles(node.ip, filePaths);
|
||||
|
||||
await fs.mkdir(userBackupDir, { recursive: true });
|
||||
|
||||
// Find the encryption key for the user's department
|
||||
let encryptionKey = '';
|
||||
for (const dept in usersConfig) {
|
||||
if (usersConfig[dept].users.includes(key)) {
|
||||
encryptionKey = usersConfig[dept].key;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
for (const file in files) {
|
||||
if (files[file]) {
|
||||
const fullPath = path.join(userBackupDir, file);
|
||||
await fs.mkdir(path.dirname(fullPath), { recursive: true });
|
||||
await fs.writeFile(fullPath, files[file]); // Write the binary data
|
||||
|
||||
// Encrypt the file in place after writing it
|
||||
if (encryptionKey) {
|
||||
await encryptFileWithKey(fullPath, encryptionKey);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('Failed to read or process the JSON file:', error);
|
||||
}
|
||||
|
||||
await encryptUserSystemConfig();
|
||||
}
|
||||
|
||||
async function getAllFilePaths(dirPath) {
|
||||
let filePaths = [];
|
||||
async function recurse(currentPath) {
|
||||
const entries = await fs.readdir(currentPath, { withFileTypes: true });
|
||||
// Create promises for each entry and process them in parallel
|
||||
const entryPromises = entries.map(async (entry) => {
|
||||
const resolvedPath = path.join(currentPath, entry.name);
|
||||
if (entry.isDirectory()) {
|
||||
// If it's a directory, recurse into it
|
||||
await recurse(resolvedPath);
|
||||
} else {
|
||||
// If it's a file, add it to the file paths array
|
||||
filePaths.push(resolvedPath);
|
||||
}
|
||||
});
|
||||
// Wait for all promises to complete
|
||||
await Promise.all(entryPromises);
|
||||
}
|
||||
await recurse(dirPath);
|
||||
return filePaths;
|
||||
}
|
||||
|
||||
async function decryptUserFilesToDirectory(destinationDir) {
|
||||
await decryptUserSystemConfig(); // Ensure user config is decrypted
|
||||
const usersConfigPath = path.join(__dirname, '..', 'usersInSystem.json');
|
||||
const usersConfig = JSON.parse(await fs.readFile(usersConfigPath, 'utf8'));
|
||||
const sourceDir = path.join(__dirname, '..', 'backup_directories');
|
||||
|
||||
try {
|
||||
for (const department in usersConfig) {
|
||||
const { key, users } = usersConfig[department];
|
||||
for (const userId of users) {
|
||||
const userDir = path.join(sourceDir, userId);
|
||||
try {
|
||||
const files = await getAllFilePaths(userDir);
|
||||
for (const filePath of files) {
|
||||
const stats = await fs.stat(filePath);
|
||||
if (stats.isFile()) {
|
||||
// Extract the part of the file path after the userId
|
||||
const relativePath = path.relative(userDir, filePath);
|
||||
|
||||
// Construct the destination file path
|
||||
const destinationFilePath = path.join(destinationDir, userId, relativePath);
|
||||
await fs.mkdir(path.dirname(destinationFilePath), { recursive: true });
|
||||
|
||||
// Decrypt the file in its original location
|
||||
await decryptFileWithKey(filePath, key);
|
||||
|
||||
// Copy the decrypted file to the destination directory
|
||||
await fs.copyFile(filePath, destinationFilePath);
|
||||
console.log(`Decrypted file copied to: ${destinationFilePath}`);
|
||||
|
||||
// Re-encrypt the file in its original location (optional)
|
||||
await encryptFileWithKey(filePath, key);
|
||||
}
|
||||
}
|
||||
} catch (error) {
|
||||
console.error(`Error processing files for user ${userId}: ${error.message}`);
|
||||
}
|
||||
}
|
||||
}
|
||||
} catch (error) {
|
||||
console.error(`Error processing decryption: ${error.message}`);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
processBackup();
|
||||
decryptUserFilesToDirectory("C:\\Users\\Andrei Cerbu\\Documents\\decrypted");
|
||||
|
||||
Reference in New Issue
Block a user