unitsdb 2.2.1 → 2.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7fda421d6dfc06976fad5da00dd85f0fb1dbb3dc4865e2904a28b237530ace05
-  data.tar.gz: 9e4fdd8395fab543936c4d4b415f05c468b788d287545d3ebfd774a53a9f766c
+  metadata.gz: e2982ead6dc990098c687109bd98560cd35945c6272e01a88bcd4f1270ffdffa
+  data.tar.gz: f89db123f11036b7963fd5fc60a1e37476cd7dac43664be26c95776089479844
 SHA512:
-  metadata.gz: 8453d7f411242d4fbf035b73cee6d968505c60e884606b872317a1fe53dd93b3e603a0497c1869b1768649dbf2c23a6fdcbe233dccf4eee95f482f1d440f2163
-  data.tar.gz: 5a7d1579b935b9afb7bb5e4c0ab8c25e943be6b525d401411bf97d8b3101061a0b211631af0740dcd3476fb012c2c3048dadc0399ec75f91ed425e8416bbd59e
+  metadata.gz: 811a621748d4936a1668e7f6969e5090f4da626b7ca0122bbce24ea1d3b217a533d2ea0e5759a812c22faa4d097941538e9273db4c2f84beb908ee301c0ef51a
+  data.tar.gz: 23456a3351a981a001f57476bba3815b95c93ce65008189bf1334ae87cf6c0aa57201f897301c8579e0cc9866af10dc2989eb9a58af1aad2c7e63c15d6f05e42

data/Gemfile

@@ -6,7 +6,7 @@ source "https://rubygems.org"
 gemspec
 
 gem "canon"
-gem "lutaml-model", github: "lutaml/lutaml-model", branch: "main"
+gem "lutaml-model", github: "lutaml/lutaml-model", branch: "fix/global-context-register-lookup-fallback"
 gem "nokogiri"
 gem "rake"
 gem "rspec"

data/lib/unitsdb/config.rb

@@ -2,17 +2,129 @@
 
 module Unitsdb
   class Config
+    CONTEXT_ID = :unitsdb_v2
+
     class << self
+      def context_id
+        @context_id ||= CONTEXT_ID
+      end
+
+      def register_model(klass, id:)
+        registered_models[id.to_sym] = klass
+        klass
+      end
+
+      def registered_models
+        @registered_models ||= {}
+      end
+
       def models
         @models ||= {}
       end
 
       def models=(user_models)
-        models.merge!(user_models)
+        normalized_models = user_models.each_with_object({}) do |(id, klass), result|
+          model_id = id.to_sym
+          result[model_id] = register_model(klass, id: model_id)
+        end
+
+        models.merge!(normalized_models)
       end
 
       def model_for(model_name)
