base.rb 38.6 KB
Newer Older
1
require 'active_resource/connection'
2 3
require 'cgi'
require 'set'
4 5

module ActiveResource
6 7
  # ActiveResource::Base is the main class for mapping RESTful resources as models in a Rails application.
  #
8
  # For an outline of what Active Resource is capable of, see link:files/vendor/rails/activeresource/README.html.
9 10 11 12 13 14 15
  #
  # == 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.
16
  #
P
Pratik Naik 已提交
17 18 19
  #   class Person < ActiveResource::Base
  #     self.site = "http://api.people.com:3000/"
  #   end
20
  #
21
  # Now the Person class is mapped to RESTful resources located at <tt>http://api.people.com:3000/people/</tt>, and
22
  # you can now use Active Resource's lifecycles methods to manipulate resources. In the case where you already have
P
Pratik Naik 已提交
23 24 25 26 27 28
  # 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
29 30
  #
  #
31 32 33 34
  # == Lifecycle methods
  #
  # Active Resource exposes methods for creating, finding, updating, and deleting resources
  # from REST web services.
35
  #
36
  #   ryan = Person.new(:first => 'Ryan', :last => 'Daigle')
P
Pratik Naik 已提交
37 38 39 40
  #   ryan.save                # => true
  #   ryan.id                  # => 2
  #   Person.exists?(ryan.id)  # => true
  #   ryan.exists?             # => true
41
  #
42
  #   ryan = Person.find(1)
P
Pratik Naik 已提交
43
  #   # Resource holding our newly created Person object
44
  #
45
  #   ryan.first = 'Rizzle'
P
Pratik Naik 已提交
46
  #   ryan.save                # => true
47
  #
P
Pratik Naik 已提交
48
  #   ryan.destroy             # => true
49 50 51
  #
  # 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.
52
  #
53 54 55
  # === Custom REST methods
  #
  # Since simple CRUD/lifecycle methods can't accomplish every task, Active Resource also supports
56
  # defining your own custom REST methods. To invoke them, Active Resource provides the <tt>get</tt>,
57
  # <tt>post</tt>, <tt>put</tt> and <tt>delete</tt> methods where you can specify a custom REST method
58 59 60 61
  # name to invoke.
  #
  #   # POST to the custom 'register' REST method, i.e. POST /people/new/register.xml.
  #   Person.new(:name => 'Ryan').post(:register)
62 63
  #   # => { :id => 1, :name => 'Ryan', :position => 'Clerk' }
  #
64
  #   # PUT an update by invoking the 'promote' REST method, i.e. PUT /people/1/promote.xml?position=Manager.
65
  #   Person.find(1).put(:promote, :position => 'Manager')
66
  #   # => { :id => 1, :name => 'Ryan', :position => 'Manager' }
67 68 69 70 71 72 73
  #
  #   # 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)
74
  #
75
  # For more information on using custom REST methods, see the
76 77 78 79 80
  # ActiveResource::CustomMethods documentation.
  #
  # == Validations
  #
  # You can validate resources client side by overriding validation methods in the base class.
81
  #
P
Pratik Naik 已提交
82 83 84 85 86 87 88
  #   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
89
  #
90 91 92
  # See the ActiveResource::Validations documentation for more information.
  #
  # == Authentication
93
  #
94
  # Many REST APIs will require authentication, usually in the form of basic
95 96
  # HTTP authentication.  Authentication can be specified by:
  # * putting the credentials in the URL for the +site+ variable.
97
  #
98 99 100
  #    class Person < ActiveResource::Base
  #      self.site = "http://ryan:password@api.people.com:3000/"
  #    end
101
  #
102 103 104 105 106 107 108 109
  # * defining +user+ and/or +password+ variables
  #
  #    class Person < ActiveResource::Base
  #      self.site = "http://api.people.com:3000/"
  #      self.user = "ryan"
  #      self.password = "password"
  #    end
  #
110
  # For obvious security reasons, it is probably best if such services are available
111
  # over HTTPS.
112 113
  #
  # Note: Some values cannot be provided in the URL passed to site.  e.g. email addresses
P
Pratik Naik 已提交
114
  # as usernames.  In those situations you should use the separate user and password option.
115 116 117
  # == Errors & Validation
  #
  # Error handling and validation is handled in much the same manner as you're used to seeing in
