repository.rb in travis-1.1.0

- old
+ new

@@ -27,15 +27,21 @@
         def ==(other)
           other.to_s == self
         end
       end
 
-      attributes :slug, :description, :last_build_id, :last_build_number, :last_build_state, :last_build_duration, :last_build_language, :last_build_started_at, :last_build_finished_at
+      include States
+
+      # @!parse attr_reader :slug, :description
+      attributes :slug, :description, :last_build_id, :last_build_number, :last_build_state, :last_build_duration, :last_build_started_at, :last_build_finished_at
       inspect_info :slug
 
+      time :last_build_finished_at, :last_build_started_at
+
       one  :repo
       many :repos
+      aka  :repository
 
       def public_key
         attributes["public_key"] ||= begin
           payload = session.get_raw("/repos/#{id}/key")
           Key.new(payload.fetch('key'))
@@ -52,15 +58,74 @@
 
       def encrypt(value)
         key.encrypt(value)
       end
 
-      def last_build_started_at=(time)
-        set_attribute(:last_build_started_at, time(time))
+      # @!parse attr_reader :last_build
+      def last_build
+        return unless last_build_id
+        attributes['last_build'] ||= begin
+          last_build               = session.find_one(Build, last_build_id)
+          last_build.number        = last_build_number
+          last_build.state         = last_build_state
+          last_build.duration      = last_build_duration
+          last_build.started_at    = last_build_started_at
+          last_build.finished_at   = last_build_finished_at
+          last_build.repository_id = id
+          last_build
+        end
       end
 
-      def last_build_finished_at=(time)
-        set_attribute(:last_build_finished_at, time(time))
+      def builds(params = nil)
+        return each_build unless params
+        session.find_many(Build, params.merge(:repository_id => id))
       end
+
+      def build(number)
+        builds(:number => number.to_s).first
+      end
+
+      def recent_builds
+        builds({})
+      end
+
+      def each_build(params = nil, &block)
+        return enum_for(__method__, params) unless block_given?
+        params ||= {}
+        chunk = builds(params)
+        until chunk.empty?
+          chunk.each(&block)
+          number = chunk.last.number
+          chunk  = number == '1' ? [] : builds(params.merge(:after_number => number))
+        end
+        self
+      end
+
+      def job(number)
+        build_number = number.to_s[/^\d+/]
+        build        = build(build_number)
+        job          = build.jobs.detect { |j| j.number == number } if number != build_number
+        job        ||= build.jobs.first if build.jobs.size == 1
+        job
+      end
+
+      def set_hook(flag)
+        result = session.put_raw('/hooks/', :hook => { :id => id, :active => flag })
+        result['result']
+      end
+
+      def disable
+        set_hook(false)
+      end
+
+      def enable
+        set_hook(true)
+      end
+
+      private
+
+        def state
+          last_build_state
+        end
     end
   end
 end