setup.rb

## Copyright 2012, Matias Pablo Brutti  All rights reserved.
##
## Esearchy is free software: you can redistribute it and/or modify it under 
## the terms of version 3 of the GNU Lesser General Public License as 
## published by the Free Software Foundation.
##
## Esearchy is distributed in the hope that it will be useful, but WITHOUT ANY
## WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS 
## FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for 
## more details.
##
## You should have received a copy of the GNU Lesser General Public License
## along with ESearchy.  If not, see <http://www.gnu.org/licenses/>.

unless Kernel.respond_to?(:require_relative)
  module Kernel
    def require_relative(path)
      require File.join(File.dirname(caller[0]), path.to_str)
    end
  end
end

case RUBY_PLATFORM
when /mingw|mswin/
  @slash = "\\"
else
  @slash = "/"
end

GEM_PATH="gem"
MONGO_VERSION="2.4.3"
require 'rubygems'
#require 'readline'
require 'getoptlong'

opts = GetoptLong.new(
  [ '--help', '-h', GetoptLong::NO_ARGUMENT ],
  [ '--gem-path', '-g', GetoptLong::REQUIRED_ARGUMENT ]
)

opts.each do |opt, arg|
  case opt
    when '--help' then
      puts "Esearchy Mirai Setup Application."
      puts "H  E  L  P:\n"
      puts ""
      puts "\t--help, -h"
      puts "\t\tWell I guess you know this one."
      puts ""
      puts "\t--gem-path, -g"
      puts "\t\pProvide gem path"
      puts ""
      puts "Copyright 2012 -- FreedomCoder"
      #END OF HELP
      exit(0)
    when '--gem-path' then
      GEM_PATH = arg
    else
      puts "[!] -  Unknown command. Please try again"
      exit(0)
  end
end

def gem_cmd
  if  RUBY_PLATFORM =~ /darwin|osx/
    return "sudo #{GEM_PATH}"
  else
    return "#{GEM_PATH}"
  end
end

# CHECK FOR GEMS HACK METHOD
def gem_available?(name)
  Gem::Specification.find_by_name(name)
rescue NoMethodError
  puts "[*] - esearchy requires #{name}."
  system "#{gem_cmd} update --system"
  puts "[*] - Please restart the script."
  exit(1)
rescue Gem::LoadError
  puts "[*] - esearchy requires #{name}."
  system "#{gem_cmd} install #{name}"
end

puts "-<[ ESearchy 0.3 CodeName Mirai ]>-"
puts "-<[ Setup and Configurations script ]>-"
puts ""
puts "[*] - Missing Gem installation"

if  RUBY_PLATFORM =~ /mingw|mswin/
  puts "[*] - If windows need to install win32console Gem."
  gem_available? 'win32console'
end

require_relative 'lib/esearchy/helpers/display.rb'



# MONGO METHODS
def mongo_download(os,platform)
  case os
  when /linux|osx|darwin/
    Display.msg "Downloading mongodb-#{os}-#{platform}-#{MONGO_VERSION}.tgz"
    system("curl http://fastdl.mongodb.org/#{os}/mongodb-#{os}-#{platform}-#{MONGO_VERSION}.tgz > external/mongodb.tgz")
  when /win32/
    Display.msg "Downloading mongodb-#{os}-#{platform}-#{MONGO_VERSION}.zip" 
    system("external/tools/wget.exe -O external\\mongodb.zip http://fastdl.mongodb.org/#{os}/mongodb-#{os}-#{platform}-#{MONGO_VERSION}.zip")
  end
end

def mongo_install(os)
  case os
  when /linux|darwin/
    Display.msg "Decompressing mongodb"
    system("tar xzf external/mongodb.tgz -C external")
    system("mv external/mongodb-* external/mongodb")
    Display.msg "Removing downloaded file"
    system("rm external/mongodb.tgz")
  when /mingw|mswin/
    Display.msg "Decompressing mongodb"
    system("external/tools/unzip.exe external\\mongodb*.zip -d external")
    system("move external\\mongodb-* external\\mongodb")
    Display.msg "Removing downloaded file"
    system("del external\\mongodb.zip")
  end
end

def mongo_install_service
  # runas /user:Administrator "external\mongodb\bin\mongod --install --dbpath=C:\Users\matt\.esearchy\data\db --logpath=C:\Users\matt\.esearchy\logs\mongodb.logs--logappend --serviceName=MONGODB"
  mongod = "external/mongodb/bin/mongod --install"
  dblogs = ENV['HOME'] + "/.esearchy/logs/mongodb.logs"
  dbpath = ENV['HOME'] + "/.esearchy/data/db"
  cmd = (mongod + " --dbpath=\"" + dbpath + "\" --logpath=\"" + dblogs + "\" --logappend --serviceName MONGODB").gsub("/",@slash)
  system("external\\tools\\Elevate.exe " + cmd )
  sleep(1)
