adminix 0.1.49 → 0.2

data/lib/adminix/entities.rb

@@ -0,0 +1,5 @@
+require_relative 'entities/service'
+require_relative 'entities/variable'
+require_relative 'entities/log'
+require_relative 'entities/sysload_stamp'
+require_relative 'entities/job'

data/lib/adminix/entities/job.rb

@@ -0,0 +1,54 @@
+module Adminix
+  module Entities
+    class Job
+      attr_accessor :id, :script
+      attr_reader :output, :status, :success
+
+      STATUS_IN_QUEUE = 'in_queue'.freeze
+      STATUS_COMPLETED = 'completed'.freeze
+
+      def initialize(id:, script:)
+        @id = id
+        @script = script
+        @status = STATUS_IN_QUEUE
+        @created_at = Time.now
+        @success = false
+
+        filter_script
+      end
+
+      def perform
+        @output = `/bin/bash #{Adminix.config.scripts[:run_script]} #{script}`
+        @success = $?.success?
+        @status = STATUS_COMPLETED
+        @completed_at = Time.now
+      end
+
+      def in_queue?
+        @status == STATUS_IN_QUEUE
+      end
+
+      def completed?
+        @status == STATUS_COMPLETED
+      end
+
+      def filter_script
+        ['sudo', 'shutdown', 'kill'].each do |word|
+          @script = @script.gsub('word', '')
+        end
+      end
+
+      def to_api
+        {
+          id: @id,
+          status: @status,
+          script: @script,
+          success: @success,
+          output: @output,
+          created_at: @created_at,
+          completed_at: @completed_at
+        }
+      end
+    end
+  end
+end

data/lib/adminix/entities/log.rb

@@ -0,0 +1,21 @@
+module Adminix
+  module Entities
+    class Log
+      attr_reader :message, :tag, :created_at
+
+      def initialize(message, tag = nil)
+        @message = message
+        @tag = tag || 'default'
+        @created_at = Time.now
+      end
+
+      def to_api
+        {
+          message: message,
+          tag: tag,
+          created_at: created_at
+        }
+      end
+    end
+  end
+end

data/lib/adminix/entities/service.rb

