kanamei-keystone 0.0.19 → 0.0.20

data/bin/disk_size_check.rb

@@ -0,0 +1,52 @@
+#!/usr/bin/ruby
+
+require 'rubygems'
+require 'keystone'
+require 'pit'
+require 'pp'
+
+include Keystone::Batch::Base
+
+config = Pit.get("disk_size_check")
+
+SMTP_SERVER = config["smtp"]
+MAIL_TO = [config["mailto"]]
+ERROR_MAIL_TO = MAIL_TO
+MAIL_FROM = config["mailfrom"]
+
+@os = Keystone::Os.get()
+pp @os
+
+def send_alert_mail(additional_body = "")
+  title = 'disk size alert'
+  body  = <<-BODY
+disk size alert at #{@os.hostname}(#{@os.ip_address.join(" , ")})
+#{@df_result}
+#{additional_body}
+BODY
+
+  Keystone::Mail::Send.send(
+    MAIL_FROM,
+    MAIL_TO,
+    title,
+    body,
+    {:retry_cnt=>3}
+  )
+end
+
+execute() do
+  info "start batch"
+  limit = 80
+  limit = Integer(ARGV[0]) if ARGV.size > 0
+
+  @df_result = @os.disk
+  @df_result.gsub(/(\d+)%/){|s|
+    if $1.to_i >= limit
+      warn 'log size alert'
+      send_alert_mail
+      break
+    end
+  }
+  info "finish batch"
+end
+

data/lib/keystone.rb

@@ -11,7 +11,7 @@ autoload :Moji          , 'moji'
 
 module Keystone
 
-  VERSION = '0.0.19'
+  VERSION = '0.0.20'
 
   autoload :StringUtil , 'keystone/string_util'
   autoload :Batch      , 'keystone/batch'

data/lib/keystone/mail/message.rb

