base.rb 41.2 KB
Newer Older
J
Jeremy Kemper 已提交
1
require 'active_support'
2 3
require 'active_support/core_ext/class/attribute_accessors'
require 'active_support/core_ext/class/inheritable_attributes'
J
Jeremy Kemper 已提交
4
require 'active_support/core_ext/kernel/reporting'
5 6 7
require 'active_support/core_ext/module/attr_accessor_with_default'
require 'active_support/core_ext/module/delegation'
require 'active_support/core_ext/module/aliasing'
J
Jeremy Kemper 已提交
8
require 'active_support/core_ext/object/blank'
J
Jeremy Kemper 已提交
9
require 'active_support/core_ext/object/misc'
10
require 'set'
J
Jeremy Kemper 已提交
11
require 'uri'
12

J
Joshua Peek 已提交
13
require 'active_resource/exceptions'
14

J
Joshua Peek 已提交
15
module ActiveResource
16 17
  # ActiveResource::Base is the main class for mapping RESTful resources as models in a Rails application.
  #
18
  # For an outline of what Active Resource is capable of, see link:files/vendor/rails/activeresource/README.html.
19 20 21 22 23 24 25
  #
  # == Automated mapping
  #
  # Active Resource objects represent your RESTful resources as manipulatable Ruby objects.  To map resources
  # to Ruby objects, Active Resource only needs a class name that corresponds to the resource name (e.g., the class
  # Person maps to the resources people, very similarly to Active Record) and a +site+ value, which holds the
  # URI of the resources.
26
  #
P
Pratik Naik 已提交
27 28 29
  #   class Person < ActiveResource::Base
  #     self.site = "http://api.people.com:3000/"
  #   end
30
  #
31
  # Now the Person class is mapped to RESTful resources located at <tt>http://api.people.com:3000/people/</tt>, and
P
Pratik Naik 已提交
32
  # you can now use Active Resource's lifecycle methods to manipulate resources. In the case where you already have
P
Pratik Naik 已提交
33 34 35 36 37 38
  # an existing model with the same name as the desired RESTful resource you can set the +element_name+ value.
  #
  #   class PersonResource < ActiveResource::Base
  #     self.site = "http://api.people.com:3000/"
  #     self.element_name = "person"
  #   end
39 40
  #
  #
41 42 43 44
  # == Lifecycle methods
  #
  # Active Resource exposes methods for creating, finding, updating, and deleting resources
  # from REST web services.
45
  #
46
  #   ryan = Person.new(:first => 'Ryan', :last => 'Daigle')
P
Pratik Naik 已提交
47 48 49 50
  #   ryan.save                # => true
  #   ryan.id                  # => 2
  #   Person.exists?(ryan.id)  # => true
  #   ryan.exists?             # => true
51
  #
52
  #   ryan = Person.find(1)
P
Pratik Naik 已提交
53
  #   # Resource holding our newly created Person object
54
  #
55
  #   ryan.first = 'Rizzle'
P
Pratik Naik 已提交
56
  #   ryan.save                # => true
57
  #
P
Pratik Naik 已提交
58
  #   ryan.destroy             # => true
59 60 61
  #
  # As you can see, these are very similar to Active Record's lifecycle methods for database records.
  # You can read more about each of these methods in their respective documentation.
62
  #
63 64 65
  # === Custom REST methods
  #
  # Since simple CRUD/lifecycle methods can't accomplish every task, Active Resource also supports
66
  # defining your own custom REST methods. To invoke them, Active Resource provides the <tt>get</tt>,
P
Pratik Naik 已提交
67
  # <tt>post</tt>, <tt>put</tt> and <tt>\delete</tt> methods where you can specify a custom REST method
68 69 70 71
  # name to invoke.
  #
  #   # POST to the custom 'register' REST method, i.e. POST /people/new/register.xml.
  #   Person.new(:name => 'Ryan').post(:register)
72 73
  #   # => { :id => 1, :name => 'Ryan', :position => 'Clerk' }
  #
74
  #   # PUT an update by invoking the 'promote' REST method, i.e. PUT /people/1/promote.xml?position=Manager.
75
  #   Person.find(1).put(:promote, :position => 'Manager')
76
  #   # => { :id => 1, :name => 'Ryan', :position => 'Manager' }
77 78 79 80 81 82 83
  #
  #   # GET all the positions available, i.e. GET /people/positions.xml.
  #   Person.get(:positions)
  #   # => [{:name => 'Manager'}, {:name => 'Clerk'}]
  #
  #   # DELETE to 'fire' a person, i.e. DELETE /people/1/fire.xml.
  #   Person.find(1).delete(:fire)
84
  #
85
  # For more information on using custom REST methods, see the
86 87 88 89 90
  # ActiveResource::CustomMethods documentation.
  #
  # == Validations
  #
  # You can validate resources client side by overriding validation methods in the base class.
91
  #
P
Pratik Naik 已提交
92 93 94 95 96 97 98
  #   class Person < ActiveResource::Base
  #      self.site = "http://api.people.com:3000/"
  #      protected
  #        def validate
  #          errors.add("last", "has invalid characters") unless last =~ /[a-zA-Z]*/
  #        end
  #   end
99
  #
100 101 102
  # See the ActiveResource::Validations documentation for more information.
  #
  # == Authentication
103
  #
104
  # Many REST APIs will require authentication, usually in the form of basic
105 106
  # HTTP authentication.  Authentication can be specified by:
  # * putting the credentials in the URL for the +site+ variable.
107
  #
108 109 110
  #    class Person < ActiveResource::Base
  #      self.site = "http://ryan:password@api.people.com:3000/"
  #    end
111
  #
112 113 114 115 116 117 118 119
  # * defining +user+ and/or +password+ variables
  #
  #    class Person < ActiveResource::Base
  #      self.site = "http://api.people.com:3000/"
  #      self.user = "ryan"
  #      self.password = "password"
  #    end
  #
120
  # For obvious security reasons, it is probably best if such services are available
121
  # over HTTPS.
122 123
  #
  # Note: Some values cannot be provided in the URL passed to site.  e.g. email addresses
P
Pratik Naik 已提交
124
  # as usernames.  In those situations you should use the separate user and password option.
