from __future__ import annotations

import pathlib
import re
import sys

main_path = pathlib.Path(sys.argv[1])
manifest_path = pathlib.Path(sys.argv[2])
build_path = pathlib.Path(sys.argv[3])

main_text = main_path.read_text(
    encoding="utf-8",
)

marker = (
    "SHIRE-MOBILE-SENTINEL-LIVE-ALERTS-0730"
)

anchor = (
    "        ApprovalNotifications.checkNow(this)"
)

addition = """        ApprovalNotifications.checkNow(this)

        // SHIRE-MOBILE-SENTINEL-LIVE-ALERTS-0730
        if (
            android.os.Build.VERSION.SDK_INT >= 33 &&
            checkSelfPermission(
                android.Manifest.permission.POST_NOTIFICATIONS
            ) != android.content.pm.PackageManager.PERMISSION_GRANTED
        ) {
            requestPermissions(
                arrayOf(
                    android.Manifest.permission.POST_NOTIFICATIONS
                ),
                7301
            )
        }

        SentinelMobileAlerts.initialise(this)"""

if marker not in main_text:
    if anchor not in main_text:
        raise SystemExit(
            "MainActivity notification anchor was not found."
        )

    main_text = main_text.replace(
        anchor,
        addition,
        1,
    )

main_path.write_text(
    main_text,
    encoding="utf-8",
)

manifest = manifest_path.read_text(
    encoding="utf-8",
)

permission_anchor = (
    '    <uses-permission '
    'android:name="android.permission.RECEIVE_BOOT_COMPLETED" />'
)

permission_block = """    <uses-permission android:name="android.permission.RECEIVE_BOOT_COMPLETED" />
    <uses-permission android:name="android.permission.FOREGROUND_SERVICE" />
    <uses-permission android:name="android.permission.FOREGROUND_SERVICE_SPECIAL_USE" />"""

if (
    "android.permission.FOREGROUND_SERVICE_SPECIAL_USE"
    not in manifest
):
    if permission_anchor not in manifest:
        raise SystemExit(
            "Manifest permission anchor was not found."
        )

    manifest = manifest.replace(
        permission_anchor,
        permission_block,
        1,
    )

component_marker = (
    "SHIRE-MOBILE-SENTINEL-LIVE-ALERTS-0730"
)

components = """
        <!-- SHIRE-MOBILE-SENTINEL-LIVE-ALERTS-0730 -->
        <service
            android:name=".SentinelAlertLinkService"
            android:exported="false"
            android:foregroundServiceType="specialUse"
            android:stopWithTask="false">

            <property
                android:name="android.app.PROPERTY_SPECIAL_USE_FGS_SUBTYPE"
                android:value="Private user-enabled SHiRE SENTiNEL security alert link over Tailscale" />

        </service>

        <service
            android:name=".SentinelAlertPollJobService"
            android:exported="false"
            android:permission="android.permission.BIND_JOB_SERVICE" />

        <receiver
            android:name=".SentinelAlertBootReceiver"
            android:enabled="true"
            android:exported="false">

            <intent-filter>
                <action android:name="android.intent.action.BOOT_COMPLETED" />
                <action android:name="android.intent.action.MY_PACKAGE_REPLACED" />
            </intent-filter>

        </receiver>
"""

if component_marker not in manifest:
    if "    </application>" not in manifest:
        raise SystemExit(
            "Manifest application closing tag was not found."
        )

    manifest = manifest.replace(
        "    </application>",
        components + "\n    </application>",
        1,
    )

manifest_path.write_text(
    manifest,
    encoding="utf-8",
)

build_text = build_path.read_text(
    encoding="utf-8",
)

match = re.search(
    r"versionCode\s*=\s*(\d+)",
    build_text,
)

if not match:
    raise SystemExit(
        "Android versionCode was not found."
    )

new_code = int(match.group(1)) + 1

build_text = re.sub(
    r"versionCode\s*=\s*\d+",
    f"versionCode = {new_code}",
    build_text,
    count=1,
)

build_text = re.sub(
    r'versionName\s*=\s*"[^"]+"',
    'versionName = "0.7.3-sentinel-mobile-alerts"',
    build_text,
    count=1,
)

build_path.write_text(
    build_text,
    encoding="utf-8",
)