data_maker 2.2.3 → 2.3.3

data/lib/data_maker.rb

@@ -1,12 +1,28 @@
+require 'i18n'
+require 'data_maker/utilities/array_utilities'
+require 'data_maker/utilities/module_utilities'
+
 module DataMaker
-  VERSION = "2.2.3"
+  BASE_LIB_PATH = File.expand_path("..", __FILE__)
+
+  if I18n.respond_to?(:enforce_available_locales=)
+    I18n.enforce_available_locales = false
+  end
 
-  require 'data_maker/utilities/array_utilities'
-  require 'data_maker/utilities/module_utilities'
+  I18n.load_path += Dir[File.join(BASE_LIB_PATH, 'data_maker', 'locales', '*.yml')]
 
   extend ModuleUtilities
 
-  BASE_LIB_PATH = File.expand_path("..", __FILE__)
+  class Config
+    @locale = nil
+
+    class << self
+      attr_writer :locale
+      def locale
+        @locale
+      end
+    end
+  end
 
   LETTERS = [*'a'..'z']
 
@@ -46,6 +62,15 @@ module DataMaker
     letterify(numerify(masks))
   end
 
+  def self.translate(*args)
+    options = args.last.is_a?(Hash) ? args.pop : {}
+    options[:locale] ||= DataMaker::Config.locale
+    options[:raise] = true
+    I18n.translate(*(args.push(options)))
+    rescue I18n::MissingTranslationData
+    I18n.translate(*(args.push(options.merge(locale: :en))))
+  end
+
   # Load all constants.
   Dir["#{BASE_LIB_PATH}/data_maker/**/*.rb"].sort.each do |f|
     require f

data/lib/data_maker/cn/address.rb

@@ -1,32 +1,220 @@
 module DataMaker
   module CN
     module Address
+      extend ModuleUtilities
+
       def self.street_address
-        "lastname and street suffix"
+        GenerateAddress.new.street_address
       end
 
-      def self.street_name
+      def self.city(province = nil)
+        address = GenerateAddress.new(province: province)
+        address.generate
+        address.city
       end
 
-      def self.city
+      def self.province(city = nil)
+        address = GenerateAddress.new(city: city)
+        address.generate
+        address.province
       end
 
-      def self.city_prefix
+      def self.postal_code(province = nil)
+        address = GenerateAddress.new(province: province)
+        address.generate
+        address.postal_code
       end
 
-      def self.city_suffix
+      def self.address(province: nil, city: nil, district: nil, struct: false)
+        address  = GenerateAddress.new(province: province, city: city, district: district)
+        struct ? address.address_struct : address.address_string
       end
 
-      def self.unit_no
-      end
+      class GenerateAddress
+        attr_accessor :province, :city, :district, :street_address, :locale, :options
 
-      def self.postcode
-      end
+        def initialize(*args)
+          self.options  = (args.last.is_a?(::Hash) ? args.last : {}).delete_if { |k, v| v.nil? }
+          self.district = options[:district]
+          self.city     = options[:city]
+          self.province = options[:province]
+          self.locale   = options[:locale] || :zh
+          unless options.empty?
+            validate
+          end
+        end
 
-      def self.province
-      end
+        def generate
+          if options.empty?
+            # Generate a random address from a province
+            self.province = provinces.sample
+            self.city     = province_cities.sample
+            self.district = city_districts.nil? ? nil : city_districts.sample
+          end
+
+          if options[:province] && options[:city].nil? && options[:district].nil?
+            self.city     = province_cities.sample
+            self.district = city_districts.nil? ? nil : city_districts.sample
+          end
+
+          if options[:province].nil? && options[:city] && options[:district].nil?
+            self.province = city_province
+            self.district = city_districts.nil? ? nil : city_districts.sample
+          end
+        end
+
+        def street_address
+          [unit_no, street_name, street_suffix].join
+        end
+
+        def address_string
+          generate
+          translate
+          build_address
+        end
+
+        def address_struct
+          generate
+          translate
+          OpenStruct.new(street_address: street_address,
+                         city:           city,
+                         district:       district,
+                         province:       province,
+                         postal_code:    postal_code
+                        )
+        end
+
+        def postal_code
+          postal_code = nil
+          province_postal_codes = DataMaker::CN::Address::PROVINCE_POSTAL_CODES
+          province_postal_codes.to_set.select do |province_postal_code|
+            p, pc = province_postal_code.split(",")
+            p == province ? postal_code = pc : nil
+          end
+          postal_code
+        end
+
+        private
+
+        def translate
+          DataMaker::Config.locale = locale
+          values = { province: province, city: city, district: district }
+          values.each do |key, value|
+            self.send("#{key}=", DataMaker.translate(['data_maker', 'address', key, value].join(".")))
+          end
+        end
+
+        def validate
+          if province
+            unless province_exists?
+              raise ArgumentError, "The province #{province} does not exist!"
+            end
+          end
+
+          if city
+            unless city_exists?
+              raise ArgumentError, "The city #{city} does not exist!"
+            end
+          end
+
+          if district
+            unless district_exists?
+              raise ArgumentError, "The district #{district} does not exist!"
+            end
+          end
+
+          if city.nil? && district
+            raise ArgumentError, "You must always pass a city with a district!"
+          end
+
+          if province && city
+            unless province_has_city?
+              raise ArgumentError, "The province #{province} does not have the city #{city}"
+            end
+          end
+
+          if city && district
+            unless city_has_district?
+              raise ArgumentError, "The city #{city} does not have the district #{district}"
+            end
+          end
+        end
+
+        def provinces
+          DataMaker::CN::Address::PROVINCES
+        end
+
+        def cities
+          DataMaker::CN::Address::CITIES
+        end
+
+        def districts
+          DataMaker::CN::Address::DISTRICTS
+        end
+
+        def build_address
+          address = []
+          address << street_address
+          address << city
+          address << district
+          address << province
+          address << postal_code
+          address.compact.join(", ")
+        end
+
+        def street_name
+          DataMaker::CN::Name::LAST_NAMES.sample
+        end
+
+        def street_suffix
+          DataMaker::CN::Address::STREET_SUFFIXES.sample
+        end
+
+        def unit_no
+          masks = "#" * rand(1..2)
+          DataMaker.numerify(masks)
+        end
+
+        def city_districts
+          raise ArgumentError, "Please pass a city!" if city.nil?
+          d = DataMaker::CN::Address.const_get("#{city}_districts".upcase)
+          d.empty? ? nil : d
+        end
+
+        def city_province
+          province = nil
+          city_provinces = DataMaker::CN::Address::CITY_PROVINCES
+          city_provinces.to_set.select do |city_province|
+            c, p = city_province.split(",")
+            c == city ? province = p : nil
+          end
+          province
+        end
+
+        def province_cities
+          raise ArgumentError, "Please pass a province!" if province.nil?
+          DataMaker::CN::Address.const_get("#{province}_cities".upcase)
+        end
+
+        def province_exists?
+          provinces.to_set.include?(province)
+        end
+
+        def city_exists?
+          cities.to_set.include?(city)
+        end
+
+        def district_exists?
+          districts.to_set.include?(district)
+        end
+
+        def province_has_city?
+          province_cities.to_set.include?(city)
+        end
 
