ShareService.java

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */

package org.pieShare.pieShareApp.service.shareService;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.StandardCopyOption;
import java.util.concurrent.ConcurrentHashMap;
import org.pieShare.pieShareApp.model.pieFile.PieFile;
import org.pieShare.pieShareApp.service.comparerService.api.ILocalFileCompareService;
import org.pieShare.pieShareApp.service.fileService.api.IFileService;
import org.pieShare.pieShareApp.service.fileService.fileEncryptionService.IFileEncryptionService;
import org.pieShare.pieShareApp.service.fileService.fileListenerService.api.IFileWatcherService;
import org.pieShare.pieTools.pieUtilities.service.pieLogger.PieLogger;

/**
 *
 * @author Svetoslav
 */
public class ShareService implements IShareService{
	
	private ILocalFileCompareService comparerService;
	private IFileService fileService;
	private IFileEncryptionService fileEncryptionService;
	private IFileWatcherService fileWatcherService;
	
	private ConcurrentHashMap<PieFile, Boolean> preparedFiles;

	public void init() {
		this.preparedFiles = new ConcurrentHashMap<>();
	}

	public void setFileEncryptionService(IFileEncryptionService fileEncryptionService) {
		this.fileEncryptionService = fileEncryptionService;
	}

	public void setFileWatcherService(IFileWatcherService fileWatcherService) {
		this.fileWatcherService = fileWatcherService;
	}

	public void setFileService(IFileService fileService) {
		this.fileService = fileService;
	}

	public void setComparerService(ILocalFileCompareService comparerService) {
		this.comparerService = comparerService;
	}
	
	@Override
	public File prepareFile(PieFile file) throws NoLocalFileException
	{		
		if(!this.comparerService.equalsWithLocalPieFile(file)) {
			throw new NoLocalFileException("Files don't match!");
		}
		
		File localFile = this.fileService.getAbsolutePath(file).toFile();
		File localTmpFileParent = this.fileService.getAbsoluteTmpPath(file).toFile().getParentFile();
		File localEncTmpFile = new File(localTmpFileParent, file.getFileName()+".enc");
		
		synchronized(file) {
			if(this.isPrepared(file)) {			
				return localEncTmpFile;
			}

			//todo: this exception belongs into the encryption service!!!
			/*if(!localFile.exists()) {
				throw new NoLocalFileException("Local file doesn't exist!");
			}*/
			//TODO: create dirs???!!!
			this.fileEncryptionService.encryptFile(localFile, localEncTmpFile);
			this.preparedFiles.put(file, Boolean.TRUE);
		}
		
		return localEncTmpFile;
	}

	@Override
	public void localFileTransferComplete(PieFile file, boolean source) {
		try {
			if(!source) {
				File localTmpFile = this.fileService.getAbsoluteTmpPath(file).toFile();
				File localEncTmpFile = new File(localTmpFile.getParentFile(), file.getFileName()+".enc");
				File localFile = this.fileService.getAbsolutePath(file).toFile();
				
				//todo: does this belong into the fileService?
				if (!localFile.getParentFile().exists()) {
					localFile.getParentFile().mkdirs();
				}
				
				this.fileEncryptionService.decryptFile(localEncTmpFile, localTmpFile);
				
				this.fileService.setCorrectModificationDateOnTmpFile(file);
				
				this.fileWatcherService.addPieFileToModifiedList(file);
				Files.move(localTmpFile.toPath(), localFile.toPath(), StandardCopyOption.REPLACE_EXISTING);
				
				//todo: is it better to delete the enc file or not?
			}
		} catch (IOException ex) {
			PieLogger.error(this.getClass(), "Error!", ex);
		}
	}
	
	@Override
	public void revokePrepared(PieFile file) {
		synchronized(file) {
			this.preparedFiles.remove(file);
		}
	}
	
	@Override
	public boolean isPrepared(PieFile file) {
		File localTmpFileParent = this.fileService.getAbsoluteTmpPath(file).toFile().getParentFile();
		File localEncTmpFile = new File(localTmpFileParent, file.getFileName()+".enc");
		synchronized(file) {
			return this.preparedFiles.getOrDefault(file, Boolean.FALSE) && localEncTmpFile.exists();
		}
	}
}