118
  # Active Record.  Both the response code in the HTTP response and the body of the response are used to
119
  # indicate that an error occurred.
120
  #
121
  # === Resource errors
122
  #
123
  # When a GET is requested for a resource that does not exist, the HTTP <tt>404</tt> (Resource Not Found)
124 125
  # response code will be returned from the server which will raise an ActiveResource::ResourceNotFound
  # exception.
126
  #
127
  #   # GET http://api.people.com:3000/people/999.xml
P
Pratik Naik 已提交
128
  #   ryan = Person.find(999) # 404, raises ActiveResource::ResourceNotFound
129
  #
P
Pratik Naik 已提交
130
  # <tt>404</tt> is just one of the HTTP error response codes that Active Resource will handle with its own exception. The
131
  # following HTTP response codes will also result in these exceptions:
132
  #
P
Pratik Naik 已提交
133 134 135 136 137 138
  # * 200..399 - Valid response, no exception
  # * 404 - ActiveResource::ResourceNotFound
  # * 409 - ActiveResource::ResourceConflict
  # * 422 - ActiveResource::ResourceInvalid (rescued by save as validation errors)
  # * 401..499 - ActiveResource::ClientError
  # * 500..599 - ActiveResource::ServerError
139 140 141 142 143 144 145 146 147 148 149 150 151
  #
  # 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
152
  #
153
  # Active Resource supports validations on resources and will return errors if any these validations fail
154
  # (e.g., "First name can not be blank" and so on).  These types of errors are denoted in the response by
155 156
  # 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.
157
  #
158
  #   ryan = Person.find(1)
P
Pratik Naik 已提交
159 160
  #   ryan.first # => ''
  #   ryan.save  # => false
161
  #
162
  #   # When
163 164 165 166
  #   # PUT http://api.people.com:3000/people/1.xml
  #   # is requested with invalid values, the response is:
  #   #
  #   # Response (422):
167
  #   # <errors type="array"><error>First cannot be empty</error></errors>
168 169
  #   #
  #
P
Pratik Naik 已提交
170 171
  #   ryan.errors.invalid?(:first)  # => true
  #   ryan.errors.full_messages     # => ['First cannot be empty']
172
  #
173 174
  # Learn more about Active Resource's validation features in the ActiveResource::Validations documentation.
  #
175 176 177 178 179 180 181 182 183 184 185
  # === Timeouts
  #
  # Active Resource relies on HTTP to access RESTful APIs and as such is inherently susceptible to slow or
  # unresponsive servers. In such cases, your Active Resource method calls could timeout. You can control the
  # 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 已提交
186
  # This sets the +timeout+ to 5 seconds. You can adjust the +timeout+ to a value suitable for the RESTful API
187 188 189 190 191
  # 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 已提交
192 193 194
  # When a timeout occurs, an ActiveResource::TimeoutError is raised. You should rescue from
  # ActiveResource::TimeoutError in your Active Resource method calls.
  #
195 196 197
  # 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.
198
  class Base
199
    # The logger for diagnosing and tracing Active Resource calls.
200 201
    cattr_accessor :logger

202
    class << self
P
Pratik Naik 已提交
203
      # Gets the URI of the REST resources to map for this class.  The site variable is required for
P
Pratik Naik 已提交
204
      # Active Resource's mapping to work.
205
      def site
206 207 208 209 210 211 212 213 214 215 216 217 218 219 220
        # 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
        #
221 222
        if defined?(@site)
          @site
223
        elsif superclass != Object && superclass.site
224 225 226
          superclass.site.dup.freeze
        end
      end
J
Jeremy Kemper 已提交
227

228
      # Sets the URI of the REST resources to map for this class to the value in the +site+ argument.
P
Pratik Naik 已提交
229
      # The site variable is required for Active Resource's mapping to work.
230
      def site=(site)
231
        @connection = nil
232 233 234 235
        if site.nil?
          @site = nil
        else
          @site = create_site_uri_from(site)
236 237
          @user = URI.decode(@site.user) if @site.user
          @password = URI.decode(@site.password) if @site.password
238 239 240
        end
      end

P
Pratik Naik 已提交
241
      # Gets the user for REST HTTP authentication.
242 243 244 245 246 247 248 249 250
      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 已提交
251
      # Sets the user for REST HTTP authentication.
