base.rb 38.3 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 16
  #
  # == 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.
  # 
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
P
Pratik Naik 已提交
22 23 24 25 26 27 28 29
  # you can now use Active Resource's lifecycles methods to manipulate resources. In the case where you already have 
  # 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
  #  
30 31 32 33 34 35
  # 
  # == Lifecycle methods
  #
  # Active Resource exposes methods for creating, finding, updating, and deleting resources
  # from REST web services.
  # 
36 37 38 39 40
  #   ryan = Person.new(:first => 'Ryan', :last => 'Daigle')
  #   ryan.save  #=> true
  #   ryan.id  #=> 2
  #   Person.exists?(ryan.id)  #=> true
  #   ryan.exists?  #=> true
41
  # 
42
  #   ryan = Person.find(1)
43
  #   # => Resource holding our newly created Person object
44
  # 
45 46
  #   ryan.first = 'Rizzle'
  #   ryan.save  #=> true
47
  # 
48
  #   ryan.destroy  #=> true
49 50 51 52 53 54 55
  #
  # 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.
  # 
  # === 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 81
  # ActiveResource::CustomMethods documentation.
  #
  # == Validations
  #
  # You can validate resources client side by overriding validation methods in the base class.
  # 
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 93 94
  # 
  # See the ActiveResource::Validations documentation for more information.
  #
  # == Authentication
  # 
  # 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 111 112
  # For obvious security reasons, it is probably best if such services are available 
  # over HTTPS.
  # 
113 114
  # Note: Some values cannot be provided in the URL passed to site.  e.g. email addresses 
  # as usernames.  In those situations you should use the seperate 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 120 121 122
  # indicate that an error occurred.
  # 
  # === Resource errors
  # 
123
  # When a GET is requested for a resource that does not exist, the HTTP <tt>404</tt> (Resource Not Found)
124 125 126 127
  # response code will be returned from the server which will raise an ActiveResource::ResourceNotFound
  # exception.
  # 
  #   # GET http://api.people.com:3000/people/999.xml
P
Pratik Naik 已提交
128
  #   ryan = Person.find(999) # 404, raises ActiveResource::ResourceNotFound
129
  # 
130
  # <tt>404</tt> is just one of the HTTP error response codes that ActiveResource will handle with its own exception. The
131 132
  # following HTTP response codes will also result in these exceptions:
  # 
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 152 153 154
  #
  # 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
  # 
  # Active Resource supports validations on resources and will return errors if any these validations fail
  # (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 159 160 161 162 163 164 165 166
  # 
  #   ryan = Person.find(1)
  #   ryan.first #=> ''
  #   ryan.save  #=> false
  #
  #   # When 
  #   # 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 170 171 172 173 174
  #   #
  #
  #   ryan.errors.invalid?(:first)  #=> true
  #   ryan.errors.full_messages  #=> ['First cannot be empty']
  # 
  # 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
203 204
      # Gets the URI of the REST resources to map for this class.  The site variable is required 
      # ActiveResource'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 229
      # Sets the URI of the REST resources to map for this class to the value in the +site+ argument.
      # The site variable is required ActiveResource'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 283 284 285 286
      def format=(mime_type_reference_or_format)
        format = mime_type_reference_or_format.is_a?(Symbol) ? 
          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.
291 292 293 294
      def format # :nodoc:
        read_inheritable_attribute("format") || ActiveResource::Formats[:xml]
      end

295 296 297 298 299 300 301 302 303 304 305 306 307 308 309
      # Sets the number of seconds after which requests to the REST API should time out.
      def timeout=(timeout)
        @connection = nil
        @timeout = timeout
      end

      # Gets tthe number of seconds after which requests to the REST API should time out.
      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 335
      attr_accessor_with_default(:collection_name) { element_name.pluralize } #:nodoc:
      attr_accessor_with_default(:primary_key, 'id') #:nodoc:
      
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

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

370
      alias_method :set_prefix, :prefix=  #:nodoc:
371

372 373
      alias_method :set_element_name, :element_name=  #:nodoc:
      alias_method :set_collection_name, :collection_name=  #:nodoc:
374

375 376 377 378
      # 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 已提交
379
      # +prefix_options+ - A hash to add a prefix to the request for nested URLs (e.g., <tt>:account_id => 19</tt>
380
      #                    would yield a URL like <tt>/accounts/19/purchases.xml</tt>).
P
Pratik Naik 已提交
381
      # +query_options+ - A hash to add items to the query string for the request.
