/*
 * Copyright 2013 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.logstorage.script;

import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Set;

import org.araqne.api.ScriptContext;
import org.araqne.logstorage.backup.StorageBackupJob;
import org.araqne.logstorage.backup.StorageBackupProgressMonitor;
import org.araqne.logstorage.backup.StorageBackupType;
import org.araqne.logstorage.backup.StorageFile;

/**
 * @since 2.2.7
 * @author xeraph
 * 
 */
class BackupProgressPrinter implements StorageBackupProgressMonitor {
	private ScriptContext context;
	private boolean disabled = false;

	public BackupProgressPrinter(ScriptContext context) {
		this.context = context;
	}

	public boolean isDisabled() {
		return disabled;
	}

	public void setDisabled(boolean disabled) {
		this.disabled = disabled;
	}

	@Override
	public void onBeginTable(StorageBackupJob job, String tableName) {
		if (!disabled)
			context.println(getTimestamp() + ">> " + getType(job) + " table [" + tableName + "]");
	}

	@Override
	public void onCompleteTable(StorageBackupJob job, String tableName) {
		if (!disabled)
			context.println(getTimestamp() + "<< " + getType(job) + " table [" + tableName + "]");
	}

	@Override
	public void onBeginFile(StorageBackupJob job, String tableName, String fileName, long fileLength) {
		if (!disabled)
			context.println(getTimestamp() + "> " + getType(job) + " file [" + tableName + ":" + fileName + "]");
	}

	@Override
	public void onCompleteFile(StorageBackupJob job, String tableName, String fileName, long fileLength, Throwable error) {
		if (!disabled)
			context.println(getTimestamp() + "< " + getType(job) + " file [" + tableName + ":" + fileName + "]");
	}

	@Override
	public void onBeginJob(StorageBackupJob job) {
		if (!disabled)
			context.println(getTimestamp() + getType(job) + " started table " + getTableNames(job) + ", ["
					+ formatNumber(job.getTotalBytes()) + "] bytes");
	}

	@Override
	public void onCompleteJob(StorageBackupJob job) {
		if (!disabled) {
			context.println(getTimestamp() + getType(job) + " completed, table " + getTableNames(job) + ", ["
					+ formatNumber(job.getTotalBytes()) + "] bytes");
			context.println("Press ctrl-c to exit monitor");
		}
	}

	private String getTimestamp() {
		SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
		return "[" + df.format(new Date()) + "] ";
	}

	private String getType(StorageBackupJob job) {
		return job.getRequest().getType().toString().toLowerCase();
	}

	private Set<String> getTableNames(StorageBackupJob job) {
		if (job.getRequest().getType() == StorageBackupType.BACKUP)
			return job.getStorageFiles().keySet();
		else
			return job.getMediaFiles().keySet();
	}

	private String formatNumber(long bytes) {
		DecimalFormat formatter = new DecimalFormat("###,###");
		return formatter.format(bytes);
	}
}