kaya 0.0.1

data/lib/kaya/support/files_cleanner.rb

@@ -0,0 +1,136 @@
+require 'fileutils'
+
+module Kaya
+  module Support
+    class FilesCleanner
+
+      # Delete all kaya_reports html files
+      def self.start!
+        begin
+          self.delete_kaya_reports_dir
+        rescue
+          false
+        end
+
+      end
+
+      def self.delete_file file_name
+        begin
+          File.delete("#{file_name}") and true
+        rescue
+          false
+        end
+      end
+
+      def self.delete_kaya_reports_dir
+        location = "#{Dir.pwd}/kaya/kaya_reports"
+        FileUtils.rm_rf(location)
+        Dir.mkdir(location)
+      end
+
+      def self.delete_report_which_has text
+        text = text.to_s if text.respond_to? :to_s
+        report = all_kaya_reports.select do |file|
+          file.include? text
+        end.first
+
+        delete_file(report)
+      end
+
+      def self.delete_html_report_for result_id
+        file = all_kaya_reports.select do |file|
+          file.include? result_id
+        end.first
+        delete_file(file) if file
+      end
+
+      def self.all_kaya_reports
+        Dir["#{Dir.pwd}/kaya/temp/*.*"].select do |file|
+          !file.scan(/kaya_report_\d+\.html/).empty?
+        end
+      end
+
+      # Deletes all kaya html reports
+      def self.delete_kaya_reports
+        # Get all html report files
+        (Kaya::Support::Git.reset_hard and Kaya::Support::Git.pull) if Kaya::Support::Configuration.use_git?
+        begin
+          self.delete_all_kaya_reports
+        rescue
+          false
+        end
+      end
+
+      # Deletes all kaya html reports files
+      # @return [Boolean] if has deleted reports
+      def self.delete_all_kaya_reports
+        not all_kaya_reports.each do |file|
+          self.delete_file(file)
+        end.empty?
+
+      end
+
+      # Deletes kaya execution output files
+      # @return [Boolean] for success
+      def self.delete_console_outputs_files
+        (Kaya::Support::Git.reset_hard and Kaya::Support::Git.pull) if Kaya::Support::Configuration.use_git?
+        begin
+          self.delete_all_console_output_files
+          true
+        rescue
+          false
+        end
+      end
+
+      # Deletes all kaya execution output files
+      # @return [Boolean] if has deleted files
+      def self.delete_all_console_output_files
+        not all_console_output_reports.each do |file|
+          delete_file(file)
+        end.empty?
+      end
+
+      def self.all_console_output_reports
+        Dir["#{Dir.pwd}/kaya/temp/*.*"].select do |file|
+          !file.scan(/kaya_co_\d+\.out/).empty?
+        end
+      end
+
+      def self.delete_console_output_for result_id
+        file=all_console_output_reports.select do |file|
+          file.include? result_id
+        end.first
+        delete_file(file) if file
+      end
+
+      def self.clear_sidekiq_log
+        sidekiq_file_path = "#{Dir.pwd}/kaya/sidekiq_log"
+        if File.exist? sidekiq_file_path
+          File.delete(sidekiq_file_path)
+          File.open(sidekiq_file_path, "a+"){}
+        end
+      end
+
+      def self.clear_kaya_log
+        kaya_log_file_path = "#{Dir.pwd}/kaya/kaya_log"
+        if File.exist? kaya_log_file_path
+          File.delete(kaya_log_file_path)
+          File.open(kaya_log_file_path, "a+"){}
+        end
+      end
+
+      # Deletes kaya folder. Used by 'bye command'
+      # @return [Boolean] for success
+      def self.delete_kaya_folder
+        begin
+          location = "#{Dir.pwd}/kaya"
+          FileUtils.rm_rf(location)
+          true
+        rescue
+          false
+        end
+      end
+
+    end
+  end
+end

data/lib/kaya/support/git.rb

