669ba6fc8d
* Fix the post-upgrade message showing the gift message instead of the normal upgrade message. * Link back to Danbooru so people don't stay on Safebooru and get confused when they can't see hidden posts.
65 lines
1.3 KiB
Ruby
65 lines
1.3 KiB
Ruby
class UserUpgradesController < ApplicationController
|
|
before_action :member_only, only: [:show]
|
|
helper_method :user
|
|
skip_before_action :verify_authenticity_token, only: [:create]
|
|
|
|
def create
|
|
if params[:stripeToken]
|
|
create_stripe
|
|
end
|
|
end
|
|
|
|
def new
|
|
end
|
|
|
|
def show
|
|
end
|
|
|
|
def user
|
|
if params[:user_id]
|
|
User.find(params[:user_id])
|
|
else
|
|
CurrentUser.user
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def create_stripe
|
|
@user = user
|
|
|
|
if params[:desc] == "Upgrade to Gold"
|
|
level = User::Levels::GOLD
|
|
cost = 2000
|
|
elsif params[:desc] == "Upgrade to Platinum"
|
|
level = User::Levels::PLATINUM
|
|
cost = 4000
|
|
elsif params[:desc] == "Upgrade Gold to Platinum" && @user.level == User::Levels::GOLD
|
|
level = User::Levels::PLATINUM
|
|
cost = 2000
|
|
else
|
|
raise "Invalid desc"
|
|
end
|
|
|
|
begin
|
|
charge = Stripe::Charge.create(
|
|
:amount => cost,
|
|
:currency => "usd",
|
|
:card => params[:stripeToken],
|
|
:description => params[:desc]
|
|
)
|
|
@user.promote_to!(level, is_upgrade: true)
|
|
flash[:success] = true
|
|
rescue Stripe::CardError => e
|
|
DanbooruLogger.log(e)
|
|
flash[:error] = e.message
|
|
end
|
|
|
|
if @user == CurrentUser.user
|
|
redirect_to user_upgrade_path
|
|
else
|
|
redirect_to user_upgrade_path(user_id: params[:user_id])
|
|
end
|
|
end
|
|
end
|