district_cn 1.0.0

data/lib/district_cn/act_as_area_field.rb

@@ -0,0 +1,24 @@
+module DistrictCn
+  module ActAsAreaField
+    module ActiveRecord
+
+      def act_as_area_field(*attributes)
+        attributes.each do |attribute|
+          class_eval <<-EVAL
+              alias_method :_#{attribute}, :#{attribute}
+              def #{attribute}
+                val = _#{attribute}
+                return val if val.blank?
+
+                unless @_#{attribute} && val.eql?(@_#{attribute}.value)
+                  @_#{attribute} = DistrictCn.code(val) 
+                end
+                @_#{attribute}
+              end
+          EVAL
+        end
+      end
+
+    end
+  end
+end

data/lib/district_cn/as_options.rb

@@ -0,0 +1,46 @@
+module DistrictCn
+  module AsOptions
+    OPTIONS = {
+      :selected_province  => nil,
+      :selected_city      => nil,
+      :selected_district  => nil,
+      :selected_provinces => [],
+      :selected_cities    => [],
+      :selected_districts => []
+    }
+
+    def as_options
+      options = OPTIONS.dup
+      options[:selected_province] = province && [province_name, province_id]
+      options[:selected_city]     = city && [city_name,city_id]
+      options[:selected_district] = district && [district_name, district_id]
+
+      options[:selected_provinces] = selected_provinces 
+      options[:selected_cities]    = selected_cities 
+      options[:selected_districts] = selected_districts 
+      options
+    end
+
+    private
+    def selected_provinces
+      self.class.data.map do |province_id,province_hash|
+        [province_hash[:text],province_id]
+      end
+    end
+
+    def selected_cities
+      return [] unless province
+      province[:children].map do |city_id,city_hash|
+        [city_hash[:text],city_id]
+      end
+    end
+
+    def selected_districts
+      return [] unless city 
+      city[:children].map do |district_id,district_hash| 
+        [district_hash[:text],district_id]
+      end
+    end
+
+  end
+end

data/lib/district_cn/code.rb

@@ -0,0 +1,107 @@
+module DistrictCn
+  class Code
+    include AsOptions
+
+    REGULAR = /(\d{2})(\d{2})(\d{2})/ 
+    attr_reader :value,:id
+
+    def initialize(id)
+      @value = id
+      @id = id.to_s
+    end
+
+    def name
+      get && get[:text]
+    end
+
+    def area_name(mark="-")
+      [province_name,city_name,district_name].compact.join(mark) 
+    end
+
+    def children
+      return [] unless get && get[:children]
+
+      get[:children].map{|id,attr| [attr[:text],id] }
+    end
+
+
+    def get
+      @get ||= province? && province
+      @get ||= city? && city
+      @get ||= district? && district
+      @get || nil
+    end
+
+    def province?
+      !!(province_id && city_id.nil?)
+    end
+
+    def city?
+      !!(province_id && city_id && district_id.nil?)
+    end
+
+    def district?
+      !!(province_id && city_id && district_id)
+    end
+
+    def province
+      @province ||=
+      province_id && self.class.data && self.class.data[province_id]
+    end
+
+    def city
+      city_id && province && province[:children][city_id]
+    end
+
+    def district
+      district_id && city && city[:children][district_id]
+    end
+
+    def province_name
+      province && province[:text]
+    end
+
+    def city_name
+      city && city[:text]
+    end
+
+    def district_name
+      district && district[:text]
+    end
+
+    def province_id
+      @province_id ||= 
+        if match && segment_blank?(match[1])
+          match[1].ljust(6, '0')
+        end
+    end
+
+    def city_id 
+      @city_id ||= 
+        if match && segment_blank?(match[2])
+          "#{match[1]}#{match[2]}".ljust(6, '0')
+        end
+    end
+
+    def district_id
+      @district_id ||= 
+        if match && segment_blank?(match[3])
+          id
+        end
+    end
+
+    private
+    def match 
+      @match ||= id.match(REGULAR)
+    end
+
+    def segment_blank?(segment)
+      segment != "00"
+    end
+
+    def self.data
+      @data ||= Db::File.instance.tree
+    end
+
+  end
+end

data/lib/district_cn/db.rb

