Change how we evaluate if prefix or apm

Author: mwatson

Src/StackifyLib/Logger.cs

@@ -304,45 +304,16 @@ public static List<TraceFrame> GetCurrentStackTrace(string declaringClassName, i
             return frames;
         }
 
-        private static bool? _PrefixEnabled = null;
 
         public static bool PrefixEnabled()
         {
-            if (_PrefixEnabled != null)
-                return _PrefixEnabled.Value;
-
-            var variable = Environment.GetEnvironmentVariable("StackSquatchUpdated", EnvironmentVariableTarget.Machine);
-
-
-            if (!string.IsNullOrEmpty(variable))
-            {
-
-                DateTime updated;
-
-                if (DateTime.TryParse(variable, out updated))
-                {
-                    if (updated > DateTime.UtcNow.AddHours(-24))
-                    {
-                        StackifyLib.Utils.StackifyAPILogger.Log("Prefix enabled", true);
-                        _PrefixEnabled = true;
-                    }
-                    else
-                    {
-                        _PrefixEnabled = false;
-                    }
-                }
-                else
-                {
-                    _PrefixEnabled = false;
-                }
-                
-            }
-            else
-            {
-                _PrefixEnabled = false;
-            }
-
-            return _PrefixEnabled.Value;
+            return StackifyLib.Utils.PrefixOrAPM.GetProfilerType() == PrefixOrAPM.ProfilerType.Prefix;
         }
+
+        /// <summary>
+        /// Scans to see if Prefix is running
+        /// </summary>
+        /// <returns></returns>
+      
     }
 }

Src/StackifyLib/StackifyLib.csproj

@@ -77,6 +77,7 @@
     <Compile Include="Utils\ErrorGovernor.cs" />
     <Compile Include="Utils\HelperFunctions.cs" />
     <Compile Include="Utils\HttpClient.cs" />
+    <Compile Include="Utils\PrefixOrAPM.cs" />
     <Compile Include="Utils\RegistryHelper.cs" />
     <Compile Include="Utils\SequentialGuid.cs" />
     <Compile Include="Utils\StackifyAPILogger.cs" />

Src/StackifyLib/Utils/PrefixOrAPM.cs

@@ -0,0 +1,92 @@
+using System;
+using System.Collections.Generic;
+using System.Data.SqlTypes;
+using System.Diagnostics;
+using System.Linq;
+using System.Text;
+
+namespace StackifyLib.Utils
+{
+    internal class PrefixOrAPM
+    {
+        internal enum ProfilerType
+        {
+            None,
+            Prefix,
+            APM
+        }
+
+        static DateTime _LastCheck = DateTime.MinValue;
+        static ProfilerType _LastProfilerType = ProfilerType.None;
+        private static bool _ScanProcessException = false;
+        internal static ProfilerType GetProfilerType()
+        {
+            if (_LastCheck > DateTime.UtcNow.AddMinutes(-1))
+            {
+                return _LastProfilerType;
+            }
+            _LastCheck = DateTime.UtcNow;
+            bool foundProcess = false;
+
+            if (!_ScanProcessException)
+            {
+                try
+                {
+                    foreach (var process in Process.GetProcesses().ToList())
+                    {
+                        if (process.Id <= 0)
+                            continue;
+
+                        try
+                        {
+                            switch (process?.ProcessName?.ToLower().Replace(".vshost", ""))
+                            {
+                                case "devdashservice":
+                                case "stackifytracernotifier":
+                                case "devdashtestconsole":
+                                    _LastProfilerType = ProfilerType.Prefix;
+                                    foundProcess = true;
+                                    break;
+                                case "stackifymonitoringservice":
+                                case "monitortestconsole":
+                                    _LastProfilerType = ProfilerType.APM;
+                                    foundProcess = true;
+                                    break;
+                            }
+
+                            if (foundProcess) break;
+                        }
+                        catch
+                        {
+                            _ScanProcessException = true;
+                        }
+
+                    }
+                }
+                catch
+                {
+                    _ScanProcessException = true;
+                }
+            }
+            
+
+            //fall back to see if this has been set
+            if (!foundProcess && _ScanProcessException)
+            {
+                var stackifyPath = Environment.GetEnvironmentVariable("StackifyPath", EnvironmentVariableTarget.Machine);
+
+                if (!string.IsNullOrEmpty(stackifyPath) && (stackifyPath.IndexOf("prefix", StringComparison.CurrentCultureIgnoreCase) > -1 || stackifyPath.IndexOf("devdash", StringComparison.CurrentCultureIgnoreCase) > -1))
+                {
+                    _LastProfilerType = ProfilerType.Prefix;
+                }
+                else 
+                {
+                    _LastProfilerType = ProfilerType.APM;
+                }
+
+            }
+
+            return _LastProfilerType;
+        }
+    }
+}