/*******************************************************************************
 * Copyright Technophobia Ltd 2012
 * 
 * This file is part of the Substeps Eclipse Plugin.
 * 
 * The Substeps Eclipse Plugin is free software: you can redistribute it and/or modify
 * it under the terms of the Eclipse Public License v1.0.
 * 
 * The Substeps Eclipse Plugin is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * Eclipse Public License for more details.
 * 
 * You should have received a copy of the Eclipse Public License
 * along with the Substeps Eclipse Plugin.  If not, see <http://www.eclipse.org/legal/epl-v10.html>.
 ******************************************************************************/
package com.technophobia.substeps.junit.ui;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.text.MessageFormat;

import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.debug.core.ILaunchConfiguration;
import org.eclipse.debug.core.ILaunchConfigurationWorkingCopy;
import org.eclipse.jface.dialogs.ErrorDialog;
import org.eclipse.swt.widgets.Shell;

import com.technophobia.eclipse.launcher.config.SubstepsLaunchConfigurationConstants;
import com.technophobia.eclipse.ui.Notifier;
import com.technophobia.substeps.FeatureRunnerPlugin;
import com.technophobia.substeps.model.structure.SubstepsTestElement;
import com.technophobia.substeps.supplier.Supplier;

public class FailedTestFirstTestRelauncher extends TestRelauncher {

    private final Shell shell;
    final Supplier<SubstepsRunSession> testRunSession;


    public FailedTestFirstTestRelauncher(final Supplier<SubstepsRunSession> testRunSession, final Shell shell,
            final Notifier<String> infoMessageNotifier) {
        super(testRunSession, shell, infoMessageNotifier);
        this.testRunSession = testRunSession;
        this.shell = shell;
    }


    @Override
    protected ILaunchConfiguration prepareLaunchConfigForRelaunch(final ILaunchConfiguration launchConfiguration) {
        try {
            final String oldName = launchConfiguration.getName();
            final String oldFailuresFilename = launchConfiguration.getAttribute(
                    SubstepsLaunchConfigurationConstants.ATTR_FAILURES_NAMES, (String) null);
            String configName;
            if (oldFailuresFilename != null) {
                configName = oldName;
            } else {
                configName = MessageFormat.format(
                        SubstepsFeatureMessages.SubstepsFeatureTestRunnerViewPart_rerunFailedFirstLaunchConfigName,
                        oldName);
            }
            final ILaunchConfigurationWorkingCopy tmp = launchConfiguration.copy(configName);
            tmp.setAttribute(SubstepsLaunchConfigurationConstants.ATTR_FAILURES_NAMES, createFailureNamesFile());
            return tmp;
        } catch (final CoreException ex) {
            ErrorDialog.openError(shell, SubstepsFeatureMessages.SubstepsFeatureTestRunnerViewPart_error_cannotrerun,
                    ex.getMessage(), ex.getStatus());
            return null;
        }
    }


    private String createFailureNamesFile() throws CoreException {
        try {
            final File file = File.createTempFile("testFailures", ".txt"); //$NON-NLS-1$ //$NON-NLS-2$
            file.deleteOnExit();
            final SubstepsTestElement[] failures = testRunSession.get().getAllFailedTestElements();
            BufferedWriter bw = null;
            try {
                bw = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(file), "UTF-8")); //$NON-NLS-1$
                for (final SubstepsTestElement testElement : failures) {
                    bw.write(testElement.getTestName());
                    bw.newLine();
                }
            } finally {
                if (bw != null) {
                    bw.close();
                }
            }
            return file.getAbsolutePath();
        } catch (final IOException e) {
            throw new CoreException(new Status(IStatus.ERROR, FeatureRunnerPlugin.PLUGIN_ID, IStatus.ERROR, "", e)); //$NON-NLS-1$
        }
    }
}