runner.rb in configure-s3-website-1.4.0

- old
+ new

@@ -1,17 +1,31 @@
 module ConfigureS3Website
   class Runner
     def self.run(options, standard_input = STDIN)
       S3Client.configure_website options
+      maybe_create_or_update_cloudfront options, standard_input
+    end
+
+    private
+
+    def self.maybe_create_or_update_cloudfront(options, standard_input)
       unless user_already_has_cf_configured options
         CloudFrontClient.create_distribution_if_user_agrees options, standard_input
+        return
       end
+      if user_already_has_cf_configured(options) and user_has_custom_cf_dist_config(options)
+        CloudFrontClient.apply_distribution_config options
+        return
+      end
     end
 
-    private
-
     def self.user_already_has_cf_configured(options)
       config_source = options[:config_source]
       config_source.cloudfront_distribution_id
+    end
+
+    def self.user_has_custom_cf_dist_config(options)
+      config_source = options[:config_source]
+      config_source.cloudfront_distribution_config
     end
   end
 end