base.rb 34.9 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.
  # 
17 18 19
  #     class Person < ActiveResource::Base
  #       self.site = "http://api.people.com:3000/"
  #     end
20 21 22 23 24 25 26 27 28
  # 
  # Now the Person class is mapped to RESTful resources located at <tt>http://api.people.com:3000/people/</tt>, and
  # you can now use Active Resource's lifecycles methods to manipulate resources.  
  # 
  # == Lifecycle methods
  #
  # Active Resource exposes methods for creating, finding, updating, and deleting resources
  # from REST web services.
  # 
29 30 31 32 33
  #   ryan = Person.new(:first => 'Ryan', :last => 'Daigle')
  #   ryan.save  #=> true
  #   ryan.id  #=> 2
  #   Person.exists?(ryan.id)  #=> true
  #   ryan.exists?  #=> true
34
  # 
35
  #   ryan = Person.find(1)
36
  #   # => Resource holding our newly created Person object
37
  # 
38 39
  #   ryan.first = 'Rizzle'
  #   ryan.save  #=> true
40
  # 
41
  #   ryan.destroy  #=> true
42 43 44 45 46 47 48
  #
  # 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
49
  # defining your own custom REST methods. To invoke them, Active Resource provides the <tt>get</tt>,
50
  # <tt>post</tt>, <tt>put</tt> and <tt>delete</tt> methods where you can specify a custom REST method
51 52 53 54
  # name to invoke.
  #
  #   # POST to the custom 'register' REST method, i.e. POST /people/new/register.xml.
  #   Person.new(:name => 'Ryan').post(:register)
55 56
  #   # => { :id => 1, :name => 'Ryan', :position => 'Clerk' }
  #
57
  #   # PUT an update by invoking the 'promote' REST method, i.e. PUT /people/1/promote.xml?position=Manager.
58
  #   Person.find(1).put(:promote, :position => 'Manager')
59
  #   # => { :id => 1, :name => 'Ryan', :position => 'Manager' }
60 61 62 63 64 65 66
  #
  #   # 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)
67
  # 
68
  # For more information on using custom REST methods, see the
69 70 71 72 73 74
  # ActiveResource::CustomMethods documentation.
  #
  # == Validations
  #
  # You can validate resources client side by overriding validation methods in the base class.
  # 
75 76 77 78 79 80 81
  #     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
82 83 84 85 86 87
  # 
  # See the ActiveResource::Validations documentation for more information.
  #
  # == Authentication
  # 
  # Many REST APIs will require authentication, usually in the form of basic
88 89
  # HTTP authentication.  Authentication can be specified by:
  # * putting the credentials in the URL for the +site+ variable.
90
  # 
91 92 93
  #    class Person < ActiveResource::Base
  #      self.site = "http://ryan:password@api.people.com:3000/"
  #    end
94
  # 
95 96 97 98 99 100 101 102
  # * defining +user+ and/or +password+ variables
  #
  #    class Person < ActiveResource::Base
  #      self.site = "http://api.people.com:3000/"
  #      self.user = "ryan"
  #      self.password = "password"
  #    end
  #
103 104 105
  # For obvious security reasons, it is probably best if such services are available 
  # over HTTPS.
  # 
106 107
  # 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.
108 109 110
  # == Errors & Validation
  #
  # Error handling and validation is handled in much the same manner as you're used to seeing in
111
  # Active Record.  Both the response code in the HTTP response and the body of the response are used to
112 113 114 115
  # indicate that an error occurred.
  # 
  # === Resource errors
  # 
116
  # When a GET is requested for a resource that does not exist, the HTTP <tt>404</tt> (Resource Not Found)
117 118 119 120 121 122 123
  # response code will be returned from the server which will raise an ActiveResource::ResourceNotFound
  # exception.
  # 
  #   # GET http://api.people.com:3000/people/999.xml
  #   ryan = Person.find(999) # => Raises ActiveResource::ResourceNotFound
  #   # => Response = 404
  # 
124
  # <tt>404</tt> is just one of the HTTP error response codes that ActiveResource will handle with its own exception. The
125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148
  # following HTTP response codes will also result in these exceptions:
  # 
  # 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
  #
  # 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 
149 150
  # 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.
151 152 153 154 155 156 157 158 159 160
  # 
  #   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):
161
  #   # <errors type="array"><error>First cannot be empty</error></errors>
162 163 164 165 166 167 168
  #   #
  #
  #   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.
  #
169
  class Base
170
    # The logger for diagnosing and tracing Active Resource calls.
