rhodecode/model/update.py

# Copyright (C) 2013-2023 RhodeCode GmbH
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License, version 3
# (only), as published by the Free Software Foundation.
#
# This program 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 General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
# This program is dual-licensed. If you wish to learn more about the
# RhodeCode Enterprise Edition, including its added features, Support services,
# and proprietary license terms, please see https://rhodecode.com/licenses/

import logging
import urllib.request
import urllib.error
import urllib.parse
from packaging.version import Version

import rhodecode
from rhodecode.lib.ext_json import json
from rhodecode.model import BaseModel
from rhodecode.model.meta import Session
from rhodecode.model.settings import SettingsModel


log = logging.getLogger(__name__)


class UpdateModel(BaseModel):
    UPDATE_SETTINGS_KEY = 'update_version'
    UPDATE_URL_SETTINGS_KEY = 'rhodecode_update_url'

    @staticmethod
    def get_update_data(update_url):
        """Return the JSON update data."""
        ver = rhodecode.__version__
        log.debug('Checking for upgrade on `%s` server', update_url)
        opener = urllib.request.build_opener()
        opener.addheaders = [('User-agent', f'RhodeCode-SCM/{ver.strip()}')]
        response = opener.open(update_url)
        response_data = response.read()
        data = json.loads(response_data)
        log.debug('update server returned data')
        return data

    def get_update_url(self):
        settings = SettingsModel().get_all_settings()
        return settings.get(self.UPDATE_URL_SETTINGS_KEY)

    def store_version(self, version):
        log.debug('Storing version %s into settings', version)
        setting = SettingsModel().create_or_update_setting(
            self.UPDATE_SETTINGS_KEY, version)
        Session().add(setting)
        Session().commit()

    def get_stored_version(self, fallback=None):
        obj = SettingsModel().get_setting_by_name(self.UPDATE_SETTINGS_KEY)
        if obj:
            return obj.app_settings_value
        return fallback or '0.0.0'

    def _sanitize_version(self, version):
        """
        Cleanup our custom ver.
        e.g 4.11.0_20171204_204825_CE_default_EE_default to 4.11.0
        """
        return version.split('_')[0]

    def is_outdated(self, cur_version, latest_version=None):
        latest_version = latest_version or self.get_stored_version()
        try:
            cur_version = self._sanitize_version(cur_version)
            return Version(latest_version) > Version(cur_version)
        except Exception:
            # could be invalid version, etc
            return False