252 253 254 255 256
      def user=(user)
        @connection = nil
        @user = user
      end

P
Pratik Naik 已提交
257
      # Gets the password for REST HTTP authentication.
258 259 260 261 262 263 264 265 266
      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 已提交
267
      # Sets the password for REST HTTP authentication.
268 269 270
      def password=(password)
        @connection = nil
        @password = password
271 272
      end

P
Pratik Naik 已提交
273
      # Sets the format that attributes are sent and received in from a mime type reference:
274 275 276 277 278 279 280
      #
      #   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 已提交
281
      # Default format is <tt>:xml</tt>.
282
      def format=(mime_type_reference_or_format)
283
        format = mime_type_reference_or_format.is_a?(Symbol) ?
284 285 286
          ActiveResource::Formats[mime_type_reference_or_format] : mime_type_reference_or_format

        write_inheritable_attribute("format", format)
287
        connection.format = format if site
288
      end
289

P
Pratik Naik 已提交
290
      # Returns the current format, default is ActiveResource::Formats::XmlFormat.
P
Pratik Naik 已提交
291
      def format
292 293 294
        read_inheritable_attribute("format") || ActiveResource::Formats[:xml]
      end

295 296 297 298 299 300
      # 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 已提交
301
      # Gets the number of seconds after which requests to the REST API should time out.
302 303 304 305 306 307 308 309
      def timeout
        if defined?(@timeout)
          @timeout
        elsif superclass != Object && superclass.timeout
          superclass.timeout
        end
      end

310 311
      # 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
312
      # or not (defaults to <tt>false</tt>).
313
      def connection(refresh = false)
314 315
        if defined?(@connection) || superclass == Object
          @connection = Connection.new(site, format) if refresh || @connection.nil?
316 317
          @connection.user = user if user
          @connection.password = password if password
318
          @connection.timeout = timeout if timeout
319 320 321 322
          @connection
        else
          superclass.connection
        end
323
      end
324

325 326
      def headers
        @headers ||= {}
327 328
      end

329 330 331 332
      # 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.
      attr_accessor_with_default(:element_name)    { to_s.split("::").last.underscore } #:nodoc:

333 334
      attr_accessor_with_default(:collection_name) { element_name.pluralize } #:nodoc:
      attr_accessor_with_default(:primary_key, 'id') #:nodoc:
335

336 337
      # 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.
338 339 340
      def prefix(options={})
        default = site.path
        default << '/' unless default[-1..-1] == '/'
341
        # generate the actual method based on the current site path
342
        self.prefix = default
343 344
        prefix(options)
      end
345

346 347
      # 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.
348 349 350 351 352
      def prefix_source
        prefix # generate #prefix and #prefix_source methods first
        prefix_source
      end

353 354
      # Sets the prefix for a resource's nested URL (e.g., <tt>prefix/collectionname/1.xml</tt>).
      # Default value is <tt>site.path</tt>.
355
      def prefix=(value = '/')
356
        # Replace :placeholders with '#{embedded options[:lookups]}'
357
        prefix_call = value.gsub(/:\w+/) { |key| "\#{options[#{key}]}" }
358

359 360 361
        # Clear prefix parameters in case they have been cached
        @prefix_parameters = nil

362 363
        # Redefine the new methods.
        code = <<-end_code
364 365
          def prefix_source() "#{value}" end
          def prefix(options={}) "#{prefix_call}" end
366 367
        end_code
        silence_warnings { instance_eval code, __FILE__, __LINE__ }
368
      rescue
369
        logger.error "Couldn't set prefix: #{$!}\n  #{code}"
370
        raise
371
      end
372

373
      alias_method :set_prefix, :prefix=  #:nodoc:
374

375 376
      alias_method :set_element_name, :element_name=  #:nodoc:
      alias_method :set_collection_name, :collection_name=  #:nodoc:
377

378 379 380 381
      # Gets the element path for the given ID in +id+.  If the +query_options+ parameter is omitted, Rails
      # will split from the prefix options.
      #
      # ==== Options
P
Pratik Naik 已提交
382
      # +prefix_options+ - A hash to add a prefix to the request for nested URLs (e.g., <tt>:account_id => 19</tt>
383
      #                    would yield a URL like <tt>/accounts/19/purchases.xml</tt>).