@@ -0,0 +1,81 @@
+require 'json'
+module DistrictCn
+  module Db
+    class File 
+      attr_reader :tree,:list
+
+      def provinces
+        @provinces ||= @tree.map{|pvn_id,pvn_hash| [pvn_hash[:text],pvn_id]}
+      end
+
+      def initialize
+        @tree = {}
+        @list = {}
+        parse
+      end
+
+      class << self
+        def instance
+          @instance ||= new
+        end
+
+        def provinces
+          instance.provinces
+        end
+
+        def tree
+          instance.tree
+        end
+
+        def list
+          instance.list
+        end
+
+        private :new
+      end
+
+      private 
+      def code_regular
+        DistrictCn::Code::REGULAR
+      end
+
+      def path
+        ::File.expand_path("../../areas.json",  __FILE__)
+      end
+
+      def json_data
+        @json_data ||= JSON.parse(::File.read(path))
+      end
+
+      def parse 
+        json_data["province"].each do |province|
+          @tree[province["id"]] = {:text => province["text"],:children => {}}
+
+          @list[province["id"]] = province["text"]
+        end
+
+        json_data["city"].each do |city|
+          city["id"] =~ (code_regular)
+          province_id = $1.ljust(6, '0')
+          if @tree[province_id]
+            @tree[province_id][:children][city["id"]] = {:text => city["text"], :children => {}}
+
+            @list[city["id"]] = city["text"]
+          end
+        end
+
+        json_data["district"].each do |district|
+          district["id"] =~ (code_regular)
+          province_id = $1.ljust(6, '0')
+          city_id = "#{$1}#{$2}".ljust(6, '0')
+          if @tree[province_id] && @tree[province_id][:children][city_id]
+            @tree[province_id][:children][city_id][:children][district["id"]] = {:text => district["text"]}
+
+            @list[district["id"]] = district["text"]
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/district_cn/version.rb

@@ -0,0 +1,3 @@
+module DistrictCn
+  VERSION = "1.0.0"
+end

data/lib/district_cn.rb

@@ -0,0 +1,40 @@
+require "district_cn/version"
+require "active_support/core_ext/module/delegation"
+require "active_support/core_ext/object/blank"
+
+module DistrictCn
+  autoload :AsOptions, 'district_cn/as_options'
+  autoload :Code,      'district_cn/code'
+  autoload :Db,        'district_cn/db'
+
+  begin
+    if defined?(ActiveRecord)
+      autoload :ActAsAreaField,'district_cn/act_as_area_field'
+      ActiveRecord::Base.extend(ActAsAreaField::ActiveRecord)
+    end
+  rescue 
+  end
+
+  class << self
+
+    def code(id)
+      return id if id.blank?
+      Code.new(id)
+    end
+
+    def search(text,limit=10)
+      results = []
+      list.each do |id,name|
+        break if results.size.eql?(limit)
+
+        if name =~ /#{text}/ 
+          results << Code.new(id)
+        end
+      end
+      results
+    end
+
+    delegate :provinces,:tree,:list, :to => Db::File 
+  end
+
+end

data/spec/act_as_area_field_spec.rb

