Sha256: ec6059e06029f71e6c424bcefbc948acd107f536bc659aa793019d8ffb820609

Contents?: true

Size: 1.43 KB

Versions: 32

Compression:

Stored size: 1.43 KB

Contents

require "yaml"

module Lutaml
  module Model
    module Schema
      class YamlSchema
        def self.generate(klass, options = {})
          schema = generate_schema(klass)
          options[:pretty] ? schema.to_yaml : YAML.dump(schema)
        end

        def self.generate_schema(klass)
          {
            "type" => "map",
            "mapping" => generate_mapping(klass),
          }
        end

        def self.generate_mapping(klass)
          klass.attributes.each_with_object({}) do |(name, attr), mapping|
            mapping[name.to_s] = generate_attribute_schema(attr)
          end
        end

        def self.generate_attribute_schema(attr)
          if attr.type <= Lutaml::Model::Serialize
            generate_schema(attr.type)
          elsif attr.collection?
            {
              "type" => "seq",
              "sequence" => [{ "type" => get_yaml_type(attr.type) }],
            }
          else
            { "type" => get_yaml_type(attr.type) }
          end
        end

        def self.get_yaml_type(type)
          {
            Lutaml::Model::Type::String => "str",
            Lutaml::Model::Type::Integer => "int",
            Lutaml::Model::Type::Boolean => "bool",
            Lutaml::Model::Type::Float => "float",
            Lutaml::Model::Type::Decimal => "float",
            Lutaml::Model::Type::Hash => "map",
          }[type] || "str" # Default to string for unknown types
        end
      end
    end
  end
end

Version data entries

32 entries across 32 versions & 1 rubygems

Version Path
lutaml-model-0.3.15 lib/lutaml/model/schema/yaml_schema.rb
lutaml-model-0.3.14 lib/lutaml/model/schema/yaml_schema.rb
lutaml-model-0.3.13 lib/lutaml/model/schema/yaml_schema.rb
lutaml-model-0.3.12 lib/lutaml/model/schema/yaml_schema.rb
lutaml-model-0.3.11 lib/lutaml/model/schema/yaml_schema.rb
lutaml-model-0.3.10 lib/lutaml/model/schema/yaml_schema.rb
lutaml-model-0.3.9 lib/lutaml/model/schema/yaml_schema.rb
lutaml-model-0.3.8 lib/lutaml/model/schema/yaml_schema.rb
lutaml-model-0.3.7 lib/lutaml/model/schema/yaml_schema.rb
lutaml-model-0.3.6 lib/lutaml/model/schema/yaml_schema.rb
lutaml-model-0.3.5 lib/lutaml/model/schema/yaml_schema.rb
lutaml-model-0.3.4 lib/lutaml/model/schema/yaml_schema.rb