lib/jekyll-s3/cli.rb in jekyll-s3-0.0.7 vs lib/jekyll-s3/cli.rb in jekyll-s3-1.0.0

@@ -8,24 +8,24 @@
       end
 
       def run(site_dir, in_headless_mode = false)
         CLI.check_configuration site_dir
         config = Jekyll::S3::ConfigLoader.load_configuration site_dir
-        new_files_count, changed_files_count, deleted_files_count = 
+        new_files_count, changed_files_count, deleted_files_count, changed_files = 
           Uploader.run(site_dir, config, in_headless_mode)
-        CLI.invalidate_cf_dist_if_configured config
+        CLI.invalidate_cf_dist_if_configured(config, changed_files)
         [new_files_count, changed_files_count, deleted_files_count]
       rescue JekyllS3Error => e
         puts e.message
         exit 1
       end
 
       private
 
-      def self.invalidate_cf_dist_if_configured(config)
+      def self.invalidate_cf_dist_if_configured(config, changed_files)
         cloudfront_configured = config['cloudfront_distribution_id'] &&
           (not config['cloudfront_distribution_id'].empty?)
-        Jekyll::Cloudfront::Invalidator.invalidate(config) if cloudfront_configured
+        Jekyll::Cloudfront::Invalidator.invalidate(config, changed_files) if cloudfront_configured
       end
 
       def self.check_configuration(site_dir)
         Jekyll::S3::ConfigLoader.check_jekyll_project site_dir
         Jekyll::S3::ConfigLoader.check_s3_configuration site_dir