-        models[model_name]
+        model_id = model_name.to_sym
+        models[model_id] || registered_models[model_id]
+      end
+
+      def register(id = context_id)
+        explicit_registers[id.to_sym]
+      end
+
+      def populate_register(id: context_id, fallback_to: [:default], substitutions: [])
+        register_id = id.to_sym
+        context(register_id)
+
+        model_register = Lutaml::Model::Register.new(register_id, fallback: fallback_to)
+        resolve_substitutions(
+          substitutions,
+          registry: build_registry,
+          fallback_to: fallback_to,
+          id: "#{register_id}_register",
+        ).each do |substitution|
+          model_register.register_global_type_substitution(**substitution)
+        end
+
+        explicit_registers[register_id] = Lutaml::Model::GlobalRegister.register(model_register)
+      end
+
+      def find_context(id)
+        Lutaml::Model::GlobalContext.context(id.to_sym)
+      end
+
+      def resolve_type(type_name, context: context_id)
+        Lutaml::Model::GlobalContext.resolve_type(type_name, context.to_sym)
+      end
+
+      def context(id = context_id, force_populate: false)
+        existing = find_context(id)
+        return existing if existing && !force_populate && populated?(id)
+
+        populate_context(id: id)
+      end
+
+      def populate_context(id: context_id, fallback_to: [:default], substitutions: [])
+        Lutaml::Model::GlobalContext.unregister_context(id) if find_context(id)
+
+        opts = { registry: build_registry, fallback_to: fallback_to, id: id }
+        context = Lutaml::Model::GlobalContext.create_context(
+          substitutions: resolve_substitutions(substitutions, **opts),
+          **opts,
+        )
+        mark_populated!(id)
+        context
+      end
+
+      def resolve_substitutions(substitutions, registry:, fallback_to:, id:)
+        resolution_context = Lutaml::Model::TypeContext.derived(
+          id: "#{id}_substitution_resolution",
+          registry: registry,
+          fallback_to: fallback_to,
+        )
+
+        Array(substitutions).map do |substitution|
+          from_key = substitution[:from_type] || substitution[:from]
+          to_key = substitution[:to_type] || substitution[:to]
+
+          {
+            from_type: resolve_substitution_type(from_key, resolution_context),
+            to_type: resolve_substitution_type(to_key, resolution_context),
+          }
+        end
+      end
+
+      def resolve_substitution_type(value, resolution_context)
+        return value if value.is_a?(Class)
+
+        Lutaml::Model::TypeResolver.resolve(value, resolution_context)
+      end
+
+      def build_registry
+        registry = Lutaml::Model::TypeRegistry.new
+        registered_models.each { |model_id, klass| registry.register(model_id, klass) }
+        registry
+      end
+
+      def populated?(context_id)
+        @populated_for&.[](context_id.to_sym)
+      end
+
+      def mark_populated!(context_id)
+        @populated_for ||= {}
+        @populated_for[context_id.to_sym] = true
+      end
+
+      def explicit_registers
+        @explicit_registers ||= {}
       end
     end
   end

data/lib/unitsdb/database.rb

@@ -4,6 +4,15 @@ module Unitsdb
   class Database < Lutaml::Model::Serializable
     # model Config.model_for(:units)
 
+    DATABASE_FILES = {
+      "prefixes" => "prefixes.yaml",
+      "dimensions" => "dimensions.yaml",
+      "units" => "units.yaml",
+      "quantities" => "quantities.yaml",
+      "unit_systems" => "unit_systems.yaml",
+    }.freeze
+    SUPPORTED_SCHEMA_VERSION = "2.0.0"
+
     attribute :schema_version, :string
     attribute :version, :string
     attribute :units, Unit, collection: true
@@ -290,25 +299,21 @@ module Unitsdb
       invalid_refs
     end
 
-    def self.from_db(dir_path)
-      # If dir_path is a relative path, make it relative to the current working directory
-      db_path = dir_path
-      puts "Database directory path: #{db_path}"
+    def self.from_db(dir_path, context: Unitsdb::Config.context_id)
+      context_id = context.to_sym
+      if context_id == Unitsdb::Config.context_id &&
+          Unitsdb::Config.find_context(context_id).nil?
+        Unitsdb::Config.context(context_id)
+      end
 
-      # Check if the directory exists
+      db_path = File.expand_path(dir_path.to_s)
       unless Dir.exist?(db_path)
         raise Errors::DatabaseNotFoundError,
               "Database directory not found: #{db_path}"
       end
 
-      # Define required files
-      required_files = %w[prefixes.yaml dimensions.yaml units.yaml
-                          quantities.yaml unit_systems.yaml]
-      yaml_files = required_files.map { |file| File.join(dir_path, file) }
-
-      # Check if all required files exist
-      missing_files = required_files.reject do |file|
-        File.exist?(File.join(dir_path, file))
+      missing_files = DATABASE_FILES.values.reject do |filename|
+        File.exist?(File.join(db_path, filename))
       end
 
       if missing_files.any?
@@ -316,99 +321,92 @@ module Unitsdb
               "Missing required database files: #{missing_files.join(', ')}"
       end
 