@@ -0,0 +1,211 @@
+require 'securerandom'
+
+module Adminix
+  module Entities
+    class Service
+      attr_reader :id, :logs, :load_stamps, :jobs, :device_id, :performed_actions
+      attr_accessor :variables
+
+      def initialize(opts = {})
+        @id = opts[:id]
+        @device_id_path = "#{Helpers::Command.home}/.config/adminix/device_id"
+        @variables = []
+        @jobs = []
+        @logs = []
+        @load_stamps = []
+        @send_logs = false
+        @locked = false
+        @send_system_load = false
+        @performed_actions = {
+          watcher_started: true,
+          watcher_updated: false,
+          process_stopped: false,
+          process_started: false,
+          process_updated: false
+        }
+
+        define_device_id
+      end
+
+      def remove_completed_jobs
+        @jobs.each do |job|
+          @jobs -= [job] if job.completed?
+        end
+      end
+
+      def clean_sent_logs
+        return unless @send_logs
+        @send_logs = false
+        @logs = []
+      end
+
+      def clean_load_stamps
+        return unless @send_system_load
+        @send_system_load = false
+        @load_stamps = []
+      end
+
+      def add_new_jobs(list)
+        list.each do |data|
+          @jobs << Entities::Job.new(
+            id: data['id'],
+            script: data['script']
+          )
+        end
+      end
+
+      def add_logs(logs)
+        return if locked?
+        @logs += logs
+      end
+
+      def shift_logs(number)
+        @logs.shift(number)
+      end
+
+      def add_variable(new_var)
+        var = @variables.find { |v| v.key == new_var.key }
+        if var
+          index = @variables.index(var)
+          @variables[index] = new_var
+        else
+          @variables << new_var
+        end
+      end
+
+      def add_system_load(sys_load)
+        @load_stamps << sys_load
+      end
+
+      def shift_load_stamps(number)
+        @load_stamps.shift(number)
+      end
+
+      def allow_sending_logs
+        @send_logs = true
+        return if @logs.count.zero?
+        Adminix.logger.info("Sending #{@logs.count} new logs")
+      end
+
+      def allow_sending_system_load
+        @send_system_load = true
+        return if @load_stamps.count.zero?
+        Adminix.logger.info("Sending #{@load_stamps.count} new system load stamps")
+      end
+
+      def ensure_credentials_exists!
+        return unless id_defined?
+        Helpers::Output.display_error_and_exit('Credentials are not provided')
+      end
+
+      def define_device_id
+        if Helpers::Files.file_exists?(@device_id_path)
+          read_device_id
+        else
+          generate_device_id
+        end
+      end
+
+      def read_device_id
+        content = Helpers::Files.read_plain_file(@device_id_path)
+                                .split("\n").first.to_s
+        if content.empty?
+          Helpers::Output.display_error_and_exit("Device ID can't be empty")
+        else
+          @device_id = content
+        end
+      end
+
+      def generate_device_id(id = nil)
+        @device_id = id || SecureRandom.uuid
+        Helpers::Files.write_plain_file(@device_id_path, @device_id)
+      end
+
+      # TODO: real total memory and disk
+      def sync_payload
+        {
+          device_id: @device_id,
+          watcher_started: @performed_actions[:watcher_started],
+          watcher_updated: @performed_actions[:watcher_updated],
+          process_stopped: @performed_actions[:process_stopped],
+          process_started: @performed_actions[:process_started],
+          process_updated: @performed_actions[:process_updated],
+          completed_jobs: completed_jobs_to_api,
+          logs: logs_to_api,
+          system_load: {
+            stamps: load_stamps_to_api,
+            total: {
+              memory: 1024,
+              disk: 10_240
+            }
+          }
+        }
+      end
+
+      def new_jobs
+        @jobs.select(&:in_queue?)
+      end
+
+      def completed_jobs
+        @jobs.select(&:completed?)
+      end
+
+      def completed_jobs_to_api
+        completed_jobs.map(&:to_api)
+      end
+
+      def logs_to_api
+        @send_logs ? @logs.map(&:to_api) : []
+      end
+
+      def load_stamps_to_api
+        @send_system_load ? @load_stamps.map(&:to_api) : []
+      end
+
+      def reset_performed_actions
+        @performed_actions.keys.map { |key| @performed_actions[key] = false }
+      end
+
+      def define_performed_action(key, value)
+        @performed_actions[key] = value
+      end
+
+      def bash_variables(file = nil)
+        output = ''
+
+        if file
+          content = (variables.map { |v| "#{v.key}=#{v.value}" }).join("\n")
+          Helpers::Files.rm(file)
+          Helpers::Files.write_plain_file(file, content)
+          output = file
+        else
+          variables.each { |v| output << "#{v.to_bash_export}\n" }
+          output << "# Run this command to configure your shell:\n" \
+            '# eval $(adminix env)'
+        end
+
+        output
+      end
+
+      def id_defined?
+        !(id || '').empty?
+      end
+
+      def new_jobs?
+        new_jobs.any?
+      end
+
+      def locked?
+        @locked == true
+      end
+
+      def lock
+        @locked = true
+      end
+
+      def unlock
+        @locked = false
+      end
+    end
+  end
+end

data/lib/adminix/entities/sysload_stamp.rb

@@ -0,0 +1,37 @@
+module Adminix
+  module Entities
+    class SysloadStamp
+
+      attr_reader :memory_usage_mb, :cpu_load_percents, :disk_usage_mb,
+                  :temperature, :created_at
+
+      def initialize
+        create_stamp
+        @created_at = Time.now
+      end
+
+      def create_stamp
+        case Adminix.os
+        when 'linux'
+          @memory_usage_mb, = Helpers::Command.debian_memory_load
+          @cpu_load_percents = Helpers::Command.debian_cpu_load
+          @disk_usage_mb, = Helpers::Command.debian_disk_usage
+          @temperature = Helpers::Command.debian_temperature
+        when 'mac'
+          @cpu_load_percents = Helpers::Command.darwin_cpu_load
+          @disk_usage_mb, = Helpers::Command.darwin_disk_usage
+        end
+      end
+
+      def to_api
+        {
+          memory: memory_usage_mb,
+          cpu: cpu_load_percents,
+          disk: disk_usage_mb,
+          temperature: temperature,
+          created_at: created_at
+        }
+      end
+    end
+  end
+end