P
Pratik Naik 已提交
384
      # +query_options+ - A hash to add items to the query string for the request.
385 386
      #
      # ==== Examples
387
      #   Post.element_path(1)
388 389
      #   # => /posts/1.xml
      #
390
      #   Comment.element_path(1, :post_id => 5)
391 392
      #   # => /posts/5/comments/1.xml
      #
393
      #   Comment.element_path(1, :post_id => 5, :active => 1)
394 395
      #   # => /posts/5/comments/1.xml?active=1
      #
396
      #   Comment.element_path(1, {:post_id => 5}, {:active => 1})
397
      #   # => /posts/5/comments/1.xml?active=1
398 399 400
      #
      def element_path(id, prefix_options = {}, query_options = nil)
        prefix_options, query_options = split_options(prefix_options) if query_options.nil?
401
        "#{prefix(prefix_options)}#{collection_name}/#{id}.#{format.extension}#{query_string(query_options)}"
402
      end
403

404 405 406 407
      # 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 已提交
408 409 410
      # * +prefix_options+ - A hash to add a prefix to the request for nested URL's (e.g., <tt>:account_id => 19</tt>
      #   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.
411 412 413 414 415
      #
      # ==== Examples
      #   Post.collection_path
      #   # => /posts.xml
      #
416
      #   Comment.collection_path(:post_id => 5)
417 418
      #   # => /posts/5/comments.xml
      #
419
      #   Comment.collection_path(:post_id => 5, :active => 1)
420 421
      #   # => /posts/5/comments.xml?active=1
      #
422
      #   Comment.collection_path({:post_id => 5}, {:active => 1})
423
      #   # => /posts/5/comments.xml?active=1
424 425 426
      #
      def collection_path(prefix_options = {}, query_options = nil)
        prefix_options, query_options = split_options(prefix_options) if query_options.nil?
427
        "#{prefix(prefix_options)}#{collection_name}.#{format.extension}#{query_string(query_options)}"
428
      end
429

430
      alias_method :set_primary_key, :primary_key=  #:nodoc:
431

P
Pratik Naik 已提交
432
      # Creates a new resource instance and makes a request to the remote service
433
      # that it be saved, making it equivalent to the following simultaneous calls:
434 435 436 437
      #
      #   ryan = Person.new(:first => 'ryan')
      #   ryan.save
      #
P
Pratik Naik 已提交
438 439 440 441
      # 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>.
442 443 444 445
      #
      # ==== Examples
      #   Person.create(:name => 'Jeremy', :email => 'myname@nospam.com', :enabled => true)
      #   my_person = Person.find(:first)
P
Pratik Naik 已提交
446
      #   my_person.email # => myname@nospam.com
447 448
      #
      #   dhh = Person.create(:name => 'David', :email => 'dhh@nospam.com', :enabled => true)
P
Pratik Naik 已提交
449 450
      #   dhh.valid? # => true
      #   dhh.new?   # => false
451 452 453
      #
      #   # 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 已提交
454 455
      #   that_guy.valid? # => false
      #   that_guy.new?   # => true
456
      def create(attributes = {})
457
        returning(self.new(attributes)) { |res| res.save }
458 459
      end

P
Pratik Naik 已提交
460
      # Core method for finding resources.  Used similarly to Active Record's +find+ method.
461
      #
462
      # ==== Arguments
463
      # The first argument is considered to be the scope of the query.  That is, how many
464 465
      # resources are returned from the request.  It can be one of the following.
      #
P
Pratik Naik 已提交
466 467
      # * <tt>:one</tt> - Returns a single resource.
      # * <tt>:first</tt> - Returns the first resource found.
468
      # * <tt>:last</tt> - Returns the last resource found.
P
Pratik Naik 已提交
469
      # * <tt>:all</tt> - Returns every resource that matches the request.
470
      #
471
      # ==== Options
472
      #
P
Pratik Naik 已提交
473 474
      # * <tt>:from</tt> - Sets the path or custom method that resources will be fetched from.
      # * <tt>:params</tt> - Sets query and prefix (nested URL) parameters.
475 476
      #
      # ==== Examples
477
      #   Person.find(1)
478 479
      #   # => GET /people/1.xml
      #
480
      #   Person.find(:all)
481 482
      #   # => GET /people.xml
      #
483
      #   Person.find(:all, :params => { :title => "CEO" })