@@ -0,0 +1,67 @@
+#encoding: utf-8
+require 'spec_helper'
+silence_warnings do
+  ActiveRecord::Migration.verbose = false
+  ActiveRecord::Base.logger = Logger.new(nil)
+  ActiveRecord::Base.establish_connection(:adapter => "sqlite3", :database => ":memory:")
+end
+
+ActiveRecord::Base.connection.instance_eval do
+  create_table :companies do |t|
+    t.string :name
+    t.timestamps
+  end
+end
+
+class Company < ActiveRecord::Base
+  attr_accessor :region_code
+  attr_accessor :loc_code
+  attr_accessible :region_code
+
+  act_as_area_field :region_code
+  validates :region_code, presence: true
+end
+
+describe DistrictCn::ActAsAreaField do
+  subject { Company.new }
+  it "should return nil when attribute region_code is nil" do
+    expect(subject.region_code).to be_nil
+  end
+
+  it "should return \"\" when attribute region_code eql \"\"" do
+    subject.region_code = ""
+    expect(subject.region_code).to eq("")
+  end
+
+  it "should return DistrictCn::Code instance" do
+    subject.region_code = 331002
+    expect(subject.region_code).to be_instance_of(DistrictCn::Code)
+  end
+
+  it "should cached" do
+    subject.region_code = 331002
+    expect(subject.region_code.object_id).to eq(subject.region_code.object_id)
+  end
+
+  it "331002 and \"\"331002\"\" shoud have diff cache " do
+    subject.region_code = 331002
+    cache1 = subject.region_code
+    subject.region_code = "331002"
+    cache2 = subject.region_code
+    expect(cache1.object_id).not_to eq(cache2.object_id)
+  end
+
+  it "should return real value without acts_as_area_field" do
+    subject.loc_code = 331002
+    expect(subject.loc_code).to eq(331002)
+    expect(subject.loc_code).not_to be_instance_of(DistrictCn::Code)
+  end
+
+  it "should be unvalid when region_code is blank" do
+    subject.region_code = nil
+    expect(subject).not_to be_valid
+    expect(subject.save).to be_false
+  end
+
+end
+

data/spec/code_spec.rb

@@ -0,0 +1,104 @@
+# encoding: utf-8
+require 'spec_helper'
+describe DistrictCn::Code do
+  let!(:code_331002){ DistrictCn::Code.new(331002) }
+
+  it ".value should return real value" do
+    expect(code_331002.value).to eq(331002)
+    expect(DistrictCn::Code.new("331002").value).to eq("331002")
+  end
+
+  it ".id should return region_code.to_s" do
+    expect(code_331002.id).to eq("331002")
+    expect(DistrictCn::Code.new("331002").id).to eq("331002")
+  end
+
+  it "should return province_id" do
+      expect(DistrictCn::Code.new("00000a").province_id).to be_nil
+      expect(DistrictCn::Code.new("000000").province_id).to be_nil
+      expect(DistrictCn::Code.new("330000").province_id).to eq("330000")
+      expect(DistrictCn::Code.new("331000").province_id).to eq("330000")
+      expect(DistrictCn::Code.new("331002").province_id).to eq("330000")
+  end
+
+  it "should return city_id" do
+    expect(DistrictCn::Code.new("00000a").city_id).to be_nil
+    expect(DistrictCn::Code.new("000000").city_id).to be_nil
+    expect(DistrictCn::Code.new("330000").city_id).to be_nil
+    expect(DistrictCn::Code.new("331000").city_id).to eq("331000")
+    expect(DistrictCn::Code.new("331002").city_id).to eq("331000")
+  end
+
+  it "should return district_id" do
+    expect(DistrictCn::Code.new("00000a").district_id).to be_nil
+    expect(DistrictCn::Code.new("000000").district_id).to be_nil
+    expect(DistrictCn::Code.new("330000").district_id).to be_nil
+    expect(DistrictCn::Code.new("331000").district_id).to be_nil
+    expect(DistrictCn::Code.new("331002").district_id).to eq("331002")
+  end
+
+  it "should return area type" do
+    expect(DistrictCn::Code.new("330000")).to be_province
+    expect(DistrictCn::Code.new("330000")).not_to be_city
+    expect(DistrictCn::Code.new("330000")).not_to be_district
+
+    expect(DistrictCn::Code.new("331000")).to be_city
+    expect(DistrictCn::Code.new("331000")).not_to be_province
+    expect(DistrictCn::Code.new("331000")).not_to be_district
+
+    expect(DistrictCn::Code.new("331002")).to be_district
+    expect(DistrictCn::Code.new("331002")).not_to be_city
+    expect(DistrictCn::Code.new("331002")).not_to be_province
+  end
+
+  it "should return privince,city,district name" do
+    expect(DistrictCn::Code.new("331002").province_name).to eq("浙江省")
+    expect(DistrictCn::Code.new("331002").city_name).to eq("台州市")
+    expect(DistrictCn::Code.new("331002").district_name).to eq("椒江区")
+
+    expect(DistrictCn::Code.new("330000").city_name).to be_nil
+    expect(DistrictCn::Code.new("330000").district_name).to be_nil
+  end
+
+  it "should get detail area" do
+    expect(DistrictCn::Code.new("330000").get[:children]).not_to be_empty
+    expect(DistrictCn::Code.new("331000").get[:children]).not_to be_empty
+    expect(DistrictCn::Code.new("331002").get[:children]).to be_nil
+  end
+
+  it "should get children" do
+    expect(DistrictCn::Code.new("330000").children).to be_include(["台州市", "331000"])
+    expect(DistrictCn::Code.new("330000").children).not_to be_include(["椒江区", "331002"])
+
+    expect(DistrictCn::Code.new("331000").children).to be_include(["椒江区", "331002"])
+    expect(DistrictCn::Code.new("331002").children).to be_empty
+  end
+
+  it "should get area name" do
+      expect(DistrictCn::Code.new("330000").area_name).to eq("浙江省")
+      expect(DistrictCn::Code.new("331000").area_name).to eq("浙江省-台州市")
+      expect(DistrictCn::Code.new("331002").area_name).to eq("浙江省-台州市-椒江区")
+  end
+
+  context "as options" do
+    it "should return selected" do
+      options = DistrictCn::Code.new("330000").as_options
+      expect(options[:selected_provinces]).not_to be_empty
+      expect(options[:selected_cities]).not_to be_empty
+      expect(options[:selected_districts]).to be_empty
+
+      options = DistrictCn::Code.new("331000").as_options
+      expect(options[:selected_provinces]).not_to be_empty
+      expect(options[:selected_cities]).not_to be_empty
+      expect(options[:selected_districts]).not_to be_empty
+
+      options = DistrictCn::Code.new("331002").as_options
+      expect(options[:selected_provinces]).not_to be_empty
+      expect(options[:selected_cities]).not_to be_empty
+      expect(options[:selected_districts]).not_to be_empty
+    end
+
+  end
+
+end
+

