From 89a80bd73e26365c8a0cbf223d95188112804d56 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tomasz=20Go=C5=82=C4=99biowski?=
 <tgolebiowski@virtuslab.com>
Date: Fri, 27 Sep 2024 11:12:29 +0200
Subject: [PATCH 1/2] Accept non-trusted certificates option

---
 .../Settings/IUserSettingsService.cs          |  3 ++
 src/Cody.Core/Settings/UserSettingsService.cs | 10 ++++++
 .../Options/GeneralOptionsControl.xaml        | 14 +++++++-
 .../Options/GeneralOptionsControl.xaml.cs     |  1 +
 .../ViewModels/GeneralOptionsViewModel.cs     | 13 ++++++++
 .../Client/AgentClientOptions.cs              |  3 ++
 .../Client/AgentProcessConnector.cs           |  4 +++
 src/Cody.VisualStudio/CodyPackage.cs          | 33 +++++++++++++++++++
 .../Options/GeneralOptionsPage.cs             |  5 +++
 9 files changed, 85 insertions(+), 1 deletion(-)

diff --git a/src/Cody.Core/Settings/IUserSettingsService.cs b/src/Cody.Core/Settings/IUserSettingsService.cs
index b24d0cf..56ad2ef 100644
--- a/src/Cody.Core/Settings/IUserSettingsService.cs
+++ b/src/Cody.Core/Settings/IUserSettingsService.cs
@@ -9,6 +9,9 @@ public interface IUserSettingsService
         string AccessToken { get; set; }
         string ServerEndpoint { get; set; }
         string CodySettings { get; set; }
+        bool AcceptNonTrustedCert { get; set; }
+
+
         event EventHandler AuthorizationDetailsChanged;
     }
 }
diff --git a/src/Cody.Core/Settings/UserSettingsService.cs b/src/Cody.Core/Settings/UserSettingsService.cs
index 49c109a..a87c5cc 100644
--- a/src/Cody.Core/Settings/UserSettingsService.cs
+++ b/src/Cody.Core/Settings/UserSettingsService.cs
@@ -101,5 +101,15 @@ public string CodySettings
             get => GetOrDefault(nameof(CodySettings), string.Empty);
             set => Set(nameof(CodySettings), value);
         }
+
+        public bool AcceptNonTrustedCert
+        {
+            get
+            {
+                var value = GetOrDefault(nameof(AcceptNonTrustedCert), false.ToString());
+                return bool.Parse(value);
+            }
+            set => Set(nameof(AcceptNonTrustedCert), value.ToString());
+        }
     }
 }
diff --git a/src/Cody.UI/Controls/Options/GeneralOptionsControl.xaml b/src/Cody.UI/Controls/Options/GeneralOptionsControl.xaml
index c4b50a3..ff160a4 100644
--- a/src/Cody.UI/Controls/Options/GeneralOptionsControl.xaml
+++ b/src/Cody.UI/Controls/Options/GeneralOptionsControl.xaml
@@ -16,6 +16,7 @@
                     <RowDefinition Height="Auto" />
                     <RowDefinition Height="Auto" />
                     <RowDefinition Height="Auto" />
+                    <RowDefinition Height="Auto" />
                 </Grid.RowDefinitions>
 
                 <Grid.ColumnDefinitions>
@@ -42,10 +43,19 @@
                     Name="ConfigurationsTextBox"
                     />
 
+                <CheckBox
+                    Name="AcceptNonTrustedCertCheckBox"
+                    Grid.Row="2"
+                    Grid.Column="1"
+                    Margin="0 5 0 0"
+                    IsChecked="{Binding AcceptNonTrustedCert, Mode=TwoWay}"
+                    Content="Accept non-trusted certificates (requires restart)"
+                 />
+
                 <!--Get Help-->
                 <Button 
                     Margin="0 5 0 0"
-                    Grid.Row="2"
+                    Grid.Row="3"
                     Grid.Column="1"
                     Height="25"
                     Width="100"
@@ -54,6 +64,8 @@
                     Command="{Binding ActivateBetaCommand}"
                 />
 
+                
+
                 <!--Sourcegraph URL-->
                 <Label
                     Grid.Row="0"
diff --git a/src/Cody.UI/Controls/Options/GeneralOptionsControl.xaml.cs b/src/Cody.UI/Controls/Options/GeneralOptionsControl.xaml.cs
index fe9d200..c4683b0 100644
--- a/src/Cody.UI/Controls/Options/GeneralOptionsControl.xaml.cs
+++ b/src/Cody.UI/Controls/Options/GeneralOptionsControl.xaml.cs
@@ -18,6 +18,7 @@ public void ForceBindingsUpdate()
             // This is a workaround to get bindings updated. The second solution is to use NotifyPropertyChange for every TextBox in the Xaml, but current solution is a little more "clean" - everything is clearly visible in a one place.
             SourcegraphUrlTextBox.GetBindingExpression(TextBox.TextProperty)?.UpdateSource();
             ConfigurationsTextBox.GetBindingExpression(TextBox.TextProperty)?.UpdateSource();
