six-updater 0.17.1-x86-mingw32

data/README

@@ -0,0 +1,3 @@
+== six-arma-updater
+
+You should document your project here.

data/Rakefile

@@ -0,0 +1,80 @@
+# 
+# To change this template, choose Tools | Templates
+# and open the template in the editor.
+ 
+
+require 'rubygems'
+require 'rake'
+require 'rake/clean'
+require 'rake/gempackagetask'
+require 'rake/rdoctask'
+require 'rake/testtask'
+
+spec = Gem::Specification.new do |s|
+  s.platform = "x86-mingw32"
+  s.name = 'six-updater'
+  s.version = '0.17.1'
+  s.has_rdoc = true
+  s.extra_rdoc_files = ['README', 'LICENSE']
+  s.summary = 'Your summary here'
+  s.description = s.summary
+  s.author = 'Sickboy'
+  s.email = 'sb@dev-heaven.net'
+  s.add_dependency('win32-api', '= 1.4.6')
+  s.add_dependency('win32-process', '>= 0.6.0')
+  s.add_dependency('win32-shortcut', '>= 0.2.3')
+  s.add_dependency('log4r', '>= 1.1.2')
+  s.add_dependency('six-uac', '>= 0.1.0')
+  s.add_dependency('six-rsync', '>= 0.3.4')
+  s.files = %w(LICENSE README Rakefile) + Dir.glob("{bin,spec}/**/*") + Dir.glob("lib/*/**/*.rb")
+  s.require_path = "lib"
+  s.bindir = "bin"
+  s.executables = ['six-updater']
+end
+
+spec2 = Gem::Specification.new do |s|
+  s.name = 'six-updater'
+  s.version = '0.17.1'
+  s.has_rdoc = true
+  s.extra_rdoc_files = ['README', 'LICENSE']
+  s.summary = 'Your summary here'
+  s.description = s.summary
+  s.author = 'Sickboy'
+  s.email = 'sb@dev-heaven.net'
+  #s.add_dependency('win32-api', '= 1.4.6')
+  #s.add_dependency('win32-process', '>= 0.6.0')
+  #s.add_dependency('win32-shortcut', '>= 0.2.3')
+  s.add_dependency('log4r', '>= 1.1.2')
+  #s.add_dependency('six-uac', '>= 0.1.0')
+  s.add_dependency('six-rsync', '>= 0.3.4')
+  s.files = %w(LICENSE README Rakefile) + Dir.glob("{bin,spec}/**/*") + Dir.glob("lib/*/**/*.rb")
+  s.require_path = "lib"
+  s.bindir = "bin"
+  s.executables = ['six-updater']
+end
+
+Rake::GemPackageTask.new(spec) do |p|
+  p.gem_spec = spec
+  p.need_tar = true
+  p.need_zip = true
+end
+
+Rake::GemPackageTask.new(spec2) do |p|
+  p.gem_spec = spec2           
+  p.need_tar = true
+  p.need_zip = true
+end
+
+
+Rake::RDocTask.new do |rdoc|
+  files =['README', 'LICENSE', 'lib/**/*.rb']
+  rdoc.rdoc_files.add(files)
+  rdoc.main = "README" # page to start on
+  rdoc.title = "six-arma-updater Docs"
+  rdoc.rdoc_dir = 'doc/rdoc' # rdoc output folder
+  rdoc.options << '--line-numbers'
+end
+
+Rake::TestTask.new do |t|
+  t.test_files = FileList['test/**/*.rb']
+end

data/bin/six-updater

@@ -0,0 +1,17 @@
+#!/usr/bin/ruby
+# encoding: utf-8
+
+begin
+  load 'six/updater-app.rb'
+rescue => e
+  puts e.class, e.message, e.backtrace.join("\n")
+  begin
+      require 'fileutils'
+      path = File.join(Six::Updater::DATA_PATH, 'logs')
+      FileUtils.mkdir_p path unless File.exists?(path)
+      File.open(File.join(path, "six-updater-error.log"), 'w') {|f| f.puts e.class, e.message, e.backtrace.join("\n")}
+  rescue
+      puts "ERROR: #{$!}"
+  end
+  sleep 10
+end

data/lib/six/updater-app.rb

