shards 0.1.0 → 1.0.2

data/lib/shards/meta/meta.rb

@@ -0,0 +1,57 @@
+require 'shards/meta/list'
+
+module Shards
+
+  module Meta
+
+    attr_reader :config
+
+    def get_config class_name
+      yaml_file[class_name]
+    end
+
+    def run var
+      eval( config[var.to_s] )
+    end
+
+    def yaml_file
+      ENV['FORMAT_METHODS_FILE'].nil? ? default_yaml : default_yaml.merge(custom_yaml)
+    end
+
+    def default_yaml
+      YAML.load_file File.join(Shards::LIB_PATH,'settings','format_methods.yaml')
+    end
+
+    def custom_yaml
+      YAML.load_file ENV['FORMAT_METHODS_FILE']
+    end
+
+    def add text
+      @output << text
+    end
+
+    def output
+      @output.join("\n")
+    end
+
+    def print_output
+      puts output unless test_env?
+      reset_output
+    end
+
+    def reset_output
+      @output=[' ']
+    end
+
+    def test_env?
+      ENV['TEST'] || false
+    end
+
+    def method_list class_name
+      get_config(class_name).keys
+    end
+
+  end
+
+end
+

data/lib/shards/repo.rb

@@ -0,0 +1,45 @@
+require "git"
+
+module Shards
+
+  class Repo
+
+    @@times = 0
+    attr_accessor :git, :client, :root
+
+    def initialize
+      @root=ENV['ENGINEERING_ROOT_PATH']
+      @git=Git.open root
+      clean if @@times<1
+    end
+
+    def live_branch
+      ENV['LIVE_BRANCH'] || 'v5'
+    end
+
+    def clean
+      @@times+=1
+      git.reset_hard
+      git.clean(force: true, d: true)
+      git.branch(live_branch).checkout
+      git.pull('origin',live_branch)
+    end
+
+    def autocommit
+      git.add all: true
+      git.commit message
+      git.push 'origin', live_branch
+    end
+
+    def message
+      "[AUTOCOMMIT] Added shard for #{client}"
+    end
+
+    def diff
+      git.add all: true
+      git.diff
+    end
+
+  end
+end
+

data/lib/shards/shard.rb

@@ -0,0 +1,64 @@
+require 'shards/base_yaml_object'
+
+module Shards
+
+  class Shard < BaseYamlObject
+
+    method_list('Shards::Shard').each do |m|
+      define_method(m.to_sym) { run __method__ }
+    end
+
+    attr_accessor :domain, :db, :dns
+
+    def shard_list_key
+      ENV['SHARD_LIST_KEY_ROOT_PATH'].split.map do |k|
+        "['#{k}']"
+      end.join
+    end
+
+    def rest
+      @rest||=clean_basename
+    end
+
+    def clean_basename
+      shard_rest=list.clone
+      shard_rest.delete(basename)
+      shard_rest
+    end
+
+    def add_domain domain
+      @domain=domain
+      @dns=Shards::Dns.new stage: stage, domain: domain
+      @db=Shards::Db.new stage: stage, dns: dns
+    end
+
+    def connection
+      { name => db.connection }
+    end
+
+    def add_connection
+      rest.merge! connection
+      @rest=@rest.sort.to_h
+    end
+
+    def merged_connection
+      base.merge rest
+    end
+
+    def update_yaml
+      eval "@yaml#{shard_list_key.to_s}=merged_connection"
+    end
+
+    def write_domain
+      add_connection
+      update_yaml
+      write_yaml
+    end
+
+    def exist?
+      list.keys.include? name
+    end
+
+  end
+end
+

data/lib/shards/site.rb

@@ -0,0 +1,31 @@
+require 'shards/base_yaml_object'
+
+module Shards
+
+  class Site < BaseYamlObject
+
+    attr_accessor :client
+
+    method_list('Shards::Site').each do |m|
+      define_method(m.to_sym) { run __method__ }
+    end
+
+    def add_site dryrun: false
+
+      add_data
+
+      if dryrun
+        add_site_message
+      else
+        write_yaml
+      end
+
+    end
+
+    def exist?
+      !yaml.select { |x| x['domain']==domain }.empty?
+    end
+
+  end
+end
+

data/lib/shards/stage.rb

@@ -0,0 +1,64 @@
+require 'shards/shard'
+require 'shards/site'
+require 'shards/db'
+require 'shards/dns'
+require 'shards/base'
+
+module Shards
+
+  class Stage < Shards::Base
+
+    include Shards::Meta
+
+    method_list('Shards::Stage').each do |m|
+      define_method(m.to_sym) { run __method__ }
+    end
+
+    attr_accessor :name, :location_name, :shard
+
+    def initialize name, data, location_name
+
+      set_config
+      @name=name
+      @location_name=location_name
+      @variables = data
+      @output=[]
+
+    end
+
+    def shard
+      @shard||=Shards::Shard.new self
+      @shard
+    end
+
+    def add_domain domain, dryrun: false
+
+      shard.add_domain domain
+
+      if dryrun
+        add_domain_message
+      else
+        shard.write_domain
+      end
+
+    end
+
+    def get var_name
+      @variables.has_key?(var_name) ? @variables[var_name] : send(var_name)
+    end
+
+    def config_name
+      config_replace[name] || name
+    end
+
+    def config_replace
+      ENV['CONFIG_REPLACE'] ? eval(ENV['CONFIG_REPLACE']) : {}
+    end
+
+    def vars
+      @variables.keys
+    end
+
+  end
+end
+

