Agent MSI support

src/CustomAction.config

@@ -0,0 +1,11 @@
+<?xml version="1.0" encoding="utf-8" ?>
+                                                <configuration>
+                                                    <startup useLegacyV2RuntimeActivationPolicy="true">
+                                                        <supportedRuntime version="v4.0.30319"/>
+                                                        <supportedRuntime version="v4.0" sku=".NETFramework,Version=v4.5"/>
+                                                        <supportedRuntime version="v4.0" sku=".NETFramework,Version=v4.0"/>
+                                                        <supportedRuntime version="v2.0.50727"/>
+                                                        <supportedRuntime version="v2.0.50215"/>
+                                                        <supportedRuntime version="v1.1.4322"/>
+                                                    </startup>
+                                                </configuration>

src/config/config.yaml

@@ -114,3 +114,13 @@ products:
     published_binaries: [ winlogbeat.exe ]
     mutable_dirs: [ ]
     service: true
+
+  elastic-agent:
+    display_name: Elastic Agent
+    description: With Elastic Agent you can collect all forms of data from anywhere with a single unified agent per host
+    published_name: Elastic Agent
+    published_url: https://www.elastic.co/elastic-agent
+    published_binaries: [ elastic-agent.exe ]
+    mutable_dirs: [ ]
+    is_agent: true
+    service: false

src/installer/BeatPackageCompiler/AgentCustomAction.cs

@@ -0,0 +1,74 @@
+using System;
+using System.IO;
+using Microsoft.Deployment.WindowsInstaller;
+
+namespace Elastic.PackageCompiler.Beats
+{
+    public class AgentCustomAction
+    {
+        [CustomAction]
+        public static ActionResult InstallAction(Session session)
+        {
+            try
+            {
+                // If there are no install args, we stop here
+                // // (the MSI just copied the files, there will be no agent-install)
+                if (string.IsNullOrEmpty(session["INSTALLARGS"]))
+                    return ActionResult.Success;
+
+                string install_args = session["INSTALLARGS"];
+                string install_folder = Path.Combine(session["INSTALLDIR"], session["exe_folder"]);
+                System.Diagnostics.Process process = new System.Diagnostics.Process();
+                process.StartInfo.WorkingDirectory = install_folder;
+                process.StartInfo.FileName = "elastic-agent.exe";
+                process.StartInfo.Arguments = "install -f " + install_args;
+                process.StartInfo.CreateNoWindow = true;
+                process.Start();
+                process.WaitForExit();
+
+                session.Log("Agent install return code:" + process.ExitCode);
+
+                if (process.ExitCode == 0)
+                {
+                    // If agent got installed properly, we can go ahead and remove all the files installed by the MSI (best effort)
+                    new DirectoryInfo(install_folder).Delete(true);
+                }
+
+                return process.ExitCode == 0 ? ActionResult.Success : ActionResult.Failure;
+            }
+            catch
+            {
+                return ActionResult.Failure;
+            }
+        }
+
+        [CustomAction]
+        public static ActionResult UnInstallAction(Session session)
+        {
+            try
+            {
+                // If there are no (un)install args, we stop here
+                if (string.IsNullOrEmpty(session["INSTALLARGS"]))
+                    return ActionResult.Success;
+
+                string install_args = session["INSTALLARGS"];
+                System.Diagnostics.Process process = new System.Diagnostics.Process();
+                process.StartInfo.FileName = @"c:\\Program Files\\Elastic\\Agent\\elastic-agent.exe";
+                process.StartInfo.Arguments = "uninstall -f " + install_args;
+                process.StartInfo.CreateNoWindow = true;
+                process.Start();
+                process.WaitForExit();
+
+                session.Log("Agent uninstall return code:" + process.ExitCode);
+            }
+            catch (Exception ex)
+            {
+                // IMPORTANT! Uninstall will be done as best effort..
+                // We don't want to fail the MSI uninstall in case there is an issue with the agent uninstall command.
+                session.Log(ex.ToString());
+            }
+
+            return ActionResult.Success;
+        }
+    }
+}

src/installer/BeatPackageCompiler/BeatPackageCompiler.cs

@@ -31,7 +31,10 @@ static void Main(string[] args)
 
             var companyName = MagicStrings.Elastic;
             var productSetName = MagicStrings.Beats.Name;
-            var displayName = MagicStrings.Beats.Name + " " + ap.TargetName;
+
+            // A product can define a display name to be used.
+            // At the time of writing this line, elastic-agent is the only product that used it
+            var displayName = !string.IsNullOrEmpty(pc.DisplayName) ? pc.DisplayName : MagicStrings.Beats.Name + " " + ap.TargetName;
             var exeName = ap.CanonicalTargetName + MagicStrings.Ext.DotExe;
 
             // Generate UUID v5 from product properties.
@@ -156,7 +159,9 @@ static void Main(string[] args)
                     bool exclude = 
                         // .exe must be excluded for service configuration to work
                         (pc.IsWindowsService && itm.EndsWith(exeName, StringComparison.OrdinalIgnoreCase))