382 383 384 385 386 387 388 389 390 391 392 393 394
      #
      # ==== Examples
      #   Post.element_path(1) 
      #   # => /posts/1.xml
      #
      #   Comment.element_path(1, :post_id => 5) 
      #   # => /posts/5/comments/1.xml
      #
      #   Comment.element_path(1, :post_id => 5, :active => 1) 
      #   # => /posts/5/comments/1.xml?active=1
      #
      #   Comment.element_path(1, {:post_id => 5}, {:active => 1}) 
      #   # => /posts/5/comments/1.xml?active=1
395 396 397
      #
      def element_path(id, prefix_options = {}, query_options = nil)
        prefix_options, query_options = split_options(prefix_options) if query_options.nil?
398
        "#{prefix(prefix_options)}#{collection_name}/#{id}.#{format.extension}#{query_string(query_options)}"        
399
      end
400

401 402 403 404
      # 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 已提交
405 406 407
      # * +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.
408 409 410 411 412 413 414 415 416 417 418 419 420
      #
      # ==== Examples
      #   Post.collection_path
      #   # => /posts.xml
      #
      #   Comment.collection_path(:post_id => 5) 
      #   # => /posts/5/comments.xml
      #
      #   Comment.collection_path(:post_id => 5, :active => 1) 
      #   # => /posts/5/comments.xml?active=1
      #
      #   Comment.collection_path({:post_id => 5}, {:active => 1}) 
      #   # => /posts/5/comments.xml?active=1
421 422 423
      #
      def collection_path(prefix_options = {}, query_options = nil)
        prefix_options, query_options = split_options(prefix_options) if query_options.nil?
424
        "#{prefix(prefix_options)}#{collection_name}.#{format.extension}#{query_string(query_options)}"
425
      end
426

427
      alias_method :set_primary_key, :primary_key=  #:nodoc:
428

429
      # Create a new resource instance and request to the remote service
430
      # that it be saved, making it equivalent to the following simultaneous calls:
431 432 433 434 435 436
      #
      #   ryan = Person.new(:first => 'ryan')
      #   ryan.save
      #
      # The newly created resource is returned.  If a failure has occurred an
      # exception will be raised (see save).  If the resource is invalid and
437 438 439 440 441 442
      # has not been saved then valid? will return <tt>false</tt>,
      # while new? will still return <tt>true</tt>.
      #
      # ==== Examples
      #   Person.create(:name => 'Jeremy', :email => 'myname@nospam.com', :enabled => true)
      #   my_person = Person.find(:first)
P
Pratik Naik 已提交
443
      #   my_person.email # => myname@nospam.com
444 445
      #
      #   dhh = Person.create(:name => 'David', :email => 'dhh@nospam.com', :enabled => true)
P
Pratik Naik 已提交
446 447
      #   dhh.valid? # => true
      #   dhh.new?   # => false
448 449 450
      #
      #   # 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 已提交
451 452
      #   that_guy.valid? # => false
      #   that_guy.new?   # => true
453 454
      def create(attributes = {})
        returning(self.new(attributes)) { |res| res.save }        
455 456
      end

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

        case scope
504 505
          when :all   then find_every(options)
          when :first then find_every(options).first
506
          when :one   then find_one(options)
507
          else             find_single(scope, options)
508 509
        end
      end
510

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

529 530 531 532
      # 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 已提交
533
      #   Note.exists?(1) # => true
534
      #
P
Pratik Naik 已提交
535
      #   Note.exists(1349) # => false
536
      def exists?(id, options = {})
537 538 539 540 541 542 543
        if id
          prefix_options, query_options = split_options(options[:params])
          path = element_path(id, prefix_options, query_options)
          response = connection.head(path, headers)
          response.code == 200
        end
        # id && !find_single(id, options).nil?
544 545 546 547
      rescue ActiveResource::ResourceNotFound
        false
      end

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

575
        # Find a single resource from the default URL
576
        def find_single(scope, options)
577 578 579 580 581 582 583 584
          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
        
        def instantiate_collection(collection, prefix_options = {})
          collection.collect! { |record| instantiate_record(record, prefix_options) }
        end
585

586 587
        def instantiate_record(record, prefix_options = {})
          returning new(record) do |resource|
588 589
            resource.prefix_options = prefix_options
          end
590
        end
591

592

593
        # Accepts a URI and creates the site URI from that.
594
        def create_site_uri_from(site)
595
          site.is_a?(URI) ? site.dup : URI.parse(site)
596
        end
597

598
        # contains a set of the current prefix parameters.
599 600 601 602
        def prefix_parameters
          @prefix_parameters ||= prefix_source.scan(/:\w+/).map { |key| key[1..-1].to_sym }.to_set
        end