data/lib/shards/version.rb

@@ -1,3 +1,3 @@
 module Shards
-  VERSION = "0.1.0"
+  VERSION = "1.0.2"
 end

data/lib/shards/workflow/base.rb

@@ -0,0 +1,177 @@
+require 'shards/config'
+
+module Shards
+
+  module Workflow
+
+    class Base
+
+      attr_accessor :yaml, :config, :flow, :pointer, :step, :answer, :pointer_history
+
+      def initialize config
+        @config=config
+        config.repo.clean
+        @yaml= YAML.load_file file
+        @pointer_history = []
+      end
+
+      def file
+        ENV['WORKFLOW_YAML_FILE'] || default_file
+      end
+
+      def default_file
+        File.join Shards.root_path, %w(lib settings workflow.yaml)
+      end
+
+      def start wf=nil
+        wf||=self.class.name.split('::').last.downcase
+        @flow=yaml[wf]
+        @pointer=flow.keys.first
+        run_step
+      end
+
+      def run_step
+
+        @step=flow[pointer]
+        @pointer_history << pointer
+        @answer=nil
+
+        begin
+          send pointer
+          @pointer=step['correct']
+        rescue => e
+          puts e.message
+          @pointer=step['wrong']
+        end
+
+        @step=flow[pointer]
+
+        run_step if @step['correct']!='finish'
+
+      end
+
+      def ask_question
+        puts step['question']
+        @answer=$stdin.gets.chomp
+      end
+
+      def format_options title, array
+        puts title
+        puts
+        array.each_index.map do |k|
+          puts "\t#{k}) #{array[k]}"
+        end
+      end
+
+      def ask_for_exit
+        exit
+      end
+
+      def choose_one title, array
+        format_options(title, array)
+        ask_question
+        validate_in_index answer, array
+      end
+
+      def validate_min_size text, min
+        condition= text.size < min
+        wrong_text= "'%s' is shorter than '%d' characters" % [text, min]
+        validation condition, wrong_text
+      end
+
+      def validate_regex text, regex
+        condition= text !~ /#{regex}/
+        wrong_text = "'%s' has a wrong format" % text
+        validation condition, wrong_text
+      end
+
+      def validate_in_index text, array
+        condition= array.at(text.to_i).nil?
+        wrong_text = text
+        validation condition, wrong_text
+      end
+
+      def validation condition, wrong_text
+        raise_wrong_text wrong_text if condition
+      end
+
+      def raise_wrong_text wrong_text
+        raise @step['wrong_message'] % wrong_text
+      end
+
+      def previous_pointer
+        @pointer_history.at(-2)
+      end
+
+      def ask_for_repeat
+        ask_question
+        if answer.strip.downcase!='r'
+          config.repo.clean
+          raise step['wrong_message']
+        end
+        @pointer=previous_pointer
+        run_step
+
+      end
+
+      def check_shard_presence_in_shards
+        stage.add_domain domain, dryrun: true
+        raise_wrong_text stage.shard.name if stage.shard.exist?
+      end
+
+      def check_database_presence_in_server
+        raise_wrong_text [stage.shard.db.name,stage.shard.db.server] if stage.shard.db.exist?
+      end
+
+      def check_domain_dns_presence
+        raise_wrong_text stage.shard.dns.host if stage.shard.dns.exist?
+      end
+
+      def check_domain_presence_in_sites
+        @site = Shards::Site.new stage
+        site.client = client
+        raise_wrong_text site.domain if site.exist?
+      end
+
+      def write_yaml_files
+        stage.add_domain domain
+        site.add_site
+        puts config.repo.diff
+        ask_question
+        raise step['wrong_message'] if answer.strip.downcase!='y'
+      end
+
+      def commit_changes
+        puts config.repo.autocommit
+      end
+
+      def create_database
+        stage.shard.db.create
+        message_params= [stage.shard.db.name, stage.shard.db.server]
+        raise_wrong_text message_params unless stage.shard.db.exist?
+        puts step['message'] % message_params
+      end
+
+      def clean_repo_and_exit
+        puts config.repo.clean
+      end
+
+      def dns_upsert
+        stage.shard.dns.r53upsert
+        raise_wrong_text host unless stage.shard.dns.exist?
+        puts step['message'] % host
+      end
+
+      def host
+        stage.shard.dns.host
+      end
+
+      def ask_for_exit_and_restart
+        ask_question
+        raise step['wrong_message'] if answer.strip.downcase!='r'
+      end
+
+    end
+
+  end
+end