legion-data 0.1.1 → 0.2.0

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

@@ -0,0 +1,26 @@
+Sequel.migration do
+  up do
+    run "CREATE TABLE `nodes` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
+  `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',
+  `active` tinyint(1) unsigned NOT NULL DEFAULT '1',
+  `created` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated` timestamp NULL DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP,
+  PRIMARY KEY (`id`),
+  UNIQUE KEY `name` (`name`),
+  KEY `active` (`active`),
+  KEY `status` (`status`),
+  KEY `node_datacenter_id` (`datacenter_id`),
+  KEY `node_environment_id` (`environment_id`),
+  CONSTRAINT `node_datacenter_id` FOREIGN KEY (`datacenter_id`) REFERENCES `datacenters` (`id`) ON DELETE SET NULL ON UPDATE CASCADE,
+  CONSTRAINT `node_environment_id` FOREIGN KEY (`environment_id`) REFERENCES `environments` (`id`) ON DELETE SET NULL ON UPDATE CASCADE
+) ENGINE=InnoDB AUTO_INCREMENT=6 DEFAULT CHARSET=utf8;"
+  end
+
+  down do
+    drop :nodes
+  end
+end

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

@@ -0,0 +1,18 @@
+Sequel.migration do
+  up do
+    run "CREATE TABLE `settings` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
+  `key` varchar(128) NOT NULL,
+  `value` varchar(256) NOT NULL,
+  `encrypted` tinyint(1) unsigned NOT NULL DEFAULT '0',
+  `created` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated` timestamp NULL DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP,
+  PRIMARY KEY (`id`),
+  UNIQUE KEY `key` (`key`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;"
+  end
+
+  down do
+    drop :settings
+  end
+end

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

@@ -0,0 +1,25 @@
+Sequel.migration do
+  up do
+    run "CREATE TABLE `extensions` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
+  `active` tinyint(1) unsigned NOT NULL DEFAULT '1',
+  `name` varchar(128) NOT NULL,
+  `namespace` varchar(128) NOT NULL DEFAULT '',
+  `exchange` varchar(255) DEFAULT NULL,
+  `uri` varchar(256) DEFAULT NULL,
+  `user_owner` int(11) unsigned DEFAULT NULL,
+  `group_owner` int(11) unsigned DEFAULT NULL,
+  `updated` timestamp NULL DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP,
+  `created` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  PRIMARY KEY (`id`),
+  UNIQUE KEY `name_namespace` (`name`,`namespace`),
+  KEY `active` (`active`),
+  KEY `name` (`name`),
+  KEY `namespace` (`namespace`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;"
+  end
+
+  down do
+    drop :extensions
+  end
+end

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

@@ -0,0 +1,21 @@
+Sequel.migration do
+  up do
+    run "CREATE TABLE `runners` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
+  `extension_id` int(11) unsigned NOT NULL,
+  `name` varchar(256) NOT NULL DEFAULT '',
+  `namespace` varchar(256) NOT NULL DEFAULT '',
+  `active` tinyint(1) unsigned NOT NULL DEFAULT '1',
+  `queue` varchar(256) DEFAULT NULL,
+  `uri` varchar(256) DEFAULT NULL,
+  `created` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated` datetime DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP,
+  PRIMARY KEY (`id`),
+  CONSTRAINT `runner_extension_id` FOREIGN KEY (`extension_id`) REFERENCES `extensions` (`id`) ON DELETE CASCADE ON UPDATE CASCADE
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;"
+  end
+
+  down do
+    drop :runners
+  end
+end

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

@@ -0,0 +1,29 @@
+Sequel.migration do
+  up do
+    run "CREATE TABLE `functions` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
+  `name` varchar(128) NOT NULL,
+  `active` tinyint(1) unsigned NOT NULL DEFAULT '1',
+  `runner_id` int(11) unsigned NOT NULL,
+  `args` text,
+  `user_owner` int(11) unsigned DEFAULT NULL,
+  `group_owner` int(11) unsigned DEFAULT NULL,
+  `created` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated` timestamp NULL DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP,
+  PRIMARY KEY (`id`),
+  UNIQUE KEY `runner_id` (`runner_id`,`name`),
+  KEY `active` (`active`),
+  KEY `namespace` (`runner_id`),
+  KEY `name` (`name`),
+  KEY `functions_user_owner` (`user_owner`),
+  KEY `functions_group_owner` (`group_owner`),
+  CONSTRAINT `function_runner_id` FOREIGN KEY (`runner_id`) REFERENCES `runners` (`id`) ON DELETE CASCADE ON UPDATE CASCADE,
+  CONSTRAINT `functions_group_owner` FOREIGN KEY (`group_owner`) REFERENCES `groups` (`id`) ON DELETE SET NULL ON UPDATE CASCADE,
+  CONSTRAINT `functions_user_owner` FOREIGN KEY (`user_owner`) REFERENCES `users` (`id`) ON DELETE SET NULL ON UPDATE CASCADE
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;"
+  end
+
+  down do
+    drop :functions
+  end
+end

data/lib/legion/data/migrations/012_add_tasks.rb

@@ -0,0 +1,25 @@
+Sequel.migration do
+  up do
+    run "CREATE TABLE `tasks` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
+  `relationship_id` int(11) unsigned NOT NULL,
+  `status` varchar(255) NOT NULL,
+  `parent_id` int(11) unsigned DEFAULT NULL,
+  `master_id` int(11) unsigned DEFAULT NULL,
+  `args` text,
+  `results` text,
+  `created` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated` datetime DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP,
+  PRIMARY KEY (`id`),
+  KEY `status` (`status`),
+  KEY `parent_id` (`parent_id`),
+  KEY `master_id` (`master_id`),
+  CONSTRAINT `parent_id` FOREIGN KEY (`parent_id`) REFERENCES `tasks` (`id`) ON DELETE SET NULL ON UPDATE CASCADE,
+  CONSTRAINT `master_id` FOREIGN KEY (`master_id`) REFERENCES `tasks` (`id`) ON DELETE SET NULL ON UPDATE CASCADE
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;"
+  end
+
+  down do
+    drop :tasks
+  end
+end