171 172
    cattr_accessor :logger

173
    class << self
174 175
      # Gets the URI of the REST resources to map for this class.  The site variable is required 
      # ActiveResource's mapping to work.
176
      def site
177 178 179 180 181 182 183 184 185 186 187 188 189 190 191
        # 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
        #
192 193
        if defined?(@site)
          @site
194
        elsif superclass != Object && superclass.site
195 196 197
          superclass.site.dup.freeze
        end
      end
J
Jeremy Kemper 已提交
198

199 200
      # 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.
201
      def site=(site)
202
        @connection = nil
203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241
        if site.nil?
          @site = nil
        else
          @site = create_site_uri_from(site)
          @user = @site.user if @site.user
          @password = @site.password if @site.password
        end
      end

      # Gets the user for REST HTTP authentication
      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

      # Sets the user for REST HTTP authentication
      def user=(user)
        @connection = nil
        @user = user
      end

      # Gets the password for REST HTTP authentication
      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

      # Sets the password for REST HTTP authentication
      def password=(password)
        @connection = nil
        @password = password
242 243
      end

244 245 246 247 248 249 250 251 252 253 254 255 256 257
      # Sets the format that attributes are sent and received in from a mime type reference. Example:
      #
      #   Person.format = :json
      #   Person.find(1) # => GET /people/1.json
      #
      #   Person.format = ActiveResource::Formats::XmlFormat
      #   Person.find(1) # => GET /people/1.xml
      #
      # Default format is :xml.
      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)
258
        connection.format = format if site
259 260 261 262 263 264 265
      end
      
      # Returns the current format, default is ActiveResource::Formats::XmlFormat
      def format # :nodoc:
        read_inheritable_attribute("format") || ActiveResource::Formats[:xml]
      end

266 267
      # 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
268
      # or not (defaults to <tt>false</tt>).
269
      def connection(refresh = false)
270 271
        if defined?(@connection) || superclass == Object
          @connection = Connection.new(site, format) if refresh || @connection.nil?
272 273
          @connection.user = user if user
          @connection.password = password if password
274 275 276 277
          @connection
        else
          superclass.connection
        end
278
      end
279

280 281
      def headers
        @headers ||= {}
282 283
      end

284 285 286 287
      # 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:

288 289 290
      attr_accessor_with_default(:collection_name) { element_name.pluralize } #:nodoc:
      attr_accessor_with_default(:primary_key, 'id') #:nodoc:
      
291 292
      # 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.
293 294 295
      def prefix(options={})
        default = site.path
        default << '/' unless default[-1..-1] == '/'
296
        # generate the actual method based on the current site path
297
        self.prefix = default
298 299
        prefix(options)
      end
300

301 302
      # 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.
303 304 305 306 307
      def prefix_source
        prefix # generate #prefix and #prefix_source methods first
        prefix_source
      end

308 309
      # Sets the prefix for a resource's nested URL (e.g., <tt>prefix/collectionname/1.xml</tt>).
      # Default value is <tt>site.path</tt>.
310
      def prefix=(value = '/')
311
        # Replace :placeholders with '#{embedded options[:lookups]}'
312
        prefix_call = value.gsub(/:\w+/) { |key| "\#{options[#{key}]}" }
313 314 315

        # Redefine the new methods.
        code = <<-end_code
316 317
          def prefix_source() "#{value}" end
          def prefix(options={}) "#{prefix_call}" end
318 319
        end_code
        silence_warnings { instance_eval code, __FILE__, __LINE__ }
320
      rescue
321
        logger.error "Couldn't set prefix: #{$!}\n  #{code}"
322
        raise
323
      end
324

325
      alias_method :set_prefix, :prefix=  #:nodoc:
326

327 328
      alias_method :set_element_name, :element_name=  #:nodoc:
      alias_method :set_collection_name, :collection_name=  #:nodoc:
329

330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349
      # 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
      # +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.
      #
      # ==== 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
350 351 352
      #
      def element_path(id, prefix_options = {}, query_options = nil)
        prefix_options, query_options = split_options(prefix_options) if query_options.nil?
353
        "#{prefix(prefix_options)}#{collection_name}/#{id}.#{format.extension}#{query_string(query_options)}"
354
      end
355

356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375
      # Gets the collection path for the REST resources.  If the +query_options+ parameter is omitted, Rails
      # will split from the +prefix_options+.
      #
      # ==== Options
      # +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.
      #
      # ==== 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