P
Pratik Naik 已提交
125
  #
126 127 128
  # == Errors & Validation
  #
  # Error handling and validation is handled in much the same manner as you're used to seeing in
129
  # Active Record.  Both the response code in the HTTP response and the body of the response are used to
130
  # indicate that an error occurred.
131
  #
132
  # === Resource errors
133
  #
134
  # When a GET is requested for a resource that does not exist, the HTTP <tt>404</tt> (Resource Not Found)
135 136
  # response code will be returned from the server which will raise an ActiveResource::ResourceNotFound
  # exception.
137
  #
138
  #   # GET http://api.people.com:3000/people/999.xml
P
Pratik Naik 已提交
139
  #   ryan = Person.find(999) # 404, raises ActiveResource::ResourceNotFound
140
  #
P
Pratik Naik 已提交
141
  # <tt>404</tt> is just one of the HTTP error response codes that Active Resource will handle with its own exception. The
142
  # following HTTP response codes will also result in these exceptions:
P
Pratik Naik 已提交
143 144 145 146 147 148
  # 
  # * 200..399 - Valid response, no exception (other than 301, 302)
  # * 301, 302 - ActiveResource::Redirection
  # * 400 - ActiveResource::BadRequest
  # * 401 - ActiveResource::UnauthorizedAccess
  # * 403 - ActiveResource::ForbiddenAccess
P
Pratik Naik 已提交
149
  # * 404 - ActiveResource::ResourceNotFound
P
Pratik Naik 已提交
150
  # * 405 - ActiveResource::MethodNotAllowed
P
Pratik Naik 已提交
151 152 153 154
  # * 409 - ActiveResource::ResourceConflict
  # * 422 - ActiveResource::ResourceInvalid (rescued by save as validation errors)
  # * 401..499 - ActiveResource::ClientError
  # * 500..599 - ActiveResource::ServerError
P
Pratik Naik 已提交
155
  # * Other - ActiveResource::ConnectionError
156 157 158 159 160 161 162 163 164 165 166 167 168
  #
  # These custom exceptions allow you to deal with resource errors more naturally and with more precision
  # rather than returning a general HTTP error.  For example:
  #
  #   begin
  #     ryan = Person.find(my_id)
  #   rescue ActiveResource::ResourceNotFound
  #     redirect_to :action => 'not_found'
  #   rescue ActiveResource::ResourceConflict, ActiveResource::ResourceInvalid
  #     redirect_to :action => 'new'
  #   end
  #
  # === Validation errors
169
  #
P
Pratik Naik 已提交
170
  # Active Resource supports validations on resources and will return errors if any of these validations fail
171
  # (e.g., "First name can not be blank" and so on).  These types of errors are denoted in the response by
172 173
  # a response code of <tt>422</tt> and an XML representation of the validation errors.  The save operation will
  # then fail (with a <tt>false</tt> return value) and the validation errors can be accessed on the resource in question.
174
  #
175
  #   ryan = Person.find(1)
P
Pratik Naik 已提交
176 177
  #   ryan.first # => ''
  #   ryan.save  # => false
178
  #
179
  #   # When
180 181 182 183
  #   # PUT http://api.people.com:3000/people/1.xml
  #   # is requested with invalid values, the response is:
  #   #
  #   # Response (422):
184
  #   # <errors type="array"><error>First cannot be empty</error></errors>
185 186
  #   #
  #
P
Pratik Naik 已提交
187 188
  #   ryan.errors.invalid?(:first)  # => true
  #   ryan.errors.full_messages     # => ['First cannot be empty']
189
  #
190 191
  # Learn more about Active Resource's validation features in the ActiveResource::Validations documentation.
  #
192 193 194
  # === Timeouts
  #
  # Active Resource relies on HTTP to access RESTful APIs and as such is inherently susceptible to slow or
P
Pratik Naik 已提交
195
  # unresponsive servers. In such cases, your Active Resource method calls could \timeout. You can control the
196 197 198 199 200 201 202
  # amount of time before Active Resource times out with the +timeout+ variable.
  #
  #   class Person < ActiveResource::Base
  #     self.site = "http://api.people.com:3000/"
  #     self.timeout = 5
  #   end
  #
P
Pratik Naik 已提交
203
  # This sets the +timeout+ to 5 seconds. You can adjust the +timeout+ to a value suitable for the RESTful API
204 205 206 207 208
  # you are accessing. It is recommended to set this to a reasonably low value to allow your Active Resource
  # clients (especially if you are using Active Resource in a Rails application) to fail-fast (see
  # http://en.wikipedia.org/wiki/Fail-fast) rather than cause cascading failures that could incapacitate your
  # server.
  #
P
Pratik Naik 已提交
209
  # When a \timeout occurs, an ActiveResource::TimeoutError is raised. You should rescue from
P
Pratik Naik 已提交
210 211
  # ActiveResource::TimeoutError in your Active Resource method calls.
  #
212 213 214
  # Internally, Active Resource relies on Ruby's Net::HTTP library to make HTTP requests. Setting +timeout+
  # sets the <tt>read_timeout</tt> of the internal Net::HTTP instance to the same value. The default
  # <tt>read_timeout</tt> is 60 seconds on most Ruby implementations.
215
  class Base
P
Pratik Naik 已提交
216 217
    ##
    # :singleton-method:
218
    # The logger for diagnosing and tracing Active Resource calls.
219 220
    cattr_accessor :logger

221
    class << self
P
Pratik Naik 已提交
222
      # Gets the URI of the REST resources to map for this class.  The site variable is required for
P
Pratik Naik 已提交
223
      # Active Resource's mapping to work.
224
      def site
225 226 227 228 229 230 231 232 233 234 235 236 237 238 239
        # Not using superclass_delegating_reader because don't want subclasses to modify superclass instance
        #
        # With superclass_delegating_reader
        #
        #   Parent.site = 'http://anonymous@test.com'
        #   Subclass.site # => 'http://anonymous@test.com'
        #   Subclass.site.user = 'david'
        #   Parent.site # => 'http://david@test.com'
        #
        # Without superclass_delegating_reader (expected behaviour)
        #
        #   Parent.site = 'http://anonymous@test.com'
        #   Subclass.site # => 'http://anonymous@test.com'
        #   Subclass.site.user = 'david' # => TypeError: can't modify frozen object
        #
