#include "stdafx.h"
#include <LMCONS.H>     // 97/07/22 vtan: for UNLEN

#pragma hdrstop

BOOL   OnUpgradeDisableActiveDesktopFeatures();

void CreateMyCurHomeComponent(BOOL fChecked)
{
    // If there is a policy to prevent adding desktop components, then we should not attempt to add.
    // Otherwise, a new user gets an error message when they login for the first time.
    // Bug #21300 -- Fixed on 2/28/2001 -- Sankar
    if (SHRestricted(REST_NOADDDESKCOMP))
        return;
        
    //Add the base components!
    TCHAR szBuf[MAX_PATH];
    ISubscriptionMgr * psm;

    // Add a component that points to "about:home"
    LoadString(HINST_THISDLL, IDS_MY_CURRENT_HOMEPAGE, szBuf, ARRAYSIZE(szBuf));

//  98/07/14 vtan #176721: Changed the following to pass default component
//  positions to AddRemoveDesktopComponentNoUI so that the restored position may
//  be set to the default component position.

    AddRemoveDesktopComponentNoUI(TRUE, AD_APPLY_SAVE, MY_HOMEPAGE_SOURCE, szBuf, COMP_TYPE_WEBSITE, COMPONENT_DEFAULT_LEFT, COMPONENT_DEFAULT_TOP, COMPONENT_DEFAULT_WIDTH, COMPONENT_DEFAULT_HEIGHT, fChecked, IS_SPLIT);
    if (SUCCEEDED(CoCreateInstance(CLSID_SubscriptionMgr, NULL,
                CLSCTX_INPROC_SERVER, IID_ISubscriptionMgr, (void**)&psm)))
    {
        WCHAR wszName[MAX_PATH];
        //We need to zero init this structure except the cbSize field.
        SUBSCRIPTIONINFO siDefault = {sizeof(SUBSCRIPTIONINFO)};

        SHTCharToUnicode(szBuf, wszName, ARRAYSIZE(wszName));

        //This field is already initialized above.
        //siDefault.cbSize = sizeof(siDefault);
        psm->CreateSubscription(NULL, MY_HOMEPAGE_SOURCEW, wszName, CREATESUBS_NOUI, SUBSTYPE_DESKTOPURL, &siDefault);
        psm->Release();
    }
}