376 377 378
      #
      def collection_path(prefix_options = {}, query_options = nil)
        prefix_options, query_options = split_options(prefix_options) if query_options.nil?
379
        "#{prefix(prefix_options)}#{collection_name}.#{format.extension}#{query_string(query_options)}"
380
      end
381

382
      alias_method :set_primary_key, :primary_key=  #:nodoc:
383

384
      # Create a new resource instance and request to the remote service
385
      # that it be saved, making it equivalent to the following simultaneous calls:
386 387 388 389 390 391
      #
      #   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
392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413
      # 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)
      #   my_person.email
      #   # => myname@nospam.com
      #
      #   dhh = Person.create(:name => 'David', :email => 'dhh@nospam.com', :enabled => true)
      #   dhh.valid?
      #   # => true
      #   dhh.new?
      #   # => false
      #
      #   # We'll assume that there's a validation that requires the name attribute
      #   that_guy = Person.create(:name => '', :email => 'thatguy@nospam.com', :enabled => true)
      #   that_guy.valid?
      #   # => false
      #   that_guy.new?
      #   # => true
      #
414 415
      def create(attributes = {})
        returning(self.new(attributes)) { |res| res.save }        
416 417
      end

418
      # Core method for finding resources.  Used similarly to Active Record's find method.
419
      #
420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450
      # ==== 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.
      #
      # +:one+:: Returns a single resource.
      # +:first+:: Returns the first resource found.
      # +:all+:: Returns every resource that matches the request.
      # 
      # ==== Options
      # +from+:: Sets the path or custom method that resources will be fetched from.
      # +params+:: Sets query and prefix (nested URL) parameters.
      #
      # ==== 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
      #
451 452 453
      #   Person.find(:all, :from => :developers, :params => { :language => 'ruby' })
      #   # => GET /people/developers.xml?language=ruby
      #
454 455 456 457 458
      #   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
459
      def find(*arguments)
460 461
        scope   = arguments.slice!(0)
        options = arguments.slice!(0) || {}
462 463

        case scope
464 465
          when :all   then find_every(options)
          when :first then find_every(options).first
466
          when :one   then find_one(options)
467
          else             find_single(scope, options)
468 469
        end
      end
470

471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489
      # Deletes the resources with the ID in the +id+ parameter.
      #
      # ==== Options
      # All options specify prefix and query parameters.
      #
      # ==== Examples
      #   Event.delete(2)
      #   # => DELETE /events/2
      #
      #   Event.create(:name => 'Free Concert', :location => 'Community Center')
      #   my_event = Event.find(:first)
      #   # => Events (id: 7)
      #   Event.delete(my_event.id)
      #   # => DELETE /events/7
      #
      #   # Let's assume a request to events/5/cancel.xml
      #   Event.delete(params[:id])
      #   # => DELETE /events/5
      #
490 491
      def delete(id, options = {})
        connection.delete(element_path(id, options))
492 493
      end

494 495 496 497 498 499 500 501 502
      # 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...')
      #   Note.exists?(1)
      #   # => true
      #
      #   Note.exists(1349)
      #   # => false
503
      def exists?(id, options = {})
504 505 506 507 508 509 510
        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?
511 512 513 514
      rescue ActiveResource::ResourceNotFound
        false
      end

515
      private
516
        # Find every resource
517
        def find_every(options)
518
          case from = options[:from]
519 520 521 522 523 524 525 526
          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)
527
            instantiate_collection( (connection.get(path, headers) || []), prefix_options )
528
          end
529 530
        end
        
531 532 533 534 535 536 537 538
        # 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))
539
          end
540
        end
541

542
        # Find a single resource from the default URL
543
        def find_single(scope, options)
544 545 546 547 548 549 550 551
          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
552

553 554
        def instantiate_record(record, prefix_options = {})
          returning new(record) do |resource|
555 556
            resource.prefix_options = prefix_options
          end
557
        end
558

559

560
        # Accepts a URI and creates the site URI from that.
561
        def create_site_uri_from(site)
562
          site.is_a?(URI) ? site.dup : URI.parse(site)
563
        end
564

565
        # contains a set of the current prefix parameters.
566 567 568 569
        def prefix_parameters
          @prefix_parameters ||= prefix_source.scan(/:\w+/).map { |key| key[1..-1].to_sym }.to_set
        end

570
        # Builds the query string for the request.
571
        def query_string(options)
572
          "?#{options.to_query}" unless options.nil? || options.empty? 
573 574 575 576 577
        end

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

          (options || {}).each do |key, value|