603
        # Builds the query string for the request.
604
        def query_string(options)
605
          "?#{options.to_query}" unless options.nil? || options.empty? 
606 607 608 609 610
        end

        # split an option hash into two hashes, one containing the prefix options, 
        # and the other containing the leftovers.
        def split_options(options = {})
611 612 613
          prefix_options, query_options = {}, {}

          (options || {}).each do |key, value|
614 615
            next if key.blank?
            (prefix_parameters.include?(key.to_sym) ? prefix_options : query_options)[key.to_sym] = value
616
          end
617 618

          [ prefix_options, query_options ]
619
        end
620 621
    end

622 623
    attr_accessor :attributes #:nodoc:
    attr_accessor :prefix_options #:nodoc:
624

P
Pratik Naik 已提交
625
    # Constructor method for new resources; the optional +attributes+ parameter takes a hash
626 627 628 629 630 631 632 633 634 635
    # 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
636 637 638
    def initialize(attributes = {})
      @attributes     = {}
      @prefix_options = {}
639
      load(attributes)
640
    end
641

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


678 679 680 681
    # 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 已提交
682
    #   not_new.new? # => false
683 684
    #
    #   is_new = Computer.new(:brand => 'IBM', :make => 'Thinkpad', :vendor => 'IBM')
P
Pratik Naik 已提交
685
    #   is_new.new? # => true
686 687
    #
    #   is_new.save
P
Pratik Naik 已提交
688
    #   is_new.new? # => false
689
    #
690
    def new?
691 692 693
      id.nil?
    end

694
    # Get the +id+ attribute of the resource.
695
    def id
696
      attributes[self.class.primary_key]
697
    end
698

699
    # Set the +id+ attribute of the resource.
700
    def id=(id)
701
      attributes[self.class.primary_key] = id
702
    end
703

704 705 706 707 708
    # Allows ActiveResource objects to be used as parameters in ActionPack URL generation.
    def to_param
      id && id.to_s
    end

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

735
    # Tests for equality (delegates to ==).
736 737 738 739 740 741 742 743 744
    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
745
    
746 747 748 749 750
    # Duplicate the current resource without saving it.
    #
    # ==== Examples
    #   my_invoice = Invoice.create(:customer => 'That Company')
    #   next_invoice = my_invoice.dup
P
Pratik Naik 已提交
751
    #   next_invoice.new? # => true
752 753
    #
    #   next_invoice.save
P
Pratik Naik 已提交
754
    #   next_invoice == my_invoice # => false (different id attributes)
755
    #
P
Pratik Naik 已提交
756 757
    #   my_invoice.customer   # => That Company
    #   next_invoice.customer # => That Company
758
    def dup
759
      returning self.class.new do |resource|
760 761 762 763
        resource.attributes     = @attributes
        resource.prefix_options = @prefix_options
      end
    end
764

765 766 767 768 769 770 771
    # A method to save (+POST+) or update (+PUT+) a resource.  It delegates to +create+ if a new object, 
    # +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 已提交
772 773
    #   my_company.new? # => true
    #   my_company.save # sends POST /companies/ (create)
774
    #
P
Pratik Naik 已提交
775
    #   my_company.new? # => false
776
    #   my_company.size = 10
P
Pratik Naik 已提交
777
    #   my_company.save # sends PUT /companies/1 (update)
778
    def save
779
      new? ? create : update
780 781
    end

782 783 784 785 786 787
    # Deletes the resource from the remote service.
    #
    # ==== Examples
    #   my_id = 3
    #   my_person = Person.find(my_id)
    #   my_person.destroy
P
Pratik Naik 已提交
788
    #   Person.find(my_id) # 404 (Resource Not Found)
789 790
    #   
    #   new_person = Person.create(:name => 'James')
P
Pratik Naik 已提交
791
    #   new_id = new_person.id # => 7
792
    #   new_person.destroy
P
Pratik Naik 已提交
793
    #   Person.find(new_id) # 404 (Resource Not Found)
794
    def destroy
795
      connection.delete(element_path, self.class.headers)
796
    end
797

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

818 819 820 821 822 823 824
    # 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
    # ActiveSupport.
    #
P
Pratik Naik 已提交
825 826 827 828 829
    # * <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>).
    # * <tt>:skip_instruct</tt> - Toggle skipping the +instruct!+ call on the XML builder 
    #   that generates the XML declaration (default is <tt>false</tt>).
830 831 832 833 834 835 836 837 838 839 840 841 842
    #
    # ==== 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>
843 844
    def to_xml(options={})
      attributes.to_xml({:root => self.class.element_name}.merge(options))
845
    end
