Cosmos/source2/Build/Cosmos.Build.Builder/CosmosTask.cs

using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;
using System.Text;
using Cosmos.Build.Installer;
using System.IO;
using Microsoft.Win32;
using System.Windows;

namespace Cosmos.Build.Builder {
  public class CosmosTask : Task {
    protected string mCosmosPath;
    protected string mOutputPath;
    protected int mReleaseNo;
    protected string mInnoFile;

    public CosmosTask(string aCosmosPath, int aReleaseNo) {
      mCosmosPath = aCosmosPath;
      mReleaseNo = aReleaseNo;
      mInnoFile = mCosmosPath + @"\Setup2\Cosmos.iss";
    }

    protected override void DoRun() {
      mOutputPath = Path.Combine(mCosmosPath, @"Build\VSIP");
      if (Directory.Exists(mOutputPath)) {
        // Make sure no files are left, else things can be not be rebuilt and when adding
        // new items this can cause issues.
        Directory.Delete(mOutputPath, true);
      } else {
        Directory.CreateDirectory(mOutputPath);
      }

      CheckPrereqs();
      
      CompileCosmos();
      CopyTemplates();
      if (App.IsUserKit) {
        CreateUserKitScript();
      }
      CreateSetup();
      if (!App.IsUserKit) {
        RunSetup();
        WriteDevKit();
        LaunchVS();
      }

      Done();
    }
    
    protected void MsBuild(string aSlnFile, string aBuildCfg) {
      StartConsole(Paths.Windows + @"\Microsoft.NET\Framework\v4.0.30319\msbuild.exe", Quoted(aSlnFile) + @" /maxcpucount /verbosity:normal /nologo /p:Configuration=" + aBuildCfg + " /p:Platform=x86 /p:OutputPath=" + Quoted(mOutputPath));
    }