@@ -0,0 +1,161 @@
+# require "git"
+
+module Kaya
+  module Support
+    class Git
+
+          def self.branch_list
+            self.remote_branches.map do |branch|
+              branch.gsub("*","").gsub(" ","").gsub("origin/","")
+            end.select do |branch|
+              not (branch.include? "HEAD" or branch.include? "/")
+            end
+          end
+
+          def self.remote_branches
+            Kaya::Support::Console.execute("git branch -r").split("\n")
+          end
+
+          def self.fetch
+            Kaya::Support::Console.execute("git fetch")
+          end
+
+          def self.branch
+            self.branches.select{|branch| branch.include? "*"}.first.gsub("*","").gsub(" ","")
+          end
+
+          def self.actual_branch; self.branch; end
+
+          def self.branches
+            Kaya::Support::Console.execute("git branch").split("\n")
+          end
+
+          def self.git_add_commit msg=nil
+            self.add_all
+            self.commit msg
+          end
+
+          def self.add_all
+            Kaya::Support::Console.execute("git add .")
+          end
+
+          def self.add_file filename
+            Kaya::Support::Console.execute("git add #{filename}")
+          end
+
+          def self.push
+            Kaya::Support::Console.execute("git push")
+          end
+
+          def self.git_push_origin_to branch_name=nil
+            branch_name = self.branch if branch_name.nil?
+            Kaya::Support::Console.execute("git push origin #{branch_name}")
+          end
+
+          def self.reset_hard
+            Kaya::Support::Console.execute("git reset --hard")
+          end
+
+          def self.git_push_origin_to_actual_branch
+            branch_name = self.branch
+            self.git_push_origin_to branch_name
+          end
+
+          def self.git_push_origin_to_actual_branch
+            git_push_origin_to(self.actual_branch)
+          end
+
+          def self.commit msg = nil
+            # self.ensure_being_at_kaya_branch
+            msg = "KAYA COMMIT #{Time.new.strftime('%d %m %Y %H:%M:%S')}" if msg.nil?
+            Kaya::Support::Console.execute"git commit -m '#{msg}'"
+          end
+
+          def self.create_branch_and_checkout branch
+            Kaya::Support::Console.execute("git checkout -b #{branch}")
+          end
+
+          def self.delete_branch branch
+            checkout_to "master"
+            Kaya::Support::Console.execute("git branch -D #{branch}")
+          end
+
+          # Performs pull from actual branc
+          def self.pull
+            self.pull_from(self.actual_branch)
+          end
+
+          def self.pull_from(branch_name=nil)
+            branch_name = self.branch if branch_name.nil?
+            Kaya::Support::Console.execute("git pull origin #{branch_name}")
+          end
+
+          def self.return_to_branch branch
+            self.checkout_to branch
+          end
+
+          def self.checkout_to branch
+            Kaya::Support::Console.execute("git checkout #{branch}") unless self.actual_branch == branch
+          end
+
+          def self.checkout_and_pull_from branch_name=nil
+            self.checkout_to(branch_name) if branch_name
+            branch_name = self.branch if branch_name.nil?
+            self.pull_from branch_name
+          end
+
+          # Returns las commit id.
+          # This method is used by Execution class to verify changes before each execution
+          # @param [String] the name of the project (folder project)
+          # @return [String] the id for the las commit
+          def self.get_last_commit_id
+            self.commits_ids.map do |commit|
+                commit.gsub("commit ","")
+            end.first
+          end
+
+          def self.commits_ids
+            self.log.split("\n").select do |line|
+                self.is_commit_id? line
+            end
+          end
+
+          def self.commits
+            self.log.split("commit")[1..-1]
+          end
+
+          def self.is_commit_id? line
+            line.start_with? "commit "
+          end
+
+          def self.log
+            Kaya::Support::Console.execute "git log"
+          end
+
+          def self.log_last_commit
+            "Commit: #{self.commits.first}"
+          end
+
+          def self.is_there_commit_id_diff? obtained_commit
+              obtained_commit != self.last_commit_id
+          end
+
+          def self.remote_url
+
+            res = Kaya::Support::Console.execute("git config --get remote.origin.url").gsub(":","/").gsub("git@", 'http://').chop
+            res[0..-5] if res.end_with? ".git"
+          end
+
+          # Returns an Array with the existing files on .gitignore
+          # @param [Array]
+          def self.get_actual_lines
+            f = File.open("#{Dir.pwd}/.gitignore","r")
+            files_list = []
+            f.each_line do |line|
+              files_list << line.gsub("\n","").gsub(" ","")
+            end
+            files_list
+          end
+    end
+  end
+end

