559 lines
21 KiB
C#
559 lines
21 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Configuration;
|
|
using System.Diagnostics;
|
|
using System.IO;
|
|
using System.Net;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using System.Windows;
|
|
using System.Linq;
|
|
using Ionic.Zip;
|
|
using MediaBrowser.Installer.Code;
|
|
using Microsoft.Win32;
|
|
using ServiceStack.Text;
|
|
|
|
namespace MediaBrowser.Installer
|
|
{
|
|
/// <summary>
|
|
/// Interaction logic for MainWindow.xaml
|
|
/// </summary>
|
|
public partial class MainWindow : Window
|
|
{
|
|
protected PackageVersionClass PackageClass = PackageVersionClass.Release;
|
|
protected Version RequestedVersion = new Version(4,0,0,0);
|
|
protected Version ActualVersion;
|
|
protected string PackageName = "MBServer";
|
|
protected string RootSuffix = "-Server";
|
|
protected string TargetExe = "MediaBrowser.ServerApplication.exe";
|
|
protected string FriendlyName = "Media Browser Server";
|
|
protected string Archive = null;
|
|
protected bool InstallPismo = true;
|
|
protected string RootPath = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.CommonApplicationData), "MediaBrowser-Server");
|
|
|
|
protected bool IsUpdate = false;
|
|
|
|
protected bool SystemClosing = false;
|
|
|
|
protected string TempLocation = Path.Combine(Path.GetTempPath(), "MediaBrowser");
|
|
|
|
protected WebClient MainClient = new WebClient();
|
|
|
|
public MainWindow()
|
|
{
|
|
try
|
|
{
|
|
GetArgs();
|
|
InitializeComponent();
|
|
DoInstall(Archive);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
MessageBox.Show("Error: " + e.Message + " \n\n" + e.StackTrace);
|
|
}
|
|
}
|
|
|
|
private void btnCancel_Click(object sender, RoutedEventArgs e)
|
|
{
|
|
this.Close();
|
|
}
|
|
|
|
protected override void OnClosing(System.ComponentModel.CancelEventArgs e)
|
|
{
|
|
if (!SystemClosing && MessageBox.Show("Cancel Installation - Are you sure?", "Cancel", MessageBoxButton.YesNo) == MessageBoxResult.No)
|
|
{
|
|
e.Cancel = true;
|
|
}
|
|
if (MainClient.IsBusy)
|
|
{
|
|
MainClient.CancelAsync();
|
|
while (MainClient.IsBusy)
|
|
{
|
|
// wait to finish
|
|
}
|
|
}
|
|
MainClient.Dispose();
|
|
ClearTempLocation(TempLocation);
|
|
base.OnClosing(e);
|
|
}
|
|
|
|
protected void SystemClose(string message = null)
|
|
{
|
|
if (message != null)
|
|
{
|
|
MessageBox.Show(message, "Error");
|
|
}
|
|
SystemClosing = true;
|
|
this.Close();
|
|
}
|
|
|
|
protected void GetArgs()
|
|
{
|
|
//cmd line args should be name/value pairs like: product=server archive="c:\.." caller=34552
|
|
var cmdArgs = Environment.GetCommandLineArgs();
|
|
var args = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
|
|
foreach (var pair in cmdArgs)
|
|
{
|
|
var nameValue = pair.Split('=');
|
|
if (nameValue.Length == 2)
|
|
{
|
|
args[nameValue[0]] = nameValue[1];
|
|
}
|
|
}
|
|
Archive = args.GetValueOrDefault("archive", null);
|
|
if (args.GetValueOrDefault("pismo","true") == "false") InstallPismo = false;
|
|
|
|
var product = args.GetValueOrDefault("product", null) ?? ConfigurationManager.AppSettings["product"] ?? "server";
|
|
PackageClass = (PackageVersionClass) Enum.Parse(typeof (PackageVersionClass), args.GetValueOrDefault("class", null) ?? ConfigurationManager.AppSettings["class"] ?? "Release");
|
|
RequestedVersion = new Version(args.GetValueOrDefault("version", "4.0"));
|
|
|
|
var callerId = args.GetValueOrDefault("caller", null);
|
|
if (callerId != null)
|
|
{
|
|
// Wait for our caller to exit
|
|
try
|
|
{
|
|
var process = Process.GetProcessById(Convert.ToInt32(callerId));
|
|
process.WaitForExit();
|
|
}
|
|
catch (ArgumentException)
|
|
{
|
|
// wasn't running
|
|
}
|
|
|
|
IsUpdate = true;
|
|
}
|
|
|
|
//MessageBox.Show(string.Format("Called with args: product: {0} archive: {1} caller: {2}", product, Archive, callerId));
|
|
|
|
switch (product.ToLower())
|
|
{
|
|
case "mbt":
|
|
PackageName = "MBTheater";
|
|
RootSuffix = "-Theater";
|
|
TargetExe = "MediaBrowser.UI.exe";
|
|
FriendlyName = "Media Browser Theater";
|
|
break;
|
|
|
|
default:
|
|
PackageName = "MBServer";
|
|
RootSuffix = "-Server";
|
|
TargetExe = "MediaBrowser.ServerApplication.exe";
|
|
FriendlyName = "Media Browser Server";
|
|
break;
|
|
}
|
|
|
|
RootPath = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.CommonApplicationData), "MediaBrowser" + RootSuffix);
|
|
|
|
}
|
|
|
|
/// <summary>
|
|
/// Execute the install process
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
protected async Task DoInstall(string archive)
|
|
{
|
|
lblStatus.Text = string.Format("Installing {0}...", FriendlyName);
|
|
|
|
// Determine Package version
|
|
var version = archive == null ? await GetPackageVersion() : null;
|
|
ActualVersion = version != null ? version.version : new Version(3,0);
|
|
|
|
// Now try and shut down the server if that is what we are installing and it is running
|
|
var procs = Process.GetProcessesByName("mediabrowser.serverapplication");
|
|
var server = procs.Length > 0 ? procs[0] : null;
|
|
if (!IsUpdate && PackageName == "MBServer" && server != null)
|
|
{
|
|
lblStatus.Text = "Shutting Down Media Browser Server...";
|
|
using (var client = new WebClient())
|
|
{
|
|
try
|
|
{
|
|
client.UploadString("http://localhost:8096/mediabrowser/System/Shutdown", "");
|
|
try
|
|
{
|
|
server.WaitForExit(30000); //don't hang indefinitely
|
|
}
|
|
catch (ArgumentException)
|
|
{
|
|
// already gone
|
|
}
|
|
}
|
|
catch (WebException e)
|
|
{
|
|
if (e.Status == WebExceptionStatus.Timeout || e.Message.StartsWith("Unable to connect",StringComparison.OrdinalIgnoreCase)) return; // just wasn't running
|
|
|
|
MessageBox.Show("Error shutting down server. Please be sure it is not running before hitting OK.\n\n" + e.Status + "\n\n" + e.Message);
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (!IsUpdate && PackageName == "MBTheater")
|
|
{
|
|
// Uninstalling MBT - shut it down if it is running
|
|
var processes = Process.GetProcessesByName("mediabrowser.ui");
|
|
if (processes.Length > 0)
|
|
{
|
|
lblStatus.Text = "Shutting Down Media Browser Theater...";
|
|
try
|
|
{
|
|
processes[0].Kill();
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
MessageBox.Show("Unable to shutdown Media Browser Theater. Please ensure it is not running before hitting OK.\n\n" + ex.Message, "Error");
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
// Download if we don't already have it
|
|
if (archive == null)
|
|
{
|
|
lblStatus.Text = string.Format("Downloading {0} (version {1})...", FriendlyName, version.versionStr);
|
|
try
|
|
{
|
|
archive = await DownloadPackage(version);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
SystemClose("Error Downloading Package - " + e.GetType().FullName + "\n\n" + e.Message);
|
|
return;
|
|
}
|
|
}
|
|
|
|
if (archive == null) return; //we canceled or had an error that was already reported
|
|
|
|
// Extract
|
|
lblStatus.Text = "Extracting Package...";
|
|
try
|
|
{
|
|
ExtractPackage(archive);
|
|
// We're done with it so delete it (this is necessary for update operations)
|
|
try
|
|
{
|
|
File.Delete(archive);
|
|
}
|
|
catch (FileNotFoundException)
|
|
{
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
SystemClose("Error Removing Archive - " + e.GetType().FullName + "\n\n" + e.Message);
|
|
return;
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
SystemClose("Error Extracting - " + e.GetType().FullName + "\n\n" + e.Message);
|
|
return;
|
|
}
|
|
|
|
// Create shortcut
|
|
lblStatus.Text = "Creating Shortcuts...";
|
|
var fullPath = Path.Combine(RootPath, "System", TargetExe);
|
|
try
|
|
{
|
|
CreateShortcuts(fullPath);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
SystemClose("Error Creating Shortcut - "+e.GetType().FullName+"\n\n"+e.Message);
|
|
return;
|
|
}
|
|
|
|
// Install Pismo
|
|
if (InstallPismo)
|
|
{
|
|
lblStatus.Text = "Installing ISO Support...";
|
|
try
|
|
{
|
|
PismoInstall();
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
SystemClose("Error Installing Pismo - "+e.GetType().FullName+"\n\n"+e.Message);
|
|
return;
|
|
}
|
|
}
|
|
|
|
// Now delete the pismo install files
|
|
Directory.Delete(Path.Combine(RootPath, "Pismo"), true);
|
|
|
|
// And run
|
|
lblStatus.Text = string.Format("Starting {0}...", FriendlyName);
|
|
try
|
|
{
|
|
Process.Start(fullPath);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
SystemClose("Error Executing - "+fullPath+ " "+e.GetType().FullName+"\n\n"+e.Message);
|
|
return;
|
|
}
|
|
|
|
SystemClose();
|
|
|
|
}
|
|
|
|
private void PismoInstall()
|
|
{
|
|
// Kick off the Pismo installer and wait for it to end
|
|
var pismo = new Process();
|
|
pismo.StartInfo.WindowStyle = ProcessWindowStyle.Hidden;
|
|
pismo.StartInfo.FileName = Path.Combine(RootPath, "Pismo", "pfminst.exe");
|
|
pismo.StartInfo.Arguments = "install";
|
|
pismo.Start();
|
|
pismo.WaitForExit();
|
|
|
|
}
|
|
|
|
protected async Task<PackageVersionInfo> GetPackageVersion()
|
|
{
|
|
try
|
|
{
|
|
// get the package information for the server
|
|
var json = await MainClient.DownloadStringTaskAsync("http://www.mb3admin.com/admin/service/package/retrieveAll?name=" + PackageName);
|
|
var packages = JsonSerializer.DeserializeFromString<List<PackageInfo>>(json);
|
|
|
|
var version = packages[0].versions.Where(v => v.classification <= PackageClass).OrderByDescending(v => v.version).FirstOrDefault(v => v.version <= RequestedVersion);
|
|
if (version == null)
|
|
{
|
|
SystemClose("Could not locate download package. Aborting.");
|
|
return null;
|
|
}
|
|
return version;
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
SystemClose(e.GetType().FullName + "\n\n" + e.Message);
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Download our specified package to an archive in a temp location
|
|
/// </summary>
|
|
/// <returns>The fully qualified name of the downloaded package</returns>
|
|
protected async Task<string> DownloadPackage(PackageVersionInfo version)
|
|
{
|
|
var success = false;
|
|
var retryCount = 0;
|
|
var archiveFile = Path.Combine(PrepareTempLocation(), version.targetFilename);
|
|
|
|
try
|
|
{
|
|
while (!success && retryCount < 3)
|
|
{
|
|
|
|
// setup download progress and download the package
|
|
MainClient.DownloadProgressChanged += DownloadProgressChanged;
|
|
try
|
|
{
|
|
await MainClient.DownloadFileTaskAsync(version.sourceUrl, archiveFile);
|
|
success = true;
|
|
}
|
|
catch (WebException e)
|
|
{
|
|
if (e.Status == WebExceptionStatus.RequestCanceled)
|
|
{
|
|
return null;
|
|
}
|
|
if (retryCount < 3 && (e.Status == WebExceptionStatus.Timeout || e.Status == WebExceptionStatus.ConnectFailure || e.Status == WebExceptionStatus.ProtocolError))
|
|
{
|
|
Thread.Sleep(500); //wait just a sec
|
|
PrepareTempLocation(); //clear this out
|
|
retryCount++;
|
|
}
|
|
else
|
|
{
|
|
throw;
|
|
}
|
|
}
|
|
}
|
|
|
|
return archiveFile;
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
SystemClose(e.GetType().FullName + "\n\n" + e.Message);
|
|
}
|
|
return "";
|
|
|
|
}
|
|
|
|
void DownloadProgressChanged(object sender, DownloadProgressChangedEventArgs e)
|
|
{
|
|
rectProgress.Width = (this.Width * e.ProgressPercentage)/100f;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Extract the provided archive to our program root
|
|
/// It is assumed the archive is a zip file relative to that root (with all necessary sub-folders)
|
|
/// </summary>
|
|
/// <param name="archive"></param>
|
|
protected void ExtractPackage(string archive)
|
|
{
|
|
// Delete old content of system
|
|
var systemDir = Path.Combine(RootPath, "System");
|
|
var backupDir = Path.Combine(RootPath, "System.old");
|
|
if (Directory.Exists(systemDir))
|
|
{
|
|
if (Directory.Exists(backupDir)) Directory.Delete(backupDir,true);
|
|
Directory.Move(systemDir, backupDir);
|
|
}
|
|
|
|
// And extract
|
|
var retryCount = 0;
|
|
var success = false;
|
|
while (!success && retryCount < 3)
|
|
{
|
|
try
|
|
{
|
|
using (var fileStream = File.OpenRead(archive))
|
|
{
|
|
using (var zipFile = ZipFile.Read(fileStream))
|
|
{
|
|
zipFile.ExtractAll(RootPath, ExtractExistingFileAction.OverwriteSilently);
|
|
success = true;
|
|
}
|
|
}
|
|
}
|
|
catch
|
|
{
|
|
if (retryCount < 3)
|
|
{
|
|
Thread.Sleep(250);
|
|
retryCount++;
|
|
}
|
|
else
|
|
{
|
|
//Rollback
|
|
if (Directory.Exists(systemDir)) Directory.Delete(systemDir);
|
|
Directory.Move(backupDir, systemDir);
|
|
File.Delete(archive); // so we don't try again if its an update
|
|
throw;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Create a shortcut in the current user's start menu
|
|
/// Only do current user to avoid need for admin elevation
|
|
/// </summary>
|
|
/// <param name="targetExe"></param>
|
|
protected void CreateShortcuts(string targetExe)
|
|
{
|
|
// get path to all users start menu
|
|
var startMenu = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.StartMenu),"Media Browser 3");
|
|
if (!Directory.Exists(startMenu)) Directory.CreateDirectory(startMenu);
|
|
var product = new ShellShortcut(Path.Combine(startMenu, FriendlyName+".lnk")) {Path = targetExe, Description = "Run " + FriendlyName};
|
|
product.Save();
|
|
|
|
if (PackageName == "MBServer")
|
|
{
|
|
var path = Path.Combine(startMenu, "MB Dashboard.lnk");
|
|
var dashboard = new ShellShortcut(path)
|
|
{Path = @"http://localhost:8096/mediabrowser/dashboard/dashboard.html", Description = "Open the Media Browser Server Dashboard (configuration)"};
|
|
dashboard.Save();
|
|
|
|
}
|
|
CreateUninstaller(Path.Combine(Path.GetDirectoryName(targetExe) ?? "", "MediaBrowser.Uninstaller.exe")+ " "+ (PackageName == "MBServer" ? "server" : "mbt"), targetExe);
|
|
|
|
}
|
|
|
|
/// <summary>
|
|
/// Create uninstall entry in add/remove
|
|
/// </summary>
|
|
/// <param name="uninstallPath"></param>
|
|
/// <param name="targetExe"></param>
|
|
private void CreateUninstaller(string uninstallPath, string targetExe)
|
|
{
|
|
var parent = Registry.CurrentUser.OpenSubKey(@"SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall", true);
|
|
{
|
|
if (parent == null)
|
|
{
|
|
var rootParent = Registry.CurrentUser.OpenSubKey(@"SOFTWARE\Microsoft\Windows\CurrentVersion", true);
|
|
{
|
|
if (rootParent != null)
|
|
{
|
|
parent = rootParent.CreateSubKey("Uninstall");
|
|
if (parent == null)
|
|
{
|
|
MessageBox.Show("Unable to create Uninstall registry key. Program is still installed sucessfully.");
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
try
|
|
{
|
|
RegistryKey key = null;
|
|
|
|
try
|
|
{
|
|
const string guidText = "{4E76DB4E-1BB9-4A7B-860C-7940779CF7A0}";
|
|
key = parent.OpenSubKey(guidText, true) ??
|
|
parent.CreateSubKey(guidText);
|
|
|
|
if (key == null)
|
|
{
|
|
MessageBox.Show(String.Format("Unable to create uninstaller entry'{0}\\{1}'. Program is still installed successfully.", @"SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall", guidText));
|
|
return;
|
|
}
|
|
|
|
key.SetValue("DisplayName", FriendlyName);
|
|
key.SetValue("ApplicationVersion", ActualVersion);
|
|
key.SetValue("Publisher", "Media Browser Team");
|
|
key.SetValue("DisplayIcon", targetExe);
|
|
key.SetValue("DisplayVersion", ActualVersion.ToString(2));
|
|
key.SetValue("URLInfoAbout", "http://www.mediabrowser3.com");
|
|
key.SetValue("Contact", "http://community.mediabrowser.tv");
|
|
key.SetValue("InstallDate", DateTime.Now.ToString("yyyyMMdd"));
|
|
key.SetValue("UninstallString", uninstallPath);
|
|
}
|
|
finally
|
|
{
|
|
if (key != null)
|
|
{
|
|
key.Close();
|
|
}
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
MessageBox.Show("An error occurred writing uninstall information to the registry.");
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Prepare a temporary location to download to
|
|
/// </summary>
|
|
/// <returns>The path to the temporary location</returns>
|
|
protected string PrepareTempLocation()
|
|
{
|
|
ClearTempLocation(TempLocation);
|
|
Directory.CreateDirectory(TempLocation);
|
|
return TempLocation;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Clear out (delete recursively) the supplied temp location
|
|
/// </summary>
|
|
/// <param name="location"></param>
|
|
protected void ClearTempLocation(string location)
|
|
{
|
|
if (Directory.Exists(location))
|
|
{
|
|
Directory.Delete(location, true);
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|