harvested2 5.0.3

data/lib/ext/hash.rb

@@ -0,0 +1,17 @@
+# Shamelessly ripped from https://github.com/rails/rails/blob/master/activesupport/lib/active_support/core_ext/hash/keys.rb
+
+unless Hash.respond_to?(:stringify_keys)
+  class Hash
+    # Return a new hash with all keys converted to strings.
+    def stringify_keys
+      dup.stringify_keys!
+    end
+    
+    def stringify_keys!
+      keys.each do |key|
+        self[key.to_s] = delete(key)
+      end
+      self
+    end
+  end
+end

data/lib/ext/time.rb

@@ -0,0 +1,5 @@
+unless Time.respond_to?(:to_time)
+  class Time
+    def to_time; self; end
+  end
+end

data/lib/harvest/account.rb

@@ -0,0 +1,13 @@
+module Harvest
+  # The model that contains information about a client
+  #
+  # == Fields
+  # [+user+] user attributes
+  # [+accounts+] accounts attributes
+  class Account < Hashie::Mash
+    include Harvest::Model
+
+    skip_json_root true
+    api_path '/accounts'
+  end
+end

data/lib/harvest/api/account.rb

@@ -0,0 +1,25 @@
+module Harvest
+  module API
+    
+    # API Methods to contain all account actions
+    class Account < Base
+      
+      # Returns the current rate limit information
+      # @return [Harvest::RateLimitStatus]
+      def rate_limit_status
+        response = request(:get, credentials, '/account/rate_limit_status')
+        Harvest::RateLimitStatus.parse(response.body).first
+      end
+      
+      # Returns the current logged in user
+      # @return [Harvest::User]
+      def who_am_i
+        response = request(:get, credentials, '/account/who_am_i')
+        parsed = JSON.parse(response.body)
+        Harvest::User.parse(parsed).first.tap do |user|
+          user.company = parsed["company"]
+        end
+      end
+    end
+  end
+end

data/lib/harvest/api/base.rb

@@ -0,0 +1,72 @@
+module Harvest
+  module API
+    class Base
+      attr_reader :credentials
+
+      def initialize(credentials)
+        @credentials = credentials
+      end
+
+      class << self
+        def api_model(klass)
+          class_eval <<-END
+            def api_model
+              #{klass}
+            end
+          END
+        end
+      end
+
+      protected
+
+      def request(method, credentials, path, options = {})
+        params = {
+          path: path,
+          options: options,
+          method: method
+        }
+
+        httparty_options = {
+          query:  options[:query],
+          body:   options[:body],
+          format: :plain,
+          headers: {
+            "Accept"       => "application/json",
+            "Content-Type" => "application/json; charset=utf-8",
+            "User-Agent"   => "Harvested/#{Harvest::VERSION}"
+          }.update(options[:headers] || {})
+        }
+
+        credentials.set_authentication(httparty_options)
+        response = HTTParty.send(method, "#{credentials.host}#{path}",
+          httparty_options)
+
+        params[:response] = response.inspect.to_s
+
+        case response.code
+        when 200..201
+          response
+        when 400
+          raise Harvest::BadRequest.new(response, params)
+        when 401
+          raise Harvest::AuthenticationFailed.new(response, params)
+        when 404
+          raise Harvest::NotFound.new(response, params,
+            'Do you have sufficient privileges?')
+        when 500
+          raise Harvest::ServerError.new(response, params)
+        when 502
+          raise Harvest::Unavailable.new(response, params)
+        when 503
+          raise Harvest::RateLimited.new(response, params)
+        else
+          raise Harvest::InformHarvest.new(response, params)
+        end
+      end
+
+      def to_json(json)
+        parsed = String === json ? JSON.parse(json) : json
+      end
+    end
+  end
+end

data/lib/harvest/api/clients.rb

@@ -0,0 +1,10 @@
+module Harvest
+  module API
+    class Clients < Base
+      include Harvest::Behavior::Crud
+      include Harvest::Behavior::Activatable
+
+      api_model Harvest::Client
+    end
+  end
+end

data/lib/harvest/api/company.rb

@@ -0,0 +1,12 @@
+module Harvest
+  module API
+    class Company < Base
+      api_model Harvest::Company
+
+      def info
+        response = request(:get, credentials, api_model.api_path)
+        api_model.parse(response.parsed_response)
+      end
+    end
+  end
+end

data/lib/harvest/api/contacts.rb

@@ -0,0 +1,9 @@
+module Harvest
+  module API
+    class Contacts < Base
+      include Harvest::Behavior::Crud
+
+      api_model Harvest::Contact
+    end
+  end
+end

data/lib/harvest/api/expense_categories.rb

