From 93544adc003119e6e5e68d0fd3360bc35d371ec4 Mon Sep 17 00:00:00 2001
From: Luc Hayward <luchay10@gmail.com>
Date: Tue, 23 Nov 2021 12:50:24 +0200
Subject: [PATCH] Add option in code for colon suffixs

---
 .../hostar/pythonInlay/PythonParametersInlayProvider.java | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/main/java/io/github/hostar/pythonInlay/PythonParametersInlayProvider.java b/src/main/java/io/github/hostar/pythonInlay/PythonParametersInlayProvider.java
index 6dfc63d..460d9e3 100644
--- a/src/main/java/io/github/hostar/pythonInlay/PythonParametersInlayProvider.java
+++ b/src/main/java/io/github/hostar/pythonInlay/PythonParametersInlayProvider.java
@@ -29,6 +29,8 @@ public class PythonParametersInlayProvider implements InlayHintsProvider<NoSetti
 
     private static final SettingsKey<NoSettings> KEY = new SettingsKey<>(CODE_LENS_ID);
 
+    private static Boolean addColonSuffix = False;
+
     @Nullable
     @Override
     public InlayHintsCollector getCollectorFor(@NotNull PsiFile file,
@@ -189,6 +191,9 @@ private void processElement(@NotNull PsiElement element, @NotNull InlayHintsSink
 
                                                             if (!match) {
                                                                 addSink(sink, paramName, args[finalPosition]);
+                                                                if addColonSuffix{
+                                                                    paramName += ":";
+                                                                }
                                                                 existingInlaysDictionary.add(new ExistingInlay(paramName, args[finalPosition].getTextOffset()));
                                                             }
                                                     }
@@ -258,6 +263,9 @@ private void processElement(@NotNull PsiElement element, @NotNull InlayHintsSink
                                                             if (!match) {
                                                                 addSink(sink, paramName, args[position]);
                                                                 position++;
+                                                                if addColonSuffix{
+                                                                    paramName += ":";
+                                                                }
                                                                 existingInlaysDictionary.add(new ExistingInlay(paramName, finalArgs1[finalPosition1].getTextOffset()));
                                                             }
                                                         }