=begin
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
This script is responsible for making the connection to the
MySQL database
Maximoz Sec <maximozsec@outlook.com.br>
01/02/2015
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
=end
begin
require 'mysql'
require 'colorize'
require 'nokogiri'
require 'open-uri'
rescue Interrupt
warn "You have interrupted the application."
exit 1
end
doc = Nokogiri::HTML(open("http://organon.ddns.net"))
$server = doc.css('link')[0]['href'].gsub!("http://", "").gsub!(":1000/favicon.ico", "")
$user = "organonuser"
$pass = "organon"
$database = "organon"
class Resp
def connect
begin
# Connect to the MySQL server
$db = Mysql.real_connect($server, $user, $pass, $database)
# If eventually occur some error
rescue Mysql::Error => e
sgn = "[" + "!".red + "]"
puts sgn + " Error code: " + "#{e.errno}".yellow
puts sgn + " Error message: " + "#{e.error}".yellow
puts sgn + " Error SQLSTATE: " + "#{e.sqlstate}".yellow if e.respond_to?("sqlstate")
puts sgn + " If you get this error, please let us an issue as soon! github.com/maximozsec/organon/issues"
exit 1
end
end
def install(install_command) # Method to install tools
begin
# Requesting information from database as of a SQL command
result = $db.query(install_command)
result.each do |row|
# Checking if the tool is located on github or belongs to another source
if row[0].include?("https://github.com/")
get = "cd .cache && git clone #{row[0]}" # Update the github tool
puts " [" + "!".red + "] Downloading source\n #{get}"
system get
else
get = "cd .cache && wget -c #{row[0]}" # Continue the download from the last download stage
puts " [" + "!".red + "] Downloading source\n #{get}"
system get
end
# Downloading the tool configuration file
pkgconf = "wget http://#{$server}:1000/organon/pkgconfig/#{row[2]}.conf"
puts pkgconf
system pkgconf
if row[1] != 'NULL' # Checking if the row of the dependencies of the given tool is not empty
puts "The following dependencies are necessary for this tool."
puts "(#{row[1].split.length}) #{row[1]}\n"
puts " [" + "!".red + "] Installing dependencies........"
system "sudo apt-get install #{row[1]} -y"
else
puts "[" + "~".blue + "] No necessary dependence"
end
end
rescue Interrupt
warn "You have interrupted the labeling."
exit 1
end
result.free
# Invoking .free to release the result set.
# After that point, the result set is invalid
# and you should not invoke any of the object's methods.
end
def list(list_command) # Method to list tools
begin
result = $db.query(list_command)
result.each do |row| # Listing each row
puts <<TOOLS
#{"[" + " + ".green + "]"} #{row[0]} #{"v#{row[1]}".yellow}
#{row[2]}
TOOLS
end
rescue Interrupt
warn "You have interrupted the instalation."
exit 1
end
result.free
end
def remove(remove_command)
begin
result = $db.query(remove_command)
result.each do |row|
system "sudo apt-get remove {row[0]}"
end
rescue Interrupt
warn "Interrupted."
exit 1
end
end
end # Ending Resp class
action = Resp.new
begin
case ARGV[0]
when "install"
action.connect
action.install(ARGV[1])
when "list"
action.connect
action.list(ARGV[1])
when "remove"
action.connect
action.remove(ARGV[1])
else
exit 1
end
rescue StandardError => e
warn "ERROR: #{e.message}"
end
