Update Photo upload url for add user login as photo path

ruby-china · Jan 7, 2021 · 1cffe09 · 1cffe09
1 parent 67c0e0b
commit 1cffe09
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 3 deletions.
diff --git a/app/controllers/photos_controller.rb b/app/controllers/photos_controller.rb
@@ -10,7 +10,7 @@ def create
     end
 
     # 浮动窗口上传
-    @photo = Photo.new(image: params[:file], user_id: current_user.id)
+    @photo = Photo.new(image: params[:file], user: current_user)
     if @photo.save
       render json: { ok: true, url: @photo.image.url(:large) }
     else

diff --git a/app/models/photo.rb b/app/models/photo.rb
@@ -5,4 +5,5 @@ class Photo < ApplicationRecord
 
   # 封面图
   mount_uploader :image, PhotoUploader
+  after_commit :remove_image!, on: :destroy
 end
diff --git a/app/uploaders/photo_uploader.rb b/app/uploaders/photo_uploader.rb
@@ -5,7 +5,7 @@ class PhotoUploader < BaseUploader
   def filename
     if super.present?
       @name ||= SecureRandom.uuid
-      "#{Time.now.year}/#{@name}.#{file.extension.downcase}"
+      "#{model.user&.login || Time.now.year}/#{@name}.#{file.extension.downcase}"
     end
   end
 end
diff --git a/test/controllers/photos_controller_test.rb b/test/controllers/photos_controller_test.rb
@@ -11,7 +11,7 @@
     post photos_path, params: { file: file }
     assert_equal 200, response.status
     assert_equal true, response.parsed_body["ok"]
-    assert_match Regexp.new("/uploads/photo/#{Date.today.year}/[a-zA-Z0-9\\-]+.png!large"), response.parsed_body["url"]
+    assert_match Regexp.new("/uploads/photo/#{user.login}/[a-zA-Z0-9\\-]+.png!large"), response.parsed_body["url"]
   end
 
   it "POST /photos failure for blank data" do