refactoring

This commit is contained in:
Oleg Hasjanov 2021-05-07 18:08:29 +03:00
parent 54f7e87370
commit 13a257210b

View file

@ -6,7 +6,7 @@ module Api
class DomainsController < ::Api::V1::Registrant::BaseController class DomainsController < ::Api::V1::Registrant::BaseController
before_action :set_tech_flag, only: [:show] before_action :set_tech_flag, only: [:show]
LIMIT_DOMAIN_TOTAL = 3000.freeze LIMIT_DOMAIN_TOTAL = 3000
def index def index
limit = params[:limit] || 200 limit = params[:limit] || 200
@ -59,24 +59,34 @@ module Api
end end
def current_user_domains def current_user_domains
return initialization_count_of__domains if params[:tech] == 'init' initialization_count_of__domains
current_registrant_user.domains(admin: params[:tech] != 'true')
rescue CompanyRegister::NotAvailableError rescue CompanyRegister::NotAvailableError
return initialization_count_of__direct_domains if params[:tech] == 'init' initialization_count_of__direct_domains
end
def initialization_count_of__direct_domains
if params[:tech] == 'init'
if current_user_domains_total_count < LIMIT_DOMAIN_TOTAL
return current_registrant_user.direct_domains(admin: false)
end
current_registrant_user.direct_domains(admin: true)
end
current_registrant_user.direct_domains(admin: params[:tech] != 'true') current_registrant_user.direct_domains(admin: params[:tech] != 'true')
end end
def initialization_count_of__direct_domains def initialization_count_of__domains
return current_registrant_user.direct_domains(admin: false) if current_user_domains_total_count < LIMIT_DOMAIN_TOTAL if params[:tech] == 'init'
current_registrant_user.direct_domains(admin: true) if current_user_domains_total_count < LIMIT_DOMAIN_TOTAL
return current_registrant_user.domains(admin: false)
end end
def initialization_count_of__domains
return current_registrant_user.domains(admin: false) if current_user_domains_total_count < LIMIT_DOMAIN_TOTAL
current_registrant_user.domains(admin: true) current_registrant_user.domains(admin: true)
end end
current_registrant_user.domains(admin: params[:tech] != 'true')
end
end end
end end
end end