-      # Ensure we have path properly joined with filenames
-      prefixes_yaml = yaml_files[0]
-      dimensions_yaml = yaml_files[1]
-      units_yaml = yaml_files[2]
-      quantities_yaml = yaml_files[3]
-      unit_systems_yaml = yaml_files[4]
-
-      # Debug paths
-      if ENV["DEBUG"]
-        puts "[UnitsDB] Loading YAML files from directory: #{dir_path}"
-        puts "  - #{prefixes_yaml}"
-        puts "  - #{dimensions_yaml}"
-        puts "  - #{units_yaml}"
-        puts "  - #{quantities_yaml}"
-        puts "  - #{unit_systems_yaml}"
+      documents = load_database_documents(db_path)
+      schema_version = validate_schema_versions!(documents)
+      combined_hash = build_database_hash(documents, schema_version)
+
+      Lutaml::Model::GlobalContext.with_context(context_id) do
+        if Unitsdb::Config.register(context_id)
+          from_hash(combined_hash, register: context_id)
+        else
+          from_hash(combined_hash)
+        end
       end
+    end
 
-      # Load YAML files with better error handling
-      begin
-        prefixes_hash = YAML.safe_load_file(prefixes_yaml)
-        dimensions_hash = YAML.safe_load_file(dimensions_yaml)
-        units_hash = YAML.safe_load_file(units_yaml)
-        quantities_hash = YAML.safe_load_file(quantities_yaml)
-        unit_systems_hash = YAML.safe_load_file(unit_systems_yaml)
-      rescue Errno::ENOENT => e
-        raise Errors::DatabaseFileNotFoundError,
-              "Failed to read database file: #{e.message}"
-      rescue Psych::SyntaxError => e
+    def self.load_database_documents(db_path)
+      puts "[UnitsDB] Loading YAML files from directory: #{db_path}" if ENV["UNITSDB_DEBUG"]
+      DATABASE_FILES.transform_values do |filename|
+        puts "  - #{File.join(db_path, filename)}" if ENV["UNITSDB_DEBUG"]
+        load_database_yaml(File.join(db_path, filename), filename)
+      end
+    end
+
+    def self.load_database_yaml(path, filename)
+      document = YAML.safe_load_file(path)
+
+      unless document.is_a?(Hash)
         raise Errors::DatabaseFileInvalidError,
-              "Invalid YAML in database file: #{e.message}"
-      rescue StandardError => e
-        raise Errors::DatabaseLoadError, "Error loading database: #{e.message}"
+              "Invalid YAML structure in #{filename}: expected a mapping"
       end
 
-      # Verify all files have schema_version field
-      missing_schema = []
-      missing_schema << "prefixes.yaml" unless prefixes_hash.key?("schema_version")
-      missing_schema << "dimensions.yaml" unless dimensions_hash.key?("schema_version")
-      missing_schema << "units.yaml" unless units_hash.key?("schema_version")
-      missing_schema << "quantities.yaml" unless quantities_hash.key?("schema_version")
-      missing_schema << "unit_systems.yaml" unless unit_systems_hash.key?("schema_version")
+      document
+    rescue Errno::ENOENT => e
+      raise Errors::DatabaseFileNotFoundError,
+            "Failed to read database file: #{e.message}"
+    rescue Psych::SyntaxError => e
+      raise Errors::DatabaseFileInvalidError,
+            "Invalid YAML in database file: #{e.message}"
+    rescue Errors::DatabaseError
+      raise
+    rescue StandardError => e
+      raise Errors::DatabaseLoadError,
+            "Error loading database file #{filename}: #{e.message}"
+    end
+    private_class_method :load_database_documents, :load_database_yaml
 
-      if missing_schema.any?
+    def self.validate_schema_versions!(documents)
+      versions = DATABASE_FILES.each_with_object({}) do |(collection_key, filename), result|
+        document = documents.fetch(collection_key)
+        result[filename] = document.fetch("schema_version")
+      rescue KeyError
         raise Errors::DatabaseFileInvalidError,
