Added a few tests

[rails.git] / app / models / client_application.rb

diff --git a/app/models/client_application.rb b/app/models/client_application.rb
index d3799abe00a483db593d68ef2e6bc4e8d07c4de0..156eeafc7989cc15a6d03d504bef6477aa0604e8 100644 (file)
--- a/app/models/client_application.rb
+++ b/app/models/client_application.rb
@@ -1,27 +1,36 @@
-require 'oauth'
+require "oauth"
+

 class ClientApplication < ActiveRecord::Base
   belongs_to :user
 class ClientApplication < ActiveRecord::Base
   belongs_to :user

-  has_many :tokens, :class_name => "OauthToken"
-  validates_presence_of :name, :url, :key, :secret
-  validates_uniqueness_of :key
-  before_validation_on_create :generate_keys
-  
+  has_many :tokens, :class_name => "OauthToken", :dependent => :delete_all
+  has_many :access_tokens
+  has_many :oauth2_verifiers
+  has_many :oauth_tokens
+
+  validates :key, :presence => true, :uniqueness => true
+  validates :name, :url, :secret, :presence => true
+  validates :url, :format => %r{\Ahttp(s?)://(\w+:{0,1}\w*@)?(\S+)(:[0-9]+)?(/|/([\w#!:.?+=&%@!\-/]))?}i
+  validates :support_url, :allow_blank => true, :format => %r{\Ahttp(s?)://(\w+:{0,1}\w*@)?(\S+)(:[0-9]+)?(/|/([\w#!:.?+=&%@!\-/]))?}i
+  validates :callback_url, :allow_blank => true, :format => %r{\A[a-z][a-z0-9.+-]*://(\w+:{0,1}\w*@)?(\S+)(:[0-9]+)?(/|/([\w#!:.?+=&%@!\-/]))?}i
+
+  before_validation :generate_keys, :on => :create
+
+  attr_accessor :token_callback_url
+
+  def self.find_token(token_key)
+    token = OauthToken.find_by_token(token_key, :include => :client_application)
+    token if token && token.authorized?
+  end
+

   def self.verify_request(request, options = {}, &block)
   def self.verify_request(request, options = {}, &block)

-    begin
-      signature = OAuth::Signature.build(request, options, &block)
-      logger.info "Signature Base String: #{signature.signature_base_string}"
-      logger.info "Consumer: #{signature.send :consumer_key}"
-      logger.info "Token: #{signature.send :token}"
-      return false unless OauthNonce.remember(signature.request.nonce, signature.request.timestamp)
-      value = signature.verify
-      logger.info "Signature verification returned: #{value.to_s}"
-      value
-    rescue OAuth::Signature::UnknownSignatureMethod => e
-      logger.info "ERROR"+e.to_s
-      false
-    end
+    signature = OAuth::Signature.build(request, options, &block)
+    return false unless OauthNonce.remember(signature.request.nonce, signature.request.timestamp)
+    value = signature.verify
+    value
+  rescue OAuth::Signature::UnknownSignatureMethod
+    false

   end
   end

-  
+

   def self.all_permissions
     PERMISSIONS
   end
   def self.all_permissions
     PERMISSIONS
   end


@@ -29,13 +38,31 @@ class ClientApplication < ActiveRecord::Base
   def oauth_server
     @oauth_server ||= OAuth::Server.new("http://" + SERVER_URL)
   end
   def oauth_server
     @oauth_server ||= OAuth::Server.new("http://" + SERVER_URL)
   end

-  
+

   def credentials
     @oauth_client ||= OAuth::Consumer.new(key, secret)
   end
   def credentials
     @oauth_client ||= OAuth::Consumer.new(key, secret)
   end

-    
-  def create_request_token
-    RequestToken.create :client_application => self
+
+  def create_request_token(params = {})
+    params = { :client_application => self, :callback_url => token_callback_url }
+    permissions.each do |p|
+      params[p] = true
+    end
+    RequestToken.create(params)
+  end
+
+  def access_token_for_user(user)
+    unless token = access_tokens.valid.find_by(:user_id => user)
+      params = { :user => user }
+
+      permissions.each do |p|
+        params[p] = true
+      end
+
+      token = access_tokens.create(params)
+    end
+
+    token

   end
 
   # the permissions that this client would like from the user
   end
 
   # the permissions that this client would like from the user


@@ -43,17 +70,17 @@ class ClientApplication < ActiveRecord::Base
     ClientApplication.all_permissions.select { |p| self[p] }
   end
 
     ClientApplication.all_permissions.select { |p| self[p] }
   end
 

-protected
-  
+  protected
+

   # this is the set of permissions that the client can ask for. clients
   # have to say up-front what permissions they want and when users sign up they
   # can agree or not agree to each of them.
   PERMISSIONS = [:allow_read_prefs, :allow_write_prefs, :allow_write_diary,
   # this is the set of permissions that the client can ask for. clients
   # have to say up-front what permissions they want and when users sign up they
   # can agree or not agree to each of them.
   PERMISSIONS = [:allow_read_prefs, :allow_write_prefs, :allow_write_diary,

-                 :allow_write_api, :allow_read_gpx, :allow_write_gpx ]
+                 :allow_write_api, :allow_read_gpx, :allow_write_gpx,
+                 :allow_write_notes].freeze

 
   def generate_keys
 
   def generate_keys

-    @oauth_client = oauth_server.generate_consumer_credentials
-    self.key = @oauth_client.key
-    self.secret = @oauth_client.secret
+    self.key = OAuth::Helper.generate_key(40)[0, 40]
+    self.secret = OAuth::Helper.generate_key(40)[0, 40]

   end
 end
   end
 end