-                        || (isConfigFile);
+
+                        // beats config file is handled further down
+                        || (!pc.IsAgent && isConfigFile);
 
                     // this is an "include" filter
                     return ! exclude;
@@ -165,65 +170,35 @@ static void Main(string[] args)
 
             packageContents.Add(pc.IsWindowsService ? service : null);
 
-            // Add a note to the final screen and a checkbox to open the directory of .example.yml file
-            var beatConfigExampleFileName = ap.CanonicalTargetName + ".example" + MagicStrings.Ext.DotYml;
-            var beatConfigExampleFileId = beatConfigExampleFileName + "_" + (uint) beatConfigExampleFileName.GetHashCode32();
-
-            project.AddProperty(new Property("WIXUI_EXITDIALOGOPTIONALTEXT",
-                $"NOTE: Only Administrators can modify configuration files! We put an example configuration file " +
-                $"in the data directory caled {ap.CanonicalTargetName}.example.yml. Please copy this example file to " +
-                $"{ap.CanonicalTargetName}.yml and make changes according to your environment. Once {ap.CanonicalTargetName}.yml " +
-                $"is created, you can configure {ap.CanonicalTargetName} from your favorite shell (in an elevated prompt) " +
-                $"and then start {serviceDisplayName} Windows service.\r\n"));
-
-            project.AddProperty(new Property("WIXUI_EXITDIALOGOPTIONALCHECKBOX", "1"));
-            project.AddProperty(new Property("WIXUI_EXITDIALOGOPTIONALCHECKBOXTEXT",
-                $"Open {ap.CanonicalTargetName} data directory in Windows Explorer"));
-
-            // We'll open the folder for now
-            // TODO: select file in explorer window
-            project.AddProperty(new Property(
-                "WixShellExecTarget",
-                $"[$Component.{beatConfigExampleFileId}]"));
-
-            project.AddWixFragment("Wix/Product",
-                XElement.Parse(@"
-<CustomAction
-    Id=""CA_SelectExampleYamlInExplorer""
-    BinaryKey = ""WixCA""
-    DllEntry = ""WixShellExec""
-    Impersonate = ""yes""
-/>"),
-                XElement.Parse(@"
-<UI>
-    <Publish
-        Dialog=""ExitDialog""
-        Control=""Finish""
-        Event=""DoAction"" 
-        Value=""CA_SelectExampleYamlInExplorer"">WIXUI_EXITDIALOGOPTIONALCHECKBOX=1 and NOT Installed
-    </Publish>
-</UI>"));
+            // For agent, the MSI installer copies the contents of the MSI to a temp folder
+            // and then shall call the 'elastic-agent install' command.
+            // When installing, the 'elastic-agent uninstall' command.
+            if (pc.IsAgent)
+            {
+                // Passing the agent executable path to the action handler which will run it post installation
+                project.AddProperty(new Property("exe_folder", Path.Combine(ap.Version, ap.CanonicalTargetName)));
+                project.AddAction(new ManagedAction(AgentCustomAction.InstallAction, Return.check, When.After, Step.InstallFinalize, Condition.NOT_Installed));
 
-            var dataContents = new DirectoryInfo(opts.PackageInDir)
-                .GetFiles(MagicStrings.Files.AllDotYml, SearchOption.TopDirectoryOnly)
-                .Select(fi =>
-                {
-                    // rename main config file to hide it from MSI engine and keep customizations
-                    if (string.Compare(
-                        fi.Name,
-                        ap.CanonicalTargetName + MagicStrings.Ext.DotYml,
-                        StringComparison.OrdinalIgnoreCase) == 0)
-                    {
-                        var wf = new WixSharp.File(fi.FullName);
-                        wf.Attributes.Add("Name", beatConfigExampleFileName);
-                        wf.Id = new Id(beatConfigExampleFileId);
-                        return wf;
-                    }
-                    return null;
-                })
-                .ToList<WixEntity>();
+                // https://stackoverflow.com/questions/320921/how-to-add-a-wix-custom-action-that-happens-only-on-uninstall-via-msi
+                project.AddAction(new ManagedAction(AgentCustomAction.UnInstallAction, Return.check, When.After, Step.InstallFinalize, Condition.BeingUninstalledAndNotBeingUpgraded));
+            }
 
-            packageContents.AddRange(dataContents);
+            if (!pc.IsAgent)
+            {
+                // Add a note to the final screen and a checkbox to open the directory of .example.yml file
+                var beatConfigExampleFileName = ap.CanonicalTargetName.Replace("-", "_") + ".example" + MagicStrings.Ext.DotYml;
+                var beatConfigExampleFileId = beatConfigExampleFileName + "_" + (uint) beatConfigExampleFileName.GetHashCode32();
+
+                project.AddProperty(new Property("WIXUI_EXITDIALOGOPTIONALTEXT",
+                    $"NOTE: Only Administrators can modify configuration files! We put an example configuration file " +
+                    $"in the data directory named {beatConfigExampleFileName}. Please copy this example file to " +
+                    $"{ap.CanonicalTargetName}.yml and make changes according to your environment. Once {ap.CanonicalTargetName}.yml " +
+                    $"is created, you can configure {ap.CanonicalTargetName} from your favorite shell (in an elevated prompt) " +
+                    $"and then start {serviceDisplayName} Windows service.\r\n"));
+
+                HandleOpenExplorer(ap, project, beatConfigExampleFileId);
+                RenameConfigFile(opts, ap, packageContents, beatConfigExampleFileName, beatConfigExampleFileId);
+            }
 
             // Drop CLI shim on disk
             var cliShimScriptPath = Path.Combine(
@@ -280,5 +255,60 @@ static void Main(string[] args)
             else
                 Compiler.BuildMsi(project);
         }
+
+        private static void HandleOpenExplorer(ArtifactPackage ap, Project project, string beatConfigExampleFileId)
+        {
+            project.AddProperty(new Property("WIXUI_EXITDIALOGOPTIONALCHECKBOX", "1"));
+            project.AddProperty(new Property("WIXUI_EXITDIALOGOPTIONALCHECKBOXTEXT",
+                $"Open {ap.CanonicalTargetName} data directory in Windows Explorer"));
+
+            // We'll open the folder for now
+            // TODO: select file in explorer window
+            project.AddProperty(new Property(
+                "WixShellExecTarget",
+                $"[$Component.{beatConfigExampleFileId}]"));
+
+            project.AddWixFragment("Wix/Product",
+                XElement.Parse(@"
+<CustomAction
+    Id=""CA_SelectExampleYamlInExplorer""
+    BinaryKey = ""WixCA""
+    DllEntry = ""WixShellExec""
+    Impersonate = ""yes""
+/>"),
+                XElement.Parse(@"
+<UI>
+    <Publish
+        Dialog=""ExitDialog""
+        Control=""Finish""
+        Event=""DoAction"" 
+        Value=""CA_SelectExampleYamlInExplorer"">WIXUI_EXITDIALOGOPTIONALCHECKBOX=1 and NOT Installed
+    </Publish>
+</UI>"));
+        }
+
+        private static void RenameConfigFile(CmdLineOptions opts, ArtifactPackage ap, List<WixEntity> packageContents, string beatConfigExampleFileName, string beatConfigExampleFileId)
+        {
+            var dataContents = new DirectoryInfo(opts.PackageInDir)
+                .GetFiles(MagicStrings.Files.AllDotYml, SearchOption.TopDirectoryOnly)
+                .Select(fi =>
+                {
+                    // rename main config file to hide it from MSI engine and keep customizations
+                    if (string.Compare(
+                        fi.Name,
+                        ap.CanonicalTargetName + MagicStrings.Ext.DotYml,
+                        StringComparison.OrdinalIgnoreCase) == 0)
+                    {
+                        var wf = new WixSharp.File(fi.FullName);
+                        wf.Attributes.Add("Name", beatConfigExampleFileName);
+                        wf.Id = new Id(beatConfigExampleFileId);
+                        return wf;
+                    }
+                    return null;
+                })
+                .ToList<WixEntity>();
+
+            packageContents.AddRange(dataContents);
+        }
     }
 }

src/installer/BeatPackageCompiler/Properties/launchSettings.json

@@ -9,6 +9,11 @@
       "commandName": "Project",
       "commandLineArgs": "--package=filebeat-8.11.0-windows-x86_64 -v --keep-temp-files",
       "workingDirectory": "$(SolutionDir)"
+    },
+    "Agent": {
+      "commandName": "Project",
+      "commandLineArgs": "--package=elastic-agent-8.11.1-windows-x86_64 -v --keep-temp-files",
+      "workingDirectory": "$(SolutionDir)"
     }
   }
 }

src/shared/ArtifactPackage.cs

@@ -68,6 +68,12 @@ static bool FromFilenameOrUrl(string fileName, string url, out ArtifactPackage a
             Architecture = rxGroups["arch"].Value.ToLower();
             IsSnapshot = !rxGroups["snapshot"].Value.IsEmpty();
 
+            // HACK
+            if (TargetName == "agent")
+            {
+                TargetName = "elastic-agent";
+            }
+
             IsOss = TargetName.EndsWith(
                 MagicStrings.Files.DashOssSuffix,
                 StringComparison.OrdinalIgnoreCase);

src/shared/ProductConfig.cs

@@ -8,6 +8,12 @@ public sealed class ProductConfig
         [YamlMember("description")]
         public string Description { get; set; } = "(add 'description' field to config.yaml)";
 
+        [YamlMember("display_name")]
+        public string DisplayName { get; set; } = "";
+
+        [YamlMember("is_agent")]
+        public bool IsAgent { get; set; } = false;
+
         [YamlMember("published_name")]
         public string PublishedName { get; set; } = "Elastic Beats";