-              "Missing schema_version in files: #{missing_schema.join(', ')}"
+              "Missing schema_version in #{filename}"
       end
 
-      # Extract versions from each file
-      prefixes_version = prefixes_hash["schema_version"]
-      dimensions_version = dimensions_hash["schema_version"]
-      units_version = units_hash["schema_version"]
-      quantities_version = quantities_hash["schema_version"]
-      unit_systems_version = unit_systems_hash["schema_version"]
-
-      # Check if all versions match
-      versions = [
-        prefixes_version,
-        dimensions_version,
-        units_version,
-        quantities_version,
-        unit_systems_version,
-      ]
-
-      unless versions.uniq.size == 1
-        version_info = {
-          "prefixes.yaml" => prefixes_version,
-          "dimensions.yaml" => dimensions_version,
-          "units.yaml" => units_version,
-          "quantities.yaml" => quantities_version,
-          "unit_systems.yaml" => unit_systems_version,
-        }
+      unless versions.values.uniq.size == 1
         raise Errors::VersionMismatchError,
-              "Version mismatch in database files: #{version_info.inspect}"
+              "Version mismatch in database files: #{versions.inspect}"
       end
 
-      # Check if the version is supported
-      version = versions.first
-      unless version == "2.0.0"
+      version = versions.values.first
+      unless version == SUPPORTED_SCHEMA_VERSION
         raise Errors::UnsupportedVersionError,
-              "Unsupported database version: #{version}. Only version 2.0.0 is supported."
+              "Unsupported database version: #{version}. Only version #{SUPPORTED_SCHEMA_VERSION} is supported."
       end
 
-      combined_yaml = {
-        "schema_version" => prefixes_version,
-        "prefixes" => prefixes_hash["prefixes"],
-        "dimensions" => dimensions_hash["dimensions"],
-        "units" => units_hash["units"],
-        "quantities" => quantities_hash["quantities"],
-        "unit_systems" => unit_systems_hash["unit_systems"],
-      }.to_yaml
+      version
+    end
+
+    def self.build_database_hash(documents, schema_version)
+      {
+        "schema_version" => schema_version,
+      }.merge(
+        DATABASE_FILES.keys.to_h do |collection_key|
+          document = documents.fetch(collection_key)
+          [collection_key, fetch_collection!(document, collection_key)]
+        end,
+      )
+    end
 
-      from_yaml(combined_yaml)
+    def self.fetch_collection!(document, collection_key)
+      document.fetch(collection_key)
+    rescue KeyError
+      raise Errors::DatabaseFileInvalidError,
+            "Missing #{collection_key} collection in #{DATABASE_FILES.fetch(collection_key)}"
     end
+    private_class_method :validate_schema_versions!, :build_database_hash,
+                         :fetch_collection!
 
     private
 
@@ -616,8 +614,7 @@ module Unitsdb
       end
     end
 
-    def validate_reference(ref_id, ref_type, ref_path, registry, invalid_refs,
-file_type)
+    def validate_reference(ref_id, ref_type, ref_path, registry, invalid_refs, file_type)
       # Handle references that are objects with id and type (could be a hash or an object)
       if ref_id.respond_to?(:id) && ref_id.respond_to?(:type)
         id = ref_id.id
@@ -697,4 +694,6 @@ file_type)
       end
     end
   end
+
+  Config.register_model(Database, id: :database)
 end

data/lib/unitsdb/dimension.rb

@@ -43,4 +43,6 @@ module Unitsdb
     attribute :names, LocalizedString, collection: true
     attribute :references, ExternalReference, collection: true
   end
+
+  Config.register_model(Dimension, id: :dimension)
 end

data/lib/unitsdb/dimension_details.rb

@@ -16,4 +16,6 @@ module Unitsdb
     attribute :symbol, :string
     attribute :symbols, SymbolPresentations, collection: true
   end
