package com.installshield.watchpoint.event.dialog.swing;


import java.util.Vector;
import java.io.*;

import com.installshield.database.ISDatabaseException;
import com.installshield.event.*;
import com.installshield.event.ui.*;
import com.installshield.wizard.*;
import com.installshield.wizard.service.*;
import com.installshield.wizard.service.system.*;
import com.installshield.util.*;

public class PanelReboot {

    private static final String REBOOTNOW_BUTTON_VARIABLE = "IS_REBOOT_NOW";
    private static final String REBOOTLATER_BUTTON_VARIABLE = "IS_REBOOT_LATER";
    private static final String WIZARD_RESTART_VARIABLE = "IS_WIZARD_RESTART";

    boolean active;
    Boolean restartWizard;
    public void queryEnterReboot(ISQueryContext context) {

        /**Set restartWizard to true if you wish to have the 
        * wizard restart after reboot with the next bean 
        * in the sequence.
        */
        try {
            restartWizard = new Boolean(context.getServices().getISDatabase()
                    .getVariableValue(REBOOTNOW_BUTTON_VARIABLE));
        } catch (ServiceException e) {
            e.printStackTrace();
        }
        try {
                    context.getServices().getISDatabase().setVariableValue(WIZARD_RESTART_VARIABLE,restartWizard.toString());
				}
				catch (Exception e) {
					context.getServices().logEvent(this, Log.ERROR, e);
				}
				
				SystemUtilService service = null;
        try {
            service =
                (SystemUtilService)context.getService(SystemUtilService.NAME);
            context.setReturnValue(service.isRebootRequired());
        }
        catch (Throwable e) {
            e.printStackTrace();
            context.setReturnValue(false);
        }
    }

    public void queryExitReboot(ISDialogQueryContext context) {

        boolean rebootSelected = false;
        Boolean restartWizard = null;
		try {
			 restartWizard = new Boolean(	context.getServices().getISDatabase().getVariableValue(WIZARD_RESTART_VARIABLE));
		}
		catch (Exception e) {
			context.getServices().logEvent(this, Log.ERROR, e);
		}

        try {
            String rebootNow =
                context.getServices().getISDatabase().getVariableValue(
                    REBOOTNOW_BUTTON_VARIABLE);
            
            rebootSelected = (rebootNow.equalsIgnoreCase("true")) ? true : false;

            SystemUtilService service =
                (SystemUtilService)context.getService(SystemUtilService.NAME);
            service.setRebootOnExit(rebootSelected);
            
            
        }
        catch (ServiceException e) {
            e.printStackTrace();
        }
		
	  if (restartWizard.booleanValue()) {    
		try {
			SystemUtilService service = (SystemUtilService)context.getService(SystemUtilService.NAME);
			WizardBean nextBean = context.getWizard().getIterator().getNext(context.getWizard().getCurrentBean());
			// if the current bean is the last wizard bean, then just reboot
			if (nextBean != context.getWizard().getIterator().end()) {
				String launcherFileName = System.getProperty("is.launcher.file");
				if (launcherFileName != null) {
					File launchFile = new File(launcherFileName);
					if (!launchFile.exists()) {
					} else {
						String restartCommand = launcherFileName +
												" -goto " +
												nextBean.getBeanId();
						service.addSystemStartupCommand(restartCommand);
					}
				} else {
					LogUtils.getLog().logEvent(this, 
							 Log.ERROR, 
							 "Restarting the wizard after a reboot is only " +
							 "supported through the use of a native launcher.");
				}
				context.getWizard().exit(ExitCodes.WIZARD_CALLBACK_REQUIRED);
			}
		} catch (ServiceException e) {
			e.printStackTrace();
		}	
	}
    }

    public void generateOptionsEntriesReboot(ISOptionsContext context) {

        String rebootNowValue = null;
        String rebootLaterValue = null;
        String option = null;
        String panelId = context.getPanel().getName();
        Vector optionEntries = context.getOptionEntries();

        try {

            if (context.getValueType() == WizardBean.TEMPLATE_VALUE) {
                rebootNowValue =
                    rebootLaterValue =
                        LocalizedStringResolver.resolve(
                            "com.installshield.wizard.i18n.WizardResources",
                            "WizardBean.valueStr");
            }
            else {
                rebootNowValue =
                    context
                        .getWizard()
                        .getServices()
                        .getISDatabase()
                        .getVariableValue(
                        REBOOTNOW_BUTTON_VARIABLE);
                rebootLaterValue =
                    context
                        .getWizard()
                        .getServices()
                        .getISDatabase()
                        .getVariableValue(
                        REBOOTLATER_BUTTON_VARIABLE);
            }
        }
        catch (ServiceException e) {
            LogUtils.getLog().logEvent(this, Log.ERROR, e);
        }
    }
}