/**
 * Copyright 1996-2014 FoxBPM ORG.
 *
 * 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.
 * 
 * @author kenshin
 * @author ych
 */
package org.foxbpm.engine.impl.model;

import java.io.InputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;

import org.foxbpm.engine.ModelService;
import org.foxbpm.engine.exception.FoxBPMException;
import org.foxbpm.engine.impl.entity.DeploymentEntity;
import org.foxbpm.engine.impl.entity.ResourceEntity;
import org.foxbpm.engine.impl.util.ExceptionUtil;
import org.foxbpm.engine.impl.util.IoUtil;
import org.foxbpm.engine.impl.util.ReflectUtil;
import org.foxbpm.engine.repository.Deployment;
import org.foxbpm.engine.repository.DeploymentBuilder;

/**
 * @author kenshin
 */
public class DeploymentBuilderImpl implements DeploymentBuilder {

	protected ModelService modelService;
	protected DeploymentEntity deployment = new DeploymentEntity();
	protected boolean isDuplicateFilterEnabled = false;
	protected String updateDeploymentId;

	public DeploymentBuilderImpl(ModelService modelService) {
		this.modelService = modelService;
	}

	public DeploymentBuilder addInputStream(String resourceName, InputStream inputStream) {
		return addInputStream(resourceName, inputStream, -1);
	}

	 
	public DeploymentBuilder addInputStream(String resourceName, InputStream inputStream, int version) {
		if (inputStream == null) {
			throw ExceptionUtil.getException("10101001",resourceName);
		}
		byte[] bytes = IoUtil.readInputStream(inputStream, resourceName);
		ResourceEntity resource = new ResourceEntity();
		resource.setName(resourceName);
		resource.setBytes(bytes);
		if (version != -1) {
			resource.setVersion(version);
		}
		// 绑定发布Id
		resource.setDeploymentId(deployment.getId());
		deployment.addResource(resource);
		return this;
	}

	public DeploymentBuilder addClasspathResource(String resource) {
		InputStream inputStream = ReflectUtil.getResourceAsStream(resource);
		if (inputStream == null) {
			throw ExceptionUtil.getException("10101001",resource);
		}
		return addInputStream(resource, inputStream);
	}

	public DeploymentBuilder addString(String resourceName, String text) {
		if (text == null) {
			throw new FoxBPMException("text is null");
		}
		ResourceEntity resource = new ResourceEntity();
		resource.setName(resourceName);
		resource.setBytes(text.getBytes());
		// 绑定发布Id
		resource.setDeploymentId(deployment.getId());
		deployment.addResource(resource);
		return this;
	}

	public DeploymentBuilder addZipInputStream(ZipInputStream zipInputStream) {
		return addZipInputStream(zipInputStream, -1);
	}

	 
	public DeploymentBuilder addZipInputStream(ZipInputStream zipInputStream, int version) {
		try {
			ZipEntry entry = zipInputStream.getNextEntry();
			while (entry != null) {
				if (!entry.isDirectory()) {
					String entryName = entry.getName();
					byte[] bytes = IoUtil.readInputStream(zipInputStream, entryName);
					ResourceEntity resource = new ResourceEntity();
					resource.setName(entryName);
					resource.setBytes(bytes);
					if (version != -1) {
						resource.setVersion(version);
					}
					// 绑定发布Id
					resource.setDeploymentId(deployment.getId());
					deployment.addResource(resource);
				}
				entry = zipInputStream.getNextEntry();
			}
		} catch (Exception e) {
			throw new FoxBPMException("problem reading zip input stream", e);
		}
		return this;
	}

	public DeploymentBuilder name(String name) {
		deployment.setName(name);
		return this;
	}

	public DeploymentBuilder enableDuplicateFiltering() {
		isDuplicateFilterEnabled = true;
		return this;
	}

	public Deployment deploy() {
		return modelService.deploy(this);
	}

	// getters and setters
	// //////////////////////////////////////////////////////

	public DeploymentEntity getDeployment() {
		return deployment;
	}

	public boolean isDuplicateFilterEnabled() {
		return isDuplicateFilterEnabled;
	}

	public DeploymentBuilder updateDeploymentId(String updateDeploymentId) {
		deployment.setId(updateDeploymentId);
		deployment.setUpdateDeploymentId(updateDeploymentId);
		this.updateDeploymentId = updateDeploymentId;
		return this;
	}

}