581 582
            next if key.blank?
            (prefix_parameters.include?(key.to_sym) ? prefix_options : query_options)[key.to_sym] = value
583
          end
584 585

          [ prefix_options, query_options ]
586
        end
587 588
    end

589 590
    attr_accessor :attributes #:nodoc:
    attr_accessor :prefix_options #:nodoc:
591

592 593 594 595 596 597 598 599 600 601 602
    # Constructor method for new resources; the optional +attributes+ parameter takes a +Hash+
    # 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
603 604 605
    def initialize(attributes = {})
      @attributes     = {}
      @prefix_options = {}
606
      load(attributes)
607
    end
608

609 610 611 612 613 614 615 616 617 618 619 620 621 622 623
    # 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')
    #   not_new.new?
    #   # => false
    #
    #   is_new = Computer.new(:brand => 'IBM', :make => 'Thinkpad', :vendor => 'IBM')
    #   is_new.new?
    #   # => true
    #
    #   is_new.save
    #   is_new.new?
    #   # => false
    #
624
    def new?
625 626 627
      id.nil?
    end

628
    # Get the +id+ attribute of the resource.
629
    def id
630
      attributes[self.class.primary_key]
631
    end
632

633
    # Set the +id+ attribute of the resource.
634
    def id=(id)
635
      attributes[self.class.primary_key] = id
636
    end
637

638 639 640 641 642
    # Allows ActiveResource objects to be used as parameters in ActionPack URL generation.
    def to_param
      id && id.to_s
    end

643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664
    # Test for equality.  Resource are equal if and only if +other+ is the same object or 
    # is an instance of the same class, is not +new?+, and has the same +id+.
    #
    # ==== 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
    #
665 666 667 668
    def ==(other)
      other.equal?(self) || (other.instance_of?(self.class) && !other.new? && other.id == id)
    end

669
    # Tests for equality (delegates to ==).
670 671 672 673 674 675 676 677 678
    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
679
    
680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695
    # Duplicate the current resource without saving it.
    #
    # ==== Examples
    #   my_invoice = Invoice.create(:customer => 'That Company')
    #   next_invoice = my_invoice.dup
    #   next_invoice.new?
    #   # => true
    #
    #   next_invoice.save
    #   next_invoice == my_invoice
    #   # => false (different id attributes)
    #
    #   my_invoice.customer
    #   # => That Company
    #   next_invoice.customer
    #   # => That Company
696
    def dup
697
      returning self.class.new do |resource|
698 699 700 701
        resource.attributes     = @attributes
        resource.prefix_options = @prefix_options
      end
    end
702

703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719
    # 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)
    #   my_company.new?
    #   # => true
    #   my_company.save
    #   # => POST /companies/ (create)
    #
    #   my_company.new?
    #   # => false
    #   my_company.size = 10
    #   my_company.save
    #   # => PUT /companies/1 (update)
720
    def save
721
      new? ? create : update
722 723
    end

724 725 726 727 728 729 730 731 732 733 734 735 736 737 738
    # Deletes the resource from the remote service.
    #
    # ==== Examples
    #   my_id = 3
    #   my_person = Person.find(my_id)
    #   my_person.destroy
    #   Person.find(my_id)
    #   # => 404 (Resource Not Found)
    #   
    #   new_person = Person.create(:name => 'James')
    #   new_id = new_person.id 
    #   # => 7
    #   new_person.destroy
    #   Person.find(new_id)
    #   # => 404 (Resource Not Found)
739
    def destroy
740
      connection.delete(element_path, self.class.headers)
741
    end
742

743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761
    # Evaluates to <tt>true</tt> if this resource is not +new?+ and is
    # 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)
    #   that_guy.exists?
    #   # => true
    #
    #   that_lady = Person.new(:name => 'Paul Bean')
    #   that_lady.exists?
    #   # => false
    #
    #   guys_id = that_guy.id
    #   Person.delete(guys_id)
    #   that_guy.exists?
    #   # => false
762
    def exists?
763
      !new? && self.class.exists?(id, :params => prefix_options)
764 765
    end

766 767 768 769 770 771 772 773 774
    # 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.
    #
    # indent:: Set the indent level for the XML output (default is +2+).
    # dasherize:: Boolean option to determine whether or not element names should
775
    #             replace underscores with dashes (default is <tt>false</tt>).
776
    # skip_instruct::  Toggle skipping the +instruct!+ call on the XML builder 
777
    #                  that generates the XML declaration (default is <tt>false</tt>).
778 779 780 781 782 783 784 785 786 787 788 789 790
    #
    # ==== 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>
