query_builder 0.0.1

data/lib/query_builder/core/statement.rb

@@ -0,0 +1,55 @@
+# encoding: utf-8
+
+module QueryBuilder::Core
+
+  # Abstract base class for CQL statemens (roots of AST)
+  #
+  # The statement can belong to some context (keyspace, table).
+  # It is created with several clauses (like WHERE, SET etc.)
+  #
+  # @abstract
+  #
+  class Statement < Base
+
+    attribute :context, default: nil
+
+    # @!method initialize(type, *attributes, &block)
+    # Initializes the statement with a type, attributes and block
+    #
+    # @param (see QueryBuilder::Clause)
+    # @param [Proc] block
+    #   The block that returns an array of clauses for the statement
+    #
+    def initialize(*)
+      @clauses = block_given? ? Array[*yield] : []
+      super
+    end
+
+    # @!method clauses(type = nil)
+    # Returns array of chunks of code for the statement's clauses
+    #
+    # @param [Symbol] type The optional type to select clauses by
+    #
+    # @return [Array<String>]
+    #
+    def clauses(cond = nil)
+      (cond ? @clauses.select { |item| cond.equal? item.type } : @clauses)
+        .map(&:to_s)
+    end
+
+    # @!method <<(clause)
+    # Returns a new statement where the clause is added
+    #
+    # @param [QueryBuilder::Clause] clause
+    #
+    # @return [QueryBuilder::Statement]
+    #
+    def <<(clause)
+      new_clauses = @clauses.dup
+      new_clauses.delete clause
+      self.class.new(attributes) { new_clauses + [clause] }
+    end
+
+  end # class Statement
+
+end # module QueryBuilder::Core

data/lib/query_builder/cql.rb

@@ -0,0 +1,80 @@
+# encoding: utf-8
+
+module QueryBuilder
+
+  require_relative "cql/operators"
+  require_relative "cql/contexts"
+  require_relative "cql/modifiers"
+  require_relative "cql/statements"
+
+  # The builder for CQL-specific statements
+  #
+  module CQL
+
+    # Returns the context of Cassandra keyspace
+    #
+    # @param [#to_s] name The name of the keyspace
+    #
+    # @param [QueryBuilder::CQL::Contexts::Keyspace]
+    #
+    def self.keyspace(name)
+      Contexts::Keyspace.new(name: name)
+    end
+
+    # Returns the context of Cassandra user
+    #
+    # @param [#to_s] name The name of the user
+    #
+    # @param [QueryBuilder::CQL::Contexts::User]
+    #
+    def self.user(name)
+      Contexts::User.new(name: name)
+    end
+
+    # Returns the context of Cassandra users role
+    #
+    # @param [#to_s] name The name of the role
+    #
+    # @param [QueryBuilder::CQL::Contexts::Role]
+    #
+    def self.role(name)
+      Contexts::Role.new(name: name)
+    end
+
+    # Returns the context of Cassandra root permission
+    #
+    # @param [#to_s, nil] name The name of the permission
+    #
+    # @param [QueryBuilder::CQL::Contexts::Permission]
+    #
+    def self.permission(name = nil)
+      Contexts::Permission.new(name: name)
+    end
+
+    # Builds the 'LIST USERS' CQL statement
+    #
+    # @return [QueryBuilder::Statements::ListUsers]
+    #
+    def self.users
+      Statements::ListUsers.new
+    end
+
+    # Builds the 'LIST ROLES' CQL statement
+    #
+    # @return [QueryBuilder::Statements::ListRoles]
+    #
+    def self.roles
+      Statements::ListRoles.new
+    end
+
+    # Builds the 'BATCH' CQL statement
+    #
+    # @return [QueryBuilder::Statements::Batch]
+    #
+    def self.batch
+      Statements::Batch.new
+    end
+
+  end # module CQL
+
+end # module QueryBuilder

data/lib/query_builder/cql/contexts.rb

@@ -0,0 +1,21 @@
+# encoding: utf-8
+
+module QueryBuilder
+
+  module CQL
+
+    # The collection of CQL-specific contexts for statements
+    #
+    module Contexts
+
+      # The base class for CQL-specific contexts
+      #
+      Base = Core::Base
+
+      Dir[File.expand_path("../contexts/*.rb", __FILE__)].each(&method(:load))
+
+    end # module Contexts
+
+  end # module CQL
+
+end # module QueryBuilder

data/lib/query_builder/cql/contexts/aggregate.rb

