thai_geodata 0.1.0 → 0.2.0

data/lib/thai_geodata/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ThaiGeodata
-  VERSION = '0.1.0'
+  VERSION = '0.2.0'
 end

data/lib/thai_geodata.rb

@@ -3,24 +3,105 @@
 require 'json'
 require_relative 'thai_geodata/version'
 
-# ThaiGeodata provides in-memory lookup of Thailand’s provinces, districts
-# and subdistricts (with postal codes) from static JSON data.
+# ThaiGeodata provides in-memory lookup of Thailand’s provinces, districts,
+# subdistricts, and combined geography records (with postal codes) from static JSON data.
 module ThaiGeodata
   DATA_PATH = File.join(__dir__, 'thai_geodata', 'data')
 
-  def self.provinces
-    @provinces ||= load_json('provinces.json')
-  end
+  class << self
+    # Raw data loaders
+    def provinces
+      @provinces ||= load_json('provinces.json')
+    end
 
-  def self.districts
-    @districts ||= load_json('districts.json')
-  end
+    def districts
+      @districts ||= load_json('districts.json')
+    end
 
-  def self.subdistricts
-    @subdistricts ||= load_json('subdistricts.json')
-  end
+    def subdistricts
+      @subdistricts ||= load_json('subdistricts.json')
+    end
+
+    def geography
+      @geography ||= load_json('geography.json')
+    end
+
+    # Lookup helpers
+
+    # Find a province by code (Integer or String), or Thai/English name
+    def find_province(code_or_name)
+      provinces.find do |p|
+        match_code_or_name?(p, 'provinceCode', 'provinceNameTh', 'provinceNameEn', code_or_name)
+      end
+    end
+
+    # Find a district by code (Integer or String), or Thai/English name
+    def find_district(code_or_name)
+      districts.find do |d|
+        match_code_or_name?(d, 'districtCode', 'districtNameTh', 'districtNameEn', code_or_name)
+      end
+    end
+
+    # Find a subdistrict by code (Integer or String), or Thai/English name
+    def find_subdistrict(code_or_name)
+      subdistricts.find do |s|
+        match_code_or_name?(s, 'subdistrictCode', 'subdistrictNameTh', 'subdistrictNameEn', code_or_name)
+      end
+    end
+
+    # List all districts for a given province (code or name)
+    def districts_for_province(province_code_or_name)
+      prov = find_province(province_code_or_name)
+      return [] unless prov
+
+      districts.select { |d| d['provinceCode'] == prov['provinceCode'] }
+    end
+
+    # List all subdistricts for a given district (code or name)
+    def subdistricts_for_district(district_code_or_name)
+      dist = find_district(district_code_or_name)
+      return [] unless dist
+
+      subdistricts.select { |s| s['districtCode'] == dist['districtCode'] }
+    end
+
+    # Get the postal code for a given subdistrict (code or name)
+    def postal_code_for_subdistrict(code_or_name)
+      sub = find_subdistrict(code_or_name)
+      sub ? sub['postalCode'] : nil
+    end
+
+    # Get a full path (province, district, subdistrict) for a subdistrict
+    def location_path_for_subdistrict(code_or_name)
+      sub = find_subdistrict(code_or_name)
+      return nil unless sub
+
+      dist = find_district(sub['districtCode'])
+      prov = find_province(sub['provinceCode'])
+
+      {
+        province: prov ? prov['provinceNameTh'] : nil,
+        district: dist ? dist['districtNameTh'] : nil,
+        subdistrict: sub['subdistrictNameTh']
+      }
+    end
+
+    private
+
+    # Load JSON file from DATA_PATH
+    def load_json(filename)
+      path = File.join(DATA_PATH, filename)
+      JSON.parse(File.read(path))
+    end
 
-  def self.load_json(file)
-    JSON.parse(File.read(File.join(DATA_PATH, file)))
+    # Helper to match by numeric code or localized/English names
+    def match_code_or_name?(hash, code_key, th_key, en_key, code_or_name)
+      if code_or_name.to_s =~ /^\d+$/
+        hash[code_key] == code_or_name.to_i
+      else
+        hash[th_key] == code_or_name || hash[en_key].casecmp(code_or_name.to_s).zero?
+      end
+    end
+    private :match_code_or_name?
   end
 end

data/test/test_thai_geodata.rb

@@ -1,10 +1,63 @@
 # frozen_string_literal: true
 
 require 'minitest/autorun'
-require_relative '../lib/thai_geodata'
+require 'thai_geodata'
 
 class ThaiGeodataTest < Minitest::Test
   def test_provinces_loaded
-    assert ThaiGeodata.provinces.any?
+    refute_empty ThaiGeodata.provinces
+  end
+
+  def test_find_province_by_code
+    bangkok = ThaiGeodata.find_province(10)
+    assert_equal 'กรุงเทพมหานคร', bangkok['provinceNameTh']
+  end
+
+  def test_find_province_by_name_en
+    cm = ThaiGeodata.find_province('Chiang Mai')
+    assert_equal 50, cm['provinceCode']
+  end
+
+  def test_find_district_by_code
+    dist = ThaiGeodata.find_district(1001)
+    assert_equal 'พระนคร', dist['districtNameTh']
+  end
+
+  def test_find_subdistrict_by_code
+    sub = ThaiGeodata.find_subdistrict(100_101)
+    assert_equal 'พระบรมมหาราชวัง', sub['subdistrictNameTh']
+  end
+
+  def test_districts_for_province
+    dists = ThaiGeodata.districts_for_province('Bangkok')
+    assert dists.any?
+    assert_equal(dists, ThaiGeodata.districts.select { |d| d['provinceCode'] == 10 })
+  end
+
+  def test_subdistricts_for_district
+    subs = ThaiGeodata.subdistricts_for_district(1001)
+    assert subs.any?
+    assert_equal(subs, ThaiGeodata.subdistricts.select { |s| s['districtCode'] == 1001 })
+  end
+
+  def test_postal_code_for_subdistrict
+    assert_equal 10_200, ThaiGeodata.postal_code_for_subdistrict(100_101)
+  end
+
+  def test_location_path_for_subdistrict
+    path = ThaiGeodata.location_path_for_subdistrict(100_101)
+    expected = {
+      province: 'กรุงเทพมหานคร',
+      district: 'พระนคร',
+      subdistrict: 'พระบรมมหาราชวัง'
+    }
+    assert_equal expected, path
+  end
+
+  def test_geography_loaded
+    refute_empty ThaiGeodata.geography
+    first = ThaiGeodata.geography.first
+    assert first.key?('provinceCode')
+    assert first.key?('postalCode')
   end
 end

data/thai_geodata.gemspec

@@ -17,4 +17,5 @@ Gem::Specification.new do |s|
   s.license     = 'MIT'
 
   s.required_ruby_version = '>= 2.7'
+  s.add_development_dependency 'minitest', '~> 5.0'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: thai_geodata
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Your Name
@@ -9,7 +9,21 @@ autorequire:
 bindir: bin
 cert_chain: []
 date: 2025-06-15 00:00:00.000000000 Z
-dependencies: []
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
 description: A Ruby gem wrapping the MIT-licensed Thailand Geography JSON dataset,
   with lookup helpers.
 email:
@@ -25,6 +39,7 @@ files:
 - Rakefile
 - lib/thai_geodata.rb
 - lib/thai_geodata/data/districts.json
+- lib/thai_geodata/data/geography.json
 - lib/thai_geodata/data/provinces.json
 - lib/thai_geodata/data/subdistricts.json
 - lib/thai_geodata/version.rb