+
+  Config.register_model(DimensionDetails, id: :dimension_details)
 end

data/lib/unitsdb/dimension_reference.rb

@@ -5,4 +5,6 @@ module Unitsdb
     attribute :id, :string
     attribute :type, :string
   end
+
+  Config.register_model(DimensionReference, id: :dimension_reference)
 end

data/lib/unitsdb/dimensions.rb

@@ -8,4 +8,6 @@ module Unitsdb
     attribute :version, :string
     attribute :dimensions, Dimension, collection: true
   end
+
+  Config.register_model(Dimensions, id: :dimensions)
 end

data/lib/unitsdb/external_reference.rb

@@ -11,4 +11,6 @@ module Unitsdb
     attribute :type, :string, values: %w[normative informative]
     attribute :authority, :string
   end
+
+  Config.register_model(ExternalReference, id: :external_reference)
 end

data/lib/unitsdb/identifier.rb

@@ -5,4 +5,6 @@ module Unitsdb
     attribute :id, :string
     attribute :type, :string
   end
+
+  Config.register_model(Identifier, id: :identifier)
 end

data/lib/unitsdb/localized_string.rb

@@ -14,4 +14,6 @@ module Unitsdb
       value&.downcase
     end
   end
+
+  Config.register_model(LocalizedString, id: :localized_string)
 end

data/lib/unitsdb/prefix.rb

@@ -23,4 +23,6 @@ module Unitsdb
     attribute :power, :integer
     attribute :references, ExternalReference, collection: true
   end
+
+  Config.register_model(Prefix, id: :prefix)
 end

data/lib/unitsdb/prefix_reference.rb

@@ -5,4 +5,6 @@ module Unitsdb
     attribute :id, :string
     attribute :type, :string
   end
+
+  Config.register_model(PrefixReference, id: :prefix_reference)
 end

data/lib/unitsdb/prefixes.rb

@@ -19,4 +19,6 @@ module Unitsdb
     attribute :version, :string
     attribute :prefixes, Prefix, collection: true
   end
+
+  Config.register_model(Prefixes, id: :prefixes)
 end

data/lib/unitsdb/quantities.rb

@@ -7,4 +7,6 @@ module Unitsdb
     attribute :version, :string
     attribute :quantities, Quantity, collection: true
   end
+
+  Config.register_model(Quantities, id: :quantities)
 end

data/lib/unitsdb/quantity.rb

@@ -12,4 +12,6 @@ module Unitsdb
     attribute :dimension_reference, DimensionReference
     attribute :references, ExternalReference, collection: true
   end
+
+  Config.register_model(Quantity, id: :quantity)
 end

data/lib/unitsdb/quantity_reference.rb

@@ -7,4 +7,6 @@ module Unitsdb
     attribute :id, :string
     attribute :type, :string
   end
+
+  Config.register_model(QuantityReference, id: :quantity_reference)
 end

data/lib/unitsdb/qudt.rb

@@ -18,6 +18,7 @@ module Unitsdb
       "qudt:unit:#{uri}"
     end
   end
+  Config.register_model(QudtUnit, id: :qudt_unit)
 
   # QUDT QuantityKind from quantitykinds vocabulary
   # Example: http://qudt.org/vocab/quantitykind/Length
@@ -33,6 +34,7 @@ module Unitsdb
       "qudt:quantitykind:#{uri}"
     end
   end
+  Config.register_model(QudtQuantityKind, id: :qudt_quantity_kind)
 
   # QUDT DimensionVector from dimensionvectors vocabulary
   # Example: http://qudt.org/vocab/dimensionvector/A0E0L1I0M0H0T0D0
@@ -52,6 +54,7 @@ module Unitsdb
       "qudt:dimensionvector:#{uri}"
     end
   end
+  Config.register_model(QudtDimensionVector, id: :qudt_dimension_vector)
 
   # QUDT SystemOfUnits from sou vocabulary
   # Example: http://qudt.org/vocab/sou/SI