240 241
        if defined?(@site)
          @site
242
        elsif superclass != Object && superclass.site
243 244 245
          superclass.site.dup.freeze
        end
      end
J
Jeremy Kemper 已提交
246

247
      # Sets the URI of the REST resources to map for this class to the value in the +site+ argument.
P
Pratik Naik 已提交
248
      # The site variable is required for Active Resource's mapping to work.
249
      def site=(site)
250
        @connection = nil
251 252 253 254
        if site.nil?
          @site = nil
        else
          @site = create_site_uri_from(site)
255 256
          @user = URI.decode(@site.user) if @site.user
          @password = URI.decode(@site.password) if @site.password
257 258 259
        end
      end

P
Pratik Naik 已提交
260
      # Gets the \user for REST HTTP authentication.
261 262 263 264 265 266 267 268 269
      def user
        # Not using superclass_delegating_reader. See +site+ for explanation
        if defined?(@user)
          @user
        elsif superclass != Object && superclass.user
          superclass.user.dup.freeze
        end
      end

P
Pratik Naik 已提交
270
      # Sets the \user for REST HTTP authentication.
271 272 273 274 275
      def user=(user)
        @connection = nil
        @user = user
      end

P
Pratik Naik 已提交
276
      # Gets the \password for REST HTTP authentication.
277 278 279 280 281 282 283 284 285
      def password
        # Not using superclass_delegating_reader. See +site+ for explanation
        if defined?(@password)
          @password
        elsif superclass != Object && superclass.password
          superclass.password.dup.freeze
        end
      end

P
Pratik Naik 已提交
286
      # Sets the \password for REST HTTP authentication.
287 288 289
      def password=(password)
        @connection = nil
        @password = password
290 291
      end

P
Pratik Naik 已提交
292
      # Sets the format that attributes are sent and received in from a mime type reference:
293 294 295 296 297 298 299
      #
      #   Person.format = :json
      #   Person.find(1) # => GET /people/1.json
      #
      #   Person.format = ActiveResource::Formats::XmlFormat
      #   Person.find(1) # => GET /people/1.xml
      #
P
Pratik Naik 已提交
300
      # Default format is <tt>:xml</tt>.
301
      def format=(mime_type_reference_or_format)
302
        format = mime_type_reference_or_format.is_a?(Symbol) ?
303 304
          ActiveResource::Formats[mime_type_reference_or_format] : mime_type_reference_or_format

305
        write_inheritable_attribute(:format, format)
306
        connection.format = format if site
307
      end
308

P
Pratik Naik 已提交
309
      # Returns the current format, default is ActiveResource::Formats::XmlFormat.
P
Pratik Naik 已提交
310
      def format
311
        read_inheritable_attribute(:format) || ActiveResource::Formats::XmlFormat
312 313
      end

314 315 316 317 318 319
      # Sets the number of seconds after which requests to the REST API should time out.
      def timeout=(timeout)
        @connection = nil
        @timeout = timeout
      end

P
Pratik Naik 已提交
320
      # Gets the number of seconds after which requests to the REST API should time out.
321 322 323 324 325 326 327 328
      def timeout
        if defined?(@timeout)
          @timeout
        elsif superclass != Object && superclass.timeout
          superclass.timeout
        end
      end

P
Pratik Naik 已提交
329 330
      # An instance of ActiveResource::Connection that is the base \connection to the remote service.
      # The +refresh+ parameter toggles whether or not the \connection is refreshed at every request
331
      # or not (defaults to <tt>false</tt>).
332
      def connection(refresh = false)
333 334
        if defined?(@connection) || superclass == Object
          @connection = Connection.new(site, format) if refresh || @connection.nil?
335 336
          @connection.user = user if user
          @connection.password = password if password
337
          @connection.timeout = timeout if timeout
338 339 340 341
          @connection
        else
          superclass.connection
        end
342
      end
343

344 345
      def headers
        @headers ||= {}
346 347
      end

348 349
      # Do not include any modules in the default element name. This makes it easier to seclude ARes objects
      # in a separate namespace without having to set element_name repeatedly.
J
Jeremy Kemper 已提交
350
      attr_accessor_with_default(:element_name)    { ActiveSupport::Inflector.underscore(to_s.split("::").last) } #:nodoc:
351

J
Jeremy Kemper 已提交
352
      attr_accessor_with_default(:collection_name) { ActiveSupport::Inflector.pluralize(element_name) } #:nodoc:
353
      attr_accessor_with_default(:primary_key, 'id') #:nodoc:
P
Pratik Naik 已提交
354 355 356
      
      # Gets the \prefix for a resource's nested URL (e.g., <tt>prefix/collectionname/1.xml</tt>)
      # This method is regenerated at runtime based on what the \prefix is set to.
357 358 359
      def prefix(options={})
        default = site.path
        default << '/' unless default[-1..-1] == '/'
360
        # generate the actual method based on the current site path
361
        self.prefix = default
362 363
        prefix(options)
      end
364

P
Pratik Naik 已提交
365 366
      # An attribute reader for the source string for the resource path \prefix.  This
      # method is regenerated at runtime based on what the \prefix is set to.
367 368 369 370 371
      def prefix_source
        prefix # generate #prefix and #prefix_source methods first
        prefix_source
      end

P
Pratik Naik 已提交
372
      # Sets the \prefix for a resource's nested URL (e.g., <tt>prefix/collectionname/1.xml</tt>).
373
      # Default value is <tt>site.path</tt>.
374
      def prefix=(value = '/')
375
        # Replace :placeholders with '#{embedded options[:lookups]}'
376
        prefix_call = value.gsub(/:\w+/) { |key| "\#{options[#{key}]}" }
377

378 379 380
        # Clear prefix parameters in case they have been cached
        @prefix_parameters = nil

381 382
        # Redefine the new methods.
        code = <<-end_code
383 384
          def prefix_source() "#{value}" end
          def prefix(options={}) "#{prefix_call}" end
385 386
        end_code
        silence_warnings { instance_eval code, __FILE__, __LINE__ }