data/lib/kaya/support/if_config.rb

@@ -0,0 +1,14 @@
+module Kaya
+  module Support
+    module IfConfig
+      def self.ip
+        begin
+          if_config_output = `ifconfig`
+          if_config_output.scan(/:(\d+\.\d+\.\d+\.\d+)/).first.first
+        rescue
+          nil
+        end
+      end
+    end
+  end
+end

data/lib/kaya/support/logo.rb

@@ -0,0 +1,23 @@
+# encoding: utf-8
+
+module Kaya
+  module Support
+    module Logo
+      def self.show
+        puts self.logo
+      end
+
+      def self.logo
+        "
+    ██╗  ██╗ █████╗ ██╗   ██╗ █████╗
+    ██║ ██╔╝██╔══██╗╚██╗ ██╔╝██╔══██╗
+    █████╔╝ ███████║ ╚████╔╝ ███████║
+    ██╔═██╗ ██╔══██║  ╚██╔╝  ██╔══██║
+    ██║  ██╗██║  ██║   ██║   ██║  ██║
+    ╚═╝  ╚═╝╚═╝  ╚═╝   ╚═╝   ╚═╝  ╚═╝
+             Version #{Kaya::VERSION}
+  "
+      end
+    end
+  end
+end

data/lib/kaya/support/logs.rb

@@ -0,0 +1,37 @@
+module Kaya
+  module Support
+    module Logs
+
+      def self.path
+        "#{Dir.pwd}/kaya/"
+      end
+
+      def self.all
+        logs = Hash.new
+        Dir.glob(path + '*_log') do |log_file|
+          name = log_file.split("/").last.gsub('.log','')
+          text = File.read(log_file)
+          logs[name] = text
+        end
+        logs
+      end
+
+      def self.read_log_file_content_for log=nil
+          path = "#{Dir.pwd}/kaya/#{log}"
+          if File.exist?("#{path}")
+            FileUtils.cp(path, "#{path}~")
+            all_content = IO.read("#{path}~")
+            content = if all_content.split("\n").size > 500
+               all_content.split("\n")[-500..-1].join("<br>")
+            else
+              all_content.gsub("\n","<br>")
+            end
+            File.delete("#{path}~")
+          else
+            content = ""
+          end
+          content
+      end
+    end
+  end
+end

data/lib/kaya/support/notification.rb