STDAPI CDeskHtmlProp_RegUnReg(BOOL bReg)
{
    HKEY    hKey;
    DWORD   userNameSize;
    TCHAR   szDeskcomp[MAX_PATH];
    TCHAR   userName[UNLEN];

//  98/07/22 vtan #202707: Problem: This code gets called for the first time when NT
//  runs. This sets up the default user profile. Anything that is added to this profile
//  is propagated to any current user when upgrading from NT 4.0 to NT 5.0. This
//  causes the DeskHtmlVersion, DeskHtmlMinorVersion and Component\0 to be replaced
//  with the default component. By replacing the version registry entries any old
//  components will not get correctly upgraded by the component reading code. It also
//  destroys the first component.

//  Solution: Prevent a default component being added at setup time by checking the
//  logged on user is "SYSTEM". If the user is anybody other than system then perform
//  the update or addition of the default component.

    userNameSize = ARRAYSIZE(userName);
    if ((GetUserName(userName, &userNameSize) != 0) && (lstrcmp(userName, TEXT("SYSTEM")) == 0))
        return(S_OK);           // an ungracious exit right here and now!

    if(bReg)
    {
        DWORD dwDisposition;
        DWORD dwDeskHtmlVersion = 0;
        DWORD dwDeskHtmlMinorVersion = 0;
        DWORD dwType;

        GetRegLocation(szDeskcomp, ARRAYSIZE(szDeskcomp), REG_DESKCOMP_COMPONENTS, NULL);

        if (ERROR_SUCCESS == RegCreateKeyEx(HKEY_CURRENT_USER, szDeskcomp, 
                                            0, NULL, 0, KEY_CREATE_SUB_KEY|KEY_QUERY_VALUE, NULL, &hKey, 
                                            &dwDisposition))
        {
            //Get the version stamp from the registry
            if(dwDisposition == REG_OPENED_EXISTING_KEY)
            {
                DWORD dwDataLength = sizeof(DWORD);
                SHQueryValueEx(hKey, REG_VAL_COMP_VERSION, NULL, &dwType, (LPBYTE)(&dwDeskHtmlVersion), &dwDataLength);
                SHQueryValueEx(hKey, REG_VAL_COMP_MINOR_VERSION, NULL, &dwType, (LPBYTE)(&dwDeskHtmlMinorVersion), &dwDataLength);
            }

            //We need to close this key before we delete it
            RegCloseKey(hKey);

            // If this branch is already there, don't set default comp.
            // Note: The differences between IE4_DESKHTML_VERSION and CUR_DESKHTML_VERSION are 
            // automatically taken care of when we read the components. So, we need to check only
            // for very old versions here.
            if (dwDeskHtmlVersion < IE4_DESKHTML_VERSION)
            {
                //Delete the existing components.
                SHDeleteKey(HKEY_CURRENT_USER, szDeskcomp);

                // Create the default active desktop configuration
                if(RegCreateKeyEx(HKEY_CURRENT_USER, szDeskcomp, 0, NULL, 0,
                            (KEY_CREATE_SUB_KEY | KEY_SET_VALUE), NULL, &hKey, &dwDisposition) == ERROR_SUCCESS)
                {
                    //We need an initial state
                    DWORD dw;

                    dw = CUR_DESKHTML_VERSION;
                    RegSetValueEx(hKey, REG_VAL_COMP_VERSION, 0, REG_DWORD, (LPBYTE)&dw, sizeof(dw));

                    dw = CUR_DESKHTML_MINOR_VERSION;
                    RegSetValueEx(hKey, REG_VAL_COMP_MINOR_VERSION, 0, REG_DWORD, (LPBYTE)&dw, sizeof(dw));
                    
                    dw = COMPSETTING_ENABLE;
                    RegSetValueEx(hKey, REG_VAL_COMP_SETTINGS, 0, REG_DWORD, (LPBYTE)&dw, sizeof(dw));

                    //Add the home page component
                    CreateMyCurHomeComponent(FALSE); //For millennium we want to disable the component.

                    RegCloseKey(hKey);
                }

                // Create the default active desktop safemode configuration
                if(RegCreateKeyEx(HKEY_CURRENT_USER, REG_DESKCOMP_SAFEMODE,
                        0, NULL, 0, KEY_CREATE_SUB_KEY, NULL, &hKey, &dwDisposition) == ERROR_SUCCESS)
                {
                    DWORD dwDisposition;
                    HKEY hKey2;

                    if(RegCreateKeyEx(hKey, REG_DESKCOMP_GENERAL_SUFFIX, 0, NULL, 0,
                                (KEY_CREATE_SUB_KEY | KEY_SET_VALUE), NULL, &hKey2, &dwDisposition) == ERROR_SUCCESS)
                    {
                        TCHAR szSafeMode[MAX_PATH];
                        GetWindowsDirectory(szSafeMode, ARRAYSIZE(szSafeMode));

                        HRESULT hr = StringCchCat(szSafeMode, ARRAYSIZE(szSafeMode), DESKTOPHTML_DEFAULT_SAFEMODE);
                        if (SUCCEEDED(hr))
                        {
                            // Show safemode.htx
                            SHRegSetPath(hKey2, NULL, REG_VAL_GENERAL_WALLPAPER, szSafeMode, 0);
                            // Don't bring up the gallery dialog box
                            dwDisposition = 0;
                            RegSetValueEx(hKey2, REG_VAL_GENERAL_VISITGALLERY, 0, REG_DWORD, (LPBYTE)&dwDisposition, sizeof(dwDisposition));
                        }
                        RegCloseKey(hKey2);
                    }
                    
                    RegCloseKey(hKey);
                }

                // Create the default scheme key
                if(RegCreateKeyEx(HKEY_CURRENT_USER, REG_DESKCOMP_SCHEME, 0, NULL, 0,
                        (KEY_CREATE_SUB_KEY | KEY_SET_VALUE), NULL, &hKey, &dwDisposition) == ERROR_SUCCESS)
                {
                    RegSetValueEx(hKey, REG_VAL_SCHEME_EDIT, 0, REG_SZ, (LPBYTE)TEXT(""), sizeof(TCHAR));
                    RegSetValueEx(hKey, REG_VAL_SCHEME_DISPLAY, 0, REG_SZ, (LPBYTE)TEXT(""), sizeof(TCHAR));
                    RegCloseKey(hKey);
                }

                // Set the components to be dirty sothat we re-generate desktop.htm
                // the first boot after installing IE4.0.
                SetDesktopFlags(COMPONENTS_DIRTY, COMPONENTS_DIRTY);

                GetRegLocation(szDeskcomp, ARRAYSIZE(szDeskcomp), REG_DESKCOMP_GENERAL, NULL);
                if (ERROR_SUCCESS == RegCreateKeyEx(HKEY_CURRENT_USER, szDeskcomp, 
                                                    0, NULL, 0, KEY_CREATE_SUB_KEY, NULL, &hKey, &dwDisposition))
                {
                    RegCloseKey(hKey);
                }
            }
            else
            {
                //See if we are upgrading from an older version like IE4.
                if (dwDeskHtmlVersion < CUR_DESKHTML_VERSION)
                {
                    // If so, save the DESKHTML_VERSION we are upgrading from.
                    // We use this later in SHGetSetSettings to decide if active desktop is ON/OFF.
                    // NOTE: The "UpgradedFrom" value is at "...\Desktop" and NOT at "..\Desktop\Components"
                    // This is because the "Components" key gets destroyed very often.
                    SHSetValue(HKEY_CURRENT_USER, REG_DESKCOMP, REG_VAL_COMP_UPGRADED_FROM,
                                REG_DWORD, (LPBYTE)&dwDeskHtmlVersion, sizeof(dwDeskHtmlVersion));
                }
                 // The major version numbers match. So check if the minor version numbers 
                // match too!
                if(dwDeskHtmlMinorVersion < CUR_DESKHTML_MINOR_VERSION)
                {
                    //Update the new Minor version number!
                    if(RegCreateKeyEx(HKEY_CURRENT_USER, szDeskcomp, 0, NULL, 0,
                            (KEY_CREATE_SUB_KEY | KEY_SET_VALUE), NULL, &hKey, &dwDisposition) == ERROR_SUCCESS)
                    {
                        DWORD   dw;
                        
                        dw = CUR_DESKHTML_MINOR_VERSION;
                        RegSetValueEx(hKey, REG_VAL_COMP_MINOR_VERSION, 0, REG_DWORD, (LPBYTE)&dw, sizeof(dw));
                        
                        RegCloseKey(hKey);
                    }
                    
                    // Add the new home page component
                    if((dwDeskHtmlVersion <= 0x10f) && (dwDeskHtmlMinorVersion <= 0x0001))
                        CreateMyCurHomeComponent(FALSE);

                    // 
                    // If this is an upgrade from W2K or earlier, we need to check if the
                    // active desktop is OFF. If so, we need to turn off all the desktop components
                    // sothat the active desktop continues to be OFF.
                    if((dwDeskHtmlVersion <= NT5_DESKHTML_VERSION) && 
                       (dwDeskHtmlMinorVersion <= NT5_DESKHTML_MINOR_VERSION))
                    {
                        OnUpgradeDisableActiveDesktopFeatures();
                    }
                    
                    // Minor version numbers do not match. So, set the dirty bit to force
                    // the regeneration of desktop.htt later when needed.
                    SetDesktopFlags(COMPONENTS_DIRTY, COMPONENTS_DIRTY);

//  98/07/16 vtan #176721/#202707: Added the following code to delete HKCU\Software\
//  Microsoft\Internet Explorer\Desktop\General\ComponentsPositioned because in NT 4.0
//  with IE 4.0 SP1 this registry entry is incorrectly incremented when the components
//  are iterated rather than when they are positioned. This resets the counter
//  in NT 5.0 where the bug has been fixed.

                    DWORD dw;
                    
                    GetRegLocation(szDeskcomp, ARRAYSIZE(szDeskcomp), REG_DESKCOMP_GENERAL, NULL);
                    if (ERROR_SUCCESS == RegCreateKeyEx(HKEY_CURRENT_USER, szDeskcomp, 0, 
                                                        NULL, 0, KEY_SET_VALUE, NULL, &hKey, &dw))
                    {
                        (LONG)RegDeleteValue(hKey, REG_VAL_GENERAL_CCOMPPOS);
                        (LONG)RegCloseKey(hKey);
                    }
                }
            }
        }
    }
    else
    {
        SHDeleteKey(HKEY_LOCAL_MACHINE, c_szRegDeskHtmlProp);
        SHDeleteKey(HKEY_CURRENT_USER, REG_DESKCOMP_COMPONENTS_ROOT);
        SHDeleteKey(HKEY_CURRENT_USER, REG_DESKCOMP_GENERAL_ROOT);
        SHDeleteKey(HKEY_CURRENT_USER, REG_DESKCOMP_SAFEMODE);
        SHDeleteKey(HKEY_CURRENT_USER, REG_DESKCOMP_SCHEME);
    }

    return S_OK;
}