@@ -65,6 +68,7 @@ module Unitsdb
       "qudt:sou:#{uri}"
     end
   end
+  Config.register_model(QudtSystemOfUnits, id: :qudt_system_of_units)
 
   # QUDT Prefix from prefixes vocabulary
   # Example: http://qudt.org/vocab/prefix/Kilo
@@ -83,6 +87,7 @@ module Unitsdb
       "qudt:prefix:#{uri}"
     end
   end
+  Config.register_model(QudtPrefix, id: :qudt_prefix)
 
   # Container for all QUDT vocabularies
   class QudtVocabularies

data/lib/unitsdb/root_unit_reference.rb

@@ -8,4 +8,6 @@ module Unitsdb
     attribute :unit_reference, UnitReference
     attribute :prefix_reference, PrefixReference
   end
+
+  Config.register_model(RootUnitReference, id: :root_unit_reference)
 end

data/lib/unitsdb/scale.rb

@@ -10,4 +10,6 @@ module Unitsdb
     attribute :short, :string
     attribute :properties, ScaleProperties
   end
+
+  Config.register_model(Scale, id: :scale)
 end

data/lib/unitsdb/scale_properties.rb

@@ -9,4 +9,6 @@ module Unitsdb
     attribute :interval, :boolean
     attribute :ratio, :boolean
   end
+
+  Config.register_model(ScaleProperties, id: :scale_properties)
 end

data/lib/unitsdb/scale_reference.rb

@@ -5,4 +5,6 @@ module Unitsdb
     attribute :id, :string
     attribute :type, :string
   end
+
+  Config.register_model(ScaleReference, id: :scale_reference)
 end

data/lib/unitsdb/scales.rb

@@ -7,4 +7,6 @@ module Unitsdb
     attribute :version, :string
     attribute :scales, Scale, collection: true
   end
+
+  Config.register_model(Scales, id: :scales)
 end

data/lib/unitsdb/si_derived_base.rb

@@ -14,4 +14,6 @@ module Unitsdb
   class SiDerivedBase < RootUnitReference
     # model Config.model_for(:si_derived_base)
   end
+
+  Config.register_model(SiDerivedBase, id: :si_derived_base)
 end

data/lib/unitsdb/symbol_presentations.rb

@@ -11,4 +11,6 @@ module Unitsdb
     attribute :mathml, :string
     attribute :unicode, :string
   end
+
+  Config.register_model(SymbolPresentations, id: :symbol_presentations)
 end

data/lib/unitsdb/ucum.rb

@@ -28,6 +28,7 @@ module Unitsdb
       "ucum:base-unit:code:#{code_sensitive}"
     end
   end
+  Config.register_model(UcumBaseUnit, id: :ucum_base_unit)
 
   #  <prefix Code="Y" CODE="YA">
   #   <name>yotta</name>
@@ -46,6 +47,7 @@ module Unitsdb
       map_content to: :content
     end
   end
+  Config.register_model(UcumPrefixValue, id: :ucum_prefix_value)
 
   class UcumPrefix < Lutaml::Model::Serializable
     attribute :code_sensitive, :string
@@ -67,6 +69,7 @@ module Unitsdb
       "ucum:prefix:code:#{code_sensitive}"
     end
   end
+  Config.register_model(UcumPrefix, id: :ucum_prefix)
 
   #  <unit Code="10*" CODE="10*" isMetric="no" class="dimless">
   #   <name>the number ten for arbitrary powers</name>
@@ -114,6 +117,7 @@ module Unitsdb
       map_attribute "Unit", to: :unit_sensitive
     end
   end
+  Config.register_model(UcumUnitValueFunction, id: :ucum_unit_value_function)
 
   class UcumUnitValue < Lutaml::Model::Serializable
     attribute :unit_sensitive, :string
@@ -131,6 +135,7 @@ module Unitsdb
       map_content to: :content
     end
   end
