Add download snippet screenshot link

Author: rossta

app/controllers/share/snippet_screenshots_controller.rb

@@ -15,6 +15,15 @@ def new
   def create
     @snippet = Snippet.find(params[:snippet_id])
     @snippet.attach_screenshot_from_base64(params[:screenshot])
-    redirect_to new_share_snippet_tweet_path(@snippet, auto: true), notice: "Screenshot successful".emojoy
+    redirect_to intent_url, notice: "Screenshot successful".emojoy
+  end
+
+  def intent_url
+    case params[:intent]
+    when "share"
+      new_share_snippet_tweet_url(@snippet, auto: true)
+    else
+      share_snippet_url(@snippet)
+    end
   end
 end

app/javascript/controllers/snippets/screenshot.ts

@@ -40,6 +40,13 @@ export default class extends Controller<HTMLFormElement> {
       const data = await this.drawScreenshot();
 
       event.detail.fetchOptions.body.append('screenshot', data);
+
+      const searchParams = new URLSearchParams(window.location.search);
+      ['auto', 'intent'].forEach((key) => {
+        if (searchParams.get(key)) {
+          event.detail.fetchOptions.body.append(key, searchParams.get(key));
+        }
+      });
     }
 
     event.detail.resume();

app/views/share/snippets/toolbar.rb

@@ -12,10 +12,19 @@ def view_template
     flex_block do
       render Share::SnippetTweets::TweetButton.new(@snippet)
 
+      a(
+        href: download_url,
+        class: "button transparent"
+      ) { "Download" }
+
       if @current_user&.can_edit?(@snippet)
-        link_to "Edit this snippet", edit_share_snippet_path(@snippet),
+        a(
+          href: edit_share_snippet_path(@snippet),
           class: "button secondary",
           data: {turbo_frame: "snippet_form"}
+        ) do
+          "Edit this snippet"
+        end
       end
     end
   end
@@ -24,7 +33,15 @@ def share_url
     if @snippet.screenshot.attached?
       new_share_snippet_tweet_path(@snippet, auto: "true")
     else
-      new_share_snippet_screenshot_path(@snippet, auto: "true")
+      new_share_snippet_screenshot_path(@snippet, auto: "true", intent: "share")
+    end
+  end
+
+  def download_url
+    if @snippet.screenshot.attached?
+      rails_blob_url(@snippet.screenshot, disposition: "attachment")
+    else
+      new_share_snippet_screenshot_path(@snippet, auto: "true", intent: "download")
     end
   end
 end

spec/requests/share/snippet_screenshots_spec.rb

@@ -0,0 +1,37 @@
+require "rails_helper"
+
+RSpec.describe "/snippets/:id/screenshot", type: :request do
+  describe "GET show" do
+    it "is successful" do
+      get share_snippet_screenshot_path(FactoryBot.create(:snippet))
+
+      expect(response).to have_http_status(:success)
+    end
+  end
+
+  describe "GET new" do
+    it "is successful" do
+      get new_share_snippet_screenshot_path(FactoryBot.create(:snippet))
+
+      expect(response).to have_http_status(:success)
+    end
+  end
+
+  describe "POST create" do
+    let(:png_data_uri) { "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAACklEQVR4nGMAAQAABQABDQottAAAAABJRU5ErkJggg==" }
+
+    it "attaches screenshot and redirects to show" do
+      post share_snippet_screenshot_path(FactoryBot.create(:snippet)), params: {screenshot: png_data_uri}
+
+      expect(response).to have_http_status(:found)
+      expect(response).to redirect_to(share_snippet_path(Snippet.last))
+    end
+
+    it "attaches screenshot and redirects to share" do
+      post share_snippet_screenshot_path(FactoryBot.create(:snippet)), params: {intent: "share", screenshot: png_data_uri}
+
+      expect(response).to have_http_status(:found)
+      expect(response).to redirect_to(new_share_snippet_tweet_url(Snippet.last, auto: true))
+    end
+  end
+end

spec/system/snippets_spec.rb

@@ -50,7 +50,7 @@
 
     click_link "Share"
 
-    expect(page).to have_content("Screenshot")
+    expect(page).to have_content("Download")
   end
 
   it "can share Snippet" do
@@ -63,6 +63,7 @@
       click_link "Share"
     end
 
-    expect(page).to have_content("Screenshot")
+    expect(page).to have_content("Download")
+    click_link "Download"
   end
 end