data/lib/legion/data/migrations/013_add_task_logs.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,
+  `entry` text NOT NULL,
+  `created` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated` datetime DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP,
+  PRIMARY KEY (`id`),
+  CONSTRAINT `task_log_task_id` FOREIGN KEY (`task_id`) REFERENCES `tasks` (`id`) ON DELETE CASCADE ON UPDATE CASCADE
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;"
+  end
+
+  down do
+    drop :task_logs
+  end
+end

data/lib/legion/data/migrations/014_add_relationships.rb

@@ -0,0 +1,27 @@
+Sequel.migration do
+  up do
+    run "CREATE TABLE `relationships` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
+  `name` varchar(255) DEFAULT NULL,
+  `active` tinyint(1) unsigned NOT NULL DEFAULT '1',
+  `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',
+  `allow_new_chains` tinyint(1) unsigned NOT NULL DEFAULT '1',
+  `conditions` text,
+  `transformation` text,
+  `created` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated` datetime DEFAULT NULL ON UPDATE CURRENT_TIMESTAMP,
+  PRIMARY KEY (`id`),
+  KEY `chain_id` (`chain_id`),
+  CONSTRAINT `function_chain_id` FOREIGN KEY (`chain_id`) REFERENCES `chains` (`id`) ON DELETE CASCADE ON UPDATE CASCADE,
+  CONSTRAINT `relationship_action_id` FOREIGN KEY (`action_id`) REFERENCES `functions` (`id`) ON DELETE CASCADE ON UPDATE CASCADE,
+  CONSTRAINT `relationship_trigger_id` FOREIGN KEY (`trigger_id`) REFERENCES `functions` (`id`) ON DELETE CASCADE ON UPDATE CASCADE
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;"
+  end
+
+  down do
+    drop :task_logs
+  end
+end

data/lib/legion/data/model.rb

@@ -1,34 +1,32 @@
 module Legion
   module Data
-    class Models
-      attr_reader :loaded_models
-      def initialize(options = {})
-        options.merge!(default_options) { |_key, v1, _v2| v1 }
-        @loaded_models = []
-        require_sequel_models if options[:auto_load]
-      end
+    module Models
+      class << self
+        attr_reader :loaded_models
+        def models
+          %w[user group extension chain relationship function task runner task_log datacenter environment node setting]
+        end
 
-      def default_options
-        { continue_on_model_fail: false,
-          auto_load:              true }
-      end
+        def load
+          Legion::Logging.info 'Loading Legion::Data::Models'
+          @loaded_models ||= []
+          require_sequel_models(models)
+        end
 
-      def models
-        %w[chain relationship function task namespace task_log datacenter environment node]
-      end
-
-      def require_sequel_models(files = models)
-        files.each { |file| load_sequel_model(file) }
-      end
+        def require_sequel_models(files = models)
+          Dir[File.dirname(__FILE__) + 'models/*.rb'].each { |file| puts file }
+          files.each { |file| load_sequel_model(file) }
+        end
 
-      def load_sequel_model(model)
-        Legion::Logging.debug("Trying to load #{model}.rb")
-        require_relative "models/#{model}"
-        @loaded_models << model
-        Legion::Logging.debug("Successfully loaded #{model}")
-      rescue LoadError => e
-        Legion::Logging.fatal("Failed to load #{model}")
-        raise e unless Legion::Settings[:data][:models][:continue_on_fail]
+        def load_sequel_model(model)
+          Legion::Logging.debug("Trying to load #{model}.rb")
+          require_relative "models/#{model}"
+          @loaded_models << model
+          Legion::Logging.debug("Successfully loaded #{model}")
+        rescue LoadError => e
+          Legion::Logging.fatal("Failed to load #{model}")
+          raise e unless Legion::Settings[:data][:models][:continue_on_fail]
+        end
       end
     end
   end

data/lib/legion/data/models/chain.rb

@@ -3,9 +3,10 @@
 module Legion
   module Data
     module Model
-      # Used for accessing legion.chains
       class Chain < Sequel::Model
         one_to_many :relationship
+        many_to_one :user_owner, class: Legion::Data::Model::User
+        many_to_one :group_owner, class: Legion::Data::Model::Group
       end
     end
   end

data/lib/legion/data/models/datacenter.rb

@@ -3,9 +3,10 @@
 module Legion
   module Data
     module Model
-      # Used for accessing legion.datacenters
       class Datacenter < Sequel::Model
         one_to_many :node
+        many_to_one :user_owner, class: Legion::Data::Model::User
+        many_to_one :group_owner, class: Legion::Data::Model::Group
       end
     end
   end

data/lib/legion/data/models/environment.rb

@@ -3,9 +3,10 @@
 module Legion
   module Data
     module Model
-      # Used for accessing legion.environments
       class Environment < Sequel::Model
         one_to_many :node
+        many_to_one :user_owner, class: Legion::Data::Model::User
+        many_to_one :group_owner, class: Legion::Data::Model::Group
       end
     end
   end

data/lib/legion/data/models/extension.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module Legion
+  module Data
+    module Model
+      class Extension < Sequel::Model
+        one_to_many :runners
+        many_to_one :user_owner, class: Legion::Data::Model::User
+        many_to_one :group_owner, class: Legion::Data::Model::Group
+      end
+    end
+  end
+end

data/lib/legion/data/models/function.rb

@@ -1,13 +1,16 @@
 # frozen_string_literal: true
 
+require_relative 'relationship'
+
 module Legion
   module Data
     module Model
-      # Used for accessing legion.functions
       class Function < Sequel::Model
-        many_to_one :namespace
-        one_to_many :trigger_functions, class: :Relationship, key: :trigger_id
-        one_to_many :action_functions, class: :Relationship, key: :action_id
+        many_to_one :runner
+        one_to_many :trigger_relationships, class: 'Legion::Data::Model::Relationship', key: :trigger_id
+        one_to_many :action_relationships, class: 'Legion::Data::Model::Relationship', key: :action_id
+        many_to_one :user_owner, class: Legion::Data::Model::User
+        many_to_one :group_owner, class: Legion::Data::Model::Group
       end
     end
   end

data/lib/legion/data/models/group.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+module Legion
+  module Data
+    module Model
+      class Group < Sequel::Model
+      end
+    end
+  end
+end

data/lib/legion/data/models/namespace.rb

@@ -3,9 +3,11 @@
 module Legion
   module Data
     module Model
-      # Used for accessing legion.namespaces
       class Namespace < Sequel::Model
         one_to_many :function
+        many_to_one :extension
+        many_to_one :user_owner, class: Legion::Data::Model::User
+        many_to_one :group_owner, class: Legion::Data::Model::Group
       end
     end
   end

data/lib/legion/data/models/node.rb

@@ -3,10 +3,10 @@
 module Legion
   module Data
     module Model
-      # Used for accessing legion.nodes
       class Node < Sequel::Model
         many_to_one :environment
         many_to_one :datacenter
+        one_to_many :task_log
       end
     end
   end

data/lib/legion/data/models/relationship.rb

@@ -1,15 +1,17 @@
 # frozen_string_literal: true
 
 require_relative 'function'
+
 module Legion
   module Data
     module Model
-      # Used for access legion.relationships
       class Relationship < Sequel::Model
         many_to_one :chain
         one_to_many :task
         many_to_one :trigger, class: Legion::Data::Model::Function
         many_to_one :action, class: Legion::Data::Model::Function
+        many_to_one :user_owner, class: Legion::Data::Model::User
+        many_to_one :group_owner, class: Legion::Data::Model::Group
       end
     end
   end

data/lib/legion/data/models/runner.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require_relative 'function'
+module Legion
+  module Data
+    module Model
+      class Runner < Sequel::Model
+        many_to_one :chain
+        one_to_many :task
+        one_to_many :functions
+        many_to_one :extension
+        many_to_one :user_owner, class: Legion::Data::Model::User
+        many_to_one :group_owner, class: Legion::Data::Model::Group
+      end
+    end
+  end
+end

data/lib/legion/data/models/setting.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+module Legion
+  module Data
+    module Model
+      class Setting < Sequel::Model
+      end
+    end
+  end
+end

data/lib/legion/data/models/task.rb

@@ -3,7 +3,6 @@
 module Legion
   module Data
     module Model
-      # Used for accessing legion.tasks
       class Task < Sequel::Model
         many_to_one :relationship
         one_to_many :task_log
@@ -11,6 +10,8 @@ module Legion
         one_to_many :children, key: :parent_id, class: self
         many_to_one :master, class: self
         one_to_many :slave, key: :master_id, class: self
+        many_to_one :user_owner, class: Legion::Data::Model::User
+        many_to_one :group_owner, class: Legion::Data::Model::Group
       end
     end
   end