387
      rescue
J
Jeremy Kemper 已提交
388
        logger.error "Couldn't set prefix: #{$!}\n  #{code}" if logger
389
        raise
390
      end
391

392
      alias_method :set_prefix, :prefix=  #:nodoc:
393

394 395
      alias_method :set_element_name, :element_name=  #:nodoc:
      alias_method :set_collection_name, :collection_name=  #:nodoc:
396

397
      # Gets the element path for the given ID in +id+.  If the +query_options+ parameter is omitted, Rails
P
Pratik Naik 已提交
398
      # will split from the \prefix options.
399 400
      #
      # ==== Options
P
Pratik Naik 已提交
401
      # +prefix_options+ - A \hash to add a \prefix to the request for nested URLs (e.g., <tt>:account_id => 19</tt>
402
      #                    would yield a URL like <tt>/accounts/19/purchases.xml</tt>).
P
Pratik Naik 已提交
403
      # +query_options+ - A \hash to add items to the query string for the request.
404 405
      #
      # ==== Examples
406
      #   Post.element_path(1)
407 408
      #   # => /posts/1.xml
      #
409
      #   Comment.element_path(1, :post_id => 5)
410 411
      #   # => /posts/5/comments/1.xml
      #
412
      #   Comment.element_path(1, :post_id => 5, :active => 1)
413 414
      #   # => /posts/5/comments/1.xml?active=1
      #
415
      #   Comment.element_path(1, {:post_id => 5}, {:active => 1})
416
      #   # => /posts/5/comments/1.xml?active=1
417 418 419
      #
      def element_path(id, prefix_options = {}, query_options = nil)
        prefix_options, query_options = split_options(prefix_options) if query_options.nil?
420
        "#{prefix(prefix_options)}#{collection_name}/#{id}.#{format.extension}#{query_string(query_options)}"
421
      end
422

423 424 425 426
      # Gets the collection path for the REST resources.  If the +query_options+ parameter is omitted, Rails
      # will split from the +prefix_options+.
      #
      # ==== Options
P
Pratik Naik 已提交
427
      # * +prefix_options+ - A hash to add a prefix to the request for nested URLs (e.g., <tt>:account_id => 19</tt>
P
Pratik Naik 已提交
428 429
      #   would yield a URL like <tt>/accounts/19/purchases.xml</tt>).
      # * +query_options+ - A hash to add items to the query string for the request.
430 431 432 433 434
      #
      # ==== Examples
      #   Post.collection_path
      #   # => /posts.xml
      #
435
      #   Comment.collection_path(:post_id => 5)
436 437
      #   # => /posts/5/comments.xml
      #
438
      #   Comment.collection_path(:post_id => 5, :active => 1)
439 440
      #   # => /posts/5/comments.xml?active=1
      #
441
      #   Comment.collection_path({:post_id => 5}, {:active => 1})
442
      #   # => /posts/5/comments.xml?active=1
443 444 445
      #
      def collection_path(prefix_options = {}, query_options = nil)
        prefix_options, query_options = split_options(prefix_options) if query_options.nil?
446
        "#{prefix(prefix_options)}#{collection_name}.#{format.extension}#{query_string(query_options)}"
447
      end
448

449
      alias_method :set_primary_key, :primary_key=  #:nodoc:
450

P
Pratik Naik 已提交
451
      # Creates a new resource instance and makes a request to the remote service
452
      # that it be saved, making it equivalent to the following simultaneous calls:
453 454 455 456
      #
      #   ryan = Person.new(:first => 'ryan')
      #   ryan.save
      #
P
Pratik Naik 已提交
457 458 459 460
      # Returns the newly created resource.  If a failure has occurred an
      # exception will be raised (see <tt>save</tt>).  If the resource is invalid and
      # has not been saved then <tt>valid?</tt> will return <tt>false</tt>,
      # while <tt>new?</tt> will still return <tt>true</tt>.
461 462 463 464
      #
      # ==== Examples
      #   Person.create(:name => 'Jeremy', :email => 'myname@nospam.com', :enabled => true)
      #   my_person = Person.find(:first)
P
Pratik Naik 已提交
465
      #   my_person.email # => myname@nospam.com
466 467
      #
      #   dhh = Person.create(:name => 'David', :email => 'dhh@nospam.com', :enabled => true)
P
Pratik Naik 已提交
468 469
      #   dhh.valid? # => true
      #   dhh.new?   # => false
470 471 472
      #
      #   # We'll assume that there's a validation that requires the name attribute
      #   that_guy = Person.create(:name => '', :email => 'thatguy@nospam.com', :enabled => true)
P
Pratik Naik 已提交
473 474
      #   that_guy.valid? # => false
      #   that_guy.new?   # => true
475
      def create(attributes = {})
J
Jeremy Kemper 已提交
476
        self.new(attributes).tap { |resource| resource.save }
477 478
      end

P
Pratik Naik 已提交
479
      # Core method for finding resources.  Used similarly to Active Record's +find+ method.
480
      #
481
      # ==== Arguments
482
      # The first argument is considered to be the scope of the query.  That is, how many
483 484
      # resources are returned from the request.  It can be one of the following.
      #
P
Pratik Naik 已提交
485 486
      # * <tt>:one</tt> - Returns a single resource.
      # * <tt>:first</tt> - Returns the first resource found.
487
      # * <tt>:last</tt> - Returns the last resource found.
P
Pratik Naik 已提交
488
      # * <tt>:all</tt> - Returns every resource that matches the request.
489
      #
490
      # ==== Options
491
      #
P
Pratik Naik 已提交
492
      # * <tt>:from</tt> - Sets the path or custom method that resources will be fetched from.
P
Pratik Naik 已提交
493
      # * <tt>:params</tt> - Sets query and \prefix (nested URL) parameters.
494 495
      #
      # ==== Examples
496
      #   Person.find(1)
497 498
      #   # => GET /people/1.xml
      #
499
      #   Person.find(:all)
500 501
      #   # => GET /people.xml
      #
502
      #   Person.find(:all, :params => { :title => "CEO" })
503 504
      #   # => GET /people.xml?title=CEO
      #
