egn.rb in egn-0.1.0

- old
+ new
@@ -1,39 +1,103 @@
-require "egn/version"
+# require "egn/version"
 
 module Egn
   PARITY_WEIGHTS = [2,4,8,5,10,9,7,3,6]
 
-  def self.generate
-    year = Random.rand(1800..2000)
-    mon = Random.rand(1..12)
-    day = Random.rand(1..31)
-    cent = year - (year % 100)
-    sex = Random.rand(1..2)
+  module Generator
+    def self.egn
+      year = Random.rand(1800..2099)
+      mon = Random.rand(1..12)
+      day = Random.rand(1..31)
+      cent = year - (year % 100)
+      sex = Random.rand(1..2)
 
-    if cent == 1800
-      mon += 20
-    elsif cent == 2000
-      mon += 40
+      if cent == 1800
+        mon += 20
+      elsif cent == 2000
+        mon += 40
+      end
+
+      region = Random.rand(0..999)
+
+      if sex == 1 && (region %2 != 0)
+        region -= 1
+      elsif sex == 2 && (region % 2 == 0)
+        region += 1
+      end
+
+      final_year = year - cent
+      egn = final_year.to_s.rjust(2, '0') + mon.to_s.rjust(2, '0') + day.to_s.rjust(2,'0') + region.to_s.rjust(3,'0')
+
+      return egn + egn_checksum(egn).to_s
     end
 
-    region = Random.rand(0..999)
+    def self.egn_checksum(egn)
+      sum = egn.split('').map(&:to_i).zip(PARITY_WEIGHTS).map { |n| n.reduce(:*) }.reduce(:+)
 
-    if sex == 1 && (region %2 != 0)
-      region -= 1
-    elsif sex == 2 && (region % 2 == 0)
-      region += 1
+      rest = sum % 11
+      rest < 10 ? rest : 0
     end
 
-    final_year = year - cent
-    egn = final_year.to_s.rjust(2, '0') + mon.to_s.rjust(2, '0') + day.to_s.rjust(2,'0') + region.to_s.rjust(3,'0')
 
-    return egn + gen_checksum(egn).to_s
   end
 
-  def self.gen_checksum(egn)
-    sum = egn.split('').map(&:to_i).zip(PARITY_WEIGHTS).map { |n| n.reduce(:*) }.reduce(:+)
+  module Validator
+    def self.egn(egn)
+      return false unless egn.length == 10
 
-    rest = sum % 11
-    rest < 10 ? rest : 0
+      year, month, day = egn.scan(/.{1,2}/)
+      month = month.to_i
+      day = day.to_i
+
+      case month
+      when (1..12)
+        year = "19#{year}"
+      when (21..32)
+        month -= 20
+        year = "20#{year}"
+      when (41..52)
+        month -= 40
+        year = "18#{year}"
+      end
+      year = year.to_i
+
+      return false unless Date.valid_date? year, month, day
+
+      checksum = Generator.egn_checksum egn[0,9]
+
+      checksum == egn[9].to_i
+    end
+
   end
+
+  class Parser
+
+    def initialize(egn)
+      return false unless egn.length == 10
+
+      @year, @month, @day = egn.scan(/.{1,2}/)
+      @month = @month.to_i
+      @day = @day.to_i
+
+      case @month
+      when (1..12)
+        @year = "19#{@year}"
+      when (21..32)
+        @month -= 20
+        @year = "20#{@year}"
+      when (41..52)
+        @month -= 40
+        @year = "18#{@year}"
+      end
+      @year = @year.to_i
+
+      raise ArgumentError, "invalid date" unless Date.valid_date? @year, @month, @day
+    end
+
+    def birth_date
+      Date.parse("#{@year}-#{@month}-#{@day}")
+    end
+
+  end
+
 end