plugins/org.eclipse.oomph.setup.installer/src/org/eclipse/oomph/setup/internal/installer/KeepInstallerUtil.java

/*
 * Copyright (c) 2015, 2016 Eike Stepper (Loehne, Germany) and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v2.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v20.html
 *
 * Contributors:
 *    Eike Stepper - initial API and implementation
 *    Yatta Solutions - [466264] Enhance UX in simple installer
 */
package org.eclipse.oomph.setup.internal.installer;

import org.eclipse.oomph.util.IOUtil;
import org.eclipse.oomph.util.OS;
import org.eclipse.oomph.util.OomphPlugin.Preference;
import org.eclipse.oomph.util.PropertiesUtil;

import org.eclipse.core.runtime.Platform;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @author Eike Stepper
 */
public final class KeepInstallerUtil
{
  public static final String KEEP_INSTALLER_DESCRIPTION = "Copy the installer to a permanent location on your disk in order to remember your settings and support self updates";

  private static final Preference PREF_KEPT = SetupInstallerPlugin.INSTANCE.getConfigurationPreference("kept");

  private static String powerShell;

  private KeepInstallerUtil()
  {
  }

  public static void createShortCut(String specialFolder, String target)
  {
    createShortCut(specialFolder, null, target, PropertiesUtil.getProductName());
  }

  public static void createShortCut(String specialFolder, String groupName, String target, String shortcutName)
  {
    try
    {
      String powerShell = KeepInstallerUtil.getPowerShell();
      if (powerShell != null)
      {
        if (groupName != null)
        {
          Runtime.getRuntime().exec(
              new String[] {
                  powerShell,
                  "-command",
                  "& { " + "$folderPath = Join-Path ([Environment]::GetFolderPath('" + specialFolder + "')) '" + groupName + "';"
                      + //
                      "[system.io.directory]::CreateDirectory($folderPath); "
                      + //
                      "$linkPath = Join-Path $folderPath '" + shortcutName + ".lnk'; $targetPath = '" + target
                      + "'; $link = (New-Object -ComObject WScript.Shell).CreateShortcut( $linkpath ); $link.TargetPath = $targetPath; $link.Save()}" });

        }
        else
        {
          Runtime.getRuntime().exec(
              new String[] {
                  powerShell,
                  "-command",
                  "& {$linkPath = Join-Path ([Environment]::GetFolderPath('" + specialFolder + "')) '" + shortcutName + ".lnk'; $targetPath = '" + target
                      + "'; $link = (New-Object -ComObject WScript.Shell).CreateShortcut( $linkpath ); $link.TargetPath = $targetPath; $link.Save()}" });

        }
      }
      // [system.io.directory]::CreateDirectory("C:\test")
    }
    catch (IOException ex)
    {
      SetupInstallerPlugin.INSTANCE.log(ex);
    }
  }

  public static void pinToTaskBar(String location, String launcherName)
  {
    try
    {
      String powerShell = KeepInstallerUtil.getPowerShell();
      if (powerShell != null)
      {
        Runtime.getRuntime().exec(
            new String[] { powerShell, "-command",
                "& { (new-object -c shell.application).namespace('" + location + "').parsename('" + launcherName + "').invokeverb('taskbarpin') }" });
      }
    }
    catch (IOException ex)
    {
      SetupInstallerPlugin.INSTANCE.log(ex);
    }
  }

  protected void keepInstaller(String launcher, boolean startMenu, boolean desktop, boolean quickLaunch)
  {
  }

  public static boolean canKeepInstaller()
  {
    if (!isInstallerKept() && OS.INSTANCE.isWin())
    {
      String launcher = OS.getCurrentLauncher(false);
      return launcher != null && launcher.startsWith(PropertiesUtil.getTmpDir());
    }

    return false;
  }

  public static String getPowerShell()
  {
    if (powerShell == null)
    {
      try
      {
        String systemRoot = System.getenv("SystemRoot");
        if (systemRoot != null)
        {
          File system32 = new File(systemRoot, "system32");
          if (system32.isDirectory())
          {
            File powerShellFolder = new File(system32, "WindowsPowerShell");
            if (powerShellFolder.isDirectory())
            {
              File[] versions = powerShellFolder.listFiles();
              if (versions != null)
              {
                for (File version : versions)
                {
                  try
                  {
                    File executable = new File(version, "powershell.exe");
                    if (executable.isFile())
                    {
                      powerShell = executable.getAbsolutePath();
                      break;
                    }
                  }
                  catch (Exception ex)
                  {
                    //$FALL-THROUGH$
                  }
                }
              }
            }
          }
        }
      }
      catch (Exception ex)
      {
        //$FALL-THROUGH$
      }
    }

    return powerShell;
  }

  public static void keepInstaller(String targetLocation, boolean startPermanentInstaller, String launcher, boolean startMenu, boolean desktop,
      boolean quickLaunch)
  {
    File source = new File(launcher).getParentFile();
    File target = new File(targetLocation);
    IOUtil.copyTree(source, target, true);

    String launcherName = new File(launcher).getName();
    String permanentLauncher = new File(target, launcherName).getAbsolutePath();

    if (startPermanentInstaller)
    {
      // Include the application arguments in this launch.
      List<String> command = new ArrayList<String>();
      command.add(permanentLauncher);
      command.addAll(Arrays.asList(Platform.getApplicationArgs()));
      try
      {
        Runtime.getRuntime().exec(command.toArray(new String[command.size()]));
      }
      catch (Exception ex)
      {
        SetupInstallerPlugin.INSTANCE.log(ex);
      }
    }
    else
    {
      String url = target.toURI().toString();
      OS.INSTANCE.openSystemBrowser(url);
    }

    if (startMenu)
    {
      createShortCut("Programs", permanentLauncher);
    }

    if (desktop)
    {
      createShortCut("Desktop", permanentLauncher);
    }

    if (quickLaunch)
    {
      pinToTaskBar(targetLocation, launcherName);
    }

    setKeepInstaller(true);
  }

  public static boolean isInstallerKept()
  {
    return PREF_KEPT.get(false);
  }

  public static void setKeepInstaller(boolean keep)
  {
    PREF_KEPT.set(keep);
  }
}