+  Config.register_model(UcumUnitValue, id: :ucum_unit_value)
 
   class UcumUnit < Lutaml::Model::Serializable
     attribute :code_sensitive, :string
@@ -165,6 +170,7 @@ module Unitsdb
       "ucum:unit:#{k}:code:#{code_sensitive}"
     end
   end
+  Config.register_model(UcumUnit, id: :ucum_unit)
 
   class UcumNamespace < Lutaml::Xml::Namespace
     uri "http://unitsofmeasure.org/ucum-essence"
@@ -199,4 +205,5 @@ module Unitsdb
 
     # No adapter registration needed
   end
+  Config.register_model(UcumFile, id: :ucum_file)
 end

data/lib/unitsdb/unit.rb

@@ -50,4 +50,6 @@ module Unitsdb
     attribute :references, ExternalReference, collection: true
     attribute :scale_reference, ScaleReference
   end
+
+  Config.register_model(Unit, id: :unit)
 end

data/lib/unitsdb/unit_reference.rb

@@ -5,4 +5,6 @@ module Unitsdb
     attribute :id, :string
     attribute :type, :string
   end
+
+  Config.register_model(UnitReference, id: :unit_reference)
 end

data/lib/unitsdb/unit_system.rb

@@ -10,4 +10,6 @@ module Unitsdb
     attribute :acceptable, :boolean
     attribute :references, ExternalReference, collection: true
   end
+
+  Config.register_model(UnitSystem, id: :unit_system)
 end

data/lib/unitsdb/unit_system_reference.rb

@@ -5,4 +5,6 @@ module Unitsdb
     attribute :id, :string
     attribute :type, :string
   end
+
+  Config.register_model(UnitSystemReference, id: :unit_system_reference)
 end

data/lib/unitsdb/unit_systems.rb

@@ -8,4 +8,6 @@ module Unitsdb
     attribute :version, :string
     attribute :unit_systems, UnitSystem, collection: true
   end
+
+  Config.register_model(UnitSystems, id: :unit_systems)
 end

data/lib/unitsdb/units.rb

@@ -8,4 +8,6 @@ module Unitsdb
     attribute :version, :string
     attribute :units, Unit, collection: true
   end
+
+  Config.register_model(Units, id: :units)
 end

data/lib/unitsdb/version.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module Unitsdb
-  VERSION = "2.2.1"
+  VERSION = "2.2.2"
 
   # The version of the bundled UnitsDB data (from the data/ submodule).
   # This version corresponds to a tag/commit in https://github.com/unitsml/unitsdb.

data/lib/unitsdb.rb

@@ -1,52 +1,45 @@
 # frozen_string_literal: true
 
 require "lutaml/model"
+require "unitsdb/config"
+require "unitsdb/identifier"
+require "unitsdb/localized_string"
+require "unitsdb/symbol_presentations"
+require "unitsdb/scale_properties"
+require "unitsdb/unit_reference"
+require "unitsdb/prefix_reference"
+require "unitsdb/quantity_reference"
+require "unitsdb/dimension_reference"
+require "unitsdb/unit_system_reference"
+require "unitsdb/scale_reference"
+require "unitsdb/external_reference"
+require "unitsdb/root_unit_reference"
+require "unitsdb/si_derived_base"
+require "unitsdb/dimension_details"
+require "unitsdb/dimension"
+require "unitsdb/prefix"
+require "unitsdb/unit_system"
+require "unitsdb/quantity"
+require "unitsdb/scale"
+require "unitsdb/unit"
+require "unitsdb/dimensions"
+require "unitsdb/prefixes"
+require "unitsdb/quantities"
+require "unitsdb/scales"
+require "unitsdb/unit_systems"
+require "unitsdb/units"
+require "unitsdb/database"
+require "unitsdb/qudt"
+require "unitsdb/ucum"
 
 module Unitsdb
