@echo off
setlocal

rem ============================================================
rem  Constants: edit here only
rem ============================================================
set "DomainPart=autoset.example.com"

set "ApiKey=YOUR_API_KEY"
set "CreatedDate=2025-12-11T00:00:00Z"
set "OrganizationID=YOUR_ORG_ID"

rem Registry root: HKCU or HKLM
rem   HKCU = current user only  (no admin needed, double click to run)
rem   HKLM = all users on this PC (requires admin / Run as administrator)
set "RegRoot=HKCU"


rem ============================================================
rem Extension IDs (Chrome and Edge have different IDs for same extension)
set "ExtensionIdChrome=bdeanmdeckegmfjpbnngomallcedjold"
set "ExtensionIdEdge=flggmhlpipcopffjfkpgkoljghfkmfcg"

rem Update URLs
set "ChromeUpdateUrl=https://clients2.google.com/service/update2/crx"
set "EdgeUpdateUrl=https://edge.microsoft.com/extensionwebstorebase/v1/crx"
rem ============================================================

rem Validate RegRoot
if /i "%RegRoot%"=="HKCU" set "RegRootFull=HKEY_CURRENT_USER"
if /i "%RegRoot%"=="HKLM" set "RegRootFull=HKEY_LOCAL_MACHINE"
if not defined RegRootFull (
    echo [ERROR] RegRoot must be HKCU or HKLM. Got: %RegRoot%
    pause
    exit /b 1
)

rem Registry keys
set "ChromeExtSettingsKey=%RegRootFull%\Software\Policies\Google\Chrome\ExtensionSettings\%ExtensionIdChrome%"
set "ChromePolicyKey=%RegRootFull%\Software\Policies\Google\Chrome\3rdparty\extensions\%ExtensionIdChrome%\policy"
set "EdgeExtSettingsKey=%RegRootFull%\Software\Policies\Microsoft\Edge\ExtensionSettings\%ExtensionIdEdge%"
set "EdgePolicyKey3rd=%RegRootFull%\Software\Policies\Microsoft\Edge\3rdparty\extensions\%ExtensionIdEdge%\policy"


echo ==============================
echo  Edge + Chrome Extension Initializer
echo ==============================
echo   Registry root: %RegRoot% (%RegRootFull%)
echo.


rem ============================================================
rem 1. Generate UserEmail (whoami /upn, then fallbacks)
rem Priority : whoami /upn, USERNAME, COMPUTERNAME, HOSTNAME, LOGONSERVER
rem ============================================================
echo [1] Generating UserEmail ...

set "UserEmail="
for /f "usebackq delims=" %%i in (`whoami /upn 2^>nul`) do set "UserEmail=%%i"
if defined UserEmail (
    echo   Got from whoami /upn
) else if defined USERNAME (
    set "UserEmail=%USERNAME%@%DomainPart%"
    echo   whoami /upn failed, using USERNAME.
) else (
    echo   USERNAME not defined. Trying hostname...
    if defined COMPUTERNAME (
        set "TempUser=%COMPUTERNAME%"
    ) else if defined HOSTNAME (
        set "TempUser=%HOSTNAME%"
    ) else (
        echo   COMPUTERNAME, HOSTNAME not defined. Trying LOGONSERVER...
        if defined LOGONSERVER (
            set "TempUser=%LOGONSERVER%"
        ) else (
            echo   LOGONSERVER not defined. Using unknown.
            set "TempUser=unknown"
        )
    )
    set "UserEmail=%TempUser%@%DomainPart%"
)

echo   UserEmail = %UserEmail%
echo.


rem ============================================================
rem 2. Get UserPc (hostname) not persisted
rem ============================================================
echo [2] Reading hostname for UserPc ...