rescue
  Display.error "Something went wrong installing MongoDB as a service."
end

def mongo_install?
  unless File.exists?((File.expand_path File.dirname(__FILE__) + "external").gsub("/",@slash))
    system("mkdir external")
    unless File.exists?((File.expand_path File.dirname(__FILE__) + "external/mongodb").gsub("/",@slash))
      case RUBY_PLATFORM
      when /linux/
        Display.msg "IMPORTANT: curl must be installed and on $PATH."
        RUBY_PLATFORM =~ /x86_64/ ? 
          mongo_download("linux", "x86_64") :
          mongo_download("linux", "i686")
        mongo_install("linux")
      when /darwin/
        RUBY_PLATFORM =~ /x86_64/ ?
          mongo_download("osx", "x86_64") :
          mongo_download("osx", "i386")
        mongo_install("darwin")
      when /mingw|mswin/
        Display.msg "IMPORTANT: curl must be installed and on $PATH."
        RUBY_PLATFORM =~ /x86_64/ ?
          mongo_download("win32", "x86_64") :
          mongo_download("win32", "i386")
        mongo_install("mswin")
      end
    end
  end
end

# CHECK FOR BASIC CONFIG STRUCTURE METHOD
# TODO: Might need to implement a few changes here for Windows.
def configure_esearchy
  #Check & create folders
  unless File.exists?((ENV["HOME"] + "/.esearchy").gsub("/",@slash))
    Display.msg "Running for the first time."
    Display.msg "Generating environment"
    system(("mkdir " + ENV["HOME"] + "/.esearchy").gsub("/",@slash))
  end
  unless File.exists?((ENV["HOME"] + "/.esearchy/config").gsub("/",@slash))
    File.open((ENV['HOME'] + "/.esearchy/config").gsub("/",@slash), "w" ) do |line|
      # A few defaults. Although this can all be overwritten at runtime. 
      line << "{"
      line << "\t\"maxhits\" : 1000,\n"
      line << "\t\"yahookey\" : \"AwgiZ8rV34Ejo9hDAsmE925sNwU0iwXoFxBSEky8wu1viJqXjwyPP7No9DYdCaUW28y0.i8pyTh4\",\n" 
      line << "\t\"bingkey\" : \"2KVYFdqxGbncMWuC2+Hd6KHnR181NvEaA6XqDc+npBI=\",\n"
      line << "\t\"dburl\" : \"localhost\",\n"
      line << "\t\"dbport\" : 27017,\n"
      line << "\t\"dbname\" : \"esearchy\"\n,"
      case RUBY_PLATFORM
      when /linux|darwin/
        line << "\t\"editor\" : \"vim\",\n"
        line << "\t\"antiword\" :\"/usr/bin/antiword\"\n"
      when /mingw|mswin/
        line << "\t\"editor\" : \"notepad.exe\",\n"
        line << "\t\"antiword\" : \"C:\\antiword\\antiword.exe\"\n"
      end
      line << "}"
  end
  end
  unless File.exists?((ENV["HOME"] + "/.esearchy/data").gsub("/",@slash))
    system(("mkdir " + ENV["HOME"] + "/.esearchy/data").gsub("/",@slash))
    unless File.exists?((ENV["HOME"] + "/.esearchy/data/db").gsub("/",@slash))
      system(("mkdir " + ENV["HOME"] + "/.esearchy/data/db").gsub("/",@slash))
    end
  end
  unless File.exists?((ENV["HOME"] + "/.esearchy/plugins").gsub("/",@slash))
    system(("mkdir " + ENV["HOME"] + "/.esearchy/plugins").gsub("/",@slash))
  end
  unless File.exists?((ENV["HOME"] + "/.esearchy/logs").gsub("/",@slash))
    system(("mkdir " + ENV["HOME"] + "/.esearchy/logs").gsub("/",@slash))
  end
end


#Current gem needs
Display.msg "Installing Bundler"
gem_available? "bundler"

Display.msg "Installing gems"
system "bundle install"

#Add any other dependencies installations, if you add any later on.
# gem_available? 'xxx'

Display.msg "Installing mongodb"
mongo_install?

Display.msg "Setup esearchy initial configuration"
configure_esearchy

if RUBY_PLATFORM =~ /mingw|mswin/
  Display.msg "Installing mongodb as service"
  mongo_install_service
end