846

847 848 849 850
    # A method to reload the attributes of this object from the remote web service.
    #
    # ==== Examples
    #   my_branch = Branch.find(:first)
P
Pratik Naik 已提交
851
    #   my_branch.name # => "Wislon Raod"
852 853 854
    #   
    #   # Another client fixes the typo...
    #
P
Pratik Naik 已提交
855
    #   my_branch.name # => "Wislon Raod"
856
    #   my_branch.reload
P
Pratik Naik 已提交
857
    #   my_branch.name # => "Wilson Road"
858
    def reload
859
      self.load(self.class.find(to_param, :params => @prefix_options).attributes)
860 861
    end

862
    # A method to manually load attributes from a hash. Recursively loads collections of
P
Pratik Naik 已提交
863
    # resources.  This method is called in +initialize+ and +create+ when a hash of attributes
864 865 866 867 868 869
    # is provided.
    #
    # ==== Examples
    #   my_attrs = {:name => 'J&J Textiles', :industry => 'Cloth and textiles'}
    #
    #   the_supplier = Supplier.find(:first)
P
Pratik Naik 已提交
870
    #   the_supplier.name # => 'J&M Textiles'
871 872 873 874 875 876 877 878 879 880 881
    #   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
882
    def load(attributes)
J
Jeremy Kemper 已提交
883
      raise ArgumentError, "expected an attributes Hash, got #{attributes.inspect}" unless attributes.is_a?(Hash)
884
      @prefix_options, attributes = split_options(attributes)
885 886 887 888 889 890 891
      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
892 893
              resource = find_or_create_resource_for(key)
              resource.new(value)
894 895 896 897
            else
              value.dup rescue value
          end
      end
898 899
      self
    end
900
    
P
Pratik Naik 已提交
901
    # For checking <tt>respond_to?</tt> without searching the attributes (which is faster).
902 903
    alias_method :respond_to_without_attributes?, :respond_to?

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

922 923 924 925
    protected
      def connection(refresh = false)
        self.class.connection(refresh)
      end
926

927
      # Update the resource on the remote service.
928
      def update
929
        returning connection.put(element_path(prefix_options), to_xml, self.class.headers) do |response|
930 931
          load_attributes_from_response(response)
        end
932
      end
933

934
      # Create (i.e., save to the remote service) the new resource.
935
      def create
936
        returning connection.post(collection_path, to_xml, self.class.headers) do |response|
937
          self.id = id_from_response(response)
938
          load_attributes_from_response(response)
939
        end
940
      end
941 942
      
      def load_attributes_from_response(response)
943
        if response['Content-Length'] != "0" && response.body.strip.size > 0
944
          load(self.class.format.decode(response.body))
945
        end
946
      end
947

948
      # Takes a response from a typical create post and pulls the ID out
949 950 951 952
      def id_from_response(response)
        response['Location'][/\/([^\/]*?)(\.\w+)?$/, 1]
      end

953
      def element_path(options = nil)
954
        self.class.element_path(to_param, options || prefix_options)
955 956 957 958 959 960
      end

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

961
    private
962
      # Tries to find a resource for a given collection name; if it fails, then the resource is created
963 964 965
      def find_or_create_resource_for_collection(name)
        find_or_create_resource_for(name.to_s.singularize)
      end
966
      
967 968 969 970 971 972 973 974 975 976 977 978 979 980
      # 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

981
      # Tries to find a resource for a given name; if it fails, then the resource is created
982 983
      def find_or_create_resource_for(name)
        resource_name = name.to_s.camelize
984 985 986
        ancestors = self.class.name.split("::")
        if ancestors.size > 1
          find_resource_in_modules(resource_name, ancestors)
987 988 989
        else
          self.class.const_get(resource_name)
        end
990 991 992
      rescue NameError
        resource = self.class.const_set(resource_name, Class.new(ActiveResource::Base))
        resource.prefix = self.class.prefix
993
        resource.site   = self.class.site
994 995 996
        resource
      end

997
      def split_options(options = {})
998
        self.class.send!(:split_options, options)
999 1000
      end

1001
      def method_missing(method_symbol, *arguments) #:nodoc:
1002
        method_name = method_symbol.to_s
1003

1004 1005 1006 1007
        case method_name.last
          when "="
            attributes[method_name.first(-1)] = arguments.first
          when "?"
T
 
Tobias Lütke 已提交
1008
            attributes[method_name.first(-1)]
1009
          else
1010
            attributes.has_key?(method_name) ? attributes[method_name] : super
1011 1012 1013
        end
      end
  end
J
Jeremy Kemper 已提交
1014
end