legion-data 0.1.0

data/lib/legion/data/connection.rb

@@ -0,0 +1,30 @@
+require 'sequel'
+
+module Legion
+  module Data
+    class Connection
+      attr_accessor :database
+      def initialize(options = {})
+        options.merge!(default_options) { |_key, v1, _v2| v1 }
+        return unless options[:auto_connect]
+
+        jruby(options) if RUBY_ENGINE == 'jruby'
+        mri(options) unless RUBY_ENGINE == 'jruby'
+      end
+
+      def default_options
+        { auto_connect: true }
+      end
+
+      def jruby(_options = {})
+        require 'legion/data/connections/jdbc'
+        @database = Legion::Data::Connections::JDBC.new
+      end
+
+      def mri(_options = {})
+        require 'legion/data/connections/mysql2'
+        @database = Legion::Data::Connections::MySQL2.new
+      end
+    end
+  end
+end

data/lib/legion/data/connections/base.rb

@@ -0,0 +1,45 @@
+module Legion
+  module Data
+    module Connections
+      class Base
+        def check_gem(gem = adapter)
+          Gem::Specification.find_by_name(gem)
+        rescue Gem::MissingSpecError, ArgumentError
+          false
+        end
+
+        def creds_builder(credentials = {})
+          creds = {}
+          creds = creds.merge(default_creds) if creds.is_a? Hash
+          creds = creds.merge(Legion::Settings[:data][:mysql]) if Legion::Settings[:data][:mysql].is_a? Hash
+          creds = creds.merge(credentials) if credentials.is_a? Hash
+          creds
+        end
+
+        def connect(adapter, creds, _options = {})
+          creds[:adapter] = adapter
+          Sequel.connect(creds)
+        end
+
+        def settings(options, _connection = @connection)
+          raise ArgumentError unless options.is_a? Hash
+        end
+
+        def debug_logging(enabled = true, connection = @connection, _options = {})
+          if enabled
+            connection.loggers = Legion::Logging
+            connection.sql_log_level = :debug
+          elsif enabled == false
+            connection.loggers = nil
+            connection.sql_log_level = :debug
+          end
+          connection
+        end
+
+        def adapter
+          'mysql2'
+        end
+      end
+    end
+  end
+end

data/lib/legion/data/connections/jdbc.rb

@@ -0,0 +1,21 @@
+module Legion
+  module Data
+    module Connections
+      class JDBC
+        attr_accessor :connection
+        def initialize
+          Legion::Logging.debug('Connecting to MySQL with JBDC Connector')
+          connection_string = build_connection_string
+          @connection = Sequel.connect(connection_string, max_connetions: @args[:max_connections])
+          Legion::Logging.info("Legion is connected to database #{args[:database]}")
+        end
+
+        def build_connection_string
+          set = Legion::Settings[:data][:mysql]
+          con_string = "jdbc:mysql://#{set[:host]}/#{set[:database]}?user=#{set[:user]}&password=#{set[:password]}"
+          con_string
+        end
+      end
+    end
+  end
+end

data/lib/legion/data/connections/mysql.rb

@@ -0,0 +1,26 @@
+require 'legion/data/connections/mysql_base'
+
+module Legion
+  module Data
+    module Connections
+      class MySQL < Legion::Data::Connections::MySQLBase
+        attr_accessor :connection
+        def initialize(_options = {})
+          Legion::Logging.debug('Connecting to MySQL with MySQL2 Connector')
+          @connection = connect
+
+          Legion::Logging.info("Legion is connected to database #{Legion::Settings[:data][:mysql][:database]}")
+          Legion::Settings[:data][:connected] = true
+        end
+
+        def adapter
+          'mysql2'
+        end
+
+        def connect
+          super(adapter, creds_builder)
+        end
+      end
+    end
+  end
+end

data/lib/legion/data/connections/mysql2.rb

@@ -0,0 +1,26 @@
+require 'legion/data/connections/mysql_base'
+
+module Legion
+  module Data
+    module Connections
+      class MySQL2 < Legion::Data::Connections::MySQLBase
+        attr_accessor :connection
+        def initialize(_options = {})
+          Legion::Logging.debug('Connecting to MySQL with MySQL2 Connector')
+          @connection = connect
+
+          Legion::Logging.info("Legion is connected to database #{Legion::Settings[:data][:mysql][:database]}")
+          Legion::Settings[:data][:connected] = true
+        end
+
+        def adapter
+          'mysql2'
+        end
+
+        def connect
+          super(adapter, creds_builder)
+        end
+      end
+    end
+  end
+end

data/lib/legion/data/connections/mysql_base.rb

@@ -0,0 +1,19 @@
+require 'legion/data/connections/base'
+
+module Legion
+  module Data
+    module Connections
+      class MySQLBase < Legion::Data::Connections::Base
+        def default_creds
+          { data: { mysql: {
+            host:            '127.0.0.1',
+            user:            'legion',
+            password:        'legion',
+            max_connections: 10,
+            database:        'legion'
+          } } }
+        end
+      end
+    end
+  end
+end

data/lib/legion/data/migration.rb