@@ -0,0 +1,91 @@
+require 'six/updater/options'
+require 'six/updater'
+
+# TODO: Should exec outside the module
+module Six
+  module Updater
+    PID_FILE = File.join(DATA_PATH, "#{COMPONENT}.pid")
+    if not defined?(Ocra)
+      #Signal.trap("EXIT") { exit_app }
+      #Signal.trap("INT") { exit_app }
+      #Signal.trap("TERM") { exit_app }
+      #Signal.trap("KILL") { exit_app }
+
+      if File.exist?(PID_FILE)
+        pid = File.open(PID_FILE) {|file| file.read}
+        pid = pid.to_i
+        found = false
+        begin
+          out = %x[tasklist]
+          out.split("\n").each do |line|
+            line[/\Aruby\.exe[\t| ]*([0-9]*)/]
+            p = $1
+            if p
+              if p.size > 0
+                found = true if pid == p.to_i
+              end
+            end
+          end
+        rescue
+          log.warn "Failed checking tasklist for #{pid}"
+          log.debug "#{$!}"
+        end
+        if found
+          log.error "The program seems to be already running (PID: #{pid}). Please wait/close the other application first."
+          log.info "If you believe this is in error, please delete #{PID_FILE}"
+          log.info "If the file doesn't exist, you are probably NOT running this software 'As Administrator'"
+          sleep 15
+          Process.exit
+        else
+          FileUtils.rm_f PID_FILE
+        end
+      end
+      File.open(PID_FILE, 'w') {|f| f.puts Process.pid}
+
+      initialize
+      mods = []
+      puts
+      log.info "Processing:"
+      @config[:folders].each_with_index do |config, index|
+        #unless config[:skip] || config[:disabled]
+        # TODO: Only display message when there are actually todo's
+        puts
+        log.info "= #{config[:folder]}"
+        mod = Mod.new(config, @config)
+        mods[index] = mod
+        @todo.each { |t| mod.send t }
+        # end
+      end
+
+      puts
+      log.info "Postprocessing:"
+      @config[:folders].each_with_index do |config, index|
+        # unless config[:skip] || config[:disabled]
+        mod = mods[index]
+        if mod.changed || @config[:force]
+          # TODO: Only display message when there are actually todo's
+          puts
+          log.info "= #{config[:folder]}"
+          @second_todo.each { |t| mod.send t }
+        end
+        #  end
+      end
+
+      puts
+      log.info "Processing general tasks:"
+      @general_todo.each { |t| Six::Updater.send t }
+
+      puts
+      log.info "Ready!"
+      beta_msg
+
+      FileUtils.rm_f PID_FILE if File.exist?(PID_FILE)
+
+      if @config[:wait]
+        puts
+        log.info "Please press ENTER to exit"
+        STDIN.gets
+      end
+    end
+  end
+end

data/lib/six/updater.rb

@@ -0,0 +1,440 @@
+# encoding: UTF-8
+
+require 'yaml'
+require 'fileutils'
+
+gem 'log4r', '>= 1.1.2'
+require 'log4r'
+require 'digest/md5'
+
+case RUBY_PLATFORM
+  when /-mingw32$/, /-mswin32$/
+    gem 'win32-api', '= 1.4.6'
+    gem 'win32-shortcut'
+    require 'win32/shortcut'
+    gem 'win32-process'
+    require 'win32/process'
+    require 'win32/registry'
+end
+
+gem 'six-rsync'
+require 'six/rsync'
+
+require 'six/updater/mod'
+#require 'sixupdater/gitrepo'
+require 'six/updater/rsyncrepo'
+
+include Six::Repositories
+
+case RUBY_VERSION
+when /1\.8\.[0-9]/
+  class Array
+    def sample
+      self[rand self.size]
+    end
+  end
+end
+
+module Six
+  # TODO: Evaluate if this module should be a class instead?
+  module Updater
+    VERSION = '0.17.1'
+    COMPONENT = 'six-updater'
+    LOG_LIMIT = 9999
+                      
+    # Configuration
+    case RUBY_PLATFORM
+      when /-mingw32$/, /-mswin32$/
+        ARMA2 = ['SOFTWARE\\Bohemia Interactive Studio\\ArmA 2', 'MAIN']
+        ARMA2_ALT = ['SOFTWARE\\Bohemia Interactive\\ArmA 2', 'InstallPath']
+        ARMA2_STEAM = ['SOFTWARE\\Valve\\Steam\\Common\\ARMA 2', 'InstallPath']
+        DESKTOP = ['Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\User Shell Folders', 'Desktop']
+    end
+
+    # Fix for WinXP
+    BASE_PATH = if defined?(TAR2RUBYSCRIPT)
+      if oldlocation
+        oldlocation
+      else
+        Dir.pwd
+      end
+    else
+      p = Dir.pwd
+      p.gsub!('\\', '/')
+      p
+    end
+    TOOL_PATH = File.join(BASE_PATH, 'tools')
+
+    case RUBY_PLATFORM
+      when /-mingw32$/, /-mswin32$/
+        TEMP_PATH = if ENV['TEMP']
+          if ENV['TEMP'].size > 0
+            File.directory?(ENV['TEMP']) ? ENV['TEMP'] : BASE_PATH
+          else
+            BASE_PATH
+          end
+        else
+          BASE_PATH
+        end
+        HOME_PATH = File.exists?(File.join(ENV['APPDATA'])) ? File.join(ENV['APPDATA']) : TEMP_PATH
+        bpath = BASE_PATH.clone
+        bpath.gsub!('/', '\\')
+        tpath = "#{File.join(bpath, 'tools')};#{File.join(bpath, 'tools', 'bin')}"
+        tpath.gsub!('/', '\\')
+        ENV['PATH'] = "#{bpath};#{tpath};#{ENV['PATH']}"
+      else
+        HOME_PATH = '~/'
+        TEMP_PATH = '/tmp'
+    end
+    DATA_PATH = File.join(HOME_PATH, COMPONENT)
+
+=begin
+    dpath = BASE_PATH
+    unless File.exists?(File.join(BASE_PATH, "legacy.txt"))
+      reg_path = nil
+      #case RUBY_VERSION
+      #when /1\.8\.[0-9]/
+      #  begin
+      #    reg_path = Win32::Registry::HKEY_CURRENT_USER.open("SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders")['AppData']
+      #  rescue
+      #  end
+      #end
+      if reg_path
+        if reg_path.size > 0
+          dpath = File.join(reg_path, 'Six-Updater') if File.directory?(reg_path)
+        end
+      elsif ENV['APPDATA']
+        if ENV['APPDATA'].size > 0
+          dpath = File.join(ENV['APPDATA'], 'Six-Updater') if File.directory?(ENV['APPDATA'])
+        end
+      end
+    end
+    DATA_PATH = dpath
+=end
+
+    lf = File.join(DATA_PATH, "logs")
+    FileUtils.mkdir_p(lf) unless File.directory?(lf)
+
+    cf = File.join(DATA_PATH, "#{COMPONENT}.yml")
+    unless File.exists?(cf)
+      cfo = File.join(BASE_PATH, "#{COMPONENT}.yml")
+      FileUtils.cp(cfo, cf) if File.exists?(cfo)
+      begin
+        FileUtils.rm_f(cfo) if File.exists?(cfo)
+      rescue
+      end
+    end
+
+    @general_todo, @second_todo = [], []
+
+    # Create config hash with default settings
+    @config = Hash.new
+    @config[:app_params] = '-noSplash -noFilePatching -showScriptErrors'
+    @config[:app_exe] = 'arma2.exe'
+    @config[:depth] = 1
+    @config[:folders] = []
+    @config[:server] = Hash.new
+    @config[:verbose] = false
+
+    # FIXME: Solve this properly!
+    @config[:defaultactions] = [:install]
+    @config[:defaultgeneralactions] = [:createshortcut]
+    @config[:defaultsecondactions] = [:changelog, :cleanup]
+
+    # Create loggers
+    @@log = Log4r::Logger.new(COMPONENT)
+    format1 = if defined?(DEBUG)
+      Log4r::PatternFormatter.new(:pattern => "[%l] %d: %m", :date_pattern => '%H:%M:%S')
+    else
+      Log4r::PatternFormatter.new(:pattern => "%m")
+    end
+    format2 = Log4r::PatternFormatter.new(:pattern => "[%l] %c %d: %m", :date_pattern => '%H:%M:%S')
+
+
+    if not defined?(Ocra)
+      # Create Outputters
+      path = File.join(DATA_PATH, 'logs')
+      FileUtils::mkdir_p path
+      o_file = Log4r::FileOutputter.new "#{COMPONENT}-file",
+        'level' => 0, # All
+      :filename => File.join(path, "#{COMPONENT}.log"),
+        'formatter' =>  format2
+      #:maxsize => 1024
+
+      @@log.outputters << o_file
+
+      o_out = Log4r::StdoutOutputter.new "#{COMPONENT}-stdout",
+        'level' => 2, # no DEBUG
+      'formatter' =>  format1
+
+      o_err = Log4r::StderrOutputter.new "#{COMPONENT}-stderr",
+        'level' => 4, # Error and Up
+      'formatter' =>  format1
+
+      @@log.outputters << o_out << o_err
+    end
+
+    module_function
+    def initialize
+      # Banner
+      log.info "Updater (v#{VERSION}) by Sickboy <sb_at_dev-heaven.net>"
+      log.info 'Run with --help for help'
+      log.warn "WARNING: Please make sure anything ArmA related has been closed / shutdown, incl explorer windows, etc"
+      log.debug "BASE_PATH: #{BASE_PATH}"
+      log.debug "TOOL_PATH: #{TOOL_PATH}"
+      log.debug "@config #{@config}"
+
+      # Process the config file
+      config = "#{COMPONENT}.yml"
+      config = File.join(DATA_PATH, config)
+      if ARGV[0]
+        # FIXME: This is not a very good catch..
+        if ARGV[0][/.*\..*/]
+          if File.exists?(ARGV[0])
+            config = ARGV[0]
+          else
+            log.warn "#{ARGV[0]} not found, trying #{config}"
+          end
+        end
+      end
+
+      unless File.exists?(config)
+        log.error "ERROR: #{config} config missing!"
+        wait
+        exit
+      end
+
+      File.open(config) { |file| @config.merge!(YAML::load(file)) }
+      @config[:folders] = [] unless @config[:folders]
+      @config[:server][:folders] = [] unless @config[:server][:folders]
+
+      # Parses CLI Parameters, merges configuration and gets todo list
+      parse_options
+
+      # Backwards compatibility
+      @config[:app_path] = @config[:armapath] if @config[:armapath]
+      @config[:app_exe] = @config[:armaexe] if @config[:armaexe]
+      @config[:app_params] = @config[:armaparams] if @config[:armaparams]
+      #@config[:git_path] = @config[:gitpath] if @config[:gitpath]
+
+      # Determine arma distribution and location
+      case RUBY_PLATFORM
+        when /-mingw32$/, /-mswin32$/
+
+        unless @config[:app_path]
+          begin
+            @config[:app_path] = Win32::Registry.open(Win32::Registry::HKEY_LOCAL_MACHINE, ARMA2_STEAM[0])[ARMA2_STEAM[1]]
+            log.info "ArmA 2 Steam Distribution detected"
+          rescue
+            begin
+              @config[:app_path] = Win32::Registry.open(Win32::Registry::HKEY_LOCAL_MACHINE, ARMA2_ALT[0])[ARMA2_ALT[1]]
+              log.info "ArmA 2 Alt Distribution detected"
+            rescue
+              begin
+                @config[:app_path] = Win32::Registry.open(Win32::Registry::HKEY_LOCAL_MACHINE, ARMA2[0])[ARMA2[1]]
+                log.info "ArmA 2 Standard Distribution detected"
+              rescue
+                log.warn "No (known) ArmA 2 Distribution detected"
+              end
+            end
+          end
+        end
+      end
+
+      # Verify installation folder
+      if @config[:app_path]
+        log.info "ArmA Path: #{@config[:app_path]}"
+        if File.exists?(@config[:app_path])
+          @config[:app_path].gsub!('\\', '/')
+        else
+          log.error "No valid ArmA installation folder found, please abort!"
+          wait
+          exit
+        end
+      else
+        log.error "No valid ArmA installation folder found, please abort!"
+        wait
+        exit
+      end
+
+      if @config[:app_modpath]
+        log.info "Installation Path: #{@config[:app_modpath]}"
+        if File.exists?(@config[:app_modpath])
+          @config[:app_modpath].gsub!('\\', '/')
+        else
+          log.error "No valid Mod installation folder found, please abort!"
+          wait
+          exit
+        end
+      else
+        @config[:app_modpath] = @config[:app_path]
+      end
+
+      @config[:folders].uniq!
+      @config[:folders].sort! do |a, b|
+        a_prio = a[:priority] ? a[:priority] : 9999
+        b_prio = b[:priority] ? b[:priority] : 9999
+        a_prio <=> b_prio
+      end
+      log.debug "@config #{@config}"
+
+      # PreProcess the config data
+      @mods = []
+      @mods << @config[:mods] if @config[:mods]
+
+      @beta = false
+      @beta_exe = false
+      @beta_exe = true if @config[:app_exe][/beta[\/|\\]arma2.*/]
+
+      @config[:folders].each do |folder|
+        @beta = true if folder[:folder] == 'beta'
+        unless folder[:disabled]
+          folder[:folder].gsub!('\\', '/')
+          if folder[:mods]
+            @mods << folder[:mods]
+          else
+            @mods << folder[:folder]
+          end
+        end
+      end
+      @mods = @mods.join(';')
+
+      # Output processed config data
+      log.info "Manager for: #{@mods}"
+      beta_msg
+
+      # Prepare stats
+      @stats = []
+      @config[:folders].each do |folder|
+        stat = Hash.new
+        @stats << stat
+        path = File.join(@config[:app_modpath], folder[:folder])
+        stat[:changelog] = []
+      end
+    end
+
+    def beta_msg
+      if @beta_exe && !@beta
+        log.warn "WARNING: You do not seem to be running the 'beta' modfolder, but have selected 'beta/arma2.exe'"
+      elsif !@beta_exe && @beta
+        log.warn "WARNING: You seem to be running the 'beta' modfolder, but have not selected 'beta/arma2.exe'"
+      end
+    end
+
+    def log
+      @@log
+    end
+
+    def exit_app
+      FileUtils.rm_f PID_FILE
+      exit
+    end
+
+    def wait
+      STDIN.gets
+    end
+
+    def sure?
+      log.info "Are you really sure? Please press Enter to continue, or close the application to abort."
+      wait
+    end
+
+    def exit
+      FileUtils.rm_f PID_FILE if File.exist?(PID_FILE)
+      Process.exit
+    end
+
+    # General Tasks
+    def joingame
+      cmd = ""
+      cmd << " -connect=#{@config[:server][:address]}" if @config[:server][:address]
+      cmd << " -port=#{@config[:server][:port]}" if @config[:server][:port]
+      cmd << " -password=#{@config[:server][:password]}" if @config[:server][:password]
+      startgame(cmd)
+    end
+
+    def startgame(cmd = nil)
+      log.info "Starting the game with #{@config[:app_params]} -mod=#{@mods}#{cmd}"
+      #system "\"#{@config[:app_exe]}\" #{@config[:app_params]} -mod=#{@mods}"
+      case RUBY_PLATFORM
+        when /-mingw32$/, /-mswin32$/
+          begin
+            struct = Process.create(
+              :app_name         => File.join(@config[:app_path], @config[:app_exe]),
+              :command_line     => " #{@config[:app_params]} -mod=#{@mods}#{cmd}",
+              :creation_flags   => Process::DETACHED_PROCESS,
+              :process_inherit  => false,
+              :thread_inherit   => false,
+              :cwd              => @config[:app_path],
+              :inherit          => false
+              #:environment      => ""
+            )
+          rescue
+            log.warn "WARNING: Something went wrong starting the app: #{@config[:app_exe]}"
+            log.debug "#{$!}"
+          end
+        else
+          Dir.chdir(@config[:app_path]) do
+            system "#{@config[:app_exe]} #{@config[:app_params]} -mod=#{@mods}#{cmd}"
+          end
+      end
+    end
+
+    def shortcut(path)
+      case RUBY_PLATFORM
+        when /-mingw32$/, /-mswin32$/
+          cwd = Dir.pwd
+          Dir.chdir path
+          mods = @mods.clone
+          mods.gsub!(/\//, '-')
+          mods2 = @mods.clone
+          mods2.gsub!('/', '\\')
+          name = "Play ArmA2 with #{mods}.lnk"
+          Win32::Shortcut.new(name) do |shortcut|
+            shortcut.description = "Start Arma2 with #{mods2}"
+            shortcut.path = File.join(@config[:app_path], @config[:app_exe])
+            shortcut.working_directory = @config[:app_path]
+            shortcut.arguments = "#{@config[:app_params]} -mod=#{mods2}"
+          end
+          str = path.clone
+          str.gsub!('/', '\\')
+          log.info "Created shortcut (in #{str}): #{name}"
+          Dir.chdir cwd
+        else
+        log.warn "Unsupported on current Platform: #{RUBY_PLATFORM}"
+      end
+    end
+
+    def createshortcut
+      shortcut @config[:app_path]
+    end
+
+    def createdesktopshortcut
+      case RUBY_PLATFORM
+        when /-mingw32$/, /-mswin32$/
+          begin
+            desktop = Win32::Registry.open(Win32::Registry::HKEY_CURRENT_USER, DESKTOP[0])[DESKTOP[1]]
+            while desktop =~ /\%(\w*)\%/ do
+              desktop.sub!("%#{$1}%", ENV[$1])
+            end
+            shortcut desktop if File.directory? desktop
+          rescue
+            log.warn "WARNING: Problem while creating shortcut! #{$!}"
+          end
+        else
+        log.warn "Unsupported on current Platform: #{RUBY_PLATFORM}"
+      end
+    end
+
+    def uninstall
+      log.warn "Uninstall feature temporary disabled. Please delete the modfolder manually for uninstall"
+=begin
+        @config[:folders].each do |config|
+          log.info "= #{config[:folder]}"
+          mod = Mod.new(config, @config)
+          mod.uninstall
+        end
+=end
+    end
+  end
+end