Integrating OpenAI Code Analysis for identifying harmful code

Engine.Tests/Analyze/Conditions/AIisBestPracticeTest.cs

@@ -0,0 +1,54 @@
+using Engine;
+using Engine.Configuration;
+using System;
+using System.Collections.Generic;
+using System.Text;
+using OpenAI_API;
+using System.Xml.Serialization;
+using System.IO;
+using System.Threading.Tasks;
+using Xunit;
+
+namespace PowerShellProtect.Analyze.Conditions.Tests
+{
+    public class AIisBestPracticeConditionTests
+    {
+        [Fact]
+        public void Analyze_ReturnsTrue_WhenOpenAIResponseRatingIsGreaterThanOrEqualToConditionAIRating()
+        {
+            // Arrange
+            var context = new ScriptContext();
+            var condition = new Condition
+            {
+                APIKey = "your-api-key",
+                AIRating = 0.5m
+            };
+            var aIisBestPracticeCondition = new AIisBestPracticeCondition();
+
+            // Act
+            var result = aIisBestPracticeCondition.Analyze(context, condition);
+
+            // Assert
+            Assert.True(result);
+        }
+
+        [Fact]
+        public void Analyze_ReturnsFalse_WhenOpenAIResponseIsNull()
+        {
+            // Arrange
+            var context = new ScriptContext();
+            var condition = new Condition
+            {
+                APIKey = "your-api-key",
+                AIRating = 0.5m
+            };
+            var aIisBestPracticeCondition = new AIisBestPracticeCondition();
+
+            // Act
+            var result = aIisBestPracticeCondition.Analyze(context, condition);
+
+            // Assert
+            Assert.False(result);
+        }
+    }
+}

Engine.Tests/Analyze/Conditions/AIisSuspiciousTest.cs

@@ -0,0 +1,54 @@
+using Engine;
+using Engine.Configuration;
+using System;
+using System.Collections.Generic;
+using System.Text;
+using OpenAI_API;
+using System.Xml.Serialization;
+using System.IO;
+using System.Threading.Tasks;
+using Xunit;
+
+namespace PowerShellProtect.Analyze.Conditions.Tests
+{
+    public class AIisSuspiciousConditionTests
+    {
+        [Fact]
+        public void Analyze_ReturnsTrue_WhenOpenAIResponseRatingIsGreaterThanOrEqualToConditionAIRating()
+        {
+            // Arrange
+            var context = new ScriptContext();
+            var condition = new Condition
+            {
+                APIKey = "your-api-key",
+                AIRating = 0.5m
+            };
+            var aiIsSuspiciousCondition = new AIisSuspiciousCondition();
+
+            // Act
+            var result = aiIsSuspiciousCondition.Analyze(context, condition);
+
+            // Assert
+            Assert.True(result);
+        }
+
+        [Fact]
+        public void Analyze_ReturnsFalse_WhenOpenAIResponseIsNull()
+        {
+            // Arrange
+            var context = new ScriptContext();
+            var condition = new Condition
+            {
+                APIKey = "your-api-key",
+                AIRating = 0.5m
+            };
+            var aiIsSuspiciousCondition = new AIisSuspiciousCondition();
+
+            // Act
+            var result = aiIsSuspiciousCondition.Analyze(context, condition);
+
+            // Assert
+            Assert.False(result);
+        }
+    }
+}

Engine.Tests/Engine.Tests.csproj

@@ -10,6 +10,7 @@
     <PackageReference Include="Microsoft.NET.Test.Sdk" Version="16.5.0" />
     <PackageReference Include="NSubstitute" Version="4.2.2" />
     <PackageReference Include="xunit" Version="2.4.0" />
+    <PackageReference Include="Moq" Version="4.18.4" />
     <PackageReference Include="xunit.runner.visualstudio" Version="2.4.0" />
     <PackageReference Include="coverlet.collector" Version="1.2.0" />
     <PackageReference Include="PowerShellStandard.Library" Version="5.1.0" />

PowerShellProtect.psd1

@@ -12,7 +12,7 @@
     RootModule        = 'PowerShellProtect.psm1'
 
     # Version number of this module.
-    ModuleVersion     = '2022.11.0'
+    ModuleVersion     = '2022.11.1'
 
     # Supported PSEditions
     # CompatiblePSEditions = @()
@@ -72,7 +72,7 @@
     FunctionsToExport = @('Install-PowerShellProtect', 'Uninstall-PowerShellProtect')
 
     # Cmdlets to export from this module, for best performance, do not use wildcards and do not delete the entry, use an empty array if there are no cmdlets to export.
-    CmdletsToExport   = @('Get-PSPConfiguration', 'Set-PSPConfiguration', 'Test-PSPConfiguration', 'New-PSPRule', 'New-PSPCondition', 'New-PSPAction', 'New-PSPConfiguration', 'Save-PSPConfiguration')
+    CmdletsToExport   = @('Get-PSPConfiguration', 'Set-PSPConfiguration', 'Test-PSPConfiguration', 'New-PSPRule', 'New-PSPCondition', 'New-PSPAction', 'New-PSPConfiguration', 'Save-PSPConfiguration','New-PSPAICondition')
 
     # Variables to export from this module
     VariablesToExport = @()

PowerShellProtect/Analyze/AIResultCondition.cs

@@ -0,0 +1,13 @@
+namespace PowerShellProtect.Analyze
+{
+    public enum AIResultCondition
+    {
+        LessThan,
+        EqualTo,
+        GreaterThan,
+        GreaterThanOrEquals,
+        LessThanOrEquals,
+        NotEqualTo
+
+    }
+}

PowerShellProtect/Analyze/Analyzer.cs

@@ -45,6 +45,8 @@ public Analyzer()
                 new VariableCondition(),
                 new MemberCondition(),
                 new ScriptStringCondition(),
+                new AIisBestPracticeCondition(),
+                new AIisSuspiciousCondition()
             }.ToDictionary(m => m.Name.ToLower(), m => m);
 
             _builtInConditions = new List<ICondition>