hackerone-client 0.17.0 → 0.21.0

data/lib/hackerone/client/activity.rb

@@ -15,6 +15,12 @@ module HackerOne
           attributes.internal
         end
 
+        def attachments
+          @attachments ||= activity.relationships.fetch(:attachments, {})
+              .fetch(:data, [])
+              .map { |attachment| HackerOne::Client::Attachment.new(attachment) }
+        end
+
         private
 
         def relationships
@@ -85,6 +91,9 @@ module HackerOne
         delegate :message, :bounty_amount, :bonus_amount, to: :attributes
       end
 
+      class ReportLocked < Activity
+      end
+
       ACTIVITY_TYPE_CLASS_MAPPING = {
         "activity-bounty-awarded" => BountyAwarded,
         "activity-swag-awarded" => SwagAwarded,
@@ -93,7 +102,8 @@ module HackerOne
         "activity-bug-triaged" => BugTriaged,
         "activity-reference-id-added" => ReferenceIdAdded,
         "activity-comment" => CommentAdded,
-        "activity-bounty-suggested" => BountySuggested
+        "activity-bounty-suggested" => BountySuggested,
+        "activity-comments-closed" => ReportLocked
       }.freeze
 
       def self.build(activity_data)

data/lib/hackerone/client/attachment.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module HackerOne
+  module Client
+    class Attachment
+      delegate :expiring_url, :file_name, :content_type, :created_at, \
+        :file_size, to: :attributes
+
+      def initialize(attachment)
+        @attachment = attachment
+      end
+
+      def id
+        @attachment[:id]
+      end
+
+      private
+
+      def attributes
+        OpenStruct.new(@attachment[:attributes])
+      end
+    end
+  end
+end

data/lib/hackerone/client/billing_balance.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module HackerOne
+  module Client
+    class BillingBalance
+      delegate :balance, to: :attributes
+
+      def initialize(billing_balance)
+        @billing_balance = OpenStruct.new billing_balance
+      end
+
+      private
+      def attributes
+        OpenStruct.new(@billing_balance[:attributes])
+      end
+    end
+  end
+end

data/lib/hackerone/client/program.rb

@@ -76,6 +76,13 @@ module HackerOne
         response_body.map { |r| Swag.new(r, self) }
       end
 
+      def balance
+        response_body = make_get_request(
+         "programs/#{id}/billing/balance"
+       )
+       BillingBalance.new(response_body).balance
+      end
+
       private
 
       def members

data/lib/hackerone/client/report.rb

@@ -26,6 +26,14 @@ module HackerOne
         duplicate
       ).map(&:to_sym).freeze
 
+      RESOLVED_STATES = %w(
+        resolved
+        not-applicable
+        informative
+        duplicate
+        spam
+      ).map(&:to_sym).freeze
+
       SEVERITY_RATINGS = %w(
         none
         low
@@ -135,6 +143,12 @@ module HackerOne
         classification_label.split("-").first
       end
 
+      def attachments
+        @attachments ||= relationships.fetch(:attachments, {})
+            .fetch(:data, [])
+            .map { |attachment| HackerOne::Client::Attachment.new(attachment) }
+      end
+
       def activities
         if ships = relationships.fetch(:activities, {}).fetch(:data, [])
           ships.map do |activity_data|
@@ -174,7 +188,7 @@ module HackerOne
       end
 
       def update_severity(rating:)
-        raise ArgumentError, "Invalid severity rating" unless SEVERITY_RATINGS.include?(rating)
+        raise ArgumentError, "Invalid severity rating" unless SEVERITY_RATINGS.include?(rating.to_s)
 
         request_body = {
           type: "severity",
@@ -292,6 +306,19 @@ module HackerOne
         HackerOne::Client::Activities.build(response_json)
       end
 
+      def lock!
+        unless RESOLVED_STATES.include? self.state.to_sym
+          raise ArgumentError, "Report must be closed before locking"
+        end
+
+        body = {
+          type: "activity-comments-closed"
+        }
+
+        response_json = make_put_request("reports/#{id}/close_comments", request_body: body)
+        HackerOne::Client::Activities.build(response_json)
+      end
+
       def assign_to_user(name)
         member = program.find_member(name)
         _assign_to(member.user.id, :user)

data/lib/hackerone/client/version.rb

@@ -2,6 +2,6 @@
 
 module Hackerone
   module Client
-    VERSION = "0.17.0"
+    VERSION = "0.21.0"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: hackerone-client
 version: !ruby/object:Gem::Version
-  version: 0.17.0
+  version: 0.21.0
 platform: ruby
 authors:
 - Neil Matatall
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-03-24 00:00:00.000000000 Z
+date: 2021-02-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -142,12 +142,18 @@ files:
 - fixtures/vcr_cassettes/award_a_bounty.yml
 - fixtures/vcr_cassettes/award_swag.yml
 - fixtures/vcr_cassettes/common_responses.yml
+- fixtures/vcr_cassettes/create_report.yml
+- fixtures/vcr_cassettes/create_report_invalid.yml
 - fixtures/vcr_cassettes/dup.yml
 - fixtures/vcr_cassettes/empty_report_list.yml
+- fixtures/vcr_cassettes/get_balance.yml
+- fixtures/vcr_cassettes/lock_report.yml
 - fixtures/vcr_cassettes/missing_report.yml
 - fixtures/vcr_cassettes/programs.yml
 - fixtures/vcr_cassettes/report.yml
 - fixtures/vcr_cassettes/report_list.yml
+- fixtures/vcr_cassettes/report_list_before.yml
+- fixtures/vcr_cassettes/report_list_triaged.yml
 - fixtures/vcr_cassettes/reporters.yml
 - fixtures/vcr_cassettes/server_error.yml
 - fixtures/vcr_cassettes/server_error_when_assigning_report_to_user.yml
@@ -165,6 +171,8 @@ files:
 - lib/hackerone/client.rb
 - lib/hackerone/client/activity.rb
 - lib/hackerone/client/address.rb
+- lib/hackerone/client/attachment.rb
+- lib/hackerone/client/billing_balance.rb
 - lib/hackerone/client/bounty.rb
 - lib/hackerone/client/group.rb
 - lib/hackerone/client/incremental/activities.rb