ruby 符合RFC 5322的消息ID生成器

Posted 2021-05-16

require "socket"

class MessageID
  UnsupportedFormatException = Class.new(StandardError)

  MESSAGE_ID_FORMAT = "<%s.%s.%s%s@%s>"
  STAMP_FIELD_FORMAT = "%Y%m%d%H%M%S%6N"

  def initialize(*ident)
    @ident = ident.compact.flatten
    @ident << Random.rand(999_999_999) if @ident.empty?
  end

  def id(format=:short)
    raise(UnsupportedFormatException) unless %I{ short long }.include?(format)
    sprintf(MESSAGE_ID_FORMAT, *fields(format == :short))
  end
  alias :to_str :id
  alias :to_s :id

  private

  def timestamp
    @timestamp ||= Time.now.utc
  end

  def random_id
    @random_id ||= Random.rand(999_999_999)
  end

  def fields(shorten)
    values = [
      timestamp.strftime(STAMP_FIELD_FORMAT).to_i,
      (shorten ? base36ify(@ident) : @ident).join("."),
      Process.pid,
      random_id,
      Socket.gethostname
    ]
    shorten ? base36ify(values) : values
  end

  def base36ify(fields)
    fields.collect { |v| v.is_a?(Integer) ? v.to_s(36) : v }
  end
end

puts MessageID.new  #=> <[e-mail]>

5.times { puts MessageID.new }
#=> <[e-mail]>
#   <[e-mail]>
#   <[e-mail]>
#   <[e-mail]>
#   <[e-mail]>

msgid = MessageID.new(20_345_678, 23_456, 23, 234)

puts msgid              #=> <[e-mail]>
puts msgid.to_s(:long)  #=> <[e-mail]>

require "date"

class MessageID
  module Decode
    STAMP_FIELD_FORMAT = "%Y%m%d%H%M%S%N"

    def self.decode(id)
      user, host = id.split("@")
      parts = user.split(".").map { |part| part.to_i(36) }
      stamp, ident = parts.shift, []
      while (parts.length > 1) do ident << parts.shift; end
      counter = parts.last
      stamp = DateTime.strptime(stamp.to_s, STAMP_FIELD_FORMAT)

      {
        stamp:   stamp.strftime("%Y-%m-%d %H:%M:%S.%6N"),
        ident:   ident,
        counter: counter,
        host:    host
      }
    end
  end
end

ids = %w{
  [e-mail]
  [e-mail]
  [e-mail]
  [e-mail]
}

ids.each { |id| p MessageID::Decode.decode(id) }