@@ -0,0 +1,28 @@
+require 'sequel/extensions/migration'
+
+module Legion
+  module Data
+    class Migration
+      def initialize(connection, options = {})
+        options.merge!(default_options) { |_key, v1, _v2| v1 }
+        migrate(options[:migration_path], connection) if options[:auto_migrate]
+      end
+
+      def default_options
+        { auto_migrate: true, migration_path: __dir__ + '/migrations' }
+      end
+
+      def migrate(path, connection)
+        Legion::Logging.debug("Running Legion::Data.migrate with path: #{path}")
+        Sequel::Migrator.run(connection, path)
+        Legion::Logging.info('Legion::Data finished migrations')
+      rescue Sequel::Migrator::Error => ex
+        Legion::Logging.error(ex.message)
+        raise ex
+      rescue StandardError => ex
+        Legion::Logging.error(ex.message)
+        raise ex unless Legion::Settings[:data][:migrations][:continue_on_fail]
+      end
+    end
+  end
+end

data/lib/legion/data/migrations/001_add_schema_columns.rb

@@ -0,0 +1,17 @@
+require 'sequel/extensions/migration'
+
+Sequel.migration do
+  up do
+    run 'ALTER TABLE `schema_info` ADD `created_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP AFTER `version`;'
+    run 'ALTER TABLE `schema_info` ADD `updated_at` TIMESTAMP  NULL  ON UPDATE CURRENT_TIMESTAMP  AFTER `created_at`;'
+    run 'ALTER TABLE `schema_info` ADD `catalog` VARCHAR(255)  NULL  DEFAULT NULL  AFTER `version`;'
+  end
+
+  down do
+    alter_table(:schema) do
+      drop_column :catalog
+      drop_column :updated_at
+      drop_column :created_at
+    end
+  end
+end

data/lib/legion/data/migrations/002_add_chains_table.rb