-  autoload :Cli, "unitsdb/cli"
-  autoload :Config, "unitsdb/config"
-  autoload :Commands, "unitsdb/commands"
-  autoload :Database, "unitsdb/database"
-  autoload :Dimension, "unitsdb/dimension"
-  autoload :DimensionDetails, "unitsdb/dimension_details"
-  autoload :DimensionReference, "unitsdb/dimension_reference"
-  autoload :Dimensions, "unitsdb/dimensions"
+  # Core models are eagerly loaded so type registrations are complete before
+  # any context or database loading happens.
+  unless RUBY_ENGINE == "opal"
+    autoload :Cli, "unitsdb/cli"
+    autoload :Commands, "unitsdb/commands"
+  end
   autoload :Errors, "unitsdb/errors"
-  autoload :ExternalReference, "unitsdb/external_reference"
-  autoload :Identifier, "unitsdb/identifier"
-  autoload :LocalizedString, "unitsdb/localized_string"
-  autoload :Prefix, "unitsdb/prefix"
-  autoload :PrefixReference, "unitsdb/prefix_reference"
-  autoload :Prefixes, "unitsdb/prefixes"
-  autoload :Quantities, "unitsdb/quantities"
-  autoload :Quantity, "unitsdb/quantity"
-  autoload :QuantityReference, "unitsdb/quantity_reference"
-  autoload :QudtUnit, "unitsdb/qudt"
-  autoload :QudtQuantityKind, "unitsdb/qudt"
-  autoload :QudtDimensionVector, "unitsdb/qudt"
-  autoload :QudtSystemOfUnits, "unitsdb/qudt"
-  autoload :QudtPrefix, "unitsdb/qudt"
-  autoload :QudtVocabularies, "unitsdb/qudt"
-  autoload :RootUnitReference, "unitsdb/root_unit_reference"
-  autoload :Scale, "unitsdb/scale"
-  autoload :ScaleProperties, "unitsdb/scale_properties"
-  autoload :ScaleReference, "unitsdb/scale_reference"
-  autoload :Scales, "unitsdb/scales"
-  autoload :SiDerivedBase, "unitsdb/si_derived_base"
-  autoload :SymbolPresentations, "unitsdb/symbol_presentations"
-  autoload :UcumBaseUnit, "unitsdb/ucum"
-  autoload :UcumPrefixValue, "unitsdb/ucum"
-  autoload :UcumPrefix, "unitsdb/ucum"
-  autoload :UcumUnitValueFunction, "unitsdb/ucum"
-  autoload :UcumUnitValue, "unitsdb/ucum"
-  autoload :UcumUnit, "unitsdb/ucum"
-  autoload :UcumFile, "unitsdb/ucum"
-  autoload :Unit, "unitsdb/unit"
-  autoload :UnitReference, "unitsdb/unit_reference"
-  autoload :UnitSystem, "unitsdb/unit_system"
-  autoload :UnitSystemReference, "unitsdb/unit_system_reference"
-  autoload :UnitSystems, "unitsdb/unit_systems"
-  autoload :Units, "unitsdb/units"
   autoload :Utils, "unitsdb/utils"
 
   class << self
@@ -56,12 +49,19 @@ module Unitsdb
     end
 
     # Returns a pre-loaded Database instance from the bundled data
-    def database
-      @database ||= Database.from_db(data_dir)
+    def database(context: Config.context_id)
+      context_id = context.to_sym
+      Config.context(context_id) if context_id == Config.context_id && Config.find_context(context_id).nil?
+      klass = Config.resolve_type(:database, context: context_id)
+      databases[context_id] ||= klass.from_db(data_dir, context: context_id)
     end
 
     private
 
+    def databases
+      @databases ||= {}
+    end
+
     def gem_dir
       @gem_dir ||= File.dirname(__dir__)
     end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: unitsdb
 version: !ruby/object:Gem::Version
-  version: 2.2.1
+  version: 2.2.2
 platform: ruby
 authors:
 - Ribose Inc.
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2026-04-03 00:00:00.000000000 Z
+date: 2026-04-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: fuzzy_match