# Copyright (C) 2016-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 click import pyramid.paster from rhodecode.lib.pyramid_utils import bootstrap, get_app_config from rhodecode.lib.db_manage import DbManage from rhodecode.lib.utils2 import get_encryption_key from rhodecode.model.db import Session log = logging.getLogger(__name__) @click.command() @click.argument('ini_path', type=click.Path(exists=True)) @click.option( '--force-yes/--force-no', default=None, help="Force yes/no to every question") @click.option( '--user', default=None, help='Initial super-admin username') @click.option( '--email', default=None, help='Initial super-admin email address.') @click.option( '--password', default=None, help='Initial super-admin password. Minimum 6 chars.') @click.option( '--api-key', help='Initial API key for the admin user') @click.option( '--repos', default=None, help='Absolute path to storage location. This is storage for all ' 'existing and future repositories, and repository groups.') @click.option( '--public-access/--no-public-access', default=None, help='Enable public access on this installation. ' 'Default is public access enabled.') @click.option( '--skip-existing-db', default=False, is_flag=True, help='Do not destroy and re-initialize the database if it already exist.') @click.option( '--apply-license-key', default=False, is_flag=True, help='Get the license key from a license file or ENV and apply during DB creation.') def main(ini_path, force_yes, user, email, password, api_key, repos, public_access, skip_existing_db, apply_license_key): return command(ini_path, force_yes, user, email, password, api_key, repos, public_access, skip_existing_db, apply_license_key) def command(ini_path, force_yes, user, email, password, api_key, repos, public_access, skip_existing_db, apply_license_key): # mapping of old parameters to new CLI from click options = dict( username=user, email=email, password=password, api_key=api_key, repos_location=repos, force_ask=force_yes, public_access=public_access ) pyramid.paster.setup_logging(ini_path) config = get_app_config(ini_path) db_uri = config['sqlalchemy.db1.url'] enc_key = get_encryption_key(config) dbmanage = DbManage(log_sql=True, dbconf=db_uri, root='.', tests=False, cli_args=options, enc_key=enc_key) if skip_existing_db and dbmanage.db_exists(): return dbmanage.create_tables(override=True) dbmanage.set_db_version() opts = dbmanage.config_prompt(None) dbmanage.create_settings(opts) dbmanage.create_default_user() dbmanage.create_admin_and_prompt() dbmanage.create_permissions() dbmanage.populate_default_permissions() if apply_license_key: try: from rc_license.models import apply_trial_license_if_missing apply_trial_license_if_missing(force=True) except ImportError: pass Session().commit() with bootstrap(ini_path, env={'RC_CMD_SETUP_RC': '1'}) as env: msg = 'Successfully initialized database, schema and default data.' print() print('*' * len(msg)) print(msg.upper()) print('*' * len(msg))