    protected bool CheckForInstall(string aCheck, bool aCanThrow) {
      return CheckForProduct(aCheck, aCanThrow, @"SOFTWARE\Classes\Installer\Products\", "ProductName");
    }
    protected bool CheckForUninstall(string aCheck, bool aCanThrow) {
      return CheckForProduct(aCheck, aCanThrow, @"SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\", "DisplayName");
    }
    protected bool CheckForProduct(string aCheck, bool aCanThrow, string aKey, string aValueName) {
      Echo("Looking for " + aCheck);
      string xCheck = aCheck.ToUpper();
      string[] xKeys;
      using (var xKey = Registry.LocalMachine.OpenSubKey(aKey, false)) {
        xKeys = xKey.GetSubKeyNames();
      }
      foreach (string xSubKey in xKeys) {
        using (var xKey = Registry.LocalMachine.OpenSubKey(aKey + xSubKey, false)) {
          string xValue = (string)xKey.GetValue(aValueName);
          if (xValue != null && xValue.ToUpper().Contains(xCheck)) {
            return true;
          }
        }
      }

      if (aCanThrow) {
        NotFound(aCheck);
      }
      return false;
    }

    protected void CheckNet35Sp1() {
      Echo("Looking for .NET 3.5 SP1");
      bool xNet35SP1Installed = false;
      using (var xKey = Registry.LocalMachine.OpenSubKey(@"SOFTWARE\Microsoft\NET Framework Setup\NDP\v3.5", false)) {
        if (xKey != null) {
          xNet35SP1Installed = (int)xKey.GetValue("SP", 0) >= 1;
        }
      }
      if (!xNet35SP1Installed) {
        NotFound(".NET 3.5 SP1");
      }
    }

    protected void CheckIsVsRunning() {
      int xSeconds = 5;
      if (App.IgnoreVS) {
        return;
      }

      Echo("Checking if Visual Studio is running.");
      if (IsRunning("devenv")) {
        Echo("--Visual Studio is running.");
        Echo("--Waiting " + xSeconds + " seconds to see if Visual Studio exits.");
        // VS doesnt exit right away and user can try devkit again after VS window has closed but is still running.
        // So we wait a few seconds first.
        if (WaitForExit("devenv", xSeconds * 1000)) {
          throw new Exception("Visual Studio is running. Please close it or kill it in task manager.");
        }
      }
    }

    protected void NotFound(string aName) {
      throw new Exception("Prerequisite '" + aName + "' not found.");
    }

    protected void CheckPrereqs() {
      Section("Checking Prerequisites");
      Echo("Note: This does not check all prerequisites, please see website for full list.");

      Echo("Checking for x86 run.");
      if (!AmRunning32Bit()) {
        throw new Exception("Builder must run as x86");
      }

      // We assume they have normal .NET stuff if user was able to build the builder...
      //Visual Studio 2010

      CheckIsVsRunning();
      CheckVs2010Sp1();
      CheckVs2010SDK();
      CheckNet35Sp1(); // Required by VMWareLib
      CheckForUninstall("Inno Setup QuickStart Pack", true);
      CheckForInstall("Microsoft Visual Studio 2010 SDK SP1", true);
      if (!CheckForInstall("VMware Workstation", false)) {
        if (!CheckForInstall("VMware Player", false)) {
          NotFound("VMWare");
        }
      }
      CheckForInstall("VMWare VIX", true);
    }

    void CheckVs2010Sp1() {
      // If user got this far, we know they have VS 2010. But we need to make sure
      // that its SP1.
      Echo("Checking for Visual Studio 2010 SP1");
      using (var xKey = Registry.LocalMachine.OpenSubKey(@"SOFTWARE\Wow6432Node\Microsoft\VisualStudio\10.0")) {
        string xDir = (string)xKey.GetValue("InstallDir");
        var xInfo = FileVersionInfo.GetVersionInfo(Path.Combine(xDir, "devenv.exe"));
        if (xInfo.ProductPrivatePart < 1) {
          throw new Exception("Visual Studio 2010 **SP1** not detected.");
        }
      }
    }

    void CheckVs2010SDK() {
      Echo("Checking for Visual Studio 2010 SDK SP1");
      using (var xKey = Registry.LocalMachine.OpenSubKey(@"SOFTWARE\Wow6432Node\Microsoft\VisualStudio\VSIP\10.0")) {
        // This tell us if SDK is installed, but not if its SP1. But if CheckVs2010Sp1() passed, then it should be SP1.
        if (xKey.GetValue("InstallDir") == null) {
          throw new Exception("Visual Studio 2010 SDK not detected.");
        }
      }
    }

    void WriteDevKit() {
      Section("Writing Dev Kit to Registry");

      // Inno deletes this from registry, so we must add this after. 
      // We let Inno delete it, so if user runs it by itself they get
      // only UserKit, and no DevKit settings.
      // HKCU instead of HKLM because builder does not run as admin.
      //
      // HKCU is not redirected.
      using (var xKey = Registry.CurrentUser.CreateSubKey(@"Software\Cosmos")) {
        xKey.SetValue("DevKit", mCosmosPath);
      }
    }

    void CreateUserKitScript() {
      Section("Creating User Kit Script");

      // Read in Cosmos.iss
      using (var xSrc = new StreamReader(mInnoFile)) {
        mInnoFile = Path.Combine(Path.GetDirectoryName(mInnoFile), "UserKit.iss");
        // Write out UserKit.iss
        using (var xDest = new StreamWriter(mInnoFile)) {
          string xLine;
          while ((xLine = xSrc.ReadLine()) != null) {
            if (xLine.StartsWith("#define ChangeSetVersion ", StringComparison.InvariantCultureIgnoreCase)) {
              xDest.WriteLine("#define ChangeSetVersion " + Quoted(mReleaseNo.ToString()));
            } else {
              xDest.WriteLine(xLine);
            }
          }
        }
      }
    }

    void CompileXSharpCompiler() {
      Section("Compiling X# Compiler");

      MsBuild(Path.Combine(mCosmosPath, @"XSharp\source\XSharp.sln"), "Debug");
    }

    void CompileXSharpSource() {
      Section("Compiling X# Sources");
      
      // XSC can do all files in path, but we do it on our own currently for better status updates.
      // When we get xsproj files we can build directly.
      var xFiles = Directory.GetFiles(mCosmosPath + @"source2\Compiler\Cosmos.Compiler.DebugStub\", "*.xs");
      foreach (var xFile in xFiles) {
        Echo("Compiling " + Path.GetFileName(xFile));
        string xDest = Path.ChangeExtension(xFile, ".cs");
        if (File.Exists(xDest)) {
          ResetReadOnly(xDest);
        }
        // We dont ref the X# asm directly because then we could not compile it without dynamic loading.
        // This way we can build it and call it directly.
        StartConsole(mOutputPath + @"\xsc.exe", Quoted(xFile) + @" Cosmos.Debug.DebugStub");
      }
    }

    void CompileCosmos() {
      Section("Compiling Cosmos");

      MsBuild(mCosmosPath + @"\source\Build.sln", "Debug");
    }

    void CopyTemplates() {
      Section("Copying Templates");

      CD(mOutputPath);
      SrcPath = Path.Combine(mCosmosPath, @"source2\VSIP\Cosmos.VS.Package\obj\x86\Debug");
      Copy("CosmosProject (C#).zip");
      Copy("CosmosKernel (C#).zip");
      Copy("CosmosProject (F#).zip");
      Copy("Cosmos.zip");
      Copy("CosmosProject (VB).zip");
      Copy("CosmosKernel (VB).zip");
      Copy(mCosmosPath + @"source2\VSIP\Cosmos.VS.XSharp\Template\XSharpFileItem.zip");
    }

    void CreateSetup() {
      Section("Creating Setup");

      string xInnoPath;
      using (var xKey = Registry.LocalMachine.OpenSubKey(@"SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\Inno Setup 5_is1", false)) {
        if (xKey == null) {
          throw new Exception("Cannot find Inno Setup.");
        }
        xInnoPath = (string)xKey.GetValue("InstallLocation");
        if (string.IsNullOrWhiteSpace(xInnoPath)) {
          throw new Exception("Cannot find Inno Setup.");
        }
      }

      string xISCC = Path.Combine(xInnoPath, "ISCC.exe");
      if (!File.Exists(xISCC)) {
        throw new Exception("Cannot find Inno setup.");
      }
      string xCfg = App.IsUserKit ? "UserKit" : "DevKit";
      StartConsole(xISCC, @"/Q " + Quoted(mInnoFile) + " /dBuildConfiguration=" + xCfg);

      if (App.IsUserKit) {
        File.Delete(mInnoFile);
      }
    }

    void LaunchVS() {
      Section("Launching Visual Studio");
     
      string xVisualStudio = Paths.ProgFiles32 + @"\Microsoft Visual Studio 10.0\Common7\IDE\devenv.exe";
      if (!File.Exists(xVisualStudio)) {
        throw new Exception("Cannot find Visual Studio.");
      }

      if (App.ResetHive) {
        Echo("Resetting hive");
        Start(xVisualStudio, @"/setup /rootsuffix Exp /ranu");
      }

      Echo("Launching Visual Studio");
      Start(xVisualStudio, mCosmosPath + @"\source\Cosmos.sln", false, true);
    }

    void RunSetup() {
      Section("Running Setup");

      if (App.UseTask) {
        // This is a hack to avoid the UAC dialog on every run which can be very disturbing if you run
        // the dev kit a lot.
        Start(@"schtasks.exe", @"/run /tn " + Quoted("CosmosSetup"), true, false);

        // Must check for start before stop, else on slow machines we exit quickly because Exit is found before
        // it starts.
        // Some slow user PCs take around 5 seconds to start up the task...
        int xSeconds = 10;
        var xTimed = DateTime.Now;
        Echo("Waiting " + xSeconds + " seconds for Setup to start.");
        if (WaitForStart("CosmosUserKit-" + mReleaseNo, xSeconds * 1000)) {
          throw new Exception("Setup did not start.");
        }
        Echo("Setup is running. " + DateTime.Now.Subtract(xTimed).ToString(@"ss\.fff"));

        // Scheduler starts it an exits, but we need to wait for the setup itself to exit before proceding
        Echo("Waiting for Setup to complete.");
        WaitForExit("CosmosUserKit-" + mReleaseNo);
      } else {
        Start(mCosmosPath + @"\Setup2\Output\CosmosUserKit-" + mReleaseNo + ".exe", @"/SILENT");
      }
    }

    void Done() {
      Section("Build Complete!");
    }
  }
}