-      def self.full_address
+        def city_has_district?
+          city_districts.to_set.include?(district)
+        end
       end
     end
   end

data/lib/data_maker/data/cn/address/aba_districts

@@ -0,0 +1,13 @@
+wenchuan
+li
+mao
+songpan
+jiuzhaigou
+jinchuan
+xiaojin
+heishui
+maerkang
+rangtang
+aba
+ruoergai
+hongyuan

data/lib/data_maker/data/cn/address/akesu_districts

@@ -0,0 +1,9 @@
+akesu
+wensu
+kuche
+shaya
+xinhe
+baicheng
+wushi
+awati
+keping

data/lib/data_maker/data/cn/address/alashanmeng_districts

@@ -0,0 +1,3 @@
+alashanzuoqi
+alashanyouqi
+ejinaqi

data/lib/data_maker/data/cn/address/aletai_districts

@@ -0,0 +1,7 @@
+aletai
+buerjin
+fuyun
+fuhai
+habahe
+qinghe
+jimunai

data/lib/data_maker/data/cn/address/ali_districts

@@ -0,0 +1,7 @@
+pulan
+zhada
+geer
+ritu
+geji
+gaize
+cuoqin

data/lib/data_maker/data/cn/address/anhui_cities

@@ -0,0 +1,17 @@
+hefei
+wuhu
+bangbu
+huainan
+maanshan
+huaibei
+tongling
+anqing
+huangshan
+chuzhou
+fuyang
+suzhou
+chaohu
+liuan
+bozhou
+chizhou
+xuancheng

data/lib/data_maker/data/cn/address/ankang_districts

@@ -0,0 +1,10 @@
+hanbin
+hanyin
+shiquan
+ningshan
+ziyang
+langao
+pingli
+zhenping
+xunyang
+baihe

data/lib/data_maker/data/cn/address/anqing_districts

@@ -0,0 +1,11 @@
+yingjiang
+daguan
+jiao
+huaining
+congyang
+qianshan
+taihu
+susong
+wangjiang
+yuexi
+tongcheng

data/lib/data_maker/data/cn/address/anshan_districts

@@ -0,0 +1,7 @@
+tiedong
+tiexi
+lishan
+qianshan
+taian
+xiuyanmanzu
+haicheng

data/lib/data_maker/data/cn/address/anshun_districts

@@ -0,0 +1,6 @@
+xixiu
+pingba
+puding
+zhenningbuyizumiaozu
+guanlingbuyizumiaozu
+ziyunmiaozubuyizu

data/lib/data_maker/data/cn/address/anyang_districts

@@ -0,0 +1,9 @@
+wenfeng
+beiguan
+yindu
+longan
+anyang
+tangyin
+hua
+neihuang
+linzhou

data/lib/data_maker/data/cn/address/aomen_cities

@@ -0,0 +1 @@
+aomen

data/lib/data_maker/data/cn/address/baicheng_districts

@@ -0,0 +1,5 @@
+taobei
+zhenlai
+tongyu
+taonan
+daan

data/lib/data_maker/data/cn/address/baise_districts

@@ -0,0 +1,12 @@
+youjiang
+tianyang
+tiandong
+pingguo
+debao
+jingxi
+napo
+lingyun
+leye
+tianlin
+xilin
+longlingezu

data/lib/data_maker/data/cn/address/baishan_districts

@@ -0,0 +1,6 @@
+badaojiang
+fusong
+jingyu
+changbaizhaoxianzu
+jiangyuan
+linjiang

data/lib/data_maker/data/cn/address/baiyin_districts

@@ -0,0 +1,5 @@
+baiyin
+pingchuan
+jingyuan
+huining
+jingtai