wakame-dolphin 0.0.2

data/lib/dolphin/mailer.rb

@@ -0,0 +1,83 @@
+# -*- coding: utf-8 -*-
+
+require 'action_mailer'
+require 'mail-iso-2022-jp'
+
+module Dolphin
+  class Mailer < ActionMailer::Base
+    ActionMailer::Base.raise_delivery_errors = true
+    case Dolphin.settings['mail']['type']
+      when 'file'
+        ActionMailer::Base.delivery_method = :file
+        ActionMailer::Base.file_settings = {
+          :location => '/var/tmp'
+        }
+      when 'tls-mail'
+        ActionMailer::Base.delivery_method = :smtp
+        ActionMailer::Base.smtp_settings = {
+          :address => Dolphin.settings['mail']['host'],
+          :port => Dolphin.settings['mail']['port'],
+          :user_name => Dolphin.settings['mail']['user_name'],
+          :password => Dolphin.settings['mail']['password'],
+          :authentication => :plain,
+          :enable_starttls_auto => true
+        }
+      when 'mail'
+        ActionMailer::Base.delivery_method = :smtp
+        ActionMailer::Base.smtp_settings = {
+          :address => Dolphin.settings['mail']['host'],
+          :port => Dolphin.settings['mail']['port'],
+        }
+    end
+
+    default :charset => 'ISO-2022-JP'
+
+
+    #
+    # ==== Examples
+    #
+    #   read_iso2022_jp_mail('/var/tmp/test@example.com')
+    #
+    def self.read_iso2022_jp_mail(path, encoding_type='UTF-8')
+      data = File.open(path, 'rb').read
+      ec = Encoding::Converter.new("ISO-2022-JP", encoding_type)
+      converted_data = ec.convert(data)
+      data = converted_data.split("\r\n\r\n")
+      header = data[0].split("\r\n")
+      body = data[1]
+
+      h = Hash.new
+      h[:date] = header[0]
+      h[:from] = header[1]
+      h[:to] = header[2]
+      h[:message_id]  = header[3]
+      h[:subject] = header[4]
+      h[:mime_version] = header[5]
+      h[:event_id] = header[9]
+      h.map{|k,v| h[k] = v.split(':')[1].strip}
+
+      h[:body] = body
+      h
+    end
+
+    def notify(send_params)
+      if send_params[:from].blank?
+        raise 'Not found from field.'
+      end
+
+      fqdn = send_params[:from].split('@')[1]
+
+      m = mail(send_params)
+      m.message_id(generate_message_id(send_params[:event_id], fqdn))
+      m.deliver
+    end
+
+    private
+
+    # Refernced function from https://github.com/mikel/mail/blob/master/lib/mail/fields/message_id_field.rb#L77-L79
+    def generate_message_id(event_id, fqdn)
+      "<#{event_id}@#{fqdn}>"
+    end
+
+  end
+end

data/lib/dolphin/manager.rb

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+
+require 'celluloid'
+
+module Dolphin
+  class Manager < Celluloid::SupervisionGroup
+    include Dolphin::Util
+
+    trap_exit :actor_died
+
+    def actor_died(actor, reason)
+      logger :info, "Actor died"
+      restart_actor(actor, "Breaked actor")
+      logger :info, "Actor restarted"
+    end
+
+    def start
+    end
+
+    def shutdown
+    end
+
+    def run_workers
+    end
+
+    def terminate_workers
+    end
+
+    def run_request_handlers
+    end
+
+    def terminate_request_handlers
+    end
+  end
+end

data/lib/dolphin/message_builder.rb

