six-updater 0.10.1

data/LICENSE

@@ -0,0 +1,3 @@
+== six-arma-updater
+
+Put appropriate LICENSE for your project here.

data/README

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

data/Rakefile

@@ -0,0 +1,49 @@
+# 
+# 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.name = 'six-updater'
+  s.version = '0.10.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('log4r', '>= 1.1.0')
+  s.add_dependency('six-rsync', '>= 0.1.0')
+  s.add_dependency('win32-process', '>= 0.6.0')
+  s.add_dependency('win32-shortcut', '>= 0.2.3')
+  # s.executables = ['your_executable_here']
+  s.files = %w(LICENSE README Rakefile) + Dir.glob("{bin,spec}/**/*") + Dir.glob("lib/*/**/*.rb")
+  s.require_path = "lib"
+  s.bindir = "bin"
+end
+
+Rake::GemPackageTask.new(spec) do |p|
+  p.gem_spec = spec
+  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/lib/six/updater-app.rb

@@ -0,0 +1,51 @@
+require 'six/updater/options'
+require 'six/updater'
+
+# TODO: Should exec outside the module
+module Six
+  module Updater
+    if not defined?(Ocra)
+      initialize
+      mods = []
+      puts
+      log.info "Processing:"
+      @config[:folders].each do |config|
+        # TODO: Only display message when there are actually todo's
+        puts
+        log.info "= #{config[:folder]}"
+        mod = Mod.new(config, @config)
+        mods << mod
+        @todo.each { |t| mod.send t }
+        #mod.install
+      end
+
+      puts
+      log.info "Postprocessing:"
+      @config[:folders].each_with_index do |config,index|
+        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
+
+      puts
+      log.info "Processing general tasks:"
+      @general_todo.each { |t| Six::Updater.send t }
+
+      puts
+      log.info "Ready!"
+
+      if @config[:wait]
+        puts
+        log.info "Please press ENTER to exit"
+        STDIN.gets
+      end
+    else
+      gem 'git'
+      require 'git'
+    end
+  end
+end

data/lib/six/updater.rb

@@ -0,0 +1,459 @@
+# encoding: UTF-8
+
+require 'yaml'
+require 'fileutils'
+require 'win32/registry'
+
+gem 'win32-shortcut'
+require 'win32/shortcut'
+gem 'log4r'
+require 'log4r'
+gem 'six-rsync'
+require 'six/rsync'
+gem 'win32-process'
+require 'win32/process'
+
+require 'digest/md5'
+require 'six/updater/mod'
+#require 'sixupdater/gitrepo'
+require 'six/updater/rsyncrepo'
+
+include Six::Repositories
+
+# Language fix for alternative localizations
+if RUBY_VERSION == "1.9.1"
+  if defined?(Ocra)
+    (Dir[File.join($:[-2], 'enc/*.so')] + Dir[File.join($:[-2], 'enc/trans/*.so')]).each do |file|
+      require file
+    end
+  end
+  #EXCLUDE = [12]
+  #(1..16).each { |i| require "enc/iso_8859_#{i}" unless EXCLUDE.include? i }
+  #require 'enc/windows_1251'
+end
+
+if RUBY_VERSION == "1.8.7"
+  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.10.1'
+    COMPONENT = 'six-updater'
+
+    # Configuration
+    ARMA2 = ['SOFTWARE\\Bohemia Interactive Studio\\ArmA 2', 'MAIN']
+    ARMA2_ALT = ['SOFTWARE\\Bohemia Interactive\\ArmA 2', 'InstallPath']
+    ARMA2_STEAM = ['SOFTWARE\\SOFTWARE\\Valve\\Steam\\', 'InstallPath']
+    # Fix for WinXP
+    BASE_PATH = if ENV['OCRA_EXECUTABLE']
+      if ENV['OCRA_EXECUTABLE'][/(.*)[\/|\\].*/]
+        p = $1
+        p.gsub!('\\', '/')
+        p
+      else
+        p = Dir.pwd
+        p.gsub!('\\', '/')
+        p
+      end
+    elsif defined?(TAR2RUBYSCRIPT)
+      if oldlocation
+        oldlocation
+      else
+        Dir.pwd
+      end
+    else
+      p = Dir.pwd
+      p.gsub!('\\', '/')
+      p
+    end
+    TOOL_PATH = File.join(BASE_PATH, 'tools')
+    LOG_LIMIT = 9999
+    # TODO: Allow downloader yml instead?
+    # TODO: Auto update always etc? :D
+
+    @general_todo, @second_todo = [], []
+
+    bpath = BASE_PATH.clone
+    bpath.gsub!('/', '\\')
+    tpath = "#{File.join(bpath, 'tools')};#{File.join(bpath, 'tools', 'bin')}"
+    tpath.gsub!('/', '\\')
+    ENV['PATH'] = "#{ENV['PATH']};#{bpath};#{tpath}"
+
+    # Create config hash with default settings
+    @config = Hash.new
+    @config[:app_params] = '-noSplash -noFilePatching'
+    @config[:app_exe] = 'arma2.exe'
+    @config[:depth] = 1
+    @config[:folders] = []
+    @config[:server] = Hash.new
+    @config[:server_path] = File.join(BASE_PATH, 'servers')
+    @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(BASE_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 log
+      @@log
+    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 zipcheck
+      begin
+        %x[7z]
+      rescue
+        log.warn "7-zip not found, attempting download!"
+        system "six-downloader.exe \"#{File.join(BASE_PATH, 'six-downloader_7z.yml').gsub('/', '\\')}\" --output \"#{TOOL_PATH.gsub('/', '\\')}\""
+        file = File.join(TOOL_PATH, "7z.exe")
+        unless FileTest.exist?(file)
+          log.error "ERROR: #{file} not found, please download from http://7-zip.org/download.html '7-Zip Command Line Version'"
+          wait
+          Process.exit
+        end
+      end
+    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]
+      log.info "Starting the game with #{@config[:app_params]} -mod=#{@mods}#{cmd}"
+      cwd = Dir.pwd
+      Dir.chdir @config[:app_path]
+      system "\"#{@config[:app_exe]}\" #{@config[:app_params]} -mod=#{@mods}#{cmd}"
+      Dir.chdir cwd
+    end
+
+    def startgame
+      log.info "Starting the game with #{@config[:app_params]} -mod=#{@mods}"
+      #system "\"#{@config[:app_exe]}\" #{@config[:app_params]} -mod=#{@mods}"
+      begin
+        struct = Process.create(
+          :app_name         => File.join(@config[:app_path], @config[:app_exe]),
+          :command_line     => "#{@config[:app_params]} -mod=#{@mods}",
+          :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]}"
+      end
+    end
+
+    def shortcut(path)
+      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
+    end
+
+    def createshortcut
+      shortcut @config[:app_path]
+    end
+
+    def createdesktopshortcut
+      shortcut "#{ENV['USERPROFILE']}\\Desktop"
+    end
+
+    def uninstall
+      if @config[:server]
+        if @config[:server][:name]
+          log.info "#{@config[:server][:name]} server"
+          path = File.join(@config[:server_path], @config[:server][:name])
+          if FileTest.exist?(path)
+            log.warn "###########"
+            log.warn "# WARNING #"
+            log.warn "###########"
+            log.warn "You are about to uninstall the specified repository"
+            sure?
+            log.warn 'Uninstalling...'
+            FileUtils.rm_rf path
+            #FileUtils.rm_f @modconfig[:changelog] if FileTest.exists?(@modconfig[:changelog])
+            #FileUtils.rm_f @modconfig[:versionfile] if FileTest.exists?(@modconfig[:versionfile])
+          end
+        end
+      end
+=begin
+        @config[:folders].each do |config|
+          log.info "= #{config[:folder]}"
+          mod = Mod.new(config, @config)
+          mod.uninstall
+        end
+=end
+    end
+
+    def setup_git
+      unless @setup_git
+        @setup_git = true
+        # Verify git existence / find git
+        unless @config[:git_path]
+          @config[:git_path] = File.join(TOOL_PATH, 'Git', 'Cmd')
+          unless FileTest.exist?(@config[:git_path])
+            begin
+              %x[git]
+              @config[:git_path] = 'PATH'
+            rescue
+              zipcheck
+              log.warn "Git not configured, nor found in path... Downloading!"
+              # download from http://alturl.com/hn84 into name; git.7z
+              pwd = Dir.pwd
+              Dir.chdir TOOL_PATH
+              file = "portablegit.7z"
+              FileUtils.rm_f file if FileTest.exist?(file)
+              system "six-downloader.exe \"#{File.join(BASE_PATH, 'six-downloader_portablegit.yml').gsub('/', '\\')}\" --output \"#{TOOL_PATH.gsub('/', '\\')}\""
+              dir = File.join(TOOL_PATH, 'Git')
+              FileUtils.rm_rf dir if FileTest.exist?(dir)
+              FileUtils.mkdir_p dir
+              log.info "Unpacking Git..."
+              %x[7z x -oGit \"#{file}\"]
+              FileUtils.rm_f file if FileTest.exist?(file)
+              Dir.chdir pwd
+            end
+          end
+        end
+        unless @config[:git_path] == 'PATH'
+          path = @config[:git_path].clone
+          path.gsub!('/','\\')
+          ENV['PATH'] = "#{ENV['PATH']};#{path}" if path.size > 0 #unless ENV['PATH'][/git/] # ??
+        end
+
+        log.info "Using Git at: #{@config[:git_path]}"
+        puts
+
+        # FIXME: This is naughty but due to portable git config setting, and this module's git version check, we must run this after initialize :/
+        gem 'git'
+        require 'git'
+      end
+    end
+
+    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(BASE_PATH, config)
+      if ARGV[0]
+        # FIXME: This is not a very good catch..
+        if ARGV[0][/.*\..*/]
+          if FileTest.exist?(ARGV[0])
+            config = ARGV[0]
+          else
+            log.warn "#{ARGV[0]} not found, trying #{config}"
+          end
+        end
+      end
+
+      unless FileTest.exist?(config)
+        log.error "ERROR: #{config} config missing!"
+        wait
+        Process.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]
+      @config[:server_path] = @config[:serverpath] if @config[:serverpath]
+
+      # Determine arma distribution and location
+      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
+
+      # Verify installation folder
+      if @config[:app_path]
+        log.info "Installation Path: #{@config[:app_path]}"
+        if FileTest.exists?(@config[:app_path])
+          @config[:app_path].gsub!('\\', '/')
+        else
+          log.error "No valid installation folder found, please abort!"
+          wait
+          Process.exit
+        end
+      else
+        log.error "No valid installation folder found, please abort!"
+        wait
+        Process.exit
+      end
+
+      # Nasty workaround for cygwin on Vista vs XP, vs acl's
+=begin
+        @config[:app_path][/\A(\w\:)/]
+        str = "#{$1}/ /six-app-root ntfs noacl 0 0"
+        etc = File.join(TOOL_PATH, 'etc')
+        FileUtils.mkdir_p etc
+        File.open(File.join(etc, 'fstab'), 'w') { |file| file.puts str }
+        ENV['six-app-root'] = '/six-app-root'
+=end
+      puts
+      if @config[:server][:repository]
+        log.info "Checking for latest updates of #{@config[:server][:name]} server info"
+        path = File.join(@config[:server_path], @config[:server][:name])
+
+        opts = {}
+        opts.merge!({:log => log}) if @config[:verbose]
+
+        case @config[:server][:repository][0]
+        when /\Agit:\/\//
+          repo = GitRepo.new(@config[:server][:repository], path)
+        when /\Arsync:\/\//
+          repo = RsyncRepo.new(@config[:server][:repository], path)
+        end
+
+        if FileTest.exist?(File.join(path, '.git')) || FileTest.exist?(File.join(path, '.rsync'))
+          repo.reset(:hard => true)
+          repo.update
+        else
+          #FileUtils::mkdir_p path
+          repo.clone
+        end
+
+        begin
+          srv = File.join(path, 'server.yml')
+          File.open(srv) do |f|
+            cfg = YAML::load(f)
+            @config[:folders].each do |f|
+              cfg[:folders].each do |f2|
+                @config[:folders] -= [f] if f2[:folder] == f[:folder]
+              end
+            end
+            @config[:folders] = cfg[:folders] + @config[:folders]
+            @config[:server] = cfg.merge @config[:server]
+          end
+        rescue
+          log.warn "WARNING: Server configured but unable to get data from repository"
+        end
+      end
+
+      @config[:folders].uniq!
+      log.debug "@config #{@config}"
+
+      # PreProcess the config data
+      @mods = []
+      @mods << @config[:mods] if @config[:mods]
+
+      @config[:folders].each do |folder|
+        folder[:folder].gsub!('\\', '/')
+        if folder[:mods]
+          @mods << folder[:mods]
+        else
+          @mods << folder[:folder]
+        end
+      end
+      @mods = @mods.join(';')
+
+      # Output processed config data
+      log.info "Manager for: #{@mods}"
+
+
+      # Prepare stats
+      @stats = []
+      @config[:folders].each do |folder|
+        stat = Hash.new
+        @stats << stat
+        path = File.join(@config[:app_path], folder[:folder])
+        stat[:changelog] = []
+      end
+    end
+  end
+end
+
+# TODO: Evaluate position
+if $0 == __FILE__
+end

data/lib/six/updater/gitrepo.rb

@@ -0,0 +1,110 @@
+# encoding: UTF-8
+
+module Six
+  module Updater
+    class GitRepo
+      def initialize(repositories, path, depth = 1)
+        @repositories = repositories
+        @path = path
+        @depth = depth
+        Six::Updater.setup_git
+      end
+
+      def open
+        repository
+      end
+
+      def repository
+        @repository || Git.open(@path)
+      end
+
+      def log
+        Six::Updater.log
+      end
+
+      def reset(opts = {})
+        repository.reset(nil, opts)
+      end
+
+      def clone(opts = {})
+        repositories = @repositories.clone
+        done = false
+        opts[:depth] = @depth
+        while repositories.size > 0 && !done
+          rep = repositories.sample
+          repositories -= [rep]
+          log.info "Trying #{rep}"
+
+          begin
+            Git.clone(rep, @path, opts)
+            Git.open(@path).config("core.autocrlf", "false")
+            done = true
+          rescue
+            log.debug "WARNING: Failed, trying other mirrors if available"
+          end
+        end
+        log.warn "WARNING: Failed, was unable to clone!" unless done
+        done
+      end
+
+      def version
+        repository.log.first.to_s
+      end
+
+      def update(opts = {})
+        done = false
+        repositories = @repositories.clone
+        while repositories.size > 0 && !done
+          cfg = nil
+          File.open(File.join(@path, '.git/config')) { |file| cfg = file.readlines }
+          rep = ''
+          cfg.map! do |entry|
+            if entry[/(\t*)url \=/]
+              rep = repositories.sample
+              repositories -= [rep]
+              "#{$1}url = #{rep}"
+            else
+              entry
+            end
+          end
+          log.info "Trying #{rep}"
+          File.open(File.join(@path, '.git/config'), 'w') { |file| file.puts cfg }
+          begin
+            repository.config("core.autocrlf", "false")
+            repository.pull_legacy
+            done = true
+          rescue
+            log.debug "WARNING: Failed, trying other mirrors if available"
+          end
+        end
+        log.warn "WARNING: Failed, was unable to update!" unless done
+        done
+      end
+
+      def status(force = false)
+        if force || !@status
+          @status = status_update
+        end
+        @status
+      end
+
+      def check_status(type, array, msg)
+        str = []
+        if array.size > 0
+          str << "#{type}:"
+          array.each { |item| str << item[0] }
+          msg << str.join("\n")
+        end
+      end
+
+      def status_update
+        msg = []
+        [
+          ['Added', repository.status.added], ['Changed', repository.status.changed],
+          ['Deleted', repository.status.deleted], ['Untracked', repository.status.untracked]
+        ].each { |entry| check_status(entry[0], entry[1], msg) }
+        msg
+      end
+    end
+  end
+end

data/lib/six/updater/mod.rb

@@ -0,0 +1,355 @@
+# encoding: UTF-8
+
+module Six
+  module Updater
+    class Mod
+      def initialize(modconfig, config)
+        @modconfig = modconfig
+        @config = config
+        @stat = Hash.new
+        @stat[:changelog] = []
+        @stat[:current], @stat[:previous] = "A", "B"
+        @path = File.join(@config[:app_path], @modconfig[:folder])
+        get_repositories
+
+        @modconfig[:folder].gsub!('\\', '/')
+        @folder_str = @modconfig[:folder].clone
+        @folder_str.gsub!('/', '-')
+
+        # TODO: Implement this into the real modfolder so it can be verified by the server you join? - 'exclude version.txt' (.gitignore)
+        if @modconfig[:versionfile]
+          @modconfig[:versionfile] = File.join(@config[:app_path], @modconfig[:folder], @modconfig[:versionfile])
+        else
+          @modconfig[:versionfile] = File.join(@config[:app_path], @modconfig[:folder], '.version.txt')
+        end
+        if @modconfig[:changelog]
+          @modconfig[:changelog] = File.join(@config[:app_path], @modconfig[:folder], @modconfig[:changelog])
+        else
+          @modconfig[:changelog] = File.join(BASE_PATH, 'changelogs', "Changelog_#{@folder_str}.txt")
+        end
+
+        unless @modconfig[:depth]
+          @modconfig[:depth] = @config[:depth]
+        end
+
+        @changed = false
+
+
+        # TODO: Handle proper Git class handling for non existent repository / not a git repos
+        # TODO: Git.open requires git to be required, which in turn requires a git binary. Should not be required for uninstall for example.
+        @installed = if FileTest.exists?(File.join(@path, '.git'))
+          #  begin
+          @repository = GitRepo.new(@modconfig[:repository], @path)
+          true
+          # rescue
+          #    log.warn "WARNING: Could not open #{@path}"
+          #   false
+          # end
+        elsif FileTest.exists?(File.join(@path, '.rsync'))
+          # begin
+          @repository = RsyncRepo.new(@modconfig[:repository], @path, :log => log)
+          true
+          # rescue
+          #   log.warn "WARNING: Could not open #{@path}"
+          #  false
+          # end
+        else
+          false
+        end
+      end
+
+      def get_repositories
+        cfg = File.join(@path, '.repositories.yml')
+        @repositories = if FileTest.exist?(cfg)
+          File.open(cfg) { |file| YAML::load(file) }
+        elsif @modconfig[:repository].class == Array
+          @modconfig[:repository]
+        else
+          [@modconfig[:repository]]
+        end
+      end
+
+      def log
+        Six::Updater.log
+      end
+
+      def changed
+        @changed
+      end
+
+      def sure?
+        Six::Updater.sure?
+      end
+
+      def log_status
+        log.warn "Status: #{@repository.status.join("\n")}" if @repository.status.size > 0
+      end
+
+      def not_a_repository
+        log.warn "Does not seem to be a repository!"
+      end
+
+      def write_version
+        @modconfig[:versionfile][/(.*)\/(.*)/]
+        folder = $1
+        FileUtils.mkdir_p folder unless FileTest.exists? folder
+        begin
+          File.open(@modconfig[:versionfile], 'w') { |file| file.puts @stat[:current] }
+          true
+        rescue
+          log.warn "WARNING: Unable to write version.txt file, please make sure everything arma is closed, etc"
+          false
+        end
+      end
+
+      # Process Comment Changelog
+      def write_changelog
+        str = "#{@stat[:current]}"
+        str += " - #{@stat[:previous]}" if @stat[:previous]
+        log.info str
+
+        history = @repository.repository.log(LOG_LIMIT)
+        @stat[:changelog] = []
+        if @stat[:previous]
+          history.between(@stat[:previous], @stat[:current]).each do |c|
+            c.message.split("\n").each { |msg| @stat[:changelog] << msg }
+          end
+        else
+          history.each do |c|
+            c.message.split("\n").each { |msg| @stat[:changelog] << msg }
+          end
+          # DONT SORT, otherwise all messed up stuff :)
+          #@stat[:changelog].sort!
+        end
+
+        # TODO: Clean empty lines etc
+        content = "== #{str}\n"
+        content += @stat[:changelog].join("\n")
+        content += "\n\n"
+        File.open(@modconfig[:changelog], 'r') { |file| content += file.read } if FileTest.exists?(@modconfig[:changelog])
+        @modconfig[:changelog][/(.*)\/(.*)/]
+        folder = $1
+        FileUtils.mkdir_p folder unless FileTest.exists? folder
+        File.open(@modconfig[:changelog], 'w') { |file| file.write content }
+      end
+
+      def changelog
+        unless @installed
+          not_a_repository
+          return
+        end
+
+        unless @stat[:previous] == @stat[:current]
+          log.info 'Playing log...'
+          @stat[:changelog].each do |msg|
+            log.debug msg # TODO: Necessary?
+            puts msg
+          end
+        end
+      end
+
+      def process(command)
+        # TODO: Exception handling, incase working folder was modified
+        log.info "Processing: #{@modconfig[:folder]} (#{command})"
+        cwd = Dir.pwd
+        if FileTest.exist?(@path)
+          Dir.chdir @path
+          system "git #{command}"
+          Dir.chdir cwd
+        else
+          log.warn "WARNING: Can't find folder. Folder does not seem to exist"
+        end
+      end
+
+      # CLEANUP Process (repack -ad / gc --agressive / --auto)
+      # TODO: run with --aggressive every 100 commits or so :O | Evaluate repack -adf   f being more aggressive?
+      # Is aggressive usable yet? http://metalinguist.wordpress.com/2007/12/06/the-woes-of-git-gc-aggressive-and-how-git-deltas-work/
+      def cleanup
+        unless @installed
+          not_a_repository
+          return
+        end
+        unless @stat[:previous] == @stat[:current]
+          log.info 'Cleaning...'
+          if @modconfig[:repository][0][/\Agit:\/\//]
+            @repository.repository.gc_legacy
+            @repository.repository.repack
+          end
+        end
+      end
+
+      # RESET process (incase everything FOOKED :D)
+      def reset
+        unless @installed
+          not_a_repository
+          return
+        end
+        log.warn "###########"
+        log.warn "# WARNING #"
+        log.warn "###########"
+        log.warn "You are about to reset and loose all local changes in the specified repository"
+        log_status
+        sure?
+        log.info 'Resetting...'
+        @repository.repository.reset(nil, {:hard => true})
+      end
+
+      # UPDATE Process
+      def update
+        unless @installed
+          not_a_repository
+          return
+        end
+
+        log.info 'Updating... This might take a while, please wait.'
+        @stat[:previous] = @repository.version
+        done = @repository.update
+        @stat[:current] = @repository.version
+
+        begin
+          @repository.status(true)
+        rescue
+          log.warn 'WARNING: Update seems to have completed but there was a status error..'
+          log_status
+        end
+
+        if @stat[:previous] == @stat[:current]
+          log.info "No updates found"
+          write_version # DO this always incase previous version.txt failed for whatever reason!
+        else
+          @changed = true
+          if write_version
+            #write_changelog
+          else
+            @repository.reset(@stat[:previous], {:hard => true})
+            @changed = false
+            @stat[:current] = @stat[:previous]
+          end
+        end
+
+        unless done
+          log.warn 'WARNING: Update was unable to complete. This could be due to a server / connection problem, or due to files in use or out of synch repositories. Please make sure ArmA is closed, retry, otherwise reset and retry'
+          log_status
+          # TODO: Add display of stats for changed files etc
+        end
+      end
+
+      # INSTALL process
+      def install
+        log.debug "Path: #{@path}"
+        if @installed
+          log.info "Already installed, trying to update..."
+          update
+        else
+          if FileTest.exist?(@path)
+            log.info "WARNING: Folder already exists but does not seem to be a repository. Not installing"
+          else
+            log.info "Installing (depth: #{@modconfig[:depth]})... This might take a while, please wait."
+            folder = ''
+            if @modconfig[:folder][/(.*)\/(.*)/]
+              basepath, subfolder = $1, $2
+              basepath = File.join(@config[:app_path], basepath)
+              FileUtils.mkdir_p basepath unless FileTest.exist?(basepath)
+              #Dir.chdir basepath
+              #folder = File.join(basepath, subfolder)
+            end
+
+            file = File.join(BASE_PATH, "#{@folder_str}.7z")
+            if FileTest.exist?(file)
+              log.info "Found #{@folder_str}.7z, using it"
+              Six::Updater.zipcheck
+              FileUtils.mkdir_p @path
+              cmd = "7z x -o\"#{@path}\" \"#{file}\""
+              log.debug cmd
+              output = %x[#{cmd}]
+              log.debug output
+              begin
+                open
+                @repository.reset(nil, {:hard => true})
+                @installed = true
+                get_repositories
+                update
+              rescue
+                log.warn "WARNING: Failed to open the repository, not installed!"
+              end
+            else
+              @repository = if @repositories[0][/\Arsync:\/\//]
+                RsyncRepo.new(@repositories, @path, @modconfig[:depth])
+              else
+                GitRepo.new(@repositories, @path, @modconfig[:depth])
+              end
+              opts = {}
+              opts.merge!({:log => log}) if @config[:verbose]
+              @installed = @repository.clone(opts)
+              unless @installed
+                log.warn "WARNING: Failed to download the repository! This could be due to a server / connection problem. Please try again later"
+              end
+            end
+
+            if @installed
+              @changed = true
+              @stat[:previous] = nil
+              begin
+                @stat[:current] = @repository.version
+                if @modconfig[:repository][0][/git:\/\//]
+                  @repository.status(true)
+                  FileUtils.rm_f @modconfig[:changelog] if FileTest.exists?(@modconfig[:changelog])
+                end
+                write_version
+                #write_changelog
+              rescue
+                log.warn 'WARNING: Post installation was unable to complete. This could be due to a server / connection problem. Please try again later'
+                FileUtils::rm_rf(@path) if FileTest.exist?(@path)
+              end
+              begin
+                userconfig
+              rescue
+                log.warn 'WARNING: Userconfig processing is unable to complete.'
+              end
+            else
+              FileUtils::rm_rf(@path) if FileTest.exist?(@path)
+            end
+          end
+        end
+      end
+
+      # UNINSTALL process
+      def uninstall
+        if FileTest.exist?(@path)
+          log.warn "###########"
+          log.warn "# WARNING #"
+          log.warn "###########"
+          log.warn "You are about to uninstall the specified repository"
+          sure?
+          log.warn 'Uninstalling...'
+          FileUtils.rm_rf @path
+          FileUtils.rm_f @modconfig[:changelog] if FileTest.exists?(@modconfig[:changelog])
+          FileUtils.rm_f @modconfig[:versionfile] if FileTest.exists?(@modconfig[:versionfile])
+        end
+        # TODO: Cleanup Userconfig ?
+      end
+
+      # Process userconfigs
+      # TODO: Add this processing to update, aka; IF userconfig has changed THEN update it
+      def userconfig
+        unless @installed
+          not_a_repository
+          return
+        end
+        # TODO: Process this on file level instead of folder level?
+        time = Time.now.to_i
+        f = @modconfig[:folder].clone
+        f.gsub!('@', '')
+        path = File.join(@path, 'userconfig')
+        uconfig = File.join(@config[:app_path], 'userconfig')
+        uconfigpath = File.join(uconfig, f)
+        if FileTest.exist?(path)
+          log.info "Found userconfig, processing..."
+          FileUtils::mv(uconfigpath, "#{uconfigpath}_#{time}") if FileTest.exist?(uconfigpath)
+          FileUtils::mkdir_p uconfig unless FileTest.exist?(uconfig)
+          FileUtils::cp_r(path, uconfigpath)
+        end
+      end
+    end
+  end
+end

data/lib/six/updater/options.rb

@@ -0,0 +1,130 @@
+# encoding: UTF-8
+
+# Docu: http://www.ruby-doc.org/stdlib/libdoc/optparse/rdoc/classes/OptionParser.html
+require 'optparse'
+
+module Six
+  module Updater
+    module_function
+    def parse_options
+      todo, general_todo, second_todo = [], [], []
+
+      options = Hash.new
+      OptionParser.new do |opts|
+        $0[/.*\/(.*)/]
+        opts.banner = "Usage: #{$1} [config.yml] [options]"
+
+        opts.on("-v", "--[no-]verbose", "Run verbosely") do |v|
+          options[:verbose] = v
+        end
+
+        opts.on("-i", "--install", "Installs the mods, unless already installed") do |bool|
+          todo << :install if bool
+        end
+
+        opts.on("-d", "--uninstall", "Uninstalls the mods, unless not installed") do |bool|
+          todo << :uninstall if bool
+        end
+
+        opts.on("-u", "--update", "Updates each mod to latest version") do |bool|
+          todo << :update if bool
+        end
+
+        opts.on("-r", "--reset", "Resets the modfolders to latest commit status. All modifications lost") do |bool|
+          todo << :reset if bool
+        end
+
+        opts.on("-z", "--userconfig", "Processes the userconfigs of each mod. Makes backups of previous files") do |bool|
+          todo << :userconfig if bool
+        end
+
+        opts.on("-l", "--changelog", "Display Changelog after update") do |bool|
+          second_todo << :changelog if bool
+        end
+
+        opts.on("-c", "--cleanup", "Cleans up the modfolders git repositories") do |bool|
+          second_todo << :cleanup if bool
+        end
+
+        opts.on("-s", "--createshortcut", "Creates shortcut to run the game with the mods, in installation folder") do |bool|
+          general_todo << :createshortcut if bool
+        end
+
+        opts.on("-k", "--createdesktopshortcut", "Creates shortcut to run the game with the mods, on the desktop") do |bool|
+          general_todo << :createdesktopshortcut if bool
+        end
+
+        opts.on("-j", "--join", "Starts the game with #{@config[:app_params]} -mod=#{@mods} -connect=#{@config[:server][:address]} -port=#{@config[:server][:port]}") do |bool|
+          general_todo << :joingame if bool
+        end
+
+        opts.on("-g", "--startgame", "Starts the game with #{@config[:app_params]} -mod=#{@mods}") do |bool|
+          general_todo << :startgame if bool
+        end
+
+        opts.on("-w", "--wait", "Waits at the end for user pressing ENTER") do |bool|
+          options[:wait] = bool
+        end
+
+        opts.on("-f", "--force", "Forces to run tasks even though no changes were detected") do |bool|
+          options[:force] = bool
+        end
+
+        opts.on("--depth I", Integer, "Clone depth, default: #{@config[:depth]}. Set to 0 to clone all history") do |s|
+          options[:depth] = s
+        end
+
+        opts.on("--mods S", String, "Additional Mods") do |s|
+          options[:mods] = s
+        end
+
+        opts.on("--armapath S", String, "Destination folder") do |s|
+          options[:armapath] = s
+        end
+
+        opts.on("--gitpath S", String, "Git binary") do |s|
+          options[:gitpath] = s
+        end
+      end.parse!
+
+      default = if (todo + second_todo + general_todo).size > 0
+        false
+      else
+        true
+      end
+
+      # TODO: Move this to Updater ?
+      @todo = if todo.size > 0
+        todo
+      else
+        log.info "No parameters given, running the default"
+        #options[:wait] = true
+        if default
+          @config[:defaultactions]
+        else
+          []
+        end
+      end
+      @general_todo = if general_todo.size > 0
+        general_todo
+      else
+        if default
+          @config[:defaultgeneralactions]
+        else
+          []
+        end
+      end
+
+      @second_todo = if second_todo.size > 0
+        second_todo
+      else
+        if default
+          @config[:defaultsecondactions]
+        else
+          []
+        end
+      end
+      @config = @config.merge(options)
+    end
+  end
+end

data/lib/six/updater/rsyncrepo.rb

@@ -0,0 +1,73 @@
+# encoding: UTF-8
+
+module Six
+  module Updater
+    class RsyncRepo
+      def initialize(repositories, path, depth = 1)
+        @repositories = repositories
+        @path = path
+        @changed = false
+        @installed = false
+      end
+
+      def log
+        Six::Updater.log
+      end
+
+      def version
+        f = File.join(@path, '.rsync', '.pack', '.repository.yml')
+        if FileTest.exist?(f)
+          File.open(f) {|file| YAML::load(file)[:version] }
+        else
+          nil
+        end
+      end
+
+      def open(opts = {})
+        repository(opts)
+      end
+
+      def reset(opts = {})
+        repository.reset(opts)
+      end
+
+      def status(force = false)
+        []
+      end
+
+      def repository(opts = {:log => log})
+        @repository || Six::Repositories::Rsync.open(@path, opts)
+      end
+
+      def clone(opts = {})
+        done = false
+        @path[/(.*)\/(.*)/]
+        opts = {:path => $1, :depth => @depth, :log => log}
+        #begin
+        @repository = Rsync.clone(@repositories, $2, opts)
+        # FIXME: Update needs to be fixed
+        done = true
+        # rescue
+        #   log.warn "WARNING: Failed, was unable to clone!" unless done
+        #log.debug "WARNING: Failed, trying other mirrors if available"
+        # end
+        done
+      end
+
+      def update(opts = {})
+        cfg = nil
+        File.open(File.join(@path, '.rsync/config.yml')) { |file| cfg = YAML::load(file) }
+        cfg[:hosts] = @repositories
+        File.open(File.join(@path, '.rsync/config.yml'), 'w') { |file| file.puts cfg.to_yaml }
+
+        # begin
+        repository.update
+        done = true
+        #  rescue
+        #   log.warn "WARNING: Failed, was unable to update!" unless done
+        #  end
+        done
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,103 @@
+--- !ruby/object:Gem::Specification 
+name: six-updater
+version: !ruby/object:Gem::Version 
+  version: 0.10.1
+platform: ruby
+authors: 
+- Sickboy
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-09-25 00:00:00 +02:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: log4r
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 1.1.0
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: six-rsync
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.1.0
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: win32-process
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.6.0
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: win32-shortcut
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.2.3
+    version: 
+description: Your summary here
+email: sb@dev-heaven.net
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README
+- LICENSE
+files: 
+- LICENSE
+- README
+- Rakefile
+- lib/six/updater/gitrepo.rb
+- lib/six/updater/mod.rb
+- lib/six/updater/options.rb
+- lib/six/updater/rsyncrepo.rb
+- lib/six/updater-app.rb
+- lib/six/updater.rb
+has_rdoc: true
+homepage: 
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Your summary here
+test_files: []
+