base.rb 32.5 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 50 51 52 53 54
  # defining your own custom REST methods. To invoke them, Active Resource provides the <tt>get</tt>,
  # <tt>post</tt>, <tt>post</tt> and <tt>put</tt> methods where you can specify a custom REST method
  # 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 88 89 90 91 92 93 94 95 96 97 98 99 100
  # 
  # See the ActiveResource::Validations documentation for more information.
  #
  # == Authentication
  # 
  # Many REST APIs will require authentication, usually in the form of basic
  # HTTP authentication.  Authentication can be specified by putting the credentials
  # in the +site+ variable of the Active Resource class you need to authenticate.
  # 
  #   class Person < ActiveResource::Base
  #     self.site = "http://ryan:password@api.people.com:3000/"
  #   end
  # 
  # For obvious security reasons, it is probably best if such services are available 
  # over HTTPS.
  # 
  # == Errors & Validation
  #
  # Error handling and validation is handled in much the same manner as you're used to seeing in
101
  # Active Record.  Both the response code in the HTTP response and the body of the response are used to
102 103 104 105
  # indicate that an error occurred.
  # 
  # === Resource errors
  # 
106
  # When a GET is requested for a resource that does not exist, the HTTP <tt>404</tt> (Resource Not Found)
107 108 109 110 111 112 113
  # 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
  # 
114
  # <tt>404</tt> is just one of the HTTP error response codes that ActiveResource will handle with its own exception. The
115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138
  # 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 
139 140
  # 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.
141 142 143 144 145 146 147 148 149 150
  # 
  #   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):
151
  #   # <errors type="array"><error>First cannot be empty</error></errors>
152 153 154 155 156 157 158
  #   #
  #
  #   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.
  #
159
  class Base
160
    # The logger for diagnosing and tracing Active Resource calls.
161 162
    cattr_accessor :logger

163
    class << self
164 165
      # Gets the URI of the REST resources to map for this class.  The site variable is required 
      # ActiveResource's mapping to work.
166 167 168
      def site
        if defined?(@site)
          @site
169
        elsif superclass != Object && superclass.site
170 171 172
          superclass.site.dup.freeze
        end
      end
J
Jeremy Kemper 已提交
173

174 175
      # 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.
176
      def site=(site)
177
        @connection = nil
178
        @site = site.nil? ? nil : create_site_uri_from(site)
179 180
      end

181 182 183 184 185 186 187 188 189 190 191 192 193 194
      # 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)
195
        connection.format = format if site
196 197 198 199 200 201 202
      end
      
      # Returns the current format, default is ActiveResource::Formats::XmlFormat
      def format # :nodoc:
        read_inheritable_attribute("format") || ActiveResource::Formats[:xml]
      end

203 204
      # 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
205
      # or not (defaults to <tt>false</tt>).
206
      def connection(refresh = false)
207 208
        if defined?(@connection) || superclass == Object
          @connection = Connection.new(site, format) if refresh || @connection.nil?
209 210 211 212
          @connection
        else
          superclass.connection
        end
213
      end
214

215 216
      def headers
        @headers ||= {}
217 218
      end

219 220 221 222
      # 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:

223 224 225
      attr_accessor_with_default(:collection_name) { element_name.pluralize } #:nodoc:
      attr_accessor_with_default(:primary_key, 'id') #:nodoc:
      
226 227
      # 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.
228 229 230
      def prefix(options={})
        default = site.path
        default << '/' unless default[-1..-1] == '/'
231
        # generate the actual method based on the current site path
232
        self.prefix = default
233 234
        prefix(options)
      end
235

236 237
      # 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.
238 239 240 241 242
      def prefix_source
        prefix # generate #prefix and #prefix_source methods first
        prefix_source
      end

243 244
      # Sets the prefix for a resource's nested URL (e.g., <tt>prefix/collectionname/1.xml</tt>).
      # Default value is <tt>site.path</tt>.
245
      def prefix=(value = '/')
246
        # Replace :placeholders with '#{embedded options[:lookups]}'
247
        prefix_call = value.gsub(/:\w+/) { |key| "\#{options[#{key}]}" }
248 249 250

        # Redefine the new methods.
        code = <<-end_code
251 252
          def prefix_source() "#{value}" end
          def prefix(options={}) "#{prefix_call}" end
253 254
        end_code
        silence_warnings { instance_eval code, __FILE__, __LINE__ }
255
      rescue
256
        logger.error "Couldn't set prefix: #{$!}\n  #{code}"
257
        raise
258
      end
259

260
      alias_method :set_prefix, :prefix=  #:nodoc:
261

262 263
      alias_method :set_element_name, :element_name=  #:nodoc:
      alias_method :set_collection_name, :collection_name=  #:nodoc:
264

265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284
      # 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
285 286 287
      #
      def element_path(id, prefix_options = {}, query_options = nil)
        prefix_options, query_options = split_options(prefix_options) if query_options.nil?
288
        "#{prefix(prefix_options)}#{collection_name}/#{id}.#{format.extension}#{query_string(query_options)}"
289
      end
290

291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310
      # 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
311 312 313
      #
      def collection_path(prefix_options = {}, query_options = nil)
        prefix_options, query_options = split_options(prefix_options) if query_options.nil?
314
        "#{prefix(prefix_options)}#{collection_name}.#{format.extension}#{query_string(query_options)}"
315
      end
316

317
      alias_method :set_primary_key, :primary_key=  #:nodoc:
318

319
      # Create a new resource instance and request to the remote service
320
      # that it be saved, making it equivalent to the following simultaneous calls:
321 322 323 324 325 326
      #
      #   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
327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348
      # 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
      #
349 350
      def create(attributes = {})
        returning(self.new(attributes)) { |res| res.save }        
351 352
      end

353
      # Core method for finding resources.  Used similarly to Active Record's find method.
354
      #
355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385
      # ==== 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
      #
386 387 388
      #   Person.find(:all, :from => :developers, :params => { :language => 'ruby' })
      #   # => GET /people/developers.xml?language=ruby
      #
389 390 391 392 393
      #   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
394
      def find(*arguments)
395 396
        scope   = arguments.slice!(0)
        options = arguments.slice!(0) || {}
397 398

        case scope
399 400
          when :all   then find_every(options)
          when :first then find_every(options).first
401
          when :one   then find_one(options)
402
          else             find_single(scope, options)
403 404
        end
      end
405

406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424
      # 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
      #
425 426
      def delete(id, options = {})
        connection.delete(element_path(id, options))
427 428
      end

429 430 431 432 433 434 435 436 437
      # 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
438 439 440 441 442 443
      def exists?(id, options = {})
        id && !find_single(id, options).nil?
      rescue ActiveResource::ResourceNotFound
        false
      end

444
      private
445
        # Find every resource
446
        def find_every(options)
447
          case from = options[:from]
448 449 450 451 452 453 454 455
          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)
456
            instantiate_collection( (connection.get(path, headers) || []), prefix_options )
457
          end
458 459
        end
        
460 461 462 463 464 465 466 467
        # 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))
468
          end
469
        end
470

471
        # Find a single resource from the default URL
472
        def find_single(scope, options)
473 474 475 476 477 478 479 480
          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
481

482 483
        def instantiate_record(record, prefix_options = {})
          returning new(record) do |resource|
484 485
            resource.prefix_options = prefix_options
          end
486
        end
487

488

489
        # Accepts a URI and creates the site URI from that.
490
        def create_site_uri_from(site)
491
          site.is_a?(URI) ? site.dup : URI.parse(site)
492
        end
493

494
        # contains a set of the current prefix parameters.
495 496 497 498
        def prefix_parameters
          @prefix_parameters ||= prefix_source.scan(/:\w+/).map { |key| key[1..-1].to_sym }.to_set
        end

499
        # Builds the query string for the request.
500
        def query_string(options)
501
          "?#{options.to_query}" unless options.nil? || options.empty? 
502 503 504 505 506
        end

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

          (options || {}).each do |key, value|
510 511
            next if key.blank?
            (prefix_parameters.include?(key.to_sym) ? prefix_options : query_options)[key.to_sym] = value
512
          end
513 514

          [ prefix_options, query_options ]
515
        end
516 517
    end

518 519
    attr_accessor :attributes #:nodoc:
    attr_accessor :prefix_options #:nodoc:
520

521 522 523 524 525 526 527 528 529 530 531
    # 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
532 533 534
    def initialize(attributes = {})
      @attributes     = {}
      @prefix_options = {}
535
      load(attributes)
536
    end
537

538 539 540 541 542 543 544 545 546 547 548 549 550 551 552
    # 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
    #
553
    def new?
554 555 556
      id.nil?
    end

557
    # Get the +id+ attribute of the resource.
558
    def id
559
      attributes[self.class.primary_key]
560
    end
561

562
    # Set the +id+ attribute of the resource.
563
    def id=(id)
564
      attributes[self.class.primary_key] = id
565
    end
566

567 568 569 570 571
    # Allows ActiveResource objects to be used as parameters in ActionPack URL generation.
    def to_param
      id && id.to_s
    end

572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593
    # 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
    #
594 595 596 597
    def ==(other)
      other.equal?(self) || (other.instance_of?(self.class) && !other.new? && other.id == id)
    end

598
    # Tests for equality (delegates to ==).
599 600 601 602 603 604 605 606 607
    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