@@ -0,0 +1,98 @@
+# -*- coding: utf-8 -*-
+require 'erubis'
+require 'extlib/blank'
+
+module Dolphin
+
+  class TemplateBuilder
+
+    include Dolphin::Helpers::Message::ZabbixHelper
+
+    def build(template_str, params)
+      template = Erubis::Eruby.new(template_str)
+      if params.is_a? Hash
+        params.each {|key, val|
+          instance_variable_set("@#{key}", val)
+        }
+      end
+      template.result(binding)
+    end
+  end
+
+  module MessageBuilder
+
+    EXT = '.erb'.freeze
+
+    class Base
+
+      include Dolphin::Util
+
+      def initialize
+      end
+
+      def build
+        raise NotImplementedError
+      end
+
+      def build_message(str, params)
+        begin
+          template = TemplateBuilder.new
+          template.build(str, params).encode("UTF-8")
+        rescue SyntaxError => e
+          logger :error, e
+        end
+      end
+    end
+
+    class Mail < Base
+
+      MESSAGE_BOUNDARY="----------------------------------------".freeze
+
+      def build(template_id, params)
+        message = ''
+
+        if template_id.blank?
+          template_id = 'default'
+        end
+
+        body_template = template(template_id)
+        if body_template.nil?
+          return nil
+        end
+
+        message = build_message(body_template, params['messages'])
+        subject, body = message.split(MESSAGE_BOUNDARY)
+        subject.strip! unless subject.nil?
+        body.strip! unless body.nil?
+
+        notification = NotificationObject.new
+        notification.subject = subject
+        notification.from = Dolphin.settings['mail']['from']
+        notification.to = params["to"]
+        notification.cc ||= params["cc"]
+        notification.bcc ||= params["bcc"]
+        notification.body = body
+        notification
+      end
+
+      private
+      def template(template_id)
+        file_path = File.join(template_path, template_file(template_id))
+        if File.exists? file_path
+          File.read(file_path, :encoding => Encoding::UTF_8)
+        else
+          logger :warn, "File not found #{file_path}"
+          nil
+        end
+      end
+
+      def template_file(template_id)
+        template_id + EXT
+      end
+
+      def template_path
+        File.join(Dolphin.templates_path, '/email')
+      end
+    end
+  end
+end

data/lib/dolphin/models/base.rb

@@ -0,0 +1,8 @@
+# -*- coding: utf-8 -*-
+
+module Dolphin
+  module Models
+    class Base
+    end
+  end
+end

data/lib/dolphin/models/cassandra/base.rb

@@ -0,0 +1,11 @@
+# -*- coding: utf-8 -*-
+module Dolphin
+  module Models::Cassandra
+    class Base
+      attr_accessor :db
+      def initialize(connection)
+        @db = connection
+      end
+    end
+  end
+end

data/lib/dolphin/models/cassandra/event.rb

@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+
+require 'multi_json'
+require 'simple_uuid'
+
+module Dolphin
+  module Models
+    module Cassandra
+      class Event < Base
+        COLUMN_FAMILY = 'events'.freeze
+        ROW_KEY = 'history'.freeze
+
+        def get(params)
+
+          options = {
+            :count => params[:count]
+          }
+
+          if params[:start_id]
+            options[:start] = params[:start_id]
+          elsif params[:start_time]
+            options[:start] = params[:start_time]
+          end
+
+          db.get(COLUMN_FAMILY, ROW_KEY, options).collect do |event| {
+            'id' => event[0].to_guid,
+            'event' => MultiJson.load(event[1]),
+            'created_at' => SimpleUUID::UUID.new(event[0].to_guid).to_time.iso8601
+          }
+          end
+        end
+
+        def put(event)
+          column_name = SimpleUUID::UUID.new(Time.now).to_guid
+          value = MultiJson.dump(event[:messages])
+          db.insert(COLUMN_FAMILY, ROW_KEY, {column_name => value})
+          column_name
+        end
+      end
+    end
+  end
+end

data/lib/dolphin/models/cassandra/notification.rb

@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*-
+
+require 'multi_json'
+
+module Dolphin
+  module Models
+    module Cassandra
+      class Notification < Base
+        def get(id)
+          res = db.get('notifications', id.to_s)
+          MultiJson.load(res['methods'])
+        end
+
+        def put(id, methods)
+          column_name = 'methods'
+          row_key = id.to_s
+          value = MultiJson.dump(methods)
+
+          db.insert('notifications', row_key, {column_name => value})
+        end
+
+        def delete(id)
+          db.remove('notifications', id)
+        end
+      end
+    end
+  end
+end

data/lib/dolphin/models/rdb/base.rb

