4 files changed, 68 insertions, 6 deletions

diff --git a/app/controllers/assets_controller.rb b/app/controllers/assets_controller.rb
index becfe13..fbede0a 100644
--- a/app/controllers/assets_controller.rb
+++ b/app/controllers/assets_controller.rb
@@ -28,6 +28,7 @@ class AssetsController < ApplicationController
   # GET /assets/new.xml
   def new
     @asset = Asset.new
+    @attach_node = Node.not_in_trash.find_by(:id => params[:node_id]) if params[:node_id].present?
 
     respond_to do |format|
       format.html # new.html.erb
@@ -44,10 +45,12 @@ class AssetsController < ApplicationController
   # POST /assets.xml
   def create
     @asset = Asset.new(asset_params)
+    attach_node = Node.not_in_trash.find_by(:id => params[:node_id]) if params[:node_id].present?
 
     respond_to do |format|
       if @asset.save
         flash[:notice] = 'Asset was successfully created.'
+        attach_to(attach_node) if attach_node
         format.html { redirect_to(@asset) }
         format.xml  { render :xml => @asset, :status => :created, :location => @asset }
       else
@@ -91,4 +94,23 @@ class AssetsController < ApplicationController
     def asset_params
       params.require(:asset).permit(:name, :upload, :creator, :source_url, :license_key)
     end
+
+    def attach_to node
+      result = node.attach_asset!(@asset, :user => current_user,
+                                   :headline => params[:headline].present?)
+      flash[:notice] =
+        if result[:attached].zero?
+          "Asset saved — it was already attached to “#{node.title}”."
+        else
+          "Asset was successfully created and attached to “#{node.title}”."
+        end
+      case result[:headline]
+      when :set           then flash[:notice] += " It is now the page's headline."
+      when :kept_existing then flash[:headline_kept_path] = node_path(node)
+      when :not_eligible  then flash[:error] = "This asset type cannot be a headline."
+      end
+    rescue LockedByAnotherUser
+      flash[:locked_by]        = node.lock_owner&.login
+      flash[:locked_node_path] = node_path(node)
+    end
 end
diff --git a/app/views/assets/edit.html.erb b/app/views/assets/edit.html.erb
index 3ef8fea..0c1dd8f 100644
--- a/app/views/assets/edit.html.erb
+++ b/app/views/assets/edit.html.erb
@@ -31,6 +31,23 @@
               ) %>
       </div>
 
+      <div class="node_description">attach to page</div>
+      <div class="node_content">
+        <div class="restore_picker">
+          <%= text_field_tag :asset_node_search_term, @attach_node&.title,
+                :placeholder => "Search for a page…", :autocomplete => "off" %>
+          <div id="asset_node_search_results" class="search_results" style="display: none"></div>
+        </div>
+        <%= hidden_field_tag :node_id, @attach_node&.id %>
+        <span class="field_hint">Optional — attaches this asset to that page, all pending versions included.</span>
+      </div>
+
+      <div class="node_description"></div>
+      <div class="node_content">
+        <label><%= check_box_tag :headline, "1" %> attach as the page's headline</label>
+        <span class="field_hint">Applies only if the page has no headline yet.</span>
+      </div>
+
       <div class="node_description">Actions</div>
       <div class="node_content node_info_group">
         <div class="node_info_group_items">
diff --git a/app/views/assets/new.html.erb b/app/views/assets/new.html.erb
index 2cf8865..7bded94 100644
--- a/app/views/assets/new.html.erb
+++ b/app/views/assets/new.html.erb
@@ -26,6 +26,23 @@
               ) %>
       </div>
 
+      <div class="node_description">attach to page</div>
+      <div class="node_content">
+        <div class="restore_picker">
+          <%= text_field_tag :asset_node_search_term, @attach_node&.title,
+                :placeholder => "Search for a page…", :autocomplete => "off" %>
+          <div id="asset_node_search_results" class="search_results" style="display: none"></div>
+        </div>
+        <%= hidden_field_tag :node_id, @attach_node&.id %>
+        <span class="field_hint">Optional — attaches this asset to that page, all pending versions included.</span>
+      </div>
+
+      <div class="node_description"></div>
+      <div class="node_content">
+        <label><%= check_box_tag :headline, "1" %> attach as the page's headline</label>
+        <span class="field_hint">Applies only if the page has no headline yet.</span>
+      </div>
+
       <div class="node_description">Actions</div>
       <div class="node_content node_info_group">
         <div class="node_info_group_items">
diff --git a/app/views/nodes/show.html.erb b/app/views/nodes/show.html.erb
index adf79dd..b981e4b 100644
--- a/app/views/nodes/show.html.erb
+++ b/app/views/nodes/show.html.erb
@@ -253,10 +253,10 @@
       <% end %>
     </div>
 
-    <% if @page.assets.any? %>
-      <% headline_asset_id = @page.headline_asset&.id %>
-      <div class="node_description">Images</div>
-      <div class="node_content node_info_group">
+    <div class="node_description">Attachments</div>
+    <div class="node_content node_info_group">
+      <% if @page.assets.any? %>
+        <% headline_asset_id = @page.headline_asset&.id %>
         <ul class="thumbnail_list">
           <% @page.assets.each do |asset| %>
             <li class="<%= "is_headline" if asset.id == headline_asset_id %>">
@@ -269,8 +269,14 @@
             </li>
           <% end %>
         </ul>
-      </div>
-    <% end %>
+      <% end %>
+      <p class="add_attachments">
+        <%= link_to new_asset_path(:node_id => @node.id), class: 'action_button' do %>
+          <%= icon("paperclip", library: "tabler", "aria-hidden": true) %> Upload new attachment
+        <% end %>
+        <span class="field_hint">To add or remove existing attachments, edit this node instead.</span>
+      </p>
+    </div>
 
     <div class="node_description">Events</div>
     <div class="node_content node_info_group">