@@ -0,0 +1,127 @@
+require 'gmail'
+
+module Kaya
+  module Support
+    class Notification
+
+      include Gmail
+
+      def initialize project_name, base_url=nil
+        @project_name = project_name
+        @base_url = base_url
+        @subject_prefix = "[Kaya] [#{@project_name}] "
+        if Kaya::Support::Configuration.notification?
+          begin
+            @email = Gmail.connect!(Kaya::Support::Configuration.notification_username,Kaya::Support::Configuration.notification_password)
+            $K_LOG.debug "Notification: Login to Gmail Succesfully"
+            $K_LOG.debug "Notification: USING NOTIFICATION TO #{Kaya::Support::Configuration.recipients}"
+          rescue => e
+            $K_LOG.error "Notification: ERROR TO CONNECT TO GMAIL #{e}".red
+            $K_LOG.error "Notification: Connecting to GMail error => #{e}"
+            @email = NoEmail.new
+          end
+        else
+          @email = NoEmail.new
+        end
+
+      end
+
+      def notificate subject, body
+        send_email subject, body, Kaya::Support::Configuration.notifications_to
+      end
+
+      def send_email message_subject, message_body, path_to_file=nil
+        message_subject = "#{@subject_prefix} #{message_subject}"
+        begin
+          email = @email.compose do
+            to Kaya::Support::Configuration.recipients
+            subject message_subject
+            text_part do
+              body message_body
+            end
+            html_part do
+              $K_LOG.debug "Notification: Attaching report file (#{path_to_file})"
+              content_type 'text/html; charset=UTF-8'
+              body "<p>#{message_body}</p>"
+            end
+            add_file path_to_file
+          end
+          email.deliver!
+          $K_LOG.debug "Notification: Email sent to (#{Kaya::Support::Configuration.recipients}) | Subject: '#{message_subject}' | Message: #{message_body}" if $K_LOG
+        rescue => e
+          $K_LOG.error "Notification: Could not sent email to (#{Kaya::Support::Configuration.recipients}) | Subject: '#{message_subject}' | Message: #{message_body} | #{e}\n #{e.backtrace}" if $K_LOG
+        end
+
+        if path_to_file
+          begin
+            File.delete path_to_file
+            $K_LOG.debug "Notification: File #{path_to_file} deleted"
+          rescue
+            $K_LOG.error "Notification: Could not delete file #{path_to_file}"
+          end
+        end
+      end
+
+
+
+      def execution_finished result
+        body = "
+    Result Summary: #{result.summary}
+
+    Command: #{result.command}
+
+    Execution name: #{result.execution_name}
+
+    Started at: #{result.started_at_formatted}
+
+    Finished at: #{result.finished_at_formatted}
+
+    Elapsed Time: #{result.elapsed_time} seconds
+
+    Custom Params: #{result.custom_params}
+
+    See log at http://#{@base_url}/kaya/results/#{result.id}/log"
+
+        if Kaya::Support::Configuration.attach_report?
+          path_to_file = "#{Dir.pwd}/kaya/temp/#{result.id}.html"
+          $K_LOG.debug "Notification: Creating file report to attach to mail (#{path_to_file})"
+          File.open("#{path_to_file}","a+"){|f| f.write result.html_report}
+          $K_LOG.debug "Notification: File created (#{path_to_file})"
+        end
+        message_subject = "Execution Finished (#{result.id}) "
+        send_email message_subject, body, path_to_file
+      end
+
+
+
+
+
+      def execution_stopped result, additional_text=nil
+        body = "Execution stopped \n#{additional_text}"
+        message_subject = "Execution stopped (#{result.id})"
+        send_email message_subject, body
+      end
+
+    end
+
+
+
+
+
+
+    class NoEmail
+
+      def initialize
+
+      end
+
+      def method_missing meth, *args
+        begin
+        rescue
+          $K_LOG.error "Notification: #{meth} invoked but email is not configured"
+        end
+      end
+
+    end
+  end
+end

data/lib/kaya/support/processes.rb

