(chocolateyGH-14) Add SourceType to config

Set it based on what is passed in
ferventcoder · Jan 22, 2015 · 1aac979 · 1aac979
1 parent b5fb7bf
commit 1aac979
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/chocolatey/infrastructure.app/configuration/ChocolateyConfiguration.cs b/src/chocolatey/infrastructure.app/configuration/ChocolateyConfiguration.cs
@@ -31,6 +31,7 @@ public ChocolateyConfiguration()
         {
             RegularOuptut = true;
             PromptForConfirmation = true;
+            SourceType = SourceType.normal;
             Information = new InformationCommandConfiguration();
             Features = new FeaturesConfiguration();
             NewCommand = new NewCommandConfiguration();
@@ -102,6 +103,7 @@ private void output_tostring(StringBuilder propertyValues, IEnumerable<PropertyI
         ///   One or more source locations set by configuration or by command line. Separated by semi-colon
         /// </summary>
         public string Sources { get; set; }
+        public SourceType SourceType { get; set; }
 
         // top level commands
 
@@ -178,7 +180,6 @@ private void output_tostring(StringBuilder propertyValues, IEnumerable<PropertyI
         ///   Configuration related specifically to Push command
         /// </summary>
         public PushCommandConfiguration PushCommand { get; private set; }
-
     }
 
     public sealed class InformationCommandConfiguration

diff --git a/src/chocolatey/infrastructure.app/runners/GenericRunner.cs b/src/chocolatey/infrastructure.app/runners/GenericRunner.cs
@@ -20,6 +20,7 @@ namespace chocolatey.infrastructure.app.runners
     using SimpleInjector;
     using attributes;
     using configuration;
+    using domain;
     using infrastructure.commands;
 
     public sealed class GenericRunner
@@ -50,6 +51,8 @@ public void run(ChocolateyConfiguration config, Container container, bool isCons
                     parseArgs.Invoke(command);
                 }
 
+                set_source_type(config);
+
                 this.Log().Debug(() => "Configuration: {0}".format_with(config.ToString()));
 
 
@@ -78,5 +81,12 @@ public void run(ChocolateyConfiguration config, Container container, bool isCons
                 }
             }
         }
+
+        private static void set_source_type(ChocolateyConfiguration config)
+        {
+            var sourceType = SourceType.normal;
+            Enum.TryParse(config.Sources, true, out sourceType);
+            config.SourceType = sourceType;
+        }
     }
 }