1 files changed, 10 insertions, 4 deletions

diff --git a/app/models/page.rb b/app/models/page.rb
index f33d88d..ab42b1c 100644
--- a/app/models/page.rb
+++ b/app/models/page.rb
@@ -220,7 +220,12 @@ class Page < ApplicationRecord
     end
 
     # Clone asset references
-    self.assets = page.assets
+    self.related_assets.delete_all
+    page.related_assets.each do |related|
+      self.related_assets.create!(:asset_id => related.asset_id,
+                                  :position => related.position,
+                                  :headline => related.headline)
+    end
 
     self.save
   end
@@ -315,18 +320,19 @@ class Page < ApplicationRecord
   end
 
   def update_assets image_ids
-
     transaction do
+      previous_headline_asset_id = self.related_assets.find_by(headline: true)&.asset_id
+
       self.related_assets.delete_all
 
       if image_ids
         image_ids.each_with_index do |id, index|
           asset = Asset.find(id)
-          self.related_assets.create!(:asset_id => asset.id, :position => index+1)
+          self.related_assets.create!(:asset_id => asset.id, :position => index+1,
+                                         :headline => asset.id == previous_headline_asset_id)
         end
       end
     end
-
   end
 
   # Installs (or re-installs) the trigger that keeps page_translations'