791 792
    def to_xml(options={})
      attributes.to_xml({:root => self.class.element_name}.merge(options))
793
    end
794

795 796 797 798 799 800 801 802 803 804 805 806 807 808
    # A method to reload the attributes of this object from the remote web service.
    #
    # ==== Examples
    #   my_branch = Branch.find(:first)
    #   my_branch.name
    #   # => Wislon Raod
    #   
    #   # Another client fixes the typo...
    #
    #   my_branch.name
    #   # => Wislon Raod
    #   my_branch.reload
    #   my_branch.name
    #   # => Wilson Road
809
    def reload
810
      self.load(self.class.find(id, :params => @prefix_options).attributes)
811 812
    end

813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833
    # 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
    # is provided.
    #
    # ==== Examples
    #   my_attrs = {:name => 'J&J Textiles', :industry => 'Cloth and textiles'}
    #
    #   the_supplier = Supplier.find(:first)
    #   the_supplier.name
    #   # => 'J&M Textiles'
    #   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
834
    def load(attributes)
J
Jeremy Kemper 已提交
835
      raise ArgumentError, "expected an attributes Hash, got #{attributes.inspect}" unless attributes.is_a?(Hash)
836
      @prefix_options, attributes = split_options(attributes)
837 838 839 840 841 842 843
      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
844 845
              resource = find_or_create_resource_for(key)
              resource.new(value)
846 847 848 849
            else
              value.dup rescue value
          end
      end
850 851
      self
    end
852 853 854 855
    
    # For checking respond_to? without searching the attributes (which is faster).
    alias_method :respond_to_without_attributes?, :respond_to?

856
    # A method to determine if an object responds to a message (e.g., a method call). In Active Resource, a +Person+ object with a
857 858
    # +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>.
859 860 861 862 863 864 865 866 867 868 869 870 871 872
    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
    
873

874 875 876 877
    protected
      def connection(refresh = false)
        self.class.connection(refresh)
      end
878

879
      # Update the resource on the remote service.
880
      def update
881
        returning connection.put(element_path(prefix_options), to_xml, self.class.headers) do |response|
882 883
          load_attributes_from_response(response)
        end
884
      end
885

886
      # Create (i.e., save to the remote service) the new resource.
887
      def create
888
        returning connection.post(collection_path, to_xml, self.class.headers) do |response|
889
          self.id = id_from_response(response)
890
          load_attributes_from_response(response)
891
        end
892
      end
893 894
      
      def load_attributes_from_response(response)
895
        if response['Content-Length'] != "0" && response.body.strip.size > 0
896
          load(self.class.format.decode(response.body))
897
        end
898
      end
899

900
      # Takes a response from a typical create post and pulls the ID out
901 902 903 904
      def id_from_response(response)
        response['Location'][/\/([^\/]*?)(\.\w+)?$/, 1]
      end

905 906 907 908 909 910 911 912
      def element_path(options = nil)
        self.class.element_path(id, options || prefix_options)
      end

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

913
    private
914
      # Tries to find a resource for a given collection name; if it fails, then the resource is created
915 916 917
      def find_or_create_resource_for_collection(name)
        find_or_create_resource_for(name.to_s.singularize)
      end
918
      
919 920 921 922 923 924 925 926 927 928 929 930 931 932
      # 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

933
      # Tries to find a resource for a given name; if it fails, then the resource is created
934 935
      def find_or_create_resource_for(name)
        resource_name = name.to_s.camelize
936 937 938
        ancestors = self.class.name.split("::")
        if ancestors.size > 1
          find_resource_in_modules(resource_name, ancestors)
939 940 941
        else
          self.class.const_get(resource_name)
        end
942 943 944
      rescue NameError
        resource = self.class.const_set(resource_name, Class.new(ActiveResource::Base))
        resource.prefix = self.class.prefix
945
        resource.site   = self.class.site
946 947 948
        resource
      end

949
      def split_options(options = {})
950
        self.class.send!(:split_options, options)
951 952
      end

953
      def method_missing(method_symbol, *arguments) #:nodoc:
954
        method_name = method_symbol.to_s
955

956 957 958 959
        case method_name.last
          when "="
            attributes[method_name.first(-1)] = arguments.first
          when "?"
T
 
Tobias Lütke 已提交
960
            attributes[method_name.first(-1)]
961
          else
962
            attributes.has_key?(method_name) ? attributes[method_name] : super
963 964 965
        end
      end
  end
J
Jeremy Kemper 已提交
966
end