505 506 507 508
      #   Person.find(:first, :from => :managers)
      #   # => GET /people/managers.xml
      #
      #   Person.find(:last, :from => :managers)
509 510
      #   # => GET /people/managers.xml
      #
511
      #   Person.find(:all, :from => "/companies/1/people.xml")
512 513
      #   # => GET /companies/1/people.xml
      #
514
      #   Person.find(:one, :from => :leader)
515 516
      #   # => GET /people/leader.xml
      #
517 518 519
      #   Person.find(:all, :from => :developers, :params => { :language => 'ruby' })
      #   # => GET /people/developers.xml?language=ruby
      #
520
      #   Person.find(:one, :from => "/companies/1/manager.xml")
521 522
      #   # => GET /companies/1/manager.xml
      #
523
      #   StreetAddress.find(1, :params => { :person_id => 1 })
524
      #   # => GET /people/1/street_addresses/1.xml
525
      def find(*arguments)
526 527
        scope   = arguments.slice!(0)
        options = arguments.slice!(0) || {}
528 529

        case scope
530 531
          when :all   then find_every(options)
          when :first then find_every(options).first
532
          when :last  then find_every(options).last
533
          when :one   then find_one(options)
534
          else             find_single(scope, options)
535 536
        end
      end
537

538 539 540
      # Deletes the resources with the ID in the +id+ parameter.
      #
      # ==== Options
P
Pratik Naik 已提交
541
      # All options specify \prefix and query parameters.
542 543
      #
      # ==== Examples
P
Pratik Naik 已提交
544
      #   Event.delete(2) # sends DELETE /events/2
545 546
      #
      #   Event.create(:name => 'Free Concert', :location => 'Community Center')
P
Pratik Naik 已提交
547 548
      #   my_event = Event.find(:first) # let's assume this is event with ID 7
      #   Event.delete(my_event.id) # sends DELETE /events/7
549 550
      #
      #   # Let's assume a request to events/5/cancel.xml
P
Pratik Naik 已提交
551
      #   Event.delete(params[:id]) # sends DELETE /events/5
552 553
      def delete(id, options = {})
        connection.delete(element_path(id, options))
554 555
      end

556 557 558 559
      # Asserts the existence of a resource, returning <tt>true</tt> if the resource is found.
      #
      # ==== Examples
      #   Note.create(:title => 'Hello, world.', :body => 'Nothing more for now...')
P
Pratik Naik 已提交
560
      #   Note.exists?(1) # => true
561
      #
P
Pratik Naik 已提交
562
      #   Note.exists(1349) # => false
563
      def exists?(id, options = {})
564 565 566 567
        if id
          prefix_options, query_options = split_options(options[:params])
          path = element_path(id, prefix_options, query_options)
          response = connection.head(path, headers)
568
          response.code.to_i == 200
569 570
        end
        # id && !find_single(id, options).nil?
571 572 573 574
      rescue ActiveResource::ResourceNotFound
        false
      end

575
      private
576
        # Find every resource
577
        def find_every(options)
578
          case from = options[:from]
579 580 581 582 583 584 585 586
          when Symbol
            instantiate_collection(get(from, options[:params]))
          when String
            path = "#{from}#{query_string(options[:params])}"
            instantiate_collection(connection.get(path, headers) || [])
          else
            prefix_options, query_options = split_options(options[:params])
            path = collection_path(prefix_options, query_options)
587
            instantiate_collection( (connection.get(path, headers) || []), prefix_options )
588
          end
589
        end
590

591 592 593 594 595 596 597 598
        # Find a single resource from a one-off URL
        def find_one(options)
          case from = options[:from]
          when Symbol
            instantiate_record(get(from, options[:params]))
          when String
            path = "#{from}#{query_string(options[:params])}"
            instantiate_record(connection.get(path, headers))
599
          end
600
        end
601

602
        # Find a single resource from the default URL
603
        def find_single(scope, options)
604 605 606 607
          prefix_options, query_options = split_options(options[:params])
          path = element_path(scope, prefix_options, query_options)
          instantiate_record(connection.get(path, headers), prefix_options)
        end
608

609 610 611
        def instantiate_collection(collection, prefix_options = {})
          collection.collect! { |record| instantiate_record(record, prefix_options) }
        end
612

613
        def instantiate_record(record, prefix_options = {})
J
Jeremy Kemper 已提交
614
          new(record).tap do |resource|
615 616
            resource.prefix_options = prefix_options
          end
617
        end
618

619

620
        # Accepts a URI and creates the site URI from that.
621
        def create_site_uri_from(site)
622
          site.is_a?(URI) ? site.dup : URI.parse(site)
623
        end
624

625
        # contains a set of the current prefix parameters.
626 627 628 629
        def prefix_parameters
          @prefix_parameters ||= prefix_source.scan(/:\w+/).map { |key| key[1..-1].to_sym }.to_set
        end

630
        # Builds the query string for the request.
631
        def query_string(options)
632
          "?#{options.to_query}" unless options.nil? || options.empty?
633 634
        end

635
        # split an option hash into two hashes, one containing the prefix options,
636 637
        # and the other containing the leftovers.
        def split_options(options = {})
638 639 640
          prefix_options, query_options = {}, {}

          (options || {}).each do |key, value|
641 642
            next if key.blank?
            (prefix_parameters.include?(key.to_sym) ? prefix_options : query_options)[key.to_sym] = value
643
          end
644 645

          [ prefix_options, query_options ]
646
        end
647 648
    end

649 650
    attr_accessor :attributes #:nodoc:
    attr_accessor :prefix_options #:nodoc:
651

P
Pratik Naik 已提交
652 653
    # Constructor method for \new resources; the optional +attributes+ parameter takes a \hash
    # of attributes for the \new resource.
654 655 656 657 658 659 660 661 662
    #
    # ==== Examples
    #   my_course = Course.new
    #   my_course.name = "Western Civilization"
    #   my_course.lecturer = "Don Trotter"
    #   my_course.save
    #
    #   my_other_course = Course.new(:name => "Philosophy: Reason and Being", :lecturer => "Ralph Cling")
    #   my_other_course.save
663 664 665
    def initialize(attributes = {})
      @attributes     = {}
      @prefix_options = {}
666
      load(attributes)
667
    end