+            AcceptNonTrustedCertCheckBox.GetBindingExpression(CheckBox.IsCheckedProperty)?.UpdateSource();
         }
     }
 }
diff --git a/src/Cody.UI/ViewModels/GeneralOptionsViewModel.cs b/src/Cody.UI/ViewModels/GeneralOptionsViewModel.cs
index 3d8be3a..18ef16d 100644
--- a/src/Cody.UI/ViewModels/GeneralOptionsViewModel.cs
+++ b/src/Cody.UI/ViewModels/GeneralOptionsViewModel.cs
@@ -65,5 +65,18 @@ public string SourcegraphUrl
             }
         }
 
+        private bool _acceptNonTrustedCert;
+        public bool AcceptNonTrustedCert
+        {
+            get => _acceptNonTrustedCert;
+            set
+            {
+                if (SetProperty(ref _acceptNonTrustedCert, value))
+                {
+                    _logger.Debug($"AcceptNonTrustedCert set:{SourcegraphUrl}");
+                }
+            }
+        }
+
     }
 }
diff --git a/src/Cody.VisualStudio/Client/AgentClientOptions.cs b/src/Cody.VisualStudio/Client/AgentClientOptions.cs
index 3d83686..dfd6f3f 100644
--- a/src/Cody.VisualStudio/Client/AgentClientOptions.cs
+++ b/src/Cody.VisualStudio/Client/AgentClientOptions.cs
@@ -1,4 +1,5 @@
 using Cody.Core.Agent;
+using Microsoft.VisualStudio.Shell.FindAllReferences;
 using System;
 using System.Collections.Generic;
 using System.Linq;
@@ -20,6 +21,8 @@ public class AgentClientOptions
         /// </summary>
         public int RemoteAgentPort { get; set; } = 3113;
 
+        public bool AcceptNonTrustedCertificates { get; set; } = false;
+
         public string AgentDirectory { get; set; }
 
         public List<object> CallbackHandlers { get; set; } = new List<object>();
diff --git a/src/Cody.VisualStudio/Client/AgentProcessConnector.cs b/src/Cody.VisualStudio/Client/AgentProcessConnector.cs
index c0bd462..fba2834 100644
--- a/src/Cody.VisualStudio/Client/AgentProcessConnector.cs
+++ b/src/Cody.VisualStudio/Client/AgentProcessConnector.cs
@@ -33,6 +33,10 @@ public void Connect(AgentClientOptions options)
             process.StartInfo.RedirectStandardInput = true;
             process.StartInfo.CreateNoWindow = true;
             process.EnableRaisingEvents = true;
+
+            if(options.AcceptNonTrustedCertificates)
+                process.StartInfo.EnvironmentVariables["NODE_TLS_REJECT_UNAUTHORIZED"] = "0";
+
             process.Exited += OnProcessExited;
             process.ErrorDataReceived += OnErrorDataReceived;
 
diff --git a/src/Cody.VisualStudio/CodyPackage.cs b/src/Cody.VisualStudio/CodyPackage.cs
index 366ea9e..40dff07 100644
--- a/src/Cody.VisualStudio/CodyPackage.cs
+++ b/src/Cody.VisualStudio/CodyPackage.cs
@@ -37,6 +37,7 @@
 using Task = System.Threading.Tasks.Task;
 using Microsoft.VisualStudio.TaskStatusCenter;
 using SolutionEvents = Microsoft.VisualStudio.Shell.Events.SolutionEvents;
+using System.Net;
 
 namespace Cody.VisualStudio
 {
@@ -217,6 +218,7 @@ private async Task InitOleMenu()
                 }
                 else
                 {
+                    
                     throw new NotSupportedException($"Cannot get {nameof(OleMenuCommandService)}");
                 }
             }
@@ -286,6 +288,7 @@ private void PrepareAgentConfiguration()
                     RestartAgentOnFailure = true,
                     ConnectToRemoteAgent = devPort != null,
                     RemoteAgentPort = portNumber,
+                    AcceptNonTrustedCertificates = UserSettingsService.AcceptNonTrustedCert,
                     Debug = true
                 };
 
@@ -376,6 +379,7 @@ private void InitializeAgent()
                 {
                     try
                     {
+                        await TestServerConnection(UserSettingsService.ServerEndpoint);
                         AgentClient.Start();
 
                         var clientConfig = GetClientInfo();
@@ -449,6 +453,35 @@ private void OnAfterCloseSolution(object sender, EventArgs e)
             }
         }
 
