ProjectVersion.java

/*
 * Copyright 2012 LinkedIn Corp.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */

package azkaban.execapp;

import java.io.File;
import java.io.IOException;
import java.util.zip.ZipFile;

import org.apache.commons.io.FileUtils;
import org.apache.log4j.Logger;

import azkaban.project.ProjectFileHandler;
import azkaban.project.ProjectLoader;
import azkaban.project.ProjectManagerException;

import azkaban.utils.FileIOUtils;
import azkaban.utils.Utils;

public class ProjectVersion implements Comparable<ProjectVersion> {
	private final int projectId;
	private final int version;
	private File installedDir;
	
	public ProjectVersion(int projectId, int version) {
		this.projectId = projectId;
		this.version = version;
	}
	
	public ProjectVersion(int projectId, int version, File installedDir) {
		this.projectId = projectId;
		this.version = version;
		this.installedDir = installedDir;
	}

	public int getProjectId() {
		return projectId;
	}

	public int getVersion() {
		return version;
	}
	
	public synchronized void setupProjectFiles(ProjectLoader projectLoader, File projectDir, Logger logger) throws ProjectManagerException, IOException {
		String projectVersion = String.valueOf(projectId) + "." + String.valueOf(version);
		if (installedDir == null) {
			installedDir = new File(projectDir, projectVersion);
		}
		
		if (!installedDir.exists()) {
			
			logger.info("First time executing new project. Setting up in directory " + installedDir.getPath());
			
			File tempDir = new File(projectDir, "_temp." + projectVersion + "." + System.currentTimeMillis());
			tempDir.mkdirs();
			ProjectFileHandler projectFileHandler = null;
			try {
				projectFileHandler = projectLoader.getUploadedFile(projectId, version);
				if ("zip".equals(projectFileHandler.getFileType())) {
					logger.info("Downloading zip file.");
					ZipFile zip = new ZipFile(projectFileHandler.getLocalFile());
					Utils.unzip(zip, tempDir);
	
					tempDir.renameTo(installedDir);
				}
				else {
					throw new IOException("The file type hasn't been decided yet.");
				}
			}
			finally {
				if (projectFileHandler != null) {
					projectFileHandler.deleteLocalFile();
				}
			}
		}
	}
	
	public synchronized void copyCreateSymlinkDirectory(File executionDir) throws IOException {
		if (installedDir == null || !installedDir.exists()) {
			throw new IOException("Installed dir doesn't exist");
		}
		else if (executionDir == null || !executionDir.exists()) {
			throw new IOException("Execution dir doesn't exist");
		}
		FileIOUtils.createDeepSymlink(installedDir, executionDir);
	}
	
	public synchronized void deleteDirectory() throws IOException {
		System.out.println("Deleting old unused project versin " + installedDir);
		if (installedDir != null && installedDir.exists()) {
			FileUtils.deleteDirectory(installedDir);
		}
	}

	@Override
	public int compareTo(ProjectVersion o) {
		if (projectId == o.projectId) {
			return version - o.version;
		}
		
		return projectId - o.projectId;
	}
}