six-updater 0.17.1-x86-mswin32

data/lib/six/updater/gitrepo.rb

@@ -0,0 +1,112 @@
+# 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"
+            log.debug "#{$!}"
+          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"
+            log.debug "#{$!}"
+          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,348 @@
+# encoding: UTF-8
+
+module Six
+  module Updater
+    class Mod
+      attr_reader :changed
+      DEFAULT_PRIORITY = 9999
+      def initialize(modconfig, config)
+        @modconfig = modconfig
+        @config = config
+        @stat = Hash.new
+        @stat[:changelog] = []
+        @stat[:current], @stat[:previous] = "A", "B"
+        @path = File.join(self.apppath, @modconfig[:folder])
+        get_repositories
+
+        @modconfig[:folder].gsub!('\\', '/')
+        @folder_str = @modconfig[:folder].clone
+        @folder_str.gsub!('/', '-')
+
+        if @modconfig[:versionfile]
+          @modconfig[:versionfile] = File.join(@config[:app_modpath], @modconfig[:folder], @modconfig[:versionfile])
+        else
+          @modconfig[:versionfile] = File.join(@config[:app_modpath], @modconfig[:folder], '.version.txt')
+        end
+
+        unless @modconfig[:depth]
+          @modconfig[:depth] = @config[:depth]
+        end
+
+        @changed = false
+
+        @installed = if FileTest.exists?(File.join(@path, '.rsync'))
+          @repository = RsyncRepo.new(@modconfig[:repository], @path, :log => log)
+          true
+        else
+          false
+        end
+      end
+
+      # TODO: Custom paths need more consideration, especially re the -mod= line
+      def apppath; @modconfig[:path] ? @modconfig[:path] : @config[:app_modpath]; end
+      def priority; @modconfig[:priority] ? @modconfig[:priority] : DEFAULT_PRIORITY; end
+      def esc(val); "\"#{val}\""; end
+      def log; Six::Updater.log; 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 disabled?
+        log.info "Mod is tagged 'disabled' in the configuration file. Skipping..." if @modconfig[:disabled]
+        @modconfig[:disabled]
+      end
+
+      def installed?
+        not_a_repository unless @installed
+        @installed
+      end
+
+      def skip?
+        log.info "Mod is tagged 'skip' in the configuration file. Skipping..." if @modconfig[:skip]
+        @modconfig[:skip]
+      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 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"
+          log.debug "#{$!}"
+          false
+        end
+      end
+
+      def changelog
+        return if self.disabled?
+        return unless self.installed?
+
+        unless @stat[:previous] == @stat[:current]
+          log.info 'Playing log...'
+          @stat[:changelog].each do |msg|
+            log.debug msg # TODO: Necessary?
+            puts msg
+          end
+        end
+      end
+
+      # RESET process (incase everything FOOKED :D)
+      def reset
+        return if self.disabled?
+        return unless self.installed?
+
+        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
+       if self.skip?
+          begin
+            userconfig(false)
+          rescue
+            log.warn 'WARNING: Userconfig processing is unable to complete.'
+            log.debug "#{$!}"
+          end
+          return
+        else
+          return if self.disabled?
+        end
+
+       return unless self.installed?
+
+        @stat[:previous] = @repository.version
+        log.info "Current Version: #{@stat[:previous]}, checking for updates..."
+        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.debug "#{$!}"
+          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
+          log.info "Applied version: #{@stat[:current]}"
+          @changed = true
+          if write_version
+            #write_changelog
+          else
+            @repository.reset(@stat[:previous], {:hard => true})
+            @changed = false
+            @stat[:current] = @stat[:previous]
+          end
+        end
+
+        # TODO: Detect missing / out of date config files, and notify user
+        begin
+          userconfig(false)
+        rescue
+          log.warn 'WARNING: Userconfig processing is unable to complete.'
+          log.debug "#{$!}"
+        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}"
+        return if self.disabled?
+        if self.installed?
+          log.info "Already installed, trying to update..."
+          update
+        else
+          if FileTest.exist?(@path)
+            log.warn "WARNING: Folder already exists but seems incompatible. Not installing."
+            log.info "If you want the software to install this mod, please delete the modfolder"
+          else
+            log.info "Installing... This might take a while, please wait."
+            folder = ''
+            if @modconfig[:folder][/(.*)\/(.*)/]
+              basepath, subfolder = $1, $2
+              basepath = File.join(@config[:app_modpath], basepath)
+              FileUtils.mkdir_p basepath unless FileTest.exist?(basepath)
+            end
+
+            file = File.join(DATA_PATH, "#{@folder_str}.7z")
+            file = File.join(BASE_PATH, "#{@folder_str}.7z") unless File.exists?(file)
+            if File.exists?(file)
+              log.info "Found #{@folder_str}.7z, using it"
+              #Six::Updater.zipcheck
+              FileUtils.mkdir_p @path
+              cmd = "7z x -o#{esc(@path)} #{esc(file)}"
+              log.debug cmd
+              output = %x[#{cmd}]
+              log.debug output
+
+              if File.exists?(File.join(@path, ".rsync", "config.yml"))
+                begin
+                  @repository = RsyncRepo.new(@modconfig[:repository], @path, :log => log)
+                  @installed = true
+                  get_repositories
+                  update
+                rescue
+                  @installed = false
+                  FileUtils::rm_rf(@path) if FileTest.exist?(@path)
+                end
+              else
+                log.warn "WARNING: Failed to open the repository, the #{@folder_str}.7z does not seem to be correct"
+                log.warn "Installing from rsync network instead"
+                log.debug "#{$!}"
+                FileUtils::rm_rf(@path) if FileTest.exist?(@path)
+              end
+            end
+
+            unless @installed
+              @repository = RsyncRepo.new(@repositories, @path)
+              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
+                write_version
+              rescue
+                log.warn 'WARNING: Post installation was unable to complete. This could be due to a server / connection problem. Please try again later'
+                log.debug "#{$!}"
+                FileUtils::rm_rf(@path) if FileTest.exist?(@path)
+              end
+              begin
+                userconfig
+              rescue
+                log.warn 'WARNING: Userconfig processing is unable to complete.'
+                log.debug "#{$!}"
+              end
+            else
+              FileUtils::rm_rf(@path) if FileTest.exist?(@path)
+            end
+          end
+        end
+      end
+
+      def cleanup
+        log.warn "CLEANUP Action unimplemented atm, please pick different action"
+      end
+
+      # UNINSTALL process
+      def uninstall
+        return if self.disabled?
+
+        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
+        end
+        # TODO: Cleanup Userconfig ?
+      end
+
+      # Process keys
+      def keys(force = true)
+        return if self.disabled? 
+        return unless self.installed?
+
+        path = File.join(@path, 'store', 'keys.tar')
+        path = File.join(@path, 'store', 'keys') unless File.exists?(path)
+        path = File.join(@path, 'keys') unless File.directory?(path)         
+        if File.exists?(path)
+          log.info "Found keys, processing..."
+          k_path = File.join(@config[:app_path], 'keys')
+          if File.directory? path
+            Dir[File.join(path, '*.bikey')].each do |key|
+              key[/(.*)\/(.*)/]
+              path, file = $1, $2
+              file[/(.*)_(.*)\.bikey/]
+              prefix, suffix = $1, $2
+              if suffix && prefix
+                if suffix[/(\D*)\d*/]
+                  suf = $1
+                  Dir[File.join(k_path, "#{prefix}_#{suf}*.bikey")].each { |delkey| FileUtils.rm_f(delkey) } if suf
+                end
+              end
+              FileUtils.cp(key, k_path)
+            end
+          else
+            # TODO: Cleanup old keys, Ruby Tar library, examine content? - tarruby doesn't build on 1.9.1. mingw
+            cmd = "7z x #{esc(path)} -o #{esc(@config[:app_path])}"
+            cmd.gsub!("/", "\\")
+            system cmd
+          end
+        end
+      end
+
+      # Process userconfigs
+      # TODO: Add this processing to update, aka; IF userconfig has changed THEN update it
+      def userconfig(force = true)
+        return if self.disabled?
+        return unless self.installed?
+
+        # 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, 'store', 'userconfig.tar')
+        path = File.join(@path, 'store', 'userconfig') unless File.exists?(path)
+        path = File.join(@path, 'userconfig') unless File.exists?(path)         
+        if File.exists?(path)
+          uconfig = File.join(@config[:app_path], 'userconfig')
+          uconfigpath = File.join(uconfig, f)
+          if !force && FileTest.exist?(uconfigpath)
+            log.debug "Userconfig folder already found, skipping"
+            return
+          end
+          log.info "Found userconfig, processing..."
+          FileUtils::mv(uconfigpath, "#{uconfigpath}_#{time}") if File.exists?(uconfigpath)
+          if File.directory? path
+            FileUtils::mkdir_p uconfig unless File.exists?(uconfig)
+            FileUtils::cp_r(path, uconfigpath)
+          else
+            cmd = "7z x #{esc(path)} -o#{esc(@config[:app_path])} -y"
+            cmd.gsub!("/", "\\")
+            system cmd
+          end
+        end
+      end
+    end
+  end
+end