484 485
      #   # => GET /people.xml?title=CEO
      #
486 487 488 489
      #   Person.find(:first, :from => :managers)
      #   # => GET /people/managers.xml
      #
      #   Person.find(:last, :from => :managers)
490 491
      #   # => GET /people/managers.xml
      #
492
      #   Person.find(:all, :from => "/companies/1/people.xml")
493 494
      #   # => GET /companies/1/people.xml
      #
495
      #   Person.find(:one, :from => :leader)
496 497
      #   # => GET /people/leader.xml
      #
498 499 500
      #   Person.find(:all, :from => :developers, :params => { :language => 'ruby' })
      #   # => GET /people/developers.xml?language=ruby
      #
501
      #   Person.find(:one, :from => "/companies/1/manager.xml")
502 503
      #   # => GET /companies/1/manager.xml
      #
504
      #   StreetAddress.find(1, :params => { :person_id => 1 })
505
      #   # => GET /people/1/street_addresses/1.xml
506
      def find(*arguments)
507 508
        scope   = arguments.slice!(0)
        options = arguments.slice!(0) || {}
509 510

        case scope
511 512
          when :all   then find_every(options)
          when :first then find_every(options).first
513
          when :last  then find_every(options).last
514
          when :one   then find_one(options)
515
          else             find_single(scope, options)
516 517
        end
      end
518

519 520 521 522 523 524
      # Deletes the resources with the ID in the +id+ parameter.
      #
      # ==== Options
      # All options specify prefix and query parameters.
      #
      # ==== Examples
P
Pratik Naik 已提交
525
      #   Event.delete(2) # sends DELETE /events/2
526 527
      #
      #   Event.create(:name => 'Free Concert', :location => 'Community Center')
P
Pratik Naik 已提交
528 529
      #   my_event = Event.find(:first) # let's assume this is event with ID 7
      #   Event.delete(my_event.id) # sends DELETE /events/7
530 531
      #
      #   # Let's assume a request to events/5/cancel.xml
P
Pratik Naik 已提交
532
      #   Event.delete(params[:id]) # sends DELETE /events/5
533 534
      def delete(id, options = {})
        connection.delete(element_path(id, options))
535 536
      end

537 538 539 540
      # 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 已提交
541
      #   Note.exists?(1) # => true
542
      #
P
Pratik Naik 已提交
543
      #   Note.exists(1349) # => false
544
      def exists?(id, options = {})
545 546 547 548
        if id
          prefix_options, query_options = split_options(options[:params])
          path = element_path(id, prefix_options, query_options)
          response = connection.head(path, headers)
549
          response.code.to_i == 200
550 551
        end
        # id && !find_single(id, options).nil?
552 553 554 555
      rescue ActiveResource::ResourceNotFound
        false
      end

556
      private
557
        # Find every resource
558
        def find_every(options)
559
          case from = options[:from]
560 561 562 563 564 565 566 567
          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)
568
            instantiate_collection( (connection.get(path, headers) || []), prefix_options )
569
          end
570
        end
571

572 573 574 575 576 577 578 579
        # 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))
580
          end
581
        end
582

583
        # Find a single resource from the default URL
584
        def find_single(scope, options)
585 586 587 588
          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
589

590 591 592
        def instantiate_collection(collection, prefix_options = {})
          collection.collect! { |record| instantiate_record(record, prefix_options) }
        end
593

594 595
        def instantiate_record(record, prefix_options = {})
          returning new(record) do |resource|
596 597
            resource.prefix_options = prefix_options
          end
598
        end
599

600

601
        # Accepts a URI and creates the site URI from that.
602
        def create_site_uri_from(site)
603
          site.is_a?(URI) ? site.dup : URI.parse(site)
604
        end
605

606
        # contains a set of the current prefix parameters.
607 608 609 610
        def prefix_parameters
          @prefix_parameters ||= prefix_source.scan(/:\w+/).map { |key| key[1..-1].to_sym }.to_set
        end

611
        # Builds the query string for the request.
612
        def query_string(options)
613
          "?#{options.to_query}" unless options.nil? || options.empty?
614 615
        end

616
        # split an option hash into two hashes, one containing the prefix options,
617 618
        # and the other containing the leftovers.
        def split_options(options = {})
619 620 621
          prefix_options, query_options = {}, {}

          (options || {}).each do |key, value|
