hachi 1.0.0 → 2.0.0

data/lib/hachi/models/base.rb

@@ -1,31 +0,0 @@
-# frozen_string_literal: true
-
-module Hachi
-  module Models
-    class Base
-      private
-
-      def validate_severity
-        return true if severity >= 1 && severity <= 3
-
-        raise ArgumentError, "severity should be 1 - 3 (1: low; 2: medium; 3: high)"
-      end
-
-      def validate_tlp
-        return true if tlp >= 0 && tlp <= 3
-
-        raise ArgumentError, "tlp should be 0 - 3 (0: white; 1: green; 2: amber; 3: red)"
-      end
-
-      def validate_status
-        return true if %w(New Updated Ignored Imported).include?(status)
-
-        raise ArgumentError, "status should be New, Updated, Ignored or Imported"
-      end
-
-      def validate_tags
-        raise ArgumentError, "tags should be an array" unless tags.is_a?(Array)
-      end
-    end
-  end
-end

data/lib/hachi/models/case.rb

@@ -1,54 +0,0 @@
-# frozen_string_literal: true
-
-module Hachi
-  module Models
-    class Case < Base
-      attr_reader :title, :description, :severity, :start_date, :owner, :flag, :tlp, :tags
-
-      def initialize(title:, description:, severity: nil, start_date: nil, owner: nil, flag: nil, tlp: nil, tags: nil)
-        @title = title
-        @description = description
-        @severity = severity
-        @start_date = start_date
-        @owner = owner
-        @flag = flag
-        @tlp = tlp
-        @tags = tags
-
-        validate_flag if flag
-        validate_severity if severity
-        validate_start_date if start_date
-        validate_tags if tags
-        validate_tlp if tlp
-      end
-
-      def payload
-        {
-          title: title,
-          description: description,
-          severity: severity,
-          startDate: start_date,
-          owner: owner,
-          flag: flag,
-          tlp: tlp,
-          tags: tags
-        }.compact
-      end
-
-      private
-
-      def validate_start_date
-        DateTime.parse(start_date)
-        true
-      rescue ArgumentError => _e
-        raise ArgumentError, "date should be Date format"
-      end
-
-      def validate_flag
-        return true if [true, false].include?(flag)
-
-        raise ArgumentError, "flag should be true or false"
-      end
-    end
-  end
-end

data/lib/hachi/models/user.rb

@@ -1,36 +0,0 @@
-# frozen_string_literal: true
-
-module Hachi
-  module Models
-    class User < Base
-      attr_reader :login, :name, :roles, :password
-
-      ROLES = %w(read write admin).freeze
-
-      def initialize(login:, name:, roles:, password:)
-        @login = login
-        @name = name
-        @roles = roles
-        @password = password
-
-        validate_roles
-      end
-
-      def payload
-        {
-          login: login,
-          name: name,
-          roles: roles,
-          password: password
-        }.compact
-      end
-
-      private
-
-      def validate_roles
-        raise ArgumentError, "roles should be an array" unless roles.is_a?(Array)
-        raise ArgumentError, "role should be one of #{ROLES.join('.')}" unless roles.all? { |role| ROLES.include? role }
-      end
-    end
-  end
-end

data/samples/01_create_an_alert.rb

@@ -1,17 +0,0 @@
-# frozen_string_literal: true
-
-$LOAD_PATH.unshift("#{__dir__}/../lib")
-
-require "hachi"
-
-api = Hachi::API.new
-
-# create a simple alert
-api.alert.create(title: "test", description: "test", type: "test", source: "test")
-
-# create an alert with artifacts
-artifacts = [
-  { data: "1.1.1.1", data_type: "ip", message: "test" },
-  { data: "github.com", data_type: "domain", tags: ["test"] }
-]
-api.alert.create(title: "test", description: "test", type: "test", source: "test", artifacts: artifacts)

data/samples/02_search_artifacts.rb

@@ -1,16 +0,0 @@
-# frozen_string_literal: true
-
-$LOAD_PATH.unshift("#{__dir__}/../lib")
-
-require "hachi"
-
-api = Hachi::API.new
-
-# search artifacts
-results = api.artifact.search(data: "1.1.1.1", data_type: "ip")
-ids = results.map { |result| result.dig("id") }
-
-ids.each do |id|
-  artifact = api.artifact.get_by_id(id)
-  p artifact
-end

data/samples/03_list_cases.rb

@@ -1,16 +0,0 @@
-# frozen_string_literal: true
-
-$LOAD_PATH.unshift("#{__dir__}/../lib")
-
-require "hachi"
-
-api = Hachi::API.new
-
-# list up cases
-results = api.case.list
-ids = results.map { |result| result.dig("id") }
-
-ids.each do |id|
-  kase = api.case.get_by_id(id)
-  p kase
-end

data/samples/04_merge_alerts.rb

@@ -1,17 +0,0 @@
-# frozen_string_literal: true
-
-$LOAD_PATH.unshift("#{__dir__}/../lib")
-
-require "hachi"
-
-def api
-  @api ||= Hachi::API.new
-end
-
-description = ARGV[0].to_s
-case_id = ARGV[1].to_s
-
-alerts = api.alert.search(description: description)
-alert_ids = alerts.map { |alert| alert.dig "id" }
-
-api.alert.merge_into_case(alert_ids, case_id)