Chaos Project

# loading settings
load './cfg.ini'

# loading classes
load './Data/Data.rb'
load './Data/Options.rb'
load './Data/Server.rb'

# loading external libraries
require './mysql'

#======================================================================
# module RMXOS
#======================================================================

module RMXOS

	#----------------------------------------------------------------------
	# Fixes strings for SQL queries and eval expressions.
	#  string - string to be converted
	# Returns: Converted string.
	#----------------------------------------------------------------------
	def self.fix_string(string)
		return string.gsub('\'') {'\\\''}
	end
  
  #----------------------------------------------------------------------
	# Backs up the data base on the back_up file
	#----------------------------------------------------------------------
  def self.back_up(server)
    sql = server.sql
    puts 'Accesing buddy_list'
    b = sql.query("SELECT * FROM buddy_list")
    buddy_list = []
    b.each_hash{|h|buddy_list.push(h)}
    puts 'Accesing guilds'
    g = sql.query("SELECT * FROM guilds")
    guilds = []
    g.each_hash{|h|guilds.push(h)}
    puts 'Accesing inbox'
    i = sql.query("SELECT * FROM inbox")
    inbox = []
    i.each_hash{|h|inbox.push(h)}
    puts 'Accesing ips'
    ip = sql.query("SELECT * FROM ips")
    ips = []
    ip.each_hash{|h|ips.push(h)}
    puts 'Accesing save_data'
    sd = sql.query("SELECT * FROM save_data")
    save_data = []
    sd.each_hash{|h|save_data.push(h)}
    puts 'Accesing user_data'
    ud = sql.query("SELECT * FROM user_data")
    user_data = []
    ud.each_hash{|h|user_data.push(h)}
    puts 'Accesing users'
    u = sql.query("SELECT * FROM users")
    users = []
    u.each_hash{|h|users.push(h)}
    puts 'Saving data'
    file = File.open('Back_Up.rxdata', 'wb')
    Marshal.dump(buddy_list, file)
    Marshal.dump(guilds, file)
    Marshal.dump(inbox, file)
    Marshal.dump(ips, file)
    Marshal.dump(save_data, file)
    Marshal.dump(user_data, file)
    Marshal.dump(users, file)
    file.close
  end
  
  #----------------------------------------------------------------------
	# Restores the data base from the back_up file
	#----------------------------------------------------------------------
  def self.restore(server)
    sql = server.sql
    puts 'Loading Data'
    file = File.open('Back_Up.rxdata', 'rb')
    puts 'Loading buddy_list'
    buddy_list = Marshal.load(file)
    puts 'Loading guilds'
    guilds = Marshal.load(file)
    puts 'Loading inbox'
    inbox = Marshal.load(file)
    puts 'Loading ips'
    ips = Marshal.load(file)
    puts 'Loading save_data'
    save_data = Marshal.load(file)
    puts 'Loading user_data'
    user_data = Marshal.load(file)
    puts 'Loading users'
    users = Marshal.load(file)
    file.close
    puts 'Updating data base'
    puts 'Updating buddy_list'
    buddy_list.each_index{|i|
    b = buddy_list[i]
    index = i+1
    b.each{|key,value|sql.query("UPDATE buddy_list SET '#{self.fix_string(key)}' = '#{self.fix_string(value)}' WHERE user_id = '#{index}'")}}
    puts 'Updating guilds'
    guilds.each_index{|i|
    index = i+1
    g = guilds[i]
    g.each{|key,value|sql.query("UPDATE guilds SET '#{self.fix_string(key)}' = '#{self.fix_string(value)}' WHERE user_id = '#{index}'")}}
    puts 'Updating inbox'
    inbox.each_index{|i|
    ib = inbox[i]
    index = i+1
    ib.each{|key,value|sql.query("UPDATE inbox SET '#{self.fix_string(key)}' = '#{self.fix_string(value)}' WHERE user_id = '#{index}'")}}
    puts 'Updating ips'
    ips.each_index{|i|
    ip = ips[i]
    index = i+1
    ip.each{|key,value|sql.query("UPDATE ips SET '#{self.fix_string(key)}' = '#{self.fix_string(value)}' WHERE user_id = '#{index}'")}}
    puts 'Updating save_data'
    save_data.each_index{|i|
    sd = save_data[i]
    index = i+1
    sd.each{|key,value|sql.query("UPDATE save_data SET '#{self.fix_string(key)}' = '#{self.fix_string(value)}' WHERE user_id = '#{index}'")}}
    puts 'Updating user_data'
    user_data.each_index{|i|
    ud = user_data[i]
    index = i+1
    ud.each{|key,value|sql.query("UPDATE user_data SET '#{self.fix_string(key)}' = '#{self.fix_string(value)}' WHERE user_id = '#{index}'")}}
    puts 'Updating users'
    users.each_index{|i|
    u = users[i]
    index = i+1
    u.each{|key,value|sql.query("UPDATE users SET '#{index}'")}}
  end
    
	
end

# create server instance
server = RMXOS::Server.new
# optimize the database
server.connect_to_database
puts 'Choose a command'
puts '1 Back up data base'
puts '2 Restore data base'
case gets.to_i
when 1 then RMXOS.back_up(server)
when 2 then RMXOS.restore(server)
else
  puts 'invalid command closing'
end
server.sql.close rescue nil

puts 'Press ENTER to continue.'
gets
puts 'Please wait...'