668

P
Pratik Naik 已提交
669 670
    # Returns a \clone of the resource that hasn't been assigned an +id+ yet and
    # is treated as a \new resource.
671
    #
P
Pratik Naik 已提交
672 673 674
    #   ryan = Person.find(1)
    #   not_ryan = ryan.clone
    #   not_ryan.new?  # => true
675 676
    #
    # Any active resource member attributes will NOT be cloned, though all other
P
Pratik Naik 已提交
677
    # attributes are.  This is to prevent the conflict between any +prefix_options+
678 679 680
    # that refer to the original parent resource and the newly cloned parent
    # resource that does not exist.
    #
P
Pratik Naik 已提交
681 682 683
    #   ryan = Person.find(1)
    #   ryan.address = StreetAddress.find(1, :person_id => ryan.id)
    #   ryan.hash = {:not => "an ARes instance"}
684
    #
P
Pratik Naik 已提交
685 686 687 688
    #   not_ryan = ryan.clone
    #   not_ryan.new?            # => true
    #   not_ryan.address         # => NoMethodError
    #   not_ryan.hash            # => {:not => "an ARes instance"}
689 690
    def clone
      # Clone all attributes except the pk and any nested ARes
J
Jeremy Kemper 已提交
691
      cloned = attributes.reject {|k,v| k == self.class.primary_key || v.is_a?(ActiveResource::Base)}.inject({}) do |attrs, (k, v)|
692 693 694 695 696 697 698 699
        attrs[k] = v.clone
        attrs
      end
      # Form the new resource - bypass initialize of resource with 'new' as that will call 'load' which
      # attempts to convert hashes into member objects and arrays into collections of objects.  We want
      # the raw objects to be cloned so we bypass load by directly setting the attributes hash.
      resource = self.class.new({})
      resource.prefix_options = self.prefix_options
J
Jeremy Kemper 已提交
700
      resource.send :instance_variable_set, '@attributes', cloned
701 702 703 704
      resource
    end


P
Pratik Naik 已提交
705
    # Returns +true+ if this object hasn't yet been saved, otherwise, returns +false+.
706 707 708
    #
    # ==== Examples
    #   not_new = Computer.create(:brand => 'Apple', :make => 'MacBook', :vendor => 'MacMall')
P
Pratik Naik 已提交
709
    #   not_new.new? # => false
710 711
    #
    #   is_new = Computer.new(:brand => 'IBM', :make => 'Thinkpad', :vendor => 'IBM')
P
Pratik Naik 已提交
712
    #   is_new.new? # => true
713 714
    #
    #   is_new.save
P
Pratik Naik 已提交
715
    #   is_new.new? # => false
716
    #
717
    def new?
718 719
      id.nil?
    end
720
    alias :new_record? :new?
721

P
Pratik Naik 已提交
722
    # Gets the <tt>\id</tt> attribute of the resource.
723
    def id
724
      attributes[self.class.primary_key]
725
    end
726

P
Pratik Naik 已提交
727
    # Sets the <tt>\id</tt> attribute of the resource.
728
    def id=(id)
729
      attributes[self.class.primary_key] = id
730
    end
731

P
Pratik Naik 已提交
732
    # Allows Active Resource objects to be used as parameters in Action Pack URL generation.
733 734 735 736
    def to_param
      id && id.to_s
    end

737
    # Test for equality.  Resource are equal if and only if +other+ is the same object or
P
Pratik Naik 已提交
738
    # is an instance of the same class, is not <tt>new?</tt>, and has the same +id+.
739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758
    #
    # ==== Examples
    #   ryan = Person.create(:name => 'Ryan')
    #   jamie = Person.create(:name => 'Jamie')
    #
    #   ryan == jamie
    #   # => false (Different name attribute and id)
    #
    #   ryan_again = Person.new(:name => 'Ryan')
    #   ryan == ryan_again
    #   # => false (ryan_again is new?)
    #
    #   ryans_clone = Person.create(:name => 'Ryan')
    #   ryan == ryans_clone
    #   # => false (Different id attributes)
    #
    #   ryans_twin = Person.find(ryan.id)
    #   ryan == ryans_twin
    #   # => true
    #
759
    def ==(other)
760
      other.equal?(self) || (other.instance_of?(self.class) && other.id == id && other.prefix_options == prefix_options)
J
Jeremy Kemper 已提交
761
    end
762

763
    # Tests for equality (delegates to ==).
764 765 766 767
    def eql?(other)
      self == other
    end

P
Pratik Naik 已提交
768
    # Delegates to id in order to allow two resources of the same type and \id to work with something like:
769 770 771 772
    #   [Person.find(1), Person.find(2)] & [Person.find(1), Person.find(4)] # => [Person.find(1)]
    def hash
      id.hash
    end
773

P
Pratik Naik 已提交
774
    # Duplicates the current resource without saving it.
775 776 777 778
    #
    # ==== Examples
    #   my_invoice = Invoice.create(:customer => 'That Company')
    #   next_invoice = my_invoice.dup
P
Pratik Naik 已提交
779
    #   next_invoice.new? # => true
780 781
    #
    #   next_invoice.save
P
Pratik Naik 已提交
782
    #   next_invoice == my_invoice # => false (different id attributes)
783
    #
P
Pratik Naik 已提交
784 785
    #   my_invoice.customer   # => That Company
    #   next_invoice.customer # => That Company
786
    def dup
J
Jeremy Kemper 已提交
787
      self.class.new.tap do |resource|
788 789 790 791
        resource.attributes     = @attributes
        resource.prefix_options = @prefix_options
      end
    end
792

P
Pratik Naik 已提交
793 794
    # Saves (+POST+) or \updates (+PUT+) a resource.  Delegates to +create+ if the object is \new,
    # +update+ if it exists. If the response to the \save includes a body, it will be assumed that this body
P
Pratik Naik 已提交
795
    # is XML for the final object as it looked after the \save (which would include attributes like +created_at+
796 797 798 799
    # that weren't part of the original submit).
    #
    # ==== Examples
    #   my_company = Company.new(:name => 'RoleModel Software', :owner => 'Ken Auer', :size => 2)
