From daf92c088c58ef0e331d887c31c02c7d9e32e430 Mon Sep 17 00:00:00 2001
From: markoz <mzorec@gmail.com>
Date: Tue, 9 Oct 2018 07:55:55 +0200
Subject: [PATCH] Updates azure task generator

---
 FlubuCore.Azure/AzureTaskGenerator.cs | 10 +++++++---
 FlubuCore.Azure/StringExtensions.cs   | 13 +++++++++++++
 2 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/FlubuCore.Azure/AzureTaskGenerator.cs b/FlubuCore.Azure/AzureTaskGenerator.cs
index 510dfda..4e08e1c 100644
--- a/FlubuCore.Azure/AzureTaskGenerator.cs
+++ b/FlubuCore.Azure/AzureTaskGenerator.cs
@@ -19,7 +19,9 @@ protected override string WriteParameter(Parameter parameter)
                 return string.Empty;
             }
 
-            return $"{parameter.ParameterType} {parameter.ParameterName} = null";
+            string parameterName = ParameterName(parameter.ParameterName);
+
+            return $"{parameter.ParameterType} {parameterName} = null";
         }
 
         protected override string WriteArgument(Argument argument)
@@ -34,10 +36,12 @@ protected override string WriteArgument(Argument argument)
                 return $@"WithArguments(""{argument.ArgumentKey}"");";
             }
 
+            var parameterName = ParameterName(argument.Parameter.ParameterName);
+
             return $@"WithArguments(""{argument.ArgumentKey}"");
-            if (!string.IsNullOrEmpty({argument.Parameter.ParameterName}))
+            if (!string.IsNullOrEmpty({parameterName}))
             {{
-                 WithArguments({argument.Parameter.ParameterName});
+                 WithArguments({parameterName});
             }}
 ";
         }
diff --git a/FlubuCore.Azure/StringExtensions.cs b/FlubuCore.Azure/StringExtensions.cs
index 8535303..c9c6c34 100644
--- a/FlubuCore.Azure/StringExtensions.cs
+++ b/FlubuCore.Azure/StringExtensions.cs
@@ -16,5 +16,18 @@ public static string FirstCharToUpper(this string value)
                 default: return value.First().ToString().ToUpper() + value.Substring(1);
             }
         }
+
+        public static IEnumerable<TSource> DistinctBy<TSource, TKey>
+            (this IEnumerable<TSource> source, Func<TSource, TKey> keySelector)
+        {
+            HashSet<TKey> seenKeys = new HashSet<TKey>();
+            foreach (TSource element in source)
+            {
+                if (seenKeys.Add(keySelector(element)))
+                {
+                    yield return element;
+                }
+            }
+        }
     }
 }