Example Axiom::Schema using Axiom objects

axiom-schema-example.rb

require 'axiom'

module Axiom
  class Schema < BasicObject
    module DSL
      def build(*args, &block)
        object = new(*args)
        object.instance_eval(&block)
        object
      end
    end

    extend DSL

    def initialize(databases = {})
      @databases = databases
    end

    def database(name, uri, &block)
      @databases[name] = Database.build(uri, &block)
      self
    end

    def [](name)
      @databases[name]
    end

    class Database < BasicObject
      extend DSL

      def initialize(uri, relations = {})
        @uri       = uri
        @relations = relations
      end

      def base_relation(name, &block)
        @relations[name] = BaseRelation.build(name, &block)
        self
      end

      def [](name)
        @relations[name]
      end
    end

    class BaseRelation < BasicObject
      extend DSL

      def initialize(name)
        @name     = name
        @relation = ::Axiom::TABLE_DUM
      end

      def attribute(name, type)
        @relation = @relation.extend { |r| r.add([name, type]) }
        self
      end

      def relation
        ::Axiom::Relation::Base.new(@name, @relation.header)
      end
    end
  end
end

schema = Axiom::Schema.build do
  database(:memory, ':memory://test') do
    base_relation(:people) do
      attribute :id,    Integer
      attribute :email, String
      attribute :name,  String
    end

    base_relation(:addresses) do
      attribute :id,        Integer
      attribute :person_id, Integer
      attribute :city,      String
      attribute :zip,       String
    und
  end
end

people    = schema[:memory][:people].relation
addresses = schema[:memory][:addresses].relation

puts people.inspect
puts addresses.inspect