@@ -0,0 +1,9 @@
+module Harvest
+  module API
+    class ExpenseCategories < Base
+      include Harvest::Behavior::Crud
+
+      api_model Harvest::ExpenseCategory
+    end
+  end
+end

data/lib/harvest/api/expenses.rb

@@ -0,0 +1,26 @@
+module Harvest
+  module API
+    class Expenses < Base
+      api_model Harvest::Expense
+      include Harvest::Behavior::Crud
+
+      def attach(expense, filename, receipt)
+        body = ""
+        body << "--__X_ATTACH_BOUNDARY__\r\n"
+        body << %Q{Content-Disposition: form-data; name="expense[receipt]"; filename="#{filename}"\r\n}
+        body << "\r\n#{receipt.read}"
+        body << "\r\n--__X_ATTACH_BOUNDARY__--\r\n\r\n"
+
+        request(
+          :post,
+          credentials,
+          "#{api_model.api_path}/#{expense.id}/receipt",
+          :headers => {
+            'Content-Type' => 'multipart/form-data; charset=utf-8; boundary=__X_ATTACH_BOUNDARY__',
+            'Content-Length' => body.length.to_s,
+          },
+          :body => body)
+      end
+    end
+  end
+end

data/lib/harvest/api/invoice_categories.rb

@@ -0,0 +1,9 @@
+module Harvest
+  module API
+    class InvoiceCategories < Base
+      include Harvest::Behavior::Crud
+
+      api_model Harvest::InvoiceCategory
+    end
+  end
+end

data/lib/harvest/api/invoice_messages.rb

@@ -0,0 +1,86 @@
+module Harvest
+  module API
+    class InvoiceMessages < Base
+      api_model Harvest::InvoiceMessage
+
+      def all(invoice, query = {})
+        response = request(:get, credentials, "/invoices/#{invoice.id}/messages", query: query)
+        response_parsed = api_model.to_json(response.parsed_response)
+
+        if response_parsed['total_pages'] > 1
+          counter = response_parsed['page']
+
+          while counter <= response_parsed['total_pages'] do
+            counter += 1
+            query = { 'page' => counter }
+
+            response_page = request(:get, credentials,
+              "/invoices/#{invoice.id}/messages",
+              query: query)
+            invoice_messages = api_model.to_json(response.parsed_response)
+            response_parsed['invoice_messages']
+              .concat(invoice_messages['invoice_messages'])
+          end
+        end
+
+        api_model.parse(response_parsed)
+      end
+
+      def create(invoice, invoice_message)
+        invoice_message = api_model.wrap(invoice_message)
+        response = request(:post, credentials, "/invoices/#{invoice.id}/messages", body: invoice_message.to_json)
+        find(invoice_message.id)
+      end
+
+      def delete(invoice, invoice_message)
+        request(:delete, credentials, "/invoices/#{invoice.id}/messages/#{invoice_message.id}")
+        message.id
+      end
+
+      # Create a message for marking an invoice as sent.
+      #
+      # @param [Harvest::InvoiceMessage] The message you want to send
+      # @return [Harvest::InvoiceMessage] The sent message
+      def mark_as_sent(invoice, invoice_message)
+        send_status_message(invoice, invoice_message, 'send')
+      end
+
+      # Create a message and mark an open invoice as closed (writing an invoice off)
+      #
+      # @param [Harvest::InvoiceMessage] The message you want to send
+      # @return [Harvest::InvoiceMessage] The sent message
+      def mark_as_closed(invoice, invoice_message)
+        send_status_message(invoice, invoice_message, 'close')
+      end
+
+      # Create a message and mark a closed (written-off) invoice as open
+      #
+      # @param [Harvest::InvoiceMessage] The message you want to send
+      # @return [Harvest::InvoiceMessage] The sent message
+      def re_open(invoice, invoice_message)
+        send_status_message(invoice, invoice_message, 're-open')
+      end
+
+      # Create a message for marking an open invoice as draft
+      #
+      # @param [Harvest::InvoiceMessage] The message you want to send
+      # @return [Harvest::InvoiceMessage] The sent message
+      def mark_as_draft(invoice_message)
+        send_status_message(invoice, invoice_message, 'draft')
+      end
+
+      private
+
+      def send_status_message(invoice, invoice_message, action)
+        invoice_message = api_model.wrap(invoice_message)
+        response = request( :post,
+                            credentials,
+                            "/invoices/#{invoice.id}/messages",
+                            event_type: action,
+                            body: invoice_message.to_json)
+
+        invoice_message
+      end
+    end
+  end
+end

data/lib/harvest/api/invoice_payments.rb