P
Pratik Naik 已提交
800 801
    #   my_company.new? # => true
    #   my_company.save # sends POST /companies/ (create)
802
    #
P
Pratik Naik 已提交
803
    #   my_company.new? # => false
804
    #   my_company.size = 10
P
Pratik Naik 已提交
805
    #   my_company.save # sends PUT /companies/1 (update)
806
    def save
807
      new? ? create : update
808 809
    end

810 811 812 813 814 815
    # Deletes the resource from the remote service.
    #
    # ==== Examples
    #   my_id = 3
    #   my_person = Person.find(my_id)
    #   my_person.destroy
P
Pratik Naik 已提交
816
    #   Person.find(my_id) # 404 (Resource Not Found)
817
    #
818
    #   new_person = Person.create(:name => 'James')
P
Pratik Naik 已提交
819
    #   new_id = new_person.id # => 7
820
    #   new_person.destroy
P
Pratik Naik 已提交
821
    #   Person.find(new_id) # 404 (Resource Not Found)
822
    def destroy
823
      connection.delete(element_path, self.class.headers)
824
    end
825

P
Pratik Naik 已提交
826
    # Evaluates to <tt>true</tt> if this resource is not <tt>new?</tt> and is
827 828 829 830 831 832 833
    # found on the remote service.  Using this method, you can check for
    # resources that may have been deleted between the object's instantiation
    # and actions on it.
    #
    # ==== Examples
    #   Person.create(:name => 'Theodore Roosevelt')
    #   that_guy = Person.find(:first)
P
Pratik Naik 已提交
834
    #   that_guy.exists? # => true
835 836
    #
    #   that_lady = Person.new(:name => 'Paul Bean')
P
Pratik Naik 已提交
837
    #   that_lady.exists? # => false
838 839 840
    #
    #   guys_id = that_guy.id
    #   Person.delete(guys_id)
P
Pratik Naik 已提交
841
    #   that_guy.exists? # => false
842
    def exists?
P
Pratik Naik 已提交
843
      !new? && self.class.exists?(to_param, :params => prefix_options)
844 845
    end

P
Pratik Naik 已提交
846
    # Converts the resource to an XML string representation.
847 848 849 850
    #
    # ==== Options
    # The +options+ parameter is handed off to the +to_xml+ method on each
    # attribute, so it has the same options as the +to_xml+ methods in
P
Pratik Naik 已提交
851
    # Active Support.
852
    #
P
Pratik Naik 已提交
853 854 855
    # * <tt>:indent</tt> - Set the indent level for the XML output (default is +2+).
    # * <tt>:dasherize</tt> - Boolean option to determine whether or not element names should
    #   replace underscores with dashes (default is <tt>false</tt>).
856
    # * <tt>:skip_instruct</tt> - Toggle skipping the +instruct!+ call on the XML builder
P
Pratik Naik 已提交
857
    #   that generates the XML declaration (default is <tt>false</tt>).
858 859 860 861 862 863 864 865 866 867 868 869 870
    #
    # ==== Examples
    #   my_group = SubsidiaryGroup.find(:first)
    #   my_group.to_xml
    #   # => <?xml version="1.0" encoding="UTF-8"?>
    #   #    <subsidiary_group> [...] </subsidiary_group>
    #
    #   my_group.to_xml(:dasherize => true)
    #   # => <?xml version="1.0" encoding="UTF-8"?>
    #   #    <subsidiary-group> [...] </subsidiary-group>
    #
    #   my_group.to_xml(:skip_instruct => true)
    #   # => <subsidiary_group> [...] </subsidiary_group>
871 872 873 874
    def to_xml(options={})
      attributes.to_xml({:root => self.class.element_name}.merge(options))
    end

875
    # Coerces to a hash for JSON encoding.
876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898
    #
    # ==== Options
    # The +options+ are passed to the +to_json+ method on each
    # attribute, so the same options as the +to_json+ methods in
    # Active Support.
    #
    # * <tt>:only</tt> - Only include the specified attribute or list of
    #   attributes in the serialized output. Attribute names must be specified
    #   as strings.
    # * <tt>:except</tt> - Do not include the specified attribute or list of
    #   attributes in the serialized output. Attribute names must be specified
    #   as strings.
    #
    # ==== Examples
    #   person = Person.new(:first_name => "Jim", :last_name => "Smith")
    #   person.to_json
    #   # => {"first_name": "Jim", "last_name": "Smith"}
    #
    #   person.to_json(:only => ["first_name"])
    #   # => {"first_name": "Jim"}
    #
    #   person.to_json(:except => ["first_name"])
    #   # => {"last_name": "Smith"}
899 900
    def as_json(options = nil)
      attributes.as_json(options)
901 902 903 904 905
    end

    # Returns the serialized string representation of the resource in the configured
    # serialization format specified in ActiveResource::Base.format. The options
    # applicable depend on the configured encoding format.
906 907
    def encode(options={})
      case self.class.format
908
        when ActiveResource::Formats::XmlFormat
909 910 911 912
          self.class.format.encode(attributes, {:root => self.class.element_name}.merge(options))
        else
          self.class.format.encode(attributes, options)
      end
913
    end
914

P
Pratik Naik 已提交
915
    # A method to \reload the attributes of this object from the remote web service.
916 917 918
    #
    # ==== Examples
    #   my_branch = Branch.find(:first)
P
Pratik Naik 已提交
919
    #   my_branch.name # => "Wislon Raod"
920
    #
921 922
    #   # Another client fixes the typo...
    #
P
Pratik Naik 已提交
923
    #   my_branch.name # => "Wislon Raod"
924
    #   my_branch.reload
P
Pratik Naik 已提交
925
    #   my_branch.name # => "Wilson Road"
926
    def reload
927
      self.load(self.class.find(to_param, :params => @prefix_options).attributes)
928 929
    end

P
Pratik Naik 已提交
930 931
    # A method to manually load attributes from a \hash. Recursively loads collections of
    # resources.  This method is called in +initialize+ and +create+ when a \hash of attributes
932 933 934 935
    # is provided.
    #
    # ==== Examples
    #   my_attrs = {:name => 'J&J Textiles', :industry => 'Cloth and textiles'}
936
    #   my_attrs = {:name => 'Marty', :colors => ["red", "green", "blue"]}