data/lib/adminix/entities/variable.rb

@@ -0,0 +1,32 @@
+module Adminix
+  module Entities
+    class Variable
+      attr_reader :key, :value
+
+      def self.all(service)
+        return [] unless Adminix.config.credentials_defined?
+
+        success, result = Helpers::NetHTTP.get("services/#{service.id}/options")
+        return [] unless success
+
+        variables = result.map do |o|
+          Variable.new(
+            key: o['key'],
+            value: o['value']
+          )
+        end
+
+        variables
+      end
+
+      def initialize(opts = {})
+        @key = opts[:key]
+        @value = opts[:value]
+      end
+
+      def to_bash_export
+        "export #{key}=\"#{value}\""
+      end
+    end
+  end
+end

data/lib/adminix/helpers.rb

@@ -1,2 +1,7 @@
-require_relative 'helpers/file'
-require_relative 'helpers/http'
+require_relative 'helpers/files'
+require_relative 'helpers/output'
+require_relative 'helpers/net_http'
+require_relative 'helpers/command'
+require_relative 'helpers/command'
+require_relative 'helpers/log_reader'
+require_relative 'helpers/systemctl'

data/lib/adminix/helpers/command.rb

@@ -0,0 +1,73 @@
+module Adminix
+  module Helpers
+    module Command
+      def self.which(arg1)
+        output = `which #{arg1}`
+        output = output.split("\n")[0]
+        output.empty? ? nil : output
+      end
+
+      def self.whoami
+        output = `whoami`
+        output.split("\n")[0]
+      end
+
+      def self.home
+        ENV['HOME']
+      end
+
+      def self.git_clone(repo, branch = 'master')
+        bin = 'git'
+        `#{bin} clone #{repo} -b #{branch}`
+      end
+
+      def mkdir_p(dirname)
+        `mkdir -p #{dirname}`
+      end
+
+      def self.run_system_command(command)
+        `#{command}`
+      end
+
+      def self.debian_memory_load
+        output = `free -m | awk 'NR==2{printf "%s/%s", $3,$2,$3*100/$2 }'`
+        output.split('/')
+      end
+
+      def self.systemctl_enabled?
+        !`which systemctl`.length.zero?
+      end
+
+      def self.darwin_memory_load
+        output = `free -m | awk 'NR==2{printf "%s/%s", $3,$2,$3*100/$2 }'`
+        output.split('/')
+      end
+
+      def self.debian_cpu_load
+        `top -bn1 | grep load | awk '{printf "%.2f\\\n", $(NF-2)}'`.to_f
+      end
+
+      def self.darwin_cpu_load
+        `ps -A -o %cpu | awk '{s+=$1} END {print s}'`.to_f
+      end
+
+      def self.debian_disk_usage
+        output = `df -h | awk '$NF=="/"{printf "%d/%d\\\n", $3,$2,$5}'`
+        output.split('/')
+      end
+
+      def self.debian_temperature
+        output = `cat /sys/class/thermal/thermal_zone*/temp`
+        temperatures = output.split("\n").map(&:to_i)
+        return nil if temperatures.empty?
+        temperatures.reduce(:+) / temperatures.size
+      end
+
+      def self.darwin_disk_usage
+        output = `df -h | awk '$NF=="/"{printf "%d/%d", $3,$2,$5}'`
+        used, disk_size = output.split('/')
+        [used.to_f * 1024, disk_size.to_f * 1024]
+      end
+    end
+  end
+end