81 lines
2.2 KiB
Ruby
81 lines
2.2 KiB
Ruby
|
module Current
|
||
|
|
thread_mattr_accessor :user
|
||
|
|
end
|
||
|
|
|
||
|
|
class ApplicationController < ActionController::Base
|
||
|
|
include DeviseTokenAuth::Concerns::SetUserByToken
|
||
|
|
include Pundit
|
||
|
|
|
||
|
|
protect_from_forgery with: :null_session
|
||
|
|
|
||
|
|
before_action :set_current_user, unless: :devise_controller?
|
||
|
|
before_action :check_subscription, unless: :devise_controller?
|
||
|
|
around_action :handle_with_exception, unless: :devise_controller?
|
||
|
|
|
||
|
|
# after_action :verify_authorized
|
||
|
|
rescue_from ActiveRecord::RecordInvalid, with: :render_record_invalid
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def current_account
|
||
|
|
@_ ||= current_user.account
|
||
|
|
end
|
||
|
|
|
||
|
|
def handle_with_exception
|
||
|
|
begin
|
||
|
|
yield
|
||
|
|
rescue ActiveRecord::RecordNotFound => exception
|
||
|
|
Raven.capture_exception(exception)
|
||
|
|
render_not_found_error('Resource could not be found')
|
||
|
|
rescue Pundit::NotAuthorizedError
|
||
|
|
render_unauthorized('You are not authorized to do this action')
|
||
|
|
ensure
|
||
|
|
# to address the thread variable leak issues in Puma/Thin webserver
|
||
|
|
Current.user = nil
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def set_current_user
|
||
|
|
@user ||= current_user
|
||
|
|
Current.user = @user
|
||
|
|
end
|
||
|
|
|
||
|
|
def current_subscription
|
||
|
|
@subscription ||= current_account.subscription
|
||
|
|
end
|
||
|
|
|
||
|
|
def render_unauthorized(message)
|
||
|
|
render json: { error: message }, status: :unauthorized
|
||
|
|
end
|
||
|
|
|
||
|
|
def render_not_found_error(message)
|
||
|
|
render json: { error: message }, status: :not_found
|
||
|
|
end
|
||
|
|
|
||
|
|
def render_could_not_create_error(message)
|
||
|
|
render json: { error: message }, status: :unprocessable_entity
|
||
|
|
end
|
||
|
|
|
||
|
|
def render_internal_server_error(message)
|
||
|
|
render json: { error: message }, status: :internal_server_error
|
||
|
|
end
|
||
|
|
|
||
|
|
def render_record_invalid(exception)
|
||
|
|
render json: {
|
||
|
|
message: "#{exception.record.errors.full_messages.join(", ")}"
|
||
|
|
}, status: :unprocessable_entity
|
||
|
|
end
|
||
|
|
|
||
|
|
def render_error_response(exception)
|
||
|
|
render json: exception.to_hash, status: exception.http_status
|
||
|
|
end
|
||
|
|
|
||
|
|
def check_subscription
|
||
|
|
if current_subscription.trial? && current_subscription.expiry < Date.current
|
||
|
|
render json: { error: 'Trial Expired'}, status: :trial_expired
|
||
|
|
elsif current_subscription.cancelled?
|
||
|
|
render json: { error: 'Account Suspended'}, status: :account_suspended
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|