+        private async Task TestServerConnection(string serverAddress)
+        {
+            if (string.IsNullOrWhiteSpace(serverAddress)) return;
+
+            using (WebClient webClient = new WebClient())
+            {
+                try
+                {
+                    await webClient.DownloadStringTaskAsync(serverAddress);
+                }
+                catch (WebException ex) when (ex.Status == WebExceptionStatus.TrustFailure)
+                {
+                    if (!UserSettingsService.AcceptNonTrustedCert)
+                    {
+                        AgentLogger.Warn("SSL certificate problem: self-signed certificate in servers certificate chain. " +
+                            "Consider using 'Accept non-trusted certificates' option in Cody settings window.");
+                    }
+                    else
+                    {
+                        AgentLogger.Warn("SSL certificate problem.");
+                    }
+                }
+                catch
+                {
+                    AgentLogger.Warn($"Connection problem with '{serverAddress}'");
+                }
+            }
+        }
+
 
         private void CurrentDomainOnUnhandledException(object sender, UnhandledExceptionEventArgs e)
         {
diff --git a/src/Cody.VisualStudio/Options/GeneralOptionsPage.cs b/src/Cody.VisualStudio/Options/GeneralOptionsPage.cs
index 07456d5..652a4e7 100644
--- a/src/Cody.VisualStudio/Options/GeneralOptionsPage.cs
+++ b/src/Cody.VisualStudio/Options/GeneralOptionsPage.cs
@@ -60,9 +60,11 @@ protected override void OnActivate(CancelEventArgs e)
 
             var codyConfig = _settingsService.CodySettings;
             var sourcegraphUrl = _settingsService.ServerEndpoint;
+            var acceptNonTrustedCert = _settingsService.AcceptNonTrustedCert;
 
             _generalOptionsViewModel.CodyConfigurations = codyConfig;
             _generalOptionsViewModel.SourcegraphUrl = sourcegraphUrl;
+            _generalOptionsViewModel.AcceptNonTrustedCert = acceptNonTrustedCert;
 
             _logger.Debug($"Is canceled:{e.Cancel}");
 
@@ -74,9 +76,11 @@ protected override void OnApply(PageApplyEventArgs args)
 
             var codyConfig = _generalOptionsViewModel.CodyConfigurations;
             var sourcegraphUrl = _generalOptionsViewModel.SourcegraphUrl;
+            var acceptNonTrustedCert = _generalOptionsViewModel.AcceptNonTrustedCert;
 
             _settingsService.CodySettings = codyConfig;
             _settingsService.ServerEndpoint = sourcegraphUrl;
+            _settingsService.AcceptNonTrustedCert = acceptNonTrustedCert;
         }
 
         protected override void OnDeactivate(CancelEventArgs e)
@@ -97,6 +101,7 @@ public override void ResetSettings()
         {
             _settingsService.CodySettings = string.Empty;
             _settingsService.ServerEndpoint = string.Empty;
+            _settingsService.AcceptNonTrustedCert = false;
 
             base.ResetSettings();
         }

From 74b1914b752ba13666f289341640fee25fda1bc6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tomasz=20Go=C5=82=C4=99biowski?=
 <tgolebiowski@virtuslab.com>
Date: Fri, 27 Sep 2024 11:25:49 +0200
Subject: [PATCH 2/2] Formating

---
 src/Cody.VisualStudio/Client/AgentProcessConnector.cs | 2 +-
 src/Cody.VisualStudio/CodyPackage.cs                  | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/Cody.VisualStudio/Client/AgentProcessConnector.cs b/src/Cody.VisualStudio/Client/AgentProcessConnector.cs
index fba2834..5b48080 100644
--- a/src/Cody.VisualStudio/Client/AgentProcessConnector.cs
+++ b/src/Cody.VisualStudio/Client/AgentProcessConnector.cs
@@ -34,7 +34,7 @@ public void Connect(AgentClientOptions options)
             process.StartInfo.CreateNoWindow = true;
             process.EnableRaisingEvents = true;
 
-            if(options.AcceptNonTrustedCertificates)
+            if (options.AcceptNonTrustedCertificates)
                 process.StartInfo.EnvironmentVariables["NODE_TLS_REJECT_UNAUTHORIZED"] = "0";
 
             process.Exited += OnProcessExited;
diff --git a/src/Cody.VisualStudio/CodyPackage.cs b/src/Cody.VisualStudio/CodyPackage.cs
index 40dff07..1459bd0 100644
--- a/src/Cody.VisualStudio/CodyPackage.cs
+++ b/src/Cody.VisualStudio/CodyPackage.cs
@@ -218,7 +218,6 @@ private async Task InitOleMenu()
                 }
                 else
                 {
-                    
                     throw new NotSupportedException($"Cannot get {nameof(OleMenuCommandService)}");
                 }
             }