622 623
            next if key.blank?
            (prefix_parameters.include?(key.to_sym) ? prefix_options : query_options)[key.to_sym] = value
624
          end
625 626

          [ prefix_options, query_options ]
627
        end
628 629
    end

630 631
    attr_accessor :attributes #:nodoc:
    attr_accessor :prefix_options #:nodoc:
632

P
Pratik Naik 已提交
633
    # Constructor method for new resources; the optional +attributes+ parameter takes a hash
634 635 636 637 638 639 640 641 642 643
    # of attributes for the new resource.
    #
    # ==== 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
644 645 646
    def initialize(attributes = {})
      @attributes     = {}
      @prefix_options = {}
647
      load(attributes)
648
    end
649

P
Pratik Naik 已提交
650
    # Returns a clone of the resource that hasn't been assigned an +id+ yet and
651 652
    # is treated as a new resource.
    #
P
Pratik Naik 已提交
653 654 655
    #   ryan = Person.find(1)
    #   not_ryan = ryan.clone
    #   not_ryan.new?  # => true
656 657
    #
    # Any active resource member attributes will NOT be cloned, though all other
P
Pratik Naik 已提交
658
    # attributes are.  This is to prevent the conflict between any +prefix_options+
659 660 661
    # that refer to the original parent resource and the newly cloned parent
    # resource that does not exist.
    #
P
Pratik Naik 已提交
662 663 664
    #   ryan = Person.find(1)
    #   ryan.address = StreetAddress.find(1, :person_id => ryan.id)
    #   ryan.hash = {:not => "an ARes instance"}
665
    #
P
Pratik Naik 已提交
666 667 668 669
    #   not_ryan = ryan.clone
    #   not_ryan.new?            # => true
    #   not_ryan.address         # => NoMethodError
    #   not_ryan.hash            # => {:not => "an ARes instance"}
670 671
    def clone
      # Clone all attributes except the pk and any nested ARes
J
Jeremy Kemper 已提交
672
      cloned = attributes.reject {|k,v| k == self.class.primary_key || v.is_a?(ActiveResource::Base)}.inject({}) do |attrs, (k, v)|
673 674 675 676 677 678 679 680
        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 已提交
681
      resource.send :instance_variable_set, '@attributes', cloned
682 683 684 685
      resource
    end


686 687 688 689
    # A method to determine if the resource a new object (i.e., it has not been POSTed to the remote service yet).
    #
    # ==== Examples
    #   not_new = Computer.create(:brand => 'Apple', :make => 'MacBook', :vendor => 'MacMall')
P
Pratik Naik 已提交
690
    #   not_new.new? # => false
691 692
    #
    #   is_new = Computer.new(:brand => 'IBM', :make => 'Thinkpad', :vendor => 'IBM')
P
Pratik Naik 已提交
693
    #   is_new.new? # => true
694 695
    #
    #   is_new.save
P
Pratik Naik 已提交
696
    #   is_new.new? # => false
697
    #
698
    def new?
699 700 701
      id.nil?
    end

702
    # Get the +id+ attribute of the resource.
703
    def id
704
      attributes[self.class.primary_key]
705
    end
706

707
    # Set the +id+ attribute of the resource.
708
    def id=(id)
709
      attributes[self.class.primary_key] = id
710
    end
711

P
Pratik Naik 已提交
712
    # Allows Active Resource objects to be used as parameters in Action Pack URL generation.
713 714 715 716
    def to_param
      id && id.to_s
    end

717
    # Test for equality.  Resource are equal if and only if +other+ is the same object or
P
Pratik Naik 已提交
718
    # is an instance of the same class, is not <tt>new?</tt>, and has the same +id+.
719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738
    #
    # ==== 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
    #
739 740 741 742
    def ==(other)
      other.equal?(self) || (other.instance_of?(self.class) && !other.new? && other.id == id)
    end

743
    # Tests for equality (delegates to ==).
744 745 746 747 748 749 750 751 752
    def eql?(other)
      self == other
    end

    # Delegates to id in order to allow two resources of the same type and id to work with something like:
    #   [Person.find(1), Person.find(2)] & [Person.find(1), Person.find(4)] # => [Person.find(1)]
    def hash
      id.hash
    end
753

