GetFilesCommand support extension

jsonzsh · Jul 20, 2021 · 198f443 · 198f443
1 parent cf76dce
commit 198f443
Showing 1 changed file with 21 additions and 1 deletion.
diff --git a/taskt/Core/Automation/Commands/GetFilesCommand.cs b/taskt/Core/Automation/Commands/GetFilesCommand.cs
@@ -25,6 +25,14 @@ public class GetFilesCommand : ScriptCommand
         [Attributes.PropertyAttributes.Remarks("")]
         public string v_SourceFolderPath { get; set; }
 
+        [XmlAttribute]
+        [Attributes.PropertyAttributes.PropertyDescription("Optional - Please indicate the extension (Default is empty and searched all files) (ex. txt, {{{vExtension}}})")]
+        [Attributes.PropertyAttributes.PropertyUIHelper(Attributes.PropertyAttributes.PropertyUIHelper.UIAdditionalHelperType.ShowVariableHelper)]
+        [Attributes.PropertyAttributes.InputSpecification("Enter or Select the extension.")]
+        [Attributes.PropertyAttributes.SampleUsage("**txt** or **{{{vExtension}}}**")]
+        [Attributes.PropertyAttributes.Remarks("")]
+        public string v_SearchExtension { get; set; }
+
         [XmlAttribute]
         [Attributes.PropertyAttributes.PropertyDescription("Assign to Variable")]
         [Attributes.PropertyAttributes.InputSpecification("Select or provide a variable from the variable list")]
@@ -46,9 +54,19 @@ public override void RunCommand(object sender)
             //apply variable logic
             var sourceFolder = v_SourceFolderPath.ConvertToUserVariable(sender);
 
+            var ext = "." + v_SearchExtension.ConvertToUserVariable(sender).ToLower();
+
             //delete folder
             //System.IO.Directory.Delete(sourceFolder, true);
-            var filesList = System.IO.Directory.GetFiles(sourceFolder).ToList();
+            List<string> filesList;
+            if (String.IsNullOrEmpty(ext))
+            {
+                filesList = System.IO.Directory.GetFiles(sourceFolder).ToList();
+            }
+            else
+            {
+                filesList = System.IO.Directory.GetFiles(sourceFolder, "*.*").Where(t => System.IO.Path.GetExtension(t).ToLower() == ext).ToList();
+            }
 
             Script.ScriptVariable newFilesList = new Script.ScriptVariable
             {
@@ -70,6 +88,8 @@ public override List<Control> Render(frmCommandEditor editor)
 
             RenderedControls.AddRange(CommandControls.CreateDefaultInputGroupFor("v_SourceFolderPath", this, editor));
 
+            RenderedControls.AddRange(CommandControls.CreateDefaultInputGroupFor("v_SearchExtension", this, editor));
+
             RenderedControls.Add(CommandControls.CreateDefaultLabelFor("v_UserVariableName", this));
             var VariableNameControl = CommandControls.CreateStandardComboboxFor("v_UserVariableName", this).AddVariableNames(editor);
             RenderedControls.AddRange(CommandControls.CreateUIHelpersFor("v_UserVariableName", this, new Control[] { VariableNameControl }, editor));