diff options
Diffstat (limited to 'app')
-rw-r--r-- | app/controllers/xrd_controller.rb | 12 | ||||
-rw-r--r-- | app/models/account.rb | 7 |
2 files changed, 14 insertions, 5 deletions
diff --git a/app/controllers/xrd_controller.rb b/app/controllers/xrd_controller.rb index 2946e9999..fa67b2baa 100644 --- a/app/controllers/xrd_controller.rb +++ b/app/controllers/xrd_controller.rb @@ -6,7 +6,7 @@ class XrdController < ApplicationController end def webfinger - @account = Account.find_by!(username: username_from_resource, domain: nil) + @account = Account.find_local!(username_from_resource) @canonical_account_uri = "acct:#{@account.username}@#{Rails.configuration.x.local_domain}" @magic_key = pem_to_magic_key(@account.keypair.public_key) rescue ActiveRecord::RecordNotFound @@ -21,10 +21,10 @@ class XrdController < ApplicationController end def username_from_resource - if params[:resource].start_with?('acct:') - params[:resource].split('@').first.gsub('acct:', '') + if resource_param.start_with?('acct:') + resource_param.split('@').first.gsub('acct:', '') else - url = Addressable::URI.parse(params[:resource]) + url = Addressable::URI.parse(resource_param) url.path.gsub('/users/', '') end end @@ -43,4 +43,8 @@ class XrdController < ApplicationController (["RSA"] + [modulus, exponent].map { |n| Base64.urlsafe_encode64(n) }).join('.') end + + def resource_param + params.require(:resource) + end end diff --git a/app/models/account.rb b/app/models/account.rb index b3917e6e8..c92bb1574 100644 --- a/app/models/account.rb +++ b/app/models/account.rb @@ -1,7 +1,7 @@ class Account < ActiveRecord::Base # Local users has_one :user, inverse_of: :account - validates :username, uniqueness: { scope: :domain } + validates :username, uniqueness: { scope: :domain, case_sensitive: false } # Avatar upload attr_reader :avatar_remote_url @@ -97,6 +97,11 @@ class Account < ActiveRecord::Base self.username end + def self.find_local!(username) + table = self.arel_table + self.where(table[:username].matches(username)).where(domain: nil).take! + end + before_create do if local? keypair = OpenSSL::PKey::RSA.new(Rails.env.test? ? 1024 : 2048) |