fluke/devops-service/db/mongo/models/stack/stack_base.rb

require "db/mongo/models/model_with_provider"
require "providers/exceptions/name_conflict"

module Devops
  module Model
    class StackBase < MongoModel

      include ModelWithProvider

      attr_accessor :id, :name, :project, :deploy_env, :stack_template, :parameters, :events, :owner, :run_list, :stack_status

      set_field_validators :id, [::Validators::FieldValidator::NotNil,
                                      ::Validators::FieldValidator::FieldType::String,
                                      ::Validators::FieldValidator::NotEmpty]

      set_field_validators :provider, [::Validators::FieldValidator::NotNil,
                                      ::Validators::FieldValidator::FieldType::String,
                                      ::Validators::FieldValidator::NotEmpty]

      set_field_validators :project, [::Validators::FieldValidator::NotNil,
                                      ::Validators::FieldValidator::FieldType::String,
                                      ::Validators::FieldValidator::NotEmpty]

      set_field_validators :deploy_env, [::Validators::FieldValidator::NotNil,
                                      ::Validators::FieldValidator::FieldType::String,
                                      ::Validators::FieldValidator::NotEmpty]

      set_field_validators :stack_template, [::Validators::FieldValidator::NotNil,
                                      ::Validators::FieldValidator::FieldType::String,
                                      ::Validators::FieldValidator::NotEmpty]

      set_field_validators :name, [::Validators::FieldValidator::Nil,
                                      ::Validators::FieldValidator::FieldType::String,
                                      ::Validators::FieldValidator::NotEmpty]

      set_field_validators :owner, [::Validators::FieldValidator::NotNil,
                                      ::Validators::FieldValidator::FieldType::String,
                                      ::Validators::FieldValidator::NotEmpty]

      set_field_validators :run_list, [::Validators::FieldValidator::NotNil,
                                      ::Validators::FieldValidator::FieldType::Array,
                                      ::Validators::FieldValidator::RunList]

      set_field_validators :tags, [::Validators::FieldValidator::Nil,
                                      ::Validators::FieldValidator::FieldType::Hash]

      def initialize attrs={}
        self.set_provider(attrs)
        self.id = attrs['id']
        self.project = attrs['project']
        self.deploy_env = attrs['deploy_env']
        self.stack_template = attrs['stack_template']
        self.name = attrs['name']
        self.parameters = attrs['parameters']
        self.owner = attrs['owner']
        self.run_list = attrs['run_list'] || []
        self.tags = attrs['tags'] || {}
        self.stack_status = attrs['stack_status']
        self
      end

      # TODO: use string keys
      def to_hash_without_id
        {
          project: project,
          deploy_env: deploy_env,
          stack_template: stack_template,
          name: name,
          parameters: parameters,
          stack_status: stack_status,
          owner: owner,
          run_list: run_list,
          tags: tags
        }.merge(provider_hash)
      end


      def create_stack_in_cloud! out
        begin
          provider_instance.create_stack(self, out)
        rescue ProviderErrors::NameConflict
          raise InvalidRecord.new "Duplicate key error: stack with name '#{id}' already exists in cloud"
        end
      end

      def delete_stack_in_cloud!
        provider_instance.delete_stack(self)
      end

      def sync!
        self.stack_status = provider_instance.stack_details(self)[:stack_status]
        self.events = provider_instance.stack_events(self)
      rescue Fog::AWS::CloudFormation::NotFound
        self.stack_status = 'NOT_FOUND'
      end

      def resources
        provider_instance.stack_resources(self)
      end

      # resource_id is logical
      def resource(resource_id)
        provider_instance.stack_resource(self, resource_id)
      end

      def template_body
        stack_template_model.template_body
      end

      def stack_template_model
        Devops::Db.connector.stack_template(stack_template)
      end

      class << self

        # attrs should include:
        # - id (String)
        # - deploy_env (String)
        # - stack_template (String)
        # - provider (String)
        # - provider_account (String)
        def create(attrs, out)
          model = new(attrs)
          model.create_stack_in_cloud!(out)
          model.sync!
          model
        end

        def build_from_bson(attrs)
          attrs['id'] = attrs["_id"]
          self.new(attrs)
        end
      end

    end
  end
end