/**
 * Copyright 2014 Eediom Inc.
 * 
 * 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 org.araqne.logdb;

import java.io.File;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.UUID;

import org.araqne.logdb.writer.LineWriter;
import org.araqne.logdb.writer.LineWriterFactory;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * 
 * @author darkluster
 * 
 */
public class PartitionOutput {
	private final Logger logger = LoggerFactory.getLogger(PartitionOutput.class);

	private String tmpPath;
	private String path;
	private FileMover mover;
	private File f;
	private LineWriter writer;

	public PartitionOutput(LineWriterFactory lineWriterFactory, String path, String tmpPath, Date day, String encoding)
			throws IOException {
		this(lineWriterFactory, path, tmpPath, day, encoding, false);
	}

	public PartitionOutput(LineWriterFactory lineWriterFactory, String path, String tmpPath, Date day, String encoding,
			boolean overwrite) throws IOException {
		try {
			this.tmpPath = tmpPath;
			this.path = convertPath(path, day);

			if (tmpPath != null) {
				File dir = new File(tmpPath);
				this.f = new File(dir, UUID.randomUUID().toString() + ".part");

				if (logger.isDebugEnabled())
					logger.debug("araqne logdb: created temp partition output file [{}] for path [{}], day [{}]", new Object[] {
							f.getAbsolutePath(), path, day });

				this.tmpPath = f.getAbsolutePath();
			} else {
				this.f = new File(this.path);
			}

			File parentFile = f.getParentFile();
			boolean created = parentFile.mkdirs();
			if (!(parentFile.exists() && parentFile.isDirectory()) && !created)
				throw new IOException("cannot create partition output directory: " + parentFile.getAbsolutePath());

			if (encoding == null)
				encoding = "utf-8";

			this.writer = lineWriterFactory.newWriter(f.getAbsolutePath());
			mover = new LocalFileMover(overwrite);
		} catch (IOException e) {
			logger.error("araqne logdb: cannot create partition output [" + f.getAbsolutePath() + "] tmp path [" + tmpPath + "]",
					e);
			throw e;
		} catch (Throwable t) {
			logger.error("araqne logdb: cannot create partition output [" + f.getAbsolutePath() + "] tmp path [" + tmpPath + "]",
					t);
			throw new IOException(t);
		}
	}

	public LineWriter getWriter() {
		return writer;
	}

	private static String convertPath(String path, Date day) {
		if (!path.contains("{") || !path.contains("}") || !path.contains(":"))
			return path;

		int lastPos = path.indexOf("{");
		String convertedPath = path.substring(0, lastPos);
		for (int f = 0; (f = path.indexOf("{", f)) != -1; f++) {
			int t = path.indexOf("}", f);
			String s = path.substring(f, t + 1);
			String time = s.substring(1, s.indexOf(":"));
			String format = s.substring(s.indexOf(":") + 1, s.length() - 1);
			convertedPath += path.substring(lastPos, f);

			Date target = new Date();
			if (time.equalsIgnoreCase("logtime"))
				target = day;

			if (format.equalsIgnoreCase("epoch"))
				convertedPath += target.getTime();
			else {
				SimpleDateFormat sdf = new SimpleDateFormat(format);
				convertedPath += sdf.format(target);
			}

			lastPos = t + 1;
		}

		convertedPath += path.substring(path.lastIndexOf("}") + 1);
		return convertedPath;
	}

	public String getPath() {
		return path;
	}

	public String getTmpPath() {
		return tmpPath;
	}

	public void close() {
		try {
			writer.close();
			if (tmpPath != null)
				mover.move(tmpPath, path);
		} catch (Throwable e) {
			logger.error("araqne logdb: cannot move file from [{}] to [{}] cause [{}]",
					new Object[] { tmpPath, path, e.getMessage() });
		}
	}
}