754 755 756 757 758
    # Duplicate the current resource without saving it.
    #
    # ==== Examples
    #   my_invoice = Invoice.create(:customer => 'That Company')
    #   next_invoice = my_invoice.dup
P
Pratik Naik 已提交
759
    #   next_invoice.new? # => true
760 761
    #
    #   next_invoice.save
P
Pratik Naik 已提交
762
    #   next_invoice == my_invoice # => false (different id attributes)
763
    #
P
Pratik Naik 已提交
764 765
    #   my_invoice.customer   # => That Company
    #   next_invoice.customer # => That Company
766
    def dup
767
      returning self.class.new do |resource|
768 769 770 771
        resource.attributes     = @attributes
        resource.prefix_options = @prefix_options
      end
    end
772

773
    # A method to save (+POST+) or update (+PUT+) a resource.  It delegates to +create+ if a new object,
774 775 776 777 778 779
    # +update+ if it is existing. If the response to the save includes a body, it will be assumed that this body
    # is XML for the final object as it looked after the save (which would include attributes like +created_at+
    # that weren't part of the original submit).
    #
    # ==== Examples
    #   my_company = Company.new(:name => 'RoleModel Software', :owner => 'Ken Auer', :size => 2)
P
Pratik Naik 已提交
780 781
    #   my_company.new? # => true
    #   my_company.save # sends POST /companies/ (create)
782
    #
P
Pratik Naik 已提交
783
    #   my_company.new? # => false
784
    #   my_company.size = 10
P
Pratik Naik 已提交
785
    #   my_company.save # sends PUT /companies/1 (update)
786
    def save
787
      new? ? create : update
788 789
    end

790 791 792 793 794 795
    # Deletes the resource from the remote service.
    #
    # ==== Examples
    #   my_id = 3
    #   my_person = Person.find(my_id)
    #   my_person.destroy
P
Pratik Naik 已提交
796
    #   Person.find(my_id) # 404 (Resource Not Found)
797
    #
798
    #   new_person = Person.create(:name => 'James')
P
Pratik Naik 已提交
799
    #   new_id = new_person.id # => 7
800
    #   new_person.destroy
P
Pratik Naik 已提交
801
    #   Person.find(new_id) # 404 (Resource Not Found)
802
    def destroy
803
      connection.delete(element_path, self.class.headers)
804
    end
805

P
Pratik Naik 已提交
806
    # Evaluates to <tt>true</tt> if this resource is not <tt>new?</tt> and is
807 808 809 810 811 812 813
    # 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 已提交
814
    #   that_guy.exists? # => true
815 816
    #
    #   that_lady = Person.new(:name => 'Paul Bean')
P
Pratik Naik 已提交
817
    #   that_lady.exists? # => false
818 819 820
    #
    #   guys_id = that_guy.id
    #   Person.delete(guys_id)
P
Pratik Naik 已提交
821
    #   that_guy.exists? # => false
822
    def exists?
P
Pratik Naik 已提交
823
      !new? && self.class.exists?(to_param, :params => prefix_options)
824 825
    end

826 827 828 829 830
    # A method to convert the the resource to an XML string.
    #
    # ==== 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 已提交
831
    # Active Support.
832
    #
P
Pratik Naik 已提交
833 834 835
    # * <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>).
836
    # * <tt>:skip_instruct</tt> - Toggle skipping the +instruct!+ call on the XML builder
P
Pratik Naik 已提交
837
    #   that generates the XML declaration (default is <tt>false</tt>).
838 839 840 841 842 843 844 845 846 847 848 849 850
    #
    # ==== 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>
851 852
    def to_xml(options={})
      attributes.to_xml({:root => self.class.element_name}.merge(options))
853
    end
854

855 856 857 858
    # A method to reload the attributes of this object from the remote web service.
    #
    # ==== Examples
    #   my_branch = Branch.find(:first)
P
Pratik Naik 已提交
859
    #   my_branch.name # => "Wislon Raod"
860
    #
861 862
    #   # Another client fixes the typo...
    #
P
Pratik Naik 已提交
863
    #   my_branch.name # => "Wislon Raod"
864
    #   my_branch.reload
P
Pratik Naik 已提交
865
    #   my_branch.name # => "Wilson Road"
866
    def reload
867
      self.load(self.class.find(to_param, :params => @prefix_options).attributes)
868 869
    end