@@ -0,0 +1,41 @@
+module Harvest
+  module API
+    class InvoicePayments < Base
+      api_model Harvest::InvoicePayment
+
+      def all(invoice, query = {})
+        response = request(:get, credentials, "/invoices/#{invoice.id}/payments", query: query)
+        response_parsed = api_model.to_json(response.parsed_response)
+
+        if response_parsed['total_pages'].to_i > 1
+          counter = response_parsed['page'].to_i
+
+          while counter <= response_parsed['total_pages'].to_i do
+            counter += 1
+            query = { 'page' => counter }
+
+            response_page = request(:get, credentials,
+              "/invoices/#{invoice.id}/payments",
+              query: query)
+            invoice_payments = api_model.to_json(response.parsed_response)
+            response_parsed['invoice_payments']
+              .concat(invoice_payments['invoice_payments'])
+          end
+        end
+
+        api_model.parse(response_parsed)
+      end
+
+      def create(invoice, invoice_payment)
+        invoice_payment = api_model.wrap(invoice_payment)
+        response = request(:post, credentials, "/invoices/#{invoice.id}/payments", body: invoice_payment.to_json)
+        find(invoice_payment.id)
+      end
+
+      def delete(invoice, invoice_payment)
+        request(:delete, credentials, "/invoices/#{invoice.id}/payments/#{invoice_payment.id}")
+        invoice_payment.id
+      end
+    end
+  end
+end

data/lib/harvest/api/invoices.rb

@@ -0,0 +1,9 @@
+module Harvest
+  module API
+    class Invoices < Base
+      api_model Harvest::Invoice
+
+      include Harvest::Behavior::Crud
+    end
+  end
+end

data/lib/harvest/api/projects.rb

@@ -0,0 +1,9 @@
+module Harvest
+  module API
+    class Projects < Base
+      api_model Harvest::Project
+
+      include Harvest::Behavior::Crud
+    end
+  end
+end

data/lib/harvest/api/task_assignments.rb

@@ -0,0 +1,75 @@
+module Harvest
+  module API
+    class TaskAssignments < Base
+      api_model Harvest::TaskAssignment
+
+      def all(project, query = {})
+        api_path = "/projects/#{project.id}/task_assignments"
+        response = request(:get, credentials, api_path, query: query)
+        response_parsed = api_model.to_json(response.parsed_response)
+
+        if response_parsed['total_pages'].to_i > 1
+          counter = response_parsed['page'].to_i
+
+          while counter <= response_parsed['total_pages'].to_i do
+            counter += 1
+            query = { 'page' => counter }
+            response_page = request(:get, credentials, api_path, query: query)
+            project_task_assignments = api_model.to_json(response.parsed_response)
+            response_parsed['task_assignments']
+              .concat(project_task_assignments['task_assignments'])
+          end
+        end
+
+        api_model.parse(response_parsed)
+      end
+
+      def find(project, task_assignment)
+        api_path = "/projects/#{project.id}/task_assignments/#{task_assignment.id}"
+        response = request(:get, credentials, api_path)
+        api_model.parse(response.parsed_response)
+      end
+
+      def create(project, task_assignment)
+        task_assignment = api_model.wrap(task_assignment)
+        api_path = "/projects/#{project.id}/task_assignments"
+        response = request(:post, credentials, api_path, body: task_assignment.to_json)
+        find(task_assignment.project_id, task_assignment.id)
+      end
+
+      def update(project, task_assignment)
+        task_assignment = api_model.wrap(task_assignment)
+        api_path = "/projects/#{project.id}/task_assignments/#{task_assignment.id}"
+        request(:put, credentials, api_path, body: task_assignment.to_json)
+        find(task_assignment.project_id, task_assignment.id)
+      end
+
+      def delete(task_assignment)
+        api_path = "/projects/#{project.id}/task_assignments/#{task_assignment.id}"
+        response = request(:delete, credentials, api_path)
+        task_assignment.id
+      end
+
+      def me(query = {})
+        api_path = "/task_assignments"
+        response = request(:get, credentials, api_path, query: query)
+        response_parsed = api_model.to_json(response.parsed_response)
+
+        if response_parsed['total_pages'].to_i > 1
+          counter = response_parsed['page'].to_i
+
+          while counter <= response_parsed['total_pages'].to_i do
+            counter += 1
+            query = { 'page' => counter }
+            response_page = request(:get, credentials, api_path, query: query)
+            task_assignments = api_model.to_json(response.parsed_response)
+            response_parsed['task_assignments']
+              .concat(task_assignments['task_assignments'])
+          end
+        end
+
+        api_model.parse(response_parsed)
+      end
+    end
+  end
+end