Added project files

Author: Jefferson Pires

ChatGPT.cs

@@ -0,0 +1,49 @@
+using OpenAI_API;
+using OpenAI_API.Completions;
+using OpenAI_API.Models;
+
+namespace JeffPires.VisualChatGPTStudio
+{
+    /// <summary>
+    /// Static class containing methods for interacting with the ChatGPT API.
+    /// </summary>
+    static class ChatGPT
+    {
+        /// <summary>
+        /// Requests a completion from the OpenAI API using the given options.
+        /// </summary>
+        /// <param name="options">The options to use for the request.</param>
+        /// <param name="request">The request to send to the API.</param>
+        /// <param name="resultHandler">The action to take when the result is received.</param>
+        /// <returns>A task representing the completion request.</returns>
+        public static async Task Request(OptionPageGrid options, string request, Action<int, CompletionResult> resultHandler)
+        {
+            OpenAIAPI api = new(options.ApiKey);
+
+            Model model = Model.DavinciText;
+
+            switch (options.Model)
+            {
+                case ModelLanguageEnum.TextCurie001:
+                    model = Model.CurieText;
+                    break;
+                case ModelLanguageEnum.TextBabbage001:
+                    model = Model.BabbageText;
+                    break;
+                case ModelLanguageEnum.TextAda001:
+                    model = Model.AdaText;
+                    break;
+                case ModelLanguageEnum.CodeDavinci:
+                    model = Model.DavinciCode;
+                    break;
+                case ModelLanguageEnum.CodeCushman:
+                    model = Model.CushmanCode;
+                    break;
+            }
+
+            CompletionRequest completionRequest = new(request, model, options.MaxTokens, options.Temperature, presencePenalty: options.PresencePenalty, frequencyPenalty: options.FrequencyPenalty, top_p: options.TopP);
+
+            await api.Completions.StreamCompletionAsync(completionRequest, resultHandler);
+        }
+    }
+}

Commands/Commands/AddComments.cs

@@ -0,0 +1,29 @@
+using EnvDTE;
+using JeffPires.VisualChatGPTStudio.Commands.Commands;
+
+namespace JeffPires.VisualChatGPTStudio
+{
+    [Command(PackageIds.AddComments)]
+    internal sealed class AddComments : BaseChatGPTCommand<AddComments>
+    {
+        protected override CommandType GetCommandType(string selectedText)
+        {
+            if (selectedText.Contains(Environment.NewLine))
+            {
+                return CommandType.Erase;
+            }
+
+            return CommandType.InsertBefore;
+        }
+
+        protected override string GetCommand(string selectedText)
+        {
+            if (selectedText.Contains(Environment.NewLine))
+            {
+                return $"Rewrite the code with comments{Environment.NewLine}{Environment.NewLine}{TextFormat.FormatSelection(selectedText)}";
+            }
+
+            return $"Comment{Environment.NewLine}{Environment.NewLine}{TextFormat.FormatSelection(selectedText)}";
+        }
+    }
+}

Commands/Commands/AddSummary.cs

@@ -0,0 +1,19 @@
+using EnvDTE;
+using JeffPires.VisualChatGPTStudio.Commands.Commands;
+
+namespace JeffPires.VisualChatGPTStudio
+{
+    [Command(PackageIds.AddSummary)]
+    internal sealed class AddSummary : BaseChatGPTCommand<AddSummary>
+    {
+        protected override CommandType GetCommandType(string selectedText)
+        {
+            return CommandType.InsertBefore;
+        }
+
+        protected override string GetCommand(string selectedText)
+        {
+            return TextFormat.FormatCommandForSummary("Only write a comment as C# summary format like\r\n\r\n{0}\r\n\r\nfor this", selectedText);
+        }
+    }
+}

Commands/Commands/AddTests.cs

@@ -0,0 +1,19 @@
+using EnvDTE;
+using JeffPires.VisualChatGPTStudio.Commands.Commands;
+
+namespace JeffPires.VisualChatGPTStudio
+{
+    [Command(PackageIds.AddTests)]
+    internal sealed class AddTests : BaseChatGPTCommand<AddTests>
+    {
+        protected override CommandType GetCommandType(string selectedText)
+        {
+            return CommandType.InsertAfter;
+        }
+
+        protected override string GetCommand(string selectedText)
+        {
+            return $"Create unit tests{Environment.NewLine}{Environment.NewLine}{TextFormat.FormatSelection(selectedText)}";
+        }
+    }
+}

Commands/Commands/AskAnything.cs