set "UserPc="
if defined COMPUTERNAME (
    set "UserPc=%COMPUTERNAME%"
    echo   Got from COMPUTERNAME
) else (
    echo   COMPUTERNAME not defined. Trying hostname command...
    for /f "usebackq delims=" %%i in (`hostname 2^>nul`) do set "UserPc=%%i"
    if not defined UserPc (
        if defined HOSTNAME (
            set "UserPc=%HOSTNAME%"
            echo   Got from HOSTNAME
        ) else (
            echo   HOSTNAME not defined. Trying LOGONSERVER...
            if defined LOGONSERVER (
                for /f "tokens=2 delims=\" %%i in ("%LOGONSERVER%") do set "UserPc=%%i"
                if not defined UserPc set "UserPc=%LOGONSERVER%"
                echo   Got from LOGONSERVER
            ) else (
                set "UserPc=unknown"
                echo   All failed. Using unknown.
            )
        )
    )
)

echo   UserPc = %UserPc%
echo.


rem ============================================================
rem 3. Chrome extension forced install + policy
rem ============================================================
echo [3] Writing Chrome ExtensionSettings and policy (%RegRoot%)...

rem Chrome ExtensionSettings subkey per extension
reg add "%ChromeExtSettingsKey%" /v "installation_mode" /t REG_SZ /d "force_installed" /f
reg add "%ChromeExtSettingsKey%" /v "update_url"        /t REG_SZ /d "%ChromeUpdateUrl%" /f

rem Chrome 3rdparty extension policy
reg add "%ChromePolicyKey%" /v "ApiKey"         /t REG_SZ /d "%ApiKey%"        /f >nul 2>&1
reg add "%ChromePolicyKey%" /v "CreatedDate"    /t REG_SZ /d "%CreatedDate%"   /f >nul 2>&1
reg add "%ChromePolicyKey%" /v "OrganizationID" /t REG_SZ /d "%OrganizationID%" /f >nul 2>&1
reg add "%ChromePolicyKey%" /v "UserEmail"      /t REG_SZ /d "%UserEmail%"     /f >nul 2>&1
reg add "%ChromePolicyKey%" /v "UserPc"         /t REG_SZ /d "%UserPc%"        /f >nul 2>&1

echo   Chrome extension and policy applied.
echo.


rem ============================================================
rem 4. Edge extension forced install + policy
rem ============================================================
echo [4] Writing Edge ExtensionSettings and policy (%RegRoot%)...

rem Edge ExtensionSettings subkey per extension
reg add "%EdgeExtSettingsKey%" /v "installation_mode" /t REG_SZ /d "force_installed" /f
reg add "%EdgeExtSettingsKey%" /v "update_url"        /t REG_SZ /d "%EdgeUpdateUrl%" /f

rem Edge 3rdparty extension policy
reg add "%EdgePolicyKey3rd%" /v "ApiKey"         /t REG_SZ /d "%ApiKey%"        /f >nul 2>&1
reg add "%EdgePolicyKey3rd%" /v "CreatedDate"    /t REG_SZ /d "%CreatedDate%"   /f >nul 2>&1
reg add "%EdgePolicyKey3rd%" /v "OrganizationID" /t REG_SZ /d "%OrganizationID%" /f >nul 2>&1
reg add "%EdgePolicyKey3rd%" /v "UserEmail"      /t REG_SZ /d "%UserEmail%"     /f >nul 2>&1
reg add "%EdgePolicyKey3rd%" /v "UserPc"         /t REG_SZ /d "%UserPc%"        /f >nul 2>&1

echo   Edge extension and policy applied.
echo.


rem ============================================================
rem 5. Output all Registry values (read from Reg)
rem ============================================================
echo [5] Registry values written:
echo.
echo   [3] Chrome ExtensionSettings:
reg query "%ChromeExtSettingsKey%" 2>nul
echo.
echo   [3] Chrome 3rdparty extension policy:
reg query "%ChromePolicyKey%" 2>nul
echo.
echo   [4] Edge ExtensionSettings:
reg query "%EdgeExtSettingsKey%" 2>nul
echo.
echo   [4] Edge 3rdparty extension policy:
reg query "%EdgePolicyKey3rd%" 2>nul
echo.
echo   regedit: HKCU ^> Software ^> Policies ^> Microsoft ^> Edge
echo           HKCU ^> Software ^> Policies ^> Google ^> Chrome
echo.


echo ==============================
echo           FINISHED
echo ==============================
echo   UserEmail = %UserEmail%
echo   UserPc    = %UserPc%
echo.

echo Press any key to exit...
pause

endlocal
exit /b