870
    # A method to manually load attributes from a hash. Recursively loads collections of
P
Pratik Naik 已提交
871
    # resources.  This method is called in +initialize+ and +create+ when a hash of attributes
872 873 874 875 876 877
    # is provided.
    #
    # ==== Examples
    #   my_attrs = {:name => 'J&J Textiles', :industry => 'Cloth and textiles'}
    #
    #   the_supplier = Supplier.find(:first)
P
Pratik Naik 已提交
878
    #   the_supplier.name # => 'J&M Textiles'
879 880 881 882 883 884 885 886 887 888 889
    #   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
890
    def load(attributes)
J
Jeremy Kemper 已提交
891
      raise ArgumentError, "expected an attributes Hash, got #{attributes.inspect}" unless attributes.is_a?(Hash)
892
      @prefix_options, attributes = split_options(attributes)
893 894 895 896 897 898 899
      attributes.each do |key, value|
        @attributes[key.to_s] =
          case value
            when Array
              resource = find_or_create_resource_for_collection(key)
              value.map { |attrs| resource.new(attrs) }
            when Hash
900 901
              resource = find_or_create_resource_for(key)
              resource.new(value)
902 903 904 905
            else
              value.dup rescue value
          end
      end
906 907
      self
    end
908

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

P
Pratik Naik 已提交
912
    # A method to determine if an object responds to a message (e.g., a method call). In Active Resource, a Person object with a
913 914
    # +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>.
915 916 917 918 919
    def respond_to?(method, include_priv = false)
      method_name = method.to_s
      if attributes.nil?
        return super
      elsif attributes.has_key?(method_name)
920
        return true
921 922 923 924 925 926 927
      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
928

929

930 931 932 933
    protected
      def connection(refresh = false)
        self.class.connection(refresh)
      end
934

935
      # Update the resource on the remote service.
936
      def update
937
        returning connection.put(element_path(prefix_options), to_xml, self.class.headers) do |response|
938 939
          load_attributes_from_response(response)
        end
940
      end
941

942
      # Create (i.e., save to the remote service) the new resource.
943
      def create
944
        returning connection.post(collection_path, to_xml, self.class.headers) do |response|
945
          self.id = id_from_response(response)
946
          load_attributes_from_response(response)
947
        end
948
      end
949

950
      def load_attributes_from_response(response)
951
        if response['Content-Length'] != "0" && response.body.strip.size > 0
952
          load(self.class.format.decode(response.body))
953
        end
954
      end
955

956
      # Takes a response from a typical create post and pulls the ID out
957 958 959 960
      def id_from_response(response)
        response['Location'][/\/([^\/]*?)(\.\w+)?$/, 1]
      end

961
      def element_path(options = nil)
962
        self.class.element_path(to_param, options || prefix_options)
963 964 965 966 967 968
      end

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

969
    private
970
      # Tries to find a resource for a given collection name; if it fails, then the resource is created
971 972 973
      def find_or_create_resource_for_collection(name)
        find_or_create_resource_for(name.to_s.singularize)
      end
974

975 976 977 978 979 980 981 982 983 984 985 986 987 988
      # 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

989
      # Tries to find a resource for a given name; if it fails, then the resource is created
990 991
      def find_or_create_resource_for(name)
        resource_name = name.to_s.camelize
992 993 994
        ancestors = self.class.name.split("::")
        if ancestors.size > 1
          find_resource_in_modules(resource_name, ancestors)
995 996 997
        else
          self.class.const_get(resource_name)
        end
998
      rescue NameError
999 1000 1001 1002 1003
        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
1004
        resource.prefix = self.class.prefix
1005
        resource.site   = self.class.site
1006 1007 1008
        resource
      end

1009
      def split_options(options = {})
1010
        self.class.send!(:split_options, options)
1011 1012
      end

1013
      def method_missing(method_symbol, *arguments) #:nodoc:
1014
        method_name = method_symbol.to_s
1015

1016 1017 1018 1019
        case method_name.last
          when "="
            attributes[method_name.first(-1)] = arguments.first
          when "?"
T
 
Tobias Lütke 已提交
1020
            attributes[method_name.first(-1)]
1021
          else
1022
            attributes.has_key?(method_name) ? attributes[method_name] : super
1023 1024 1025
        end
      end
  end
J
Jeremy Kemper 已提交
1026
end