@@ -0,0 +1,96 @@
+module Kaya
+  module Support
+    class Processes
+      def self.kill_by_result_id(result_id)
+
+        pid = pid_for(result_id)
+
+        return false if pid.nil?
+
+        childs = process_childs(pid)
+
+        unless pid.zero?
+          begin
+            childs.each{|ch_pid| kill_p(ch_pid)} unless childs.empty?
+            kill_p(pid)
+            true # Process killed
+          rescue Errno::EPERM # Denied permission to kill process
+            begin
+              kill_dash_nine(pid) and true # Process killed
+            rescue
+              false
+            end
+          rescue Errno::ESRCH # Process did not exist
+            false
+          end
+        else
+            puts "Cannot find process id"
+        end
+      end
+
+      def self.kill_p(pid)
+        pid = pid.to_i if pid.respond_to? :to_i
+        Process.kill('INT', pid)
+      end
+
+      def self.kill_dash_nine(pid)
+        pid = pid.to_i if pid.respond_to? :to_i
+        puts "kill -9 #{pid}" if $DEBUG
+        Kaya::Support::Console.execute "kill -9 #{pid}"
+      end
+
+      def self.pid_for(result_id)
+        result = Kaya::Results::Result.get(result_id)
+        result.pid.to_i
+      end
+
+      def self.process_running? pid
+        unless pid.nil?
+          res = Kaya::Support::Console.execute("ps -p #{pid}").split("\n").last
+          unless res.nil?
+            res.include? "#{pid}" and !res.include? "<defunct>"
+          end
+        end
+      end
+
+
+      #Recibe un PID y devuelve un array con los PID de sus hijos
+      def self.process_childs(pid)
+        Kaya::Support::Console.execute("ps -fea | grep #{pid} | grep -v grep | awk '$2!=#{pid} && $8!~/awk/ && $3==#{pid}{print $2}'").split("\n")
+      end
+
+      def self.kaya_pids
+        res = Kaya::Support::Console.execute "ps -fea | grep 'unicorn.rb -p #{Kaya::Support::Configuration.port}'"
+
+        res.split("\n").select{|lines| !lines.include? "grep"}.map{|line| line.split[1]}
+      end
+
+      def self.sidekiq_pids
+        res = Kaya::Support::Console.execute "ps -fea | grep 'sidekiq'"
+
+        res.split("\n").select{|lines| !lines.include? "grep"}.map{|line| line.split[1]}
+      end
+
+      def self.kill_all_these pids=nil
+        if pids
+          until pids.empty?
+            pid = pids.pop
+            if process_running? pid
+              begin
+                kill_dash_nine(pid)
+              rescue Errno::ESRCH
+                puts "Could not find process id #{pid} to kill"
+              end
+            end
+          end
+        end
+      end
+
+
+      def self.fork_this command
+        fork{exec("#{command}")}
+      end
+
+    end
+  end
+end

data/lib/kaya/support/query_string.rb

@@ -0,0 +1,22 @@
+module Kaya
+  module Support
+    class QueryString
+
+      attr_reader :values, :req
+
+      # req = Request
+      def initialize req
+        @req = req
+        @values = Rack::Utils.parse_nested_query(req.query_string.split("/").last)
+      end
+
+      def value_for key
+        @values[key]
+      end
+
+      def method_missing(key, *args, &block)
+        @values[key.to_s]
+      end
+    end
+  end
+end

data/lib/kaya/support/request.rb

@@ -0,0 +1,32 @@
+module Kaya
+  module Support
+    class Request
+
+      def initialize req
+        @req = req
+      end
+
+      def path
+        self.path_info
+      end
+
+      def path_info
+        @req.env["PATH_INFO"]
+      end
+
+      def ip
+        @req.env["REMOTE_ADDR"]
+      end
+
+      def method_missing(method_name, *args, &block)
+        begin
+          send("@req.#{method_name}", *args)
+        rescue
+          $K_LOG.error "#{method_name} not found for req in #{self.class}. Returning nil"
+          nil
+        end
+      end
+
+    end
+  end
+end

data/lib/kaya/support/risk.rb

@@ -0,0 +1,9 @@
+module Kaya
+  module Support
+    class Risk
+      def self.secure? value
+        value.scan(/&&|kill.*|rm -rf|rm -r \.\[\^\.\]\*|mkfs.*|.*\> \/dev\/sda|\:\(\)\{\:\|\:\&\}\;\:|wget .*|sh \.\/.*|\||grep|python \-c \'import os.*|mv .+|\> .+|\^foo\^bar|dd if=\/dev\/random of=\/dev\/sda|mkdir .*/).empty?
+      end
+    end
+  end
+end

data/lib/kaya/support/time_helper.rb

@@ -0,0 +1,11 @@
+module Kaya
+  module Support
+    class TimeHelper
+
+      def self.formatted_time_for timestamp
+        Time.at(timestamp).strftime(Kaya::Support::Configuration.formatted_datetime)
+      end
+
+    end
+  end
+end