@@ -0,0 +1,19 @@
+using EnvDTE;
+using JeffPires.VisualChatGPTStudio.Commands.Commands;
+
+namespace JeffPires.VisualChatGPTStudio
+{
+    [Command(PackageIds.AskAnything)]
+    internal sealed class AskAnything : BaseChatGPTCommand<AskAnything>
+    {
+        protected override CommandType GetCommandType(string selectedText)
+        {
+            return CommandType.InsertAfter;
+        }
+
+        protected override string GetCommand(string selectedText)
+        {
+            return TextFormat.FormatSelection(selectedText);
+        }
+    }
+}

Commands/Commands/BaseChatGPTCommand.cs

@@ -0,0 +1,217 @@
+using EnvDTE;
+using OpenAI_API.Completions;
+using System.Linq;
+using Span = Microsoft.VisualStudio.Text.Span;
+
+namespace JeffPires.VisualChatGPTStudio.Commands.Commands
+{
+    /// <summary>
+    /// Base abstract class for commands
+    /// </summary>
+    /// <typeparam name="TCommand">The type of the command.</typeparam>
+    /// <seealso cref="BaseCommand&lt;&gt;" />
+    internal abstract class BaseChatGPTCommand<TCommand> : BaseCommand<TCommand> where TCommand : class, new()
+    {
+        const string EXTENSION_NAME = "Visual chatGPT Studio";
+
+        protected DocumentView docView;
+        private string selectedText;
+        private int position;
+        private int positionStart;
+        private int positionEnd;
+        private int lineLength;
+        private bool firstInteration;
+
+        /// <summary>
+        /// Gets the options.
+        /// </summary>
+        public OptionPageGrid Options
+        {
+            get
+            {
+                return ((VisuallChatGPTStudioPackage)this.Package).Options;
+            }
+        }
+
+        /// <summary>
+        /// Gets the type of command.
+        /// </summary>
+        /// <param name="selectedText">The selected text.</param>
+        /// <returns>The type of command.</returns>
+        protected abstract CommandType GetCommandType(string selectedText);
+
+        /// <summary>
+        /// Gets the command for the given selected text.
+        /// </summary>
+        /// <param name="selectedText">The selected text.</param>
+        /// <returns>The command.</returns>
+        protected abstract string GetCommand(string selectedText);
+
+        /// <summary>
+        /// Executes asynchronously when the command is invoked and <see cref="M:Community.VisualStudio.Toolkit.BaseCommand.Execute(System.Object,System.EventArgs)" /> isn't overridden.
+        /// </summary>
+        /// <param name="e"></param>
+        /// <remarks>
+        /// Use this method instead of <see cref="M:Community.VisualStudio.Toolkit.BaseCommand.Execute(System.Object,System.EventArgs)" /> if you're invoking any async tasks by using async/await patterns.
+        /// </remarks>
+        protected override async Task ExecuteAsync(OleMenuCmdEventArgs e)
+        {
+            try
+            {
+                if (string.IsNullOrWhiteSpace(Options.ApiKey))
+                {
+                    await VS.MessageBox.ShowAsync(EXTENSION_NAME, "Please, set the OpenAI API key.", buttons: Microsoft.VisualStudio.Shell.Interop.OLEMSGBUTTON.OLEMSGBUTTON_OK);
+
+                    Package.ShowOptionPage(typeof(OptionPageGrid));
+
+                    return;
+                }
+
+                firstInteration = true;
+                lineLength = 0;
+
+                await Package.JoinableTaskFactory.SwitchToMainThreadAsync();
+
+                docView = await VS.Documents.GetActiveDocumentViewAsync();
+
+                if (docView?.TextView == null) return;
+
+                position = docView.TextView.Caret.Position.BufferPosition.Position;
+                positionStart = docView.TextView.Selection.Start.Position.Position;
+                positionEnd = docView.TextView.Selection.End.Position.Position;
+
+                selectedText = docView.TextView.Selection.StreamSelectionSpan.GetText();
+
+                if (string.IsNullOrWhiteSpace(selectedText))
+                {
+                    await VS.MessageBox.ShowAsync(EXTENSION_NAME, "Please select the code.", buttons: Microsoft.VisualStudio.Shell.Interop.OLEMSGBUTTON.OLEMSGBUTTON_OK);
+
+                    return;
+                }
+
+                if (CheckIfSelectedTwoOrMoreMethods(selectedText))
+                {
+                    await VS.MessageBox.ShowAsync(EXTENSION_NAME, "Please select one method at a time.", buttons: Microsoft.VisualStudio.Shell.Interop.OLEMSGBUTTON.OLEMSGBUTTON_OK);
+
+                    return;
+                }
+
+                await VS.StatusBar.ShowProgressAsync("Waiting chatGPT response", 1, 2);
+
+                await Request(selectedText);
+            }
+            catch (Exception ex)
+            {
+                await VS.StatusBar.ShowProgressAsync(ex.Message, 2, 2);
+
+                await VS.MessageBox.ShowAsync(EXTENSION_NAME, ex.Message, Microsoft.VisualStudio.Shell.Interop.OLEMSGICON.OLEMSGICON_WARNING, Microsoft.VisualStudio.Shell.Interop.OLEMSGBUTTON.OLEMSGBUTTON_OK);
+            }
+        }
+
+        /// <summary>
+        /// Requests the specified selected text from ChatGPT
+        /// </summary>
+        /// <param name="selectedText">The selected text.</param>
+        private async Task Request(string selectedText)
+        {
+            await ChatGPT.Request(Options, GetCommand(selectedText), ResultHandler);
+
+            await ThreadHelper.JoinableTaskFactory.SwitchToMainThreadAsync();
+
+            try
+            {
+                //Some documents don't has format
+                (await VS.GetServiceAsync<DTE, DTE>()).ExecuteCommand("Edit.FormatDocument", string.Empty);
+            }
+            catch (Exception)
+            {
+
+            }
+        }
+
+        /// <summary>
+        /// Results handler.
+        /// </summary>
+        /// <param name="index">The index.</param>
+        /// <param name="result">The result.</param>
+        private async void ResultHandler(int index, CompletionResult result)
+        {
+            if (firstInteration)
+            {
+                await VS.StatusBar.ShowProgressAsync("Receiving chatGPT response", 2, 2);
+
+                CommandType commandType = GetCommandType(selectedText);
+
+                if (commandType == CommandType.Erase)
+                {
+                    position = positionStart;
+
+                    //Erase current code
+                    _ = (docView.TextBuffer?.Replace(new Span(position, docView.TextView.Selection.StreamSelectionSpan.GetText().Length), String.Empty));
+                }
+                else if (commandType == CommandType.InsertBefore)
+                {
+                    position = positionStart;
+
+                    _ = (docView.TextBuffer?.Insert(position, Environment.NewLine));
+                }
+                else
+                {
+                    position = positionEnd;
+
+                    _ = (docView.TextBuffer?.Insert(position, Environment.NewLine));
+                }
+
+                firstInteration = false;
+            }
+
+            string resultText = result.ToString();
+
+            docView.TextBuffer?.Insert(position, resultText);
+
+            position += resultText.Length;
+
+            lineLength += resultText.Length;
+
+            if (lineLength > 160 && typeof(TCommand) == typeof(AskAnything))
+            {
+                lineLength = 0;
+                MovetoNextLine();
+            }
+        }
+
+        /// <summary>
+        /// Move to the next line.
+        /// </summary>
+        private void MovetoNextLine()
+        {
+            position = docView.TextView.Caret.Position.BufferPosition.GetContainingLine().End.Position;
+
+            docView.TextView.Caret.MoveTo(docView.TextView.Caret.Position.BufferPosition.GetContainingLine().End);
+
+            _ = (docView.TextBuffer?.Insert(position, Environment.NewLine));
+        }
+
+        /// <summary>
+        /// Check If Selected Two Or More Methods
+        /// </summary>
+        /// <param name="text">The text to check</param>
+        /// <returns>True if has Selected Two Or More Methods</returns>
+        private bool CheckIfSelectedTwoOrMoreMethods(string text)
+        {
+            string[] words = text.Split(' ');
+
+            return words.Count(w => w == "public" || w == "private" || w == "protected") >= 2;
+        }
+    }
+
+    /// <summary>
+    /// Enum to represent the different types of commands that can be used. 
+    /// </summary>
+    enum CommandType
+    {
+        Erase,
+        InsertBefore,
+        InsertAfter
+    }
+}