data/spec/db_spec.rb

@@ -0,0 +1,32 @@
+# encoding: utf-8
+require 'spec_helper'
+describe DistrictCn::Db do
+  context "file" do
+    subject { DistrictCn::Db::File.instance }
+
+    it "should be cached" do
+      expect(subject.object_id).to eq(DistrictCn::Db::File.instance.object_id)
+    end
+
+    it "tree should be present" do
+      expect(subject.tree).not_to be_empty
+      expect(subject.tree["330000"]).not_to be_nil
+      expect(subject.tree['330000'][:text]).to eq("浙江省")
+      expect(subject.tree['330000'][:children]).not_to be_empty
+    end
+
+    it "list should be present" do
+      expect(subject.list).not_to be_empty
+      expect(subject.list["330000"]).to eq("浙江省")
+      expect(subject.list["331000"]).to eq("台州市")
+      expect(subject.list["331002"]).to eq("椒江区")
+    end
+
+    it "provinces should be present" do
+      expect(subject.provinces).not_to be_empty
+      expect(subject.provinces).to be_include(["浙江省","330000"])
+      expect(subject.provinces).not_to be_include(["台州市","331000"])
+    end
+
+  end
+end

data/spec/district_cn_spec.rb

@@ -0,0 +1,22 @@
+# encoding: utf-8
+require 'spec_helper'
+describe DistrictCn do
+  it ".code should return \"\" or nil" do
+    expect(DistrictCn.code(nil)).to be_nil
+    expect(DistrictCn.code(" ")).to be_blank
+  end
+
+  it ".code should build DistrictCn::Code" do
+    expect(DistrictCn.code("331002")).to be_an_instance_of(DistrictCn::Code)
+    expect(DistrictCn.code("33100a")).to be_an_instance_of(DistrictCn::Code)
+  end
+
+  it "search" do
+    codes = DistrictCn.search("浙江")
+    expect(codes.first).to be_an_instance_of(DistrictCn::Code)
+    expect(codes.first.name).to eq("浙江省")
+
+    codes = DistrictCn.search("浙江a")
+    expect(codes).to be_blank
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,10 @@
+require "bundler"
+Bundler.setup
+
+require 'active_record'
+require File.expand_path("../../lib/district_cn",  __FILE__)
+require 'rspec/autorun'
+#require 'active_support/core_ext/kernel/reporting'
+require 'logger'
+require 'pp'
+