@@ -0,0 +1,84 @@
+require 'kconv'
+
+module Keystone
+  module Mail
+    class Message
+      attr_accessor :subject, :body, :mail_to, :mail_from, :mail_from_text
+
+      #
+      # 普通なことをしたいならTMailを使いましょう
+      #
+      def initialize(opt={})
+        @subject, @body, @mail_from_text = "", "", nil
+        set_option(opt)
+      end
+
+      def to_src
+        header = ""
+        header += create_mailfrom_header
+        header += create_mailto_header
+        header += create_subject_header
+        header += "\n"
+        header += encode_body
+        return header
+      end
+      
+      def set_option(opt)
+        @subject = opt[:subject] if opt.key?(:subject)
+        @body = opt[:body] if opt.key?(:body)
+        @body = opt[:mail_to] if opt.key?(:mail_to)
+        @body = opt[:mail_from] if opt.key?(:mail_from)
+        @mail_from_text = opt[:mail_from_text] if opt.key?(:mail_from_text)
+      end
+      
+      private
+      
+      def create_mailfrom_header
+        if @mail_from_text.blank?
+          return "From: #{@mail_from}\n"
+        else
+          return to_mail_header("From", @mail_from_text).chomp + "<#{@mail_from}>\n"
+        end        
+      end
+      
+      def create_subject_header
+        to_mail_header("Subject",@subject)
+      end
+      
+      def create_mailto_header
+        if @mail_to.kind_of?(Array)
+          return "To: #{@mail_to.join(',')}\n"
+        else
+          return "To: #{@mail_to.to_s}\n"
+        end
+      end
+      
+      def encode_mime
+        raise "mime_encode must be implemented!!"
+      end
+      
+      def encode_body
+        raise "encoded_body must be implemented!!"
+      end
+      
+      def mime_header_string
+        raise "mime_header_string must be implemented!!"
+      end
+
+      private
+
+      def to_mail_header(header_name, plain_text)
+        ret = ""
+        str = plain_text.clone
+        str.gsub!(/(\r|\n)/,"")
+        str.split(//).split_by(10).each do |ar|
+          st = encode_mime(ar.join(''))
+          st = st.split(//,1).pack('m')
+          st = st.chomp
+          ret += " #{mime_header_string}#{st}?=\n"   
+        end
+        return "#{header_name}:"+ret
+      end
+    end
+  end
+end

data/lib/keystone/mail/message_factory.rb

@@ -0,0 +1,18 @@
+module Keystone
+  module Mail
+    class MessageFactory
+      def self.create(opt)
+        message = MessageIso2022jp.new(opt)
+        if opt.key?(:encoding)
+          case opt[:encoding]
+          when :sjis
+            message = MessageSjis.new(opt)
+          else
+            raise "encoding '#{opt[:encoding]}' is not supported"
+          end
+        end
+        return message
+      end
+    end
+  end
+end

data/lib/keystone/mail/message_iso2022jp.rb

@@ -0,0 +1,126 @@
+require 'kconv'
+
+module Keystone
+  module Mail
+    #
+    # encoding 未実装
+    #
+    #
+    class MessageIso2022jp < Keystone::Mail::Message
+      def encode_mime(st)
+        Kconv.tojis(st)
+      end
+      
+      def mime_header_string
+        "=?ISO-2022-JP?B?"        
+      end      
+
+      def encode_body
+        Kconv.tojis(@body)
+      end
+
+      # #
+      # # 普通なことをしたいならTMailを使いましょう
+      # #    class Send
+      # def initialize(opt={})
+      #   @smtp_addr, @smtp_port    = "127.0.0.1", 25
+      #   @subject, @mail_from_text = "" ,nil
+      #   @retry_cnt, @encoding     = 0, :iso2022jp
+      #   set_option(opt)
+      # end
+      # 
+      # #
+      # # from could be String or Array
+      # # opt 
+      # #   :smtp_addr 
+      # #   :smtp_port 
+      # #   :retry
+      # #   :mail_from_text 
+      # #   :encoding
+      # #
+      # def self.send(from, to, subject, body, opt={})
+      #   queue = self.new(opt)
+      #   queue.mail_to = to
+      #   queue.mail_from = from
+      #   queue.subject = subject
+      #   queue.body = body
+      #   queue.send
+      # end
+      # 
+      # def send
+      #   raise "mail_to must not be blank!" if @mail_to.blank?
+      #   raise "mail_from must not be blank!" if @mail_from.blank?
+      #   header = ""
+      #   header += create_mailfrom_header(@mail_from,@mail_from_text)
+      #   header += create_mailto_header(@mail_to)
+      #   header += create_subject_header(@subject)
+      # 
+      #   case @encoding
+      #   when :iso2022jp
+      #     body_encoded = Kconv.tojis(@body)
+      #   else
+      #     raise "encoding #{encoding.to_s} can not use"
+      #   end
+      #   
+      #   src = header + "\n" + body_encoded
+      #   debug src
+      #   try_cnt = 0
+      #   begin
+      #     m = Net::SMTPSession.new(smtp_addr, smtp_port)
+      #     m.start()
+      #     m.sendmail(src ,@mail_from ,@mail_to)
+      #     m.finish
+      #   rescue => e
+      #     debug "try_cnt:#{try_cnt}"
+      #     try_cnt += 1
+      #     sleep 1
+      #     retry if @retry_cnt >= try_cnt
+      #     raise e
+      #   end
+      # end
+      # 
+      # def set_option(opt)
+      #   @smtp_addr = opt[:smtp_addr] if opt.key?(:smtp_addr)
+      #   @smtp_port = opt[:smtp_port] if opt.key?(:smtp_port)
+      #   @retry_cnt = Integer(opt[:retry_cnt]) if opt.key?(:retry_cnt)
+      #   @mail_from_text = opt[:mail_from_text] if opt.key?(:mail_from_text)
+      #   @encoding = opt[:encoding] if opt.key?(:encoding)
+      # end
+      # 
+      # private
+      # 
+      # def to_mail_header(header_name, plain_text)
+      #   ret = ""
+      #   str = plain_text.clone
+      #   str.gsub!(/(\r|\n)/,"")
+      #   str.split(//).split_by(10).each do |ar|
+      #     st = ar.join('')
+      #     st = Kconv.tojis(st)
+      #     st = st.split(//,1).pack('m'); st = st.chomp
+      #     ret += " =?ISO-2022-JP?B?#{st}?=\n"   
+      #   end
+      #   return "#{header_name}:"+ret
+      # end
+      # 
+      # def create_mailfrom_header(from, from_text)
+      #   if from_text.blank?
+      #     return "From: #{from}\n"
+      #   else
+      #     return to_mail_header("From", from_text).chomp + "<#{from}>\n"
+      #   end
+      # end
+      # 
+      # def create_subject_header(subject)
+      #   to_mail_header("Subject",subject)
+      # end
+      # 
+      # def create_mailto_header(to)
+      #   if to.class.to_s == 'Array'
+      #     return "To: #{to.join(',')}\n"
+      #   else
+      #     return "To: #{to.to_s}\n"
+      #   end
+      # end
+    end
+  end
+end

data/lib/keystone/mail/message_sjis.rb

@@ -0,0 +1,23 @@
+require 'kconv'
+
+module Keystone
+  module Mail
+    #
+    # encoding 未実装
+    #
+    #
+    class MessageSjis < Keystone::Mail::Message
+      def encode_mime(st)
+        Kconv.tosjis(st)
+      end
+      
+      def mime_header_string
+        "=?ShiftJIS?B?"        
+      end      
+
+      def encode_body
+        Kconv.tosjis(@body)
+      end      
+    end
+  end
+end

data/test/test_core_ext.rb

@@ -0,0 +1,15 @@
+require 'rubygems'
+require 'redgreen'
+require 'lib/keystone/core_ext'
+
+class TestCoreExt < Test::Unit::TestCase
+  def setup
+  end  
+  
+  def test_array_split_by
+    assert_equal([1,2,3,4,5,6,7,8,9].split_by(3),[[1, 2, 3], [4, 5, 6], [7, 8, 9]])
+    assert_equal([1,2,3,4,5,6,7,8,9,10].split_by(2),[[1,2],[3,4],[5,6],[7,8],[9,10]])
+    assert_equal([].split_by(3),[[]])
+    assert_equal([[1,2,3,4],[1,23],[2,33],[[4]]].split_by(3),[[[1, 2, 3, 4], [1, 23], [2, 33]], [[[4]]]])
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: kanamei-keystone
 version: !ruby/object:Gem::Version 
-  version: 0.0.19
+  version: 0.0.20
 platform: ruby
 authors: 
 - kanamei
@@ -22,9 +22,11 @@ extensions: []
 extra_rdoc_files: []
 
 files: 
+- bin/disk_size_check.rb
 - example/batch_sample.rb
 - example/os.rb
 - example/string_util.rb
+- test/test_core_ext.rb
 - test/test_string_util.rb
 - lib/keystone
 - lib/keystone/base.rb
@@ -40,6 +42,10 @@ files:
 - lib/keystone/core_ext/uri.rb
 - lib/keystone/core_ext.rb
 - lib/keystone/mail
+- lib/keystone/mail/message.rb
+- lib/keystone/mail/message_factory.rb
+- lib/keystone/mail/message_iso2022jp.rb
+- lib/keystone/mail/message_sjis.rb
 - lib/keystone/mail/send.rb
 - lib/keystone/mail.rb
 - lib/keystone/os