@@ -0,0 +1,12 @@
+# -*- coding: utf-8 -*-
+
+module Dolphin::Models
+  module Rdb
+    class Base
+      attr_accessor :db
+      def initialize(model_class)
+        @db = model_class
+      end
+    end
+  end
+end

data/lib/dolphin/models/rdb/event.rb

@@ -0,0 +1,47 @@
+# -*- coding: utf-8 -*-
+
+require 'multi_json'
+require 'simple_uuid'
+
+module Dolphin::Models
+  module Rdb
+    class Event < Base
+      def get(params)
+        options = {
+          :count => params[:count]
+        }
+
+        if params[:start_id]
+          options[:start] = params[:start_id]
+        elsif params[:start_time]
+          options[:start] = params[:start_time]
+        end
+
+        if options[:start]
+          event = db.find(:uuid => params[:start_id])
+          h = {}
+          h['id'] = event.uuid
+          h['event'] = MultiJson.load(event.value)
+          h['created_at'] = SimpleUUID::UUID.new(event.uuid).to_time.iso8601
+          [h]
+        else
+          db.order(Sequel.desc(:id)).find_all.collect do |event| {
+            'id' => event.uuid,
+            'event' => MultiJson.load(event.value),
+            'created_at' => SimpleUUID::UUID.new(event.uuid).to_time.iso8601
+          }
+          end
+        end
+      end
+
+      def put(event)
+        d = db.new
+        d.uuid = SimpleUUID::UUID.new(Time.now).to_guid
+        d.value = MultiJson.dump(event[:messages])
+        d.timestamp = DateTime.now
+        d.save
+        d.uuid
+      end
+    end
+  end
+end

data/lib/dolphin/models/rdb/notification.rb

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+
+require 'multi_json'
+
+module Dolphin::Models
+  module Rdb
+    class Notification < Base
+      def get(id)
+        res = db.find(:uuid => id)
+        MultiJson.load(res.value)
+      end
+
+      def put(id, methods)
+        d = db.find(:uuid => id)
+        d = db.new if d.nil?
+        d.uuid = id.to_s
+        d.value = MultiJson.dump(methods)
+        d.save
+      end
+
+      def delete(id)
+        res = db.find(:uuid => id)
+        res.destroy
+      end
+    end
+  end
+end

data/lib/dolphin/models/rdb/orm/base.rb

@@ -0,0 +1,10 @@
+# -*- coding: utf-8 -*-
+
+require 'sequel/model'
+
+module Dolphin::Models::Rdb
+  module Orm
+    class Base < Sequel::Model
+    end
+  end
+end

data/lib/dolphin/models/rdb/orm/event.rb

@@ -0,0 +1,8 @@
+# -*- coding: utf-8 -*-
+
+module Dolphin::Models::Rdb
+  module Orm
+    class Event < Sequel::Model
+    end
+  end
+end

data/lib/dolphin/models/rdb/orm/notification.rb

@@ -0,0 +1,8 @@
+# -*- coding: utf-8 -*-
+
+module Dolphin::Models::Rdb
+  module Orm
+    class Notification < Sequel::Model
+    end
+  end
+end

data/lib/dolphin/query_processor.rb

@@ -0,0 +1,50 @@
+# -*- coding: utf-8 -*-
+
+require 'celluloid'
+
+module Dolphin
+  class QueryProcessor
+    include Celluloid
+    include Dolphin::Util
+
+    def get_notification(id)
+      logger :info, "Get notification #{id}"
+      send('get_notification', id)
+    end
+
+    def put_event(event)
+      logger :info, "Put event #{event}"
+      send('put_event', event)
+    end
+
+    def get_event(params)
+      send('get_event', params)
+    end
+
+    def put_notification(notification)
+      logger :info, notification
+      notification_id = notification[:id]
+      methods = notification[:methods]
+      send('put_notification', notification_id, methods)
+    end
+
+    def delete_notification(notification)
+      logger :info, notification
+      notification_id = notification[:id]
+      send('delete_notification', notification_id)
+    end
+
+    private
+    def send(action, *args)
+      begin
+        ds = DataStore.current_store
+        ds.connect
+        ds.__send__(action, *args)
+      rescue => e
+        logger :error, e.backtrace
+        logger :error, e
+        false
+      end
+    end
+  end
+end