mihari 3.9.0 → 3.9.1

data/lib/mihari/web/controllers/base_controller.rb

@@ -1,22 +0,0 @@
-# frozen_string_literal: true
-
-require "rack/contrib/json_body_parser"
-require "sinatra"
-require "sinatra/param"
-
-module Mihari
-  module Controllers
-    class BaseController < Sinatra::Base
-      helpers Sinatra::Param
-
-      use Rack::JSONBodyParser
-
-      set :show_exceptions, false
-      set :raise_sinatra_param_exceptions, true
-
-      error Sinatra::Param::InvalidParameterError do
-        json({ error: "#{env["sinatra.error"].param} is invalid" })
-      end
-    end
-  end
-end

data/lib/mihari/web/controllers/command_controller.rb

@@ -1,26 +0,0 @@
-# frozen_string_literal: true
-
-require "safe_shell"
-
-module Mihari
-  module Controllers
-    class CommandController < BaseController
-      post "/api/command" do
-        param :command, String, required: true
-
-        command = params["command"]
-        if command.nil?
-          status 400
-          return json({ message: "command is required" })
-        end
-
-        command = command.split
-
-        output = SafeShell.execute("mihari", *command)
-        success = $?.success?
-
-        json({ output: output, success: success })
-      end
-    end
-  end
-end

data/lib/mihari/web/controllers/config_controller.rb

@@ -1,13 +0,0 @@
-# frozen_string_literal: true
-
-module Mihari
-  module Controllers
-    class ConfigController < BaseController
-      get "/api/config" do
-        report = Status.check
-
-        json report.to_camelback_keys
-      end
-    end
-  end
-end

data/lib/mihari/web/controllers/ip_address_controller.rb

@@ -1,21 +0,0 @@
-# frozen_string_literal: true
-
-module Mihari
-  module Controllers
-    class IPAddressController < BaseController
-      get "/api/ip_addresses/:ip" do
-        param :ip, String, required: true
-
-        ip = params["ip"].to_s
-
-        data = Enrichers::IPInfo.query(ip)
-        if data.nil?
-          status 404
-          json({ message: "IP:#{ip} is not found" })
-        else
-          json data.to_hash
-        end
-      end
-    end
-  end
-end

data/lib/mihari/web/controllers/sources_controller.rb

@@ -1,12 +0,0 @@
-# frozen_string_literal: true
-
-module Mihari
-  module Controllers
-    class SourcesController < BaseController
-      get "/api/sources" do
-        sources = Mihari::Alert.distinct.pluck(:source)
-        json sources
-      end
-    end
-  end
-end

data/lib/mihari/web/controllers/tags_controller.rb

@@ -1,30 +0,0 @@
-# frozen_string_literal: true
-
-module Mihari
-  module Controllers
-    class TagsController < BaseController
-      get "/api/tags" do
-        tags = Mihari::Tag.distinct.pluck(:name)
-        json tags
-      end
-
-      delete "/api/tags/:name" do
-        param :name, String, required: true
-
-        name = params["name"].to_s
-
-        begin
-          Mihari::Tag.where(name: name).destroy_all
-
-          status 204
-          body ""
-        rescue ActiveRecord::RecordNotFound
-          status 404
-
-          message = { message: "Name:#{name} is not found" }
-          json message
-        end
-      end
-    end
-  end
-end

data/lib/mihari/web/helpers/json.rb

@@ -1,53 +0,0 @@
-# frozen_string_literal: true
-
-require "awrence"
-require "multi_json"
-require "sinatra/base"
-
-module Sinatra
-  module JSON
-    class << self
-      def encode(object)
-        ::MultiJson.dump(object)
-      end
-    end
-
-    def json(object, options = {})
-      object = object.to_camelback_keys
-
-      content_type resolve_content_type(options)
-      resolve_encoder_action object, resolve_encoder(options)
-    end
-
-    private
-
-    def resolve_content_type(options = {})
-      options[:content_type] || settings.json_content_type
-    end
-
-    def resolve_encoder(options = {})
-      options[:json_encoder] || settings.json_encoder
-    end
-
-    def resolve_encoder_action(object, encoder)
-      [:encode, :generate].each do |method|
-        return encoder.send(method, object) if encoder.respond_to? method
-      end
-
-      if encoder.is_a? Symbol
-        object.__send__(encoder)
-      else
-        fail "#{encoder} does not respond to #generate nor #encode"
-      end
-    end
-  end
-
-  Base.set :json_encoder do
-    ::MultiJson
-  end
-
-  Base.set :json_content_type, :json
-
-  # Load the JSON helpers in modular style automatically
-  Base.helpers JSON
-end