608
    
609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624
    # 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
625
    def dup
626
      returning self.class.new do |resource|
627 628 629 630
        resource.attributes     = @attributes
        resource.prefix_options = @prefix_options
      end
    end
631

632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648
    # 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)
649
    def save
650
      new? ? create : update
651 652
    end

653 654 655 656 657 658 659 660 661 662 663 664 665 666 667
    # 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)
668
    def destroy
669
      connection.delete(element_path, self.class.headers)
670
    end
671

672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690
    # 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
691
    def exists?
692
      !new? && self.class.exists?(id, :params => prefix_options)
693 694
    end

695 696 697 698 699 700 701 702 703
    # 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
704
    #             replace underscores with dashes (default is <tt>false</tt>).
705
    # skip_instruct::  Toggle skipping the +instruct!+ call on the XML builder 
706
    #                  that generates the XML declaration (default is <tt>false</tt>).
707 708 709 710 711 712 713 714 715 716 717 718 719
    #
    # ==== 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>
720 721
    def to_xml(options={})
      attributes.to_xml({:root => self.class.element_name}.merge(options))
722
    end
723

724 725 726 727 728 729 730 731 732 733 734 735 736 737
    # 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
738
    def reload
739
      self.load(self.class.find(id, :params => @prefix_options).attributes)
740 741
    end

742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762
    # 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
763
    def load(attributes)
J
Jeremy Kemper 已提交
764
      raise ArgumentError, "expected an attributes Hash, got #{attributes.inspect}" unless attributes.is_a?(Hash)
765
      @prefix_options, attributes = split_options(attributes)
766 767 768 769 770 771 772
      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
773 774
              resource = find_or_create_resource_for(key)
              resource.new(value)
775 776 777 778
            else
              value.dup rescue value
          end
      end
779 780
      self
    end
781 782 783 784
    
    # For checking respond_to? without searching the attributes (which is faster).
    alias_method :respond_to_without_attributes?, :respond_to?

785
    # A method to determine if an object responds to a message (e.g., a method call). In Active Resource, a +Person+ object with a
786 787
    # +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>.
788 789 790 791 792 793 794 795 796 797 798 799 800 801
    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
    
802

803 804 805 806
    protected
      def connection(refresh = false)
        self.class.connection(refresh)
      end
807

808
      # Update the resource on the remote service.
809
      def update
810
        returning connection.put(element_path(prefix_options), to_xml, self.class.headers) do |response|
811 812
          load_attributes_from_response(response)
        end
813
      end
814

815
      # Create (i.e., save to the remote service) the new resource.
816
      def create
817
        returning connection.post(collection_path, to_xml, self.class.headers) do |response|
818
          self.id = id_from_response(response)
819
          load_attributes_from_response(response)
820
        end
821
      end
822 823
      
      def load_attributes_from_response(response)
824
        if response['Content-Length'] != "0" && response.body.strip.size > 0
825
          load(self.class.format.decode(response.body))
826
        end
827
      end
828

829
      # Takes a response from a typical create post and pulls the ID out
830 831 832 833
      def id_from_response(response)
        response['Location'][/\/([^\/]*?)(\.\w+)?$/, 1]
      end

834 835 836 837 838 839 840 841
      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

842
    private
843
      # Tries to find a resource for a given collection name; if it fails, then the resource is created
844 845 846
      def find_or_create_resource_for_collection(name)
        find_or_create_resource_for(name.to_s.singularize)
      end
847
      
848 849 850 851 852 853 854 855 856 857 858 859 860 861
      # 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

862
      # Tries to find a resource for a given name; if it fails, then the resource is created
863 864
      def find_or_create_resource_for(name)
        resource_name = name.to_s.camelize
865 866 867
        ancestors = self.class.name.split("::")
        if ancestors.size > 1
          find_resource_in_modules(resource_name, ancestors)
868 869 870
        else
          self.class.const_get(resource_name)
        end
871 872 873
      rescue NameError
        resource = self.class.const_set(resource_name, Class.new(ActiveResource::Base))
        resource.prefix = self.class.prefix
874
        resource.site   = self.class.site
875 876 877
        resource
      end

878
      def split_options(options = {})
879
        self.class.send!(:split_options, options)
880 881
      end

882
      def method_missing(method_symbol, *arguments) #:nodoc:
883
        method_name = method_symbol.to_s
884

885 886 887 888
        case method_name.last
          when "="
            attributes[method_name.first(-1)] = arguments.first
          when "?"
T
 
Tobias Lütke 已提交
889
            attributes[method_name.first(-1)]
890
          else
891
            attributes.has_key?(method_name) ? attributes[method_name] : super
892 893 894
        end
      end
  end
J
Jeremy Kemper 已提交
895
end