@@ -0,0 +1,21 @@
+Sequel.migration do
+  up do
+    run "CREATE TABLE `chains` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
+  `name` varchar(128) NOT NULL DEFAULT '',
+  `active` tinyint(1) unsigned DEFAULT '1',
+  `version` int(11) unsigned NOT NULL DEFAULT '1',
+  `created_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated_at` timestamp NULL DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP,
+  `user_owner` int(11) unsigned DEFAULT NULL,
+  `group_owner` int(11) unsigned DEFAULT NULL,
+  PRIMARY KEY (`id`),
+  UNIQUE KEY `name` (`name`),
+  KEY `active` (`active`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;"
+  end
+
+  down do
+    drop_table :chains
+  end
+end

data/lib/legion/data/migrations/003_add_datacenters_table.rb

@@ -0,0 +1,21 @@
+Sequel.migration do
+  up do
+    run "CREATE TABLE `datacenters` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
+  `active` tinyint(1) unsigned NOT NULL DEFAULT '1',
+  `version` int(11) unsigned NOT NULL DEFAULT '1',
+  `name` varchar(128) NOT NULL DEFAULT '',
+  `user_owner` int(11) unsigned DEFAULT NULL,
+  `group_owner` int(11) unsigned DEFAULT NULL,
+  `created_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated_at` timestamp NULL DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP,
+  PRIMARY KEY (`id`),
+  UNIQUE KEY `name` (`name`),
+  KEY `active` (`active`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;"
+  end
+
+  down do
+    drop_table :datacenters
+  end
+end

data/lib/legion/data/migrations/004_add_envs_table.rb

@@ -0,0 +1,21 @@
+Sequel.migration do
+  up do
+    run "CREATE TABLE `environments` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
+  `name` varchar(128) NOT NULL DEFAULT '',
+  `active` tinyint(1) unsigned NOT NULL DEFAULT '1',
+  `version` int(11) unsigned NOT NULL DEFAULT '1',
+  `user_owner` int(11) unsigned DEFAULT NULL,
+  `group_owner` int(11) unsigned DEFAULT NULL,
+  `created_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated_at` timestamp NULL DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP,
+  PRIMARY KEY (`id`),
+  UNIQUE KEY `name` (`name`),
+  KEY `active` (`active`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;"
+  end
+
+  down do
+    drop_table :environments
+  end
+end

data/lib/legion/data/migrations/005_add_functions_table.rb

@@ -0,0 +1,25 @@
+Sequel.migration do
+  up do
+    run "CREATE TABLE `functions` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
+  `active` tinyint(1) unsigned NOT NULL DEFAULT '1',
+  `version` int(11) unsigned NOT NULL DEFAULT '1',
+  `name` varchar(128) NOT NULL DEFAULT '',
+  `namespace_id` int(11) unsigned NOT NULL,
+  `args` text,
+  `user_owner` int(11) unsigned DEFAULT NULL,
+  `group_owner` int(11) unsigned DEFAULT NULL,
+  `created_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated_at` timestamp NULL DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP,
+  PRIMARY KEY (`id`),
+  UNIQUE KEY `namespace_id` (`namespace_id`,`name`),
+  KEY `active` (`active`),
+  KEY `namespace` (`namespace_id`),
+  KEY `name` (`name`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;"
+  end
+
+  down do
+    drop_table :functionss
+  end
+end

data/lib/legion/data/migrations/006_add_namespaces_table.rb

@@ -0,0 +1,24 @@
+Sequel.migration do
+  up do
+    run "CREATE TABLE `namespaces` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
+  `active` tinyint(1) unsigned NOT NULL DEFAULT '1',
+  `namespace` varchar(128) NOT NULL DEFAULT '',
+  `queue` varchar(128) NOT NULL DEFAULT '',
+  `uri` varchar(128) DEFAULT NULL,
+  `created_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated_at` timestamp NULL DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP,
+  `group_owner` int(11) unsigned DEFAULT NULL,
+  `user_owner` int(11) unsigned DEFAULT NULL,
+  PRIMARY KEY (`id`),
+  UNIQUE KEY `namespace` (`namespace`),
+  KEY `active` (`active`),
+  KEY `queue` (`queue`),
+  KEY `uri` (`uri`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;"
+  end
+
+  down do
+    drop_table :namespaces
+  end
+end

data/lib/legion/data/migrations/007_add_nodes_table.rb

@@ -0,0 +1,22 @@
+Sequel.migration do
+  up do
+    run "CREATE TABLE `nodes` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
+  `active` tinyint(1) unsigned NOT NULL DEFAULT '1',
+  `name` varchar(128) NOT NULL DEFAULT '',
+  `datacenter_id` int(11) unsigned DEFAULT NULL,
+  `environment_id` int(11) unsigned DEFAULT NULL,
+  `status` varchar(255) NOT NULL DEFAULT 'unknown',
+  `created_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated_at` timestamp NULL DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP,
+  PRIMARY KEY (`id`),
+  UNIQUE KEY `name` (`name`),
+  KEY `active` (`active`),
+  KEY `status` (`status`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;"
+  end
+
+  down do
+    drop_table :nodes
+  end
+end

data/lib/legion/data/migrations/008_add_relationships_table.rb

@@ -0,0 +1,28 @@
+Sequel.migration do
+  up do
+    run "CREATE TABLE `relationships` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
+  `active` tinyint(1) unsigned NOT NULL DEFAULT '1',
+  `name` varchar(128) DEFAULT NULL,
+  `chain_id` int(11) unsigned DEFAULT NULL,
+  `trigger_id` int(11) unsigned NOT NULL,
+  `action_id` int(11) unsigned NOT NULL,
+  `delay` int(11) unsigned NOT NULL DEFAULT '0',
+  `conditions` text,
+  `transformation` text,
+  `version` tinyint(5) unsigned NOT NULL DEFAULT '1',
+  `created_timestamp` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated_timestamp` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
+  `group_owner` int(11) unsigned DEFAULT NULL,
+  `user_owner` int(11) unsigned DEFAULT NULL,
+  PRIMARY KEY (`id`),
+  KEY `active` (`active`),
+  KEY `name` (`name`),
+  KEY `deplay` (`delay`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;"
+  end
+
+  down do
+    drop_table :relationships
+  end
+end

data/lib/legion/data/migrations/009_add_task_logs_table.rb

@@ -0,0 +1,17 @@
+Sequel.migration do
+  up do
+    run "CREATE TABLE `task_logs` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
+  `task_id` int(11) unsigned NOT NULL,
+  `node_id` int(11) unsigned NOT NULL,
+  `log` text,
+  `created_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated_at` timestamp NULL DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP,
+  PRIMARY KEY (`id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;"
+  end
+
+  down do
+    drop_table :task_logs
+  end
+end

data/lib/legion/data/migrations/010_add_tasks_table.rb

@@ -0,0 +1,20 @@
+Sequel.migration do
+  up do
+    run "CREATE TABLE `tasks` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
+  `status` varchar(64) NOT NULL DEFAULT 'created',
+  `payload` text,
+  `parent_id` int(11) unsigned DEFAULT NULL,
+  `master_id` int(11) unsigned DEFAULT NULL,
+  `version` tinyint(5) NOT NULL DEFAULT '1',
+  `created_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated_at` timestamp NULL DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP,
+  PRIMARY KEY (`id`),
+  KEY `status` (`status`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;"
+  end
+
+  down do
+    drop_table :tasks
+  end
+end

data/lib/legion/data/migrations/011_add_users_and_groups.rb

@@ -0,0 +1,28 @@
+Sequel.migration do
+  up do
+    run "CREATE TABLE `users` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
+  `active` tinyint(1) unsigned DEFAULT '1',
+  `name` varchar(128) DEFAULT NULL,
+  `version` tinyint(5) unsigned NOT NULL DEFAULT '1',
+  `created_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
+  PRIMARY KEY (`id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;"
+
+    run "CREATE TABLE `groups` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
+  `active` tinyint(1) unsigned DEFAULT '1',
+  `name` varchar(128) DEFAULT NULL,
+  `version` tinyint(5) unsigned NOT NULL DEFAULT '1',
+  `created_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated_at` timestamp NULL DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP,
+  PRIMARY KEY (`id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;"
+  end
+
+  down do
+    drop_table :groups
+    drop_table :users
+  end
+end