fluke/devops-service/db/mongo/models/server.rb

require "db/mongo/models/mongo_model"
require "hooks"
require "db/mongo/models/model_with_provider"


module Devops
  module Model
    class Server < MongoModel

      include ModelWithProvider
      include Hooks
      #params:
      #  out - container for output data
      #  deploy_info - hash with deploy data
      define_hook :before_deploy
      define_hook :after_deploy

      define_hook :before_create
      define_hook :after_create

      #params:
      #  out - container for output data
      define_hook :before_bootstrap
      define_hook :after_bootstrap

      attr_accessor :chef_node_name, :id, :remote_user, :project, :deploy_env, :private_ip, :public_ip, :created_at, :without_bootstrap, :created_by, :reserved_by, :stack, :run_list
      attr_accessor :options, :key

      types :id => {:type => String, :empty => false},
            :provider => {:type => String, :empty => false},
            :remote_user => {:type => String, :empty => false},
            :project => {:type => String, :empty => false},
            :deploy_env => {:type => String, :empty => false},
            :private_ip => {:type => String, :empty => false},
            :public_ip => {:type => String, :empty => true, :nil => true},
            :key => {:type => String, :empty => false},
            :created_by => {:type => String, :empty => false},
            :chef_node_name => {:type => String, :empty => true},
            :reserved_by => {:type => String, :empty => true},
            :stack => {:type => String, :nil => true},
            :run_list => {:type => Array, :value_type => String, :empty => true, :nil => false}

      set_validators  ::Validators::DeployEnv::RunList

      def self.fields
        ["chef_node_name", "project", "deploy_env", "provider", "remote_user", "private_ip", "public_ip", "created_at", "created_by", "key", "reserved_by"]
      end

      def initialize s={}
        self.provider = s["provider"]
        self.chef_node_name = s["chef_node_name"]
        self.id = s["_id"]
        self.remote_user = s["remote_user"]
        self.project = s["project"]
        self.deploy_env = s["deploy_env"]
        self.public_ip = s["public_ip"]
        self.private_ip = s["private_ip"]
        self.created_at = s["created_at"]
        self.created_by = s["created_by"]
        self.key = s["key"]
        self.reserved_by = s["reserved_by"]
        self.stack = s["stack"]
        self.run_list = p["run_list"] || []
      end

      def create
        res = {}
        res[:before] = self.run_hook :before_create
        Devops::Db.connector.server_insert self
        res[:after] = self.run_hook :after_create
        res
      end

      def validate!
        super
        true
      end

      def to_hash_without_id
        {
          "provider" => self.provider,
          "chef_node_name" => self.chef_node_name,
          "remote_user" => self.remote_user,
          "project" => self.project,
          "deploy_env" => self.deploy_env,
          "private_ip" => self.private_ip,
          "public_ip" => self.public_ip,
          "created_at" => self.created_at,
          "created_by" => self.created_by,
          "key" => self.key,
          "reserved_by" => self.reserved_by,
          "stack" => stack,
          "run_list" => self.run_list
        }.delete_if{|k,v| v.nil?}
      end

      def to_list_hash
        {
          "id" => self.id,
          "chef_node_name" => self.chef_node_name
        }
      end

      def self.build_from_bson s
        Server.new(s)
      end

      def info
        str = "Instance Name: #{self.chef_node_name}\n"
        str << "Instance ID: #{self.id}\n"
        str << "Private IP: #{self.private_ip}\n"
        str << "Public IP: #{self.public_ip}\n" unless self.public_ip.nil?
        str << "Remote user: #{self.remote_user}\n"
        str << "Project: #{self.project} - #{self.deploy_env}\n"
        str << "Created by: #{self.created_by}"
        str
      end

      def static?
        (self.provider == Provider::Static::PROVIDER) || false
      end

    end
  end
end