Commands/Commands/Complete.cs

@@ -0,0 +1,18 @@
+using EnvDTE;
+
+namespace JeffPires.VisualChatGPTStudio.Commands.Commands
+{
+    [Command(PackageIds.Complete)]
+    internal sealed class Complete : BaseChatGPTCommand<Complete>
+    {
+        protected override CommandType GetCommandType(string selectedText)
+        {
+            return CommandType.InsertAfter;
+        }
+
+        protected override string GetCommand(string selectedText)
+        {
+            return TextFormat.FormatForCompleteCommand("Please complete", selectedText, docView.FilePath);
+        }
+    }
+}

Commands/Commands/Explain.cs

@@ -0,0 +1,19 @@
+using EnvDTE;
+using JeffPires.VisualChatGPTStudio.Commands.Commands;
+
+namespace JeffPires.VisualChatGPTStudio
+{
+    [Command(PackageIds.Explain)]
+    internal sealed class Explain : BaseChatGPTCommand<Explain>
+    {
+        protected override CommandType GetCommandType(string selectedText)
+        {
+            return CommandType.InsertBefore;
+        }
+
+        protected override string GetCommand(string selectedText)
+        {
+            return $"Explain{Environment.NewLine}{Environment.NewLine}{TextFormat.FormatSelection(selectedText)}";
+        }
+    }
+}