937 938
    #
    #   the_supplier = Supplier.find(:first)
P
Pratik Naik 已提交
939
    #   the_supplier.name # => 'J&M Textiles'
940 941 942 943 944 945 946 947 948 949 950
    #   the_supplier.load(my_attrs)
    #   the_supplier.name('J&J Textiles')
    #
    #   # These two calls are the same as Supplier.new(my_attrs)
    #   my_supplier = Supplier.new
    #   my_supplier.load(my_attrs)
    #
    #   # These three calls are the same as Supplier.create(my_attrs)
    #   your_supplier = Supplier.new
    #   your_supplier.load(my_attrs)
    #   your_supplier.save
951
    def load(attributes)
J
Jeremy Kemper 已提交
952
      raise ArgumentError, "expected an attributes Hash, got #{attributes.inspect}" unless attributes.is_a?(Hash)
953
      @prefix_options, attributes = split_options(attributes)
954 955 956 957 958
      attributes.each do |key, value|
        @attributes[key.to_s] =
          case value
            when Array
              resource = find_or_create_resource_for_collection(key)
959
              value.map { |attrs| attrs.is_a?(String) ? attrs.dup : resource.new(attrs) }
960
            when Hash
961 962
              resource = find_or_create_resource_for(key)
              resource.new(value)
963 964 965 966
            else
              value.dup rescue value
          end
      end
967 968
      self
    end
969

P
Pratik Naik 已提交
970
    # For checking <tt>respond_to?</tt> without searching the attributes (which is faster).
971 972
    alias_method :respond_to_without_attributes?, :respond_to?

P
Pratik Naik 已提交
973
    # A method to determine if an object responds to a message (e.g., a method call). In Active Resource, a Person object with a
974 975
    # +name+ attribute can answer <tt>true</tt> to <tt>my_person.respond_to?(:name)</tt>, <tt>my_person.respond_to?(:name=)</tt>, and
    # <tt>my_person.respond_to?(:name?)</tt>.
976 977 978 979 980
    def respond_to?(method, include_priv = false)
      method_name = method.to_s
      if attributes.nil?
        return super
      elsif attributes.has_key?(method_name)
981
        return true
982 983 984 985 986 987 988
      elsif ['?','='].include?(method_name.last) && attributes.has_key?(method_name.first(-1))
        return true
      end
      # super must be called at the end of the method, because the inherited respond_to?
      # would return true for generated readers, even if the attribute wasn't present
      super
    end
989

990

991 992 993 994
    protected
      def connection(refresh = false)
        self.class.connection(refresh)
      end
995

996
      # Update the resource on the remote service.
997
      def update
J
Jeremy Kemper 已提交
998
        connection.put(element_path(prefix_options), encode, self.class.headers).tap do |response|
999 1000
          load_attributes_from_response(response)
        end
1001
      end
1002

P
Pratik Naik 已提交
1003
      # Create (i.e., \save to the remote service) the \new resource.
1004
      def create
J
Jeremy Kemper 已提交
1005
        connection.post(collection_path, encode, self.class.headers).tap do |response|
1006
          self.id = id_from_response(response)
1007
          load_attributes_from_response(response)
1008
        end
1009
      end
1010

1011
      def load_attributes_from_response(response)
1012
        if response['Content-Length'] != "0" && response.body.strip.size > 0
1013
          load(self.class.format.decode(response.body))
1014
        end
1015
      end
1016

1017
      # Takes a response from a typical create post and pulls the ID out
1018
      def id_from_response(response)
1019
        response['Location'][/\/([^\/]*?)(\.\w+)?$/, 1] if response['Location']
1020 1021
      end

1022
      def element_path(options = nil)
1023
        self.class.element_path(to_param, options || prefix_options)
1024 1025 1026 1027 1028 1029
      end

      def collection_path(options = nil)
        self.class.collection_path(options || prefix_options)
      end

1030
    private
1031
      # Tries to find a resource for a given collection name; if it fails, then the resource is created
1032
      def find_or_create_resource_for_collection(name)
J
Jeremy Kemper 已提交
1033
        find_or_create_resource_for(ActiveSupport::Inflector.singularize(name.to_s))
1034
      end
1035

1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049
      # Tries to find a resource in a non empty list of nested modules
      # Raises a NameError if it was not found in any of the given nested modules
      def find_resource_in_modules(resource_name, module_names)
        receiver = Object
        namespaces = module_names[0, module_names.size-1].map do |module_name|
          receiver = receiver.const_get(module_name)
        end
        if namespace = namespaces.reverse.detect { |ns| ns.const_defined?(resource_name) }
          return namespace.const_get(resource_name)
        else
          raise NameError
        end
      end

1050
      # Tries to find a resource for a given name; if it fails, then the resource is created
1051 1052
      def find_or_create_resource_for(name)
        resource_name = name.to_s.camelize
1053 1054 1055
        ancestors = self.class.name.split("::")
        if ancestors.size > 1
          find_resource_in_modules(resource_name, ancestors)
1056 1057 1058
        else
          self.class.const_get(resource_name)
        end
1059
      rescue NameError
1060 1061 1062 1063 1064
        if self.class.const_defined?(resource_name)
          resource = self.class.const_get(resource_name)
        else
          resource = self.class.const_set(resource_name, Class.new(ActiveResource::Base))
        end
1065
        resource.prefix = self.class.prefix
1066
        resource.site   = self.class.site
1067 1068 1069
        resource
      end

1070
      def split_options(options = {})
1071
        self.class.__send__(:split_options, options)
1072 1073
      end

1074
      def method_missing(method_symbol, *arguments) #:nodoc:
1075
        method_name = method_symbol.to_s
1076

1077 1078 1079 1080
        case method_name.last
          when "="
            attributes[method_name.first(-1)] = arguments.first
          when "?"
T
 
Tobias Lütke 已提交
1081
            attributes[method_name.first(-1)]
1082
          else
1083
            attributes.has_key?(method_name) ? attributes[method_name] : super
1084 1085 1086
        end
      end
  end
1087

J
Joshua Peek 已提交
1088 1089 1090 1091
  class Base
    include CustomMethods, Validations
  end
end