main.py

"""Entry point."""

import sys

from environs import Env
from loguru import logger

from src.app import APP
from src.config import RevancedConfig
from src.downloader.download import Downloader
from src.exceptions import AppNotFoundError, BuilderError, PatchesJsonLoadError, PatchingFailedError
from src.parser import Parser
from src.patches import Patches
from src.utils import check_java, delete_old_changelog, load_older_updates, save_patch_info, write_changelog_to_file


def get_app(config: RevancedConfig, app_name: str) -> APP:
    """Get App object."""
    env_package_name = config.env.str(f"{app_name}_PACKAGE_NAME".upper(), None)
    package_name = env_package_name or Patches.get_package_name(app_name)
    return APP(app_name=app_name, package_name=package_name, config=config)


def main() -> None:
    """Entry point."""
    env = Env()
    env.read_env()
    config = RevancedConfig(env)
    updates_info = {}
    Downloader.extra_downloads(config)
    if not config.dry_run:
        check_java()
        delete_old_changelog()
        updates_info = load_older_updates(env)

    logger.info(f"Will Patch only {config.apps}")
    for possible_app in config.apps:
        logger.info(f"Trying to build {possible_app}")
        try:
            app = get_app(config, possible_app)
            app.download_patch_resources(config)
            patcher = Patches(config, app)
            parser = Parser(patcher, config)
            app_all_patches = patcher.get_app_configs(app)
            app.download_apk_for_patching(config)
            parser.include_exclude_patch(app, app_all_patches, patcher.patches_dict)
            logger.info(app)
            updates_info = save_patch_info(app, updates_info)
            parser.patch_app(app)
        except AppNotFoundError as e:
            logger.info(e)
        except PatchesJsonLoadError:
            logger.exception("Patches.json not found")
        except PatchingFailedError as e:
            logger.exception(e)
        except BuilderError as e:
            logger.exception(f"Failed to build {possible_app} because of {e}")
    write_changelog_to_file(updates_info)


if __name__ == "__main__":
    try:
        main()
    except KeyboardInterrupt:
        logger.error("Script halted because of keyboard interrupt.")
        sys.exit(1)