@@ -0,0 +1,42 @@
+# encoding: utf-8
+
+module QueryBuilder::CQL
+
+  module Contexts
+
+    # Describes the Cassandra user-defined aggregate
+    #
+    class Aggregate < Base
+
+      attribute :keyspace, required: true
+      attribute :name, required: true
+
+      # Returns the full name of the aggregate
+      #
+      # @return [String]
+      #
+      def to_s
+        [keyspace, name].join(".")
+      end
+
+      # Builds the 'CREATE AGGREGATE' CQL statement
+      #
+      # @return [QueryBuilder::Statements::CreateAggregate]
+      #
+      def create
+        Statements::CreateAggregate.new(context: self)
+      end
+
+      # Builds the 'DROP AGGREGATE' CQL statement
+      #
+      # @return [QueryBuilder::Statements::DropAggregate]
+      #
+      def drop
+        Statements::DropAggregate.new(context: self)
+      end
+
+    end # class Aggregate
+
+  end # module Contexts
+
+end # module QueryBuilder::CQL

data/lib/query_builder/cql/contexts/column.rb

@@ -0,0 +1,59 @@
+# encoding: utf-8
+
+module QueryBuilder::CQL
+
+  module Contexts
+
+    # Describes the Cassandra table column
+    #
+    class Column < Base
+
+      attribute :table, required: true
+      attribute :name,  required: true
+
+      # Builds the 'ALTER TABLE ... ADD' CQL statement
+      #
+      # @param [#to_s] type
+      # @param [Hash] options
+      # @option options [Boolean] :static
+      #
+      # @return [QueryBuilder::Statements::AddColumn]
+      #
+      def create(type, options = {})
+        Statements::CreateColumn
+          .new(context: self, type: type, static: options[:static])
+      end
+
+      # Builds the 'ALTER TABLE ... RENAME' CQL statement
+      #
+      # @param [#to_s] name
+      #
+      # @return [QueryBuilder::Statements::RenameColumn]
+      #
+      def rename(name)
+        Statements::RenameColumn.new(context: self, name: name)
+      end
+
+      # Builds the 'ALTER TABLE ... ALTER' CQL statement
+      #
+      # @param [#to_s] type
+      #
+      # @return [QueryBuilder::Statements::AlterColumn]
+      #
+      def alter(type)
+        Statements::AlterColumn.new(context: self, type: type)
+      end
+
+      # Builds the 'ALTER TABLE ... DROP' CQL statement
+      #
+      # @return [QueryBuilder::Statements::DropColumn]
+      #
+      def drop
+        Statements::DropColumn.new(context: self)
+      end
+
+    end # class Column
+
+  end # module Contexts
+
+end # module QueryBuilder::CQL

data/lib/query_builder/cql/contexts/field.rb

@@ -0,0 +1,48 @@
+# encoding: utf-8
+
+module QueryBuilder::CQL
+
+  module Contexts
+
+    # Describes the field of Cassandra user-defined type
+    #
+    class Field < Base
+
+      attribute :type, required: true
+      attribute :name, required: true
+
+      # Builds the 'ALTER TYPE ... ADD' CQL statement
+      #
+      # @param [#to_s] type
+      #
+      # @return [QueryBuilder::Statements::CreateField]
+      #
+      def create(type)
+        Statements::CreateField.new(context: self, type: type)
+      end
+
+      # Builds the 'ALTER TYPE ... ALTER' CQL statement
+      #
+      # @param [#to_s] type
+      #
+      # @return [QueryBuilder::Statements::AlterField]
+      #
+      def alter(type)
+        Statements::AlterField.new(context: self, type: type)
+      end
+
+      # Builds the 'ALTER TYPE ... RENAME' CQL statement
+      #
+      # @param [#to_s] name
+      #
+      # @return [QueryBuilder::Statements::RenameField]
+      #
+      def rename(name)
+        Statements::RenameField.new(context: self, name: name)
+      end
+
+    end # class Field
+
+  end # module Contexts
+
+end # module QueryBuilder::CQL

data/lib/query_builder/cql/contexts/function.rb

@@ -0,0 +1,46 @@
+# encoding: utf-8
+
+module QueryBuilder::CQL
+
+  module Contexts
+
+    # Describes the Cassandra user-defined function
+    #
+    class Function < Base
+
+      attribute :keyspace, required: true
+      attribute :name, required: true
+
+      # Returns the full name of the function
+      #
+      # @return [String]
+      #
+      def to_s
+        [keyspace, name].join(".")
+      end
+
+      # Builds the 'CREATE FUNCTION' CQL statement
+      #
+      # @param [#to_s] language
+      # @param [#to_s] body
+      #
+      # @return [QueryBuilder::Statements::CreateFunction]
+      #
+      def create(language, body)
+        Statements::CreateFunction
+          .new(context: self, language: language, body: body)
+      end
+
+      # Builds the 'DROP FUNCTION' CQL statement
+      #
+      # @return [QueryBuilder::Statements::DropFunction]
+      #
+      def drop
+        Statements::DropFunction.new(context: self)
+      end
+
+    end # class Function
+
+  end # module Contexts
+
+end # module QueryBuilder::CQL

data/lib/query_builder/cql/contexts/index.rb

@@ -0,0 +1,44 @@
+# encoding: utf-8
+
+module QueryBuilder::CQL
+
+  module Contexts
+
+    # Describes the Cassandra table index
+    #
+    class Index < Base
+
+      attribute :table, required: true
+      attribute :name
+
+      # <description>
+      #
+      # @return [<type>] <description>
+      #
+      def to_s
+        [table.keyspace.name, name].join(".")
+      end
+
+      # Builds the 'CREATE INDEX' CQL statement for the current table
+      #
+      # @param [Hash] options
+      #
+      # @return [QueryBuilder::Statements::CreateIndex]
+      #
+      def create(*options)
+        Statements::CreateIndex.new(context: self).add(options)
+      end
+
+      # Builds the 'DROP INDEX' CQL statement
+      #
+      # @return [QueryBuilder::Statements::DropIndex]
+      #
+      def drop
+        Statements::DropIndex.new(context: self)
+      end
+
+    end # class Index
+
+  end # module Contexts
+
+end # module QueryBuilder::CQL

data/lib/query_builder/cql/contexts/keyspace.rb

@@ -0,0 +1,107 @@
+# encoding: utf-8
+
+module QueryBuilder::CQL
+
+  module Contexts
+
+    # Describes the Cassandra keyspace
+    #
+    class Keyspace < Base
+
+      attribute :name, required: true
+
+      # Returns the name of the keyspace
+      #
+      # @return [String]
+      #
+      def to_s
+        name.to_s
+      end
+
+      # Returns the table context
+      #
+      # @param [#to_s] name
+      #
+      # @return [QueryBuilder::CQL::Contexts::Table]
+      #
+      def table(name)
+        Table.new(keyspace: self, name: name)
+      end
+
+      # Returns the type context
+      #
+      # @param [#to_s] name
+      #
+      # @return [QueryBuilder::CQL::Contexts::Type]
+      #
+      def type(name)
+        Type.new(keyspace: self, name: name)
+      end
+
+      # Returns the function context
+      #
+      # @param [#to_s] name
+      #
+      # @return [QueryBuilder::CQL::Contexts::Function]
+      #
+      def function(name)
+        Function.new(keyspace: self, name: name)
+      end
+
+      # Returns the aggregate context
+      #
+      # @param [#to_s] name
+      #
+      # @return [QueryBuilder::CQL::Contexts::Aggregate]
+      #
+      def aggregate(name)
+        Aggregate.new(keyspace: self, name: name)
+      end
+
+      # Returns the context of Cassandra keyspace permission
+      #
+      # @param [#to_s, nil] name The name of the permission
+      #
+      # @param [QueryBuilder::CQL::Contexts::Permission]
+      #
+      def permission(name = nil)
+        Contexts::Permission.new(keyspace: self, name: name)
+      end
+
+      # Builds the 'CREATE KEYSPACE' CQL statement for the current keyspace
+      #
+      # @return [QueryBuilder::Statements::CreateKeyspace]
+      #
+      def create
+        Statements::CreateKeyspace.new(context: self)
+      end
+
+      # Builds the 'ALTER KEYSPACE' CQL statement for the current keyspace
+      #
+      # @return [QueryBuilder::CQL::Statements::AlterKeyspace]
+      #
+      def alter
+        Statements::AlterKeyspace.new(context: self)
+      end
+
+      # Builds the 'DROP KEYSPACE' CQL statement for the current keyspace
+      #
+      # @return [QueryBuilder::CQL::Statements::DropKeyspace]
+      #
+      def drop
+        Statements::DropKeyspace.new(context: self)
+      end
+
+      # Builds the 'USE' CQL statement
+      #
+      # @return [QueryBuilder::Statements::Use]
+      #
+      def use
+        Statements::Use.new(context: self)
+      end
+
+    end # class Keyspace
+
+  end # module Contexts
+
+end # module QueryBuilder::CQL::Contexts