discourse/lib/discourse_updates.rb

module DiscourseUpdates

  class << self

    def check_version
      version_info = if updated_at.nil?
        DiscourseVersionCheck.new(
          installed_version: Discourse::VERSION::STRING,
          installed_sha: (Discourse.git_version == 'unknown' ? nil : Discourse.git_version),
          installed_describe: Discourse.full_version,
          git_branch: Discourse.git_branch,
          updated_at: nil
        )
      else
        DiscourseVersionCheck.new(
          latest_version: latest_version,
          critical_updates: critical_updates_available?,
          installed_version: Discourse::VERSION::STRING,
          installed_sha: (Discourse.git_version == 'unknown' ? nil : Discourse.git_version),
          installed_describe: Discourse.full_version,
          missing_versions_count: missing_versions_count,
          git_branch: Discourse.git_branch,
          updated_at: updated_at
        )
      end

      # replace -commit_count with +commit_count
      if version_info.installed_describe =~ /-(\d+)-/
        version_info.installed_describe = version_info.installed_describe.gsub(/-(\d+)-.*/, " +#{$1}")
      end

      if SiteSetting.version_checks?

        # Handle cases when version check data is old so we report something that makes sense

        if (version_info.updated_at.nil? ||  # never performed a version check
            last_installed_version != (Discourse::VERSION::STRING) ||  # upgraded since the last version check
            (version_info.missing_versions_count == (0) && version_info.latest_version != (version_info.installed_version)) ||  # old data
            (version_info.missing_versions_count != (0) && version_info.latest_version == (version_info.installed_version)))    # old data
          Jobs.enqueue(:version_check, all_sites: true)
          version_info.version_check_pending = true
          unless version_info.updated_at.nil?
            version_info.missing_versions_count = 0
            version_info.critical_updates = false
          end
        end
      end

      version_info
    end

    # last_installed_version is the installed version at the time of the last version check
    def last_installed_version
      $redis.get last_installed_version_key
    end

    def latest_version
      $redis.get latest_version_key
    end

    def missing_versions_count
      $redis.get(missing_versions_count_key).try(:to_i)
    end

    def critical_updates_available?
      ($redis.get(critical_updates_available_key) || false) == 'true'
    end

    def updated_at
      t = $redis.get(updated_at_key)
      t ? Time.zone.parse(t) : nil
    end

    def updated_at=(time_with_zone)
      $redis.set updated_at_key, time_with_zone.as_json
    end

    ['last_installed_version', 'latest_version', 'missing_versions_count', 'critical_updates_available'].each do |name|
      eval "define_method :#{name}= do |arg|
        $redis.set #{name}_key, arg
      end"
    end

    def missing_versions=(versions)
      # delete previous list from redis
      prev_keys = $redis.lrange(missing_versions_list_key, 0, 4)
      if prev_keys
        $redis.del prev_keys
        $redis.del(missing_versions_list_key)
      end

      if versions.present?
        # store the list in redis
        version_keys = []
        versions[0, 5].each do |v|
          key = "#{missing_versions_key_prefix}:#{v['version']}"
          $redis.mapped_hmset key, v
          version_keys << key
        end
        $redis.rpush missing_versions_list_key, version_keys
      end

      versions || []
    end

    def missing_versions
      keys = $redis.lrange(missing_versions_list_key, 0, 4) # max of 5 versions
      keys.present? ? keys.map { |k| $redis.hgetall(k) } : []
    end

    private

      def last_installed_version_key
        'last_installed_version'
      end

      def latest_version_key
        'discourse_latest_version'
      end

      def critical_updates_available_key
        'critical_updates_available'
      end

      def missing_versions_count_key
        'missing_versions_count'
      end

      def updated_at_key
        'last_version_check_at'
      end

      def missing_versions_list_key
        'missing_versions'
      end

      def missing_versions_key_prefix
        'missing_version'
      end
  end
end