path: root/e-util/e-win32-defaults.c

/*
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) version 3.
 *
 * 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with the program; if not, see <http://www.gnu.org/licenses/>
 *
 *
 * Authors:
 *      Fridrich Strba <fridrich.strba@bluewin.ch>
 *
 * Copyright (C) 2010 Fridrich Strba
 *
 */

#include <windows.h>

#include "e-win32-defaults.h"
#include "e-util-private.h"

#define MAX_VALUE_NAME 4096
#define EVOBINARY "evolution.exe"
#define EUTILDLL "libeutil-0.dll"
#define CANONICALNAME "Evolution"

static gchar *
_e_win32_sanitize_path (gchar *path)
{
    gchar *p = path;
    while (*p != '\0') {
        if (G_IS_DIR_SEPARATOR (*p))
            *p = G_DIR_SEPARATOR;
        p = g_utf8_find_next_char (p, NULL);
    }
    return path;
}

static void
_e_register_mailto_structure (HKEY hKey)
{
    LONG returnValue;
    DWORD dwDisposition;
    gchar *defaultIcon = NULL;
    gchar *evolutionBinary = NULL;
    gchar *mailtoCommand = NULL;
    BYTE editFlags[4] = { 0x02, 0x00, 0x00, 0x00 };
    
    static HKEY tmp_subkey = (HKEY) INVALID_HANDLE_VALUE;
    
    if ((returnValue = RegSetValueExA (hKey, NULL, 0, REG_SZ, (const BYTE *)"URL:MailTo Protocol", strlen ("URL:MailTo Protocol") + 1)))
        return;
    if ((returnValue = RegSetValueExA (hKey, "EditFlags", 0, REG_BINARY, editFlags, G_N_ELEMENTS (editFlags))))
        return;
    if ((returnValue = RegSetValueExA (hKey, "URL Protocol", 0, REG_SZ, (const BYTE *)"", strlen ("") + 1)))
        return;

    RegFlushKey (hKey);

    if ((returnValue = RegCreateKeyExA (hKey, "DefaultIcon", 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &tmp_subkey, &dwDisposition)))
        return;

    evolutionBinary = _e_win32_sanitize_path (g_build_path (G_DIR_SEPARATOR_S, _e_get_bindir (), EVOBINARY, NULL));
    defaultIcon = g_strconcat (evolutionBinary, ",1", NULL);
    g_free (evolutionBinary);
    if ((returnValue = RegSetValueExA (tmp_subkey, NULL, 0, REG_SZ, (const BYTE *)defaultIcon, strlen (defaultIcon) + 1))) {
        g_free (defaultIcon);
        return;
    }
    g_free (defaultIcon);
    
    RegFlushKey (tmp_subkey);
    
    RegCloseKey (tmp_subkey);
    
    if ((returnValue = RegCreateKeyExA (hKey, "shell\\open\\command", 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &tmp_subkey, &dwDisposition)))
        return;

    evolutionBinary = _e_win32_sanitize_path (g_build_path (G_DIR_SEPARATOR_S, _e_get_bindir (), EVOBINARY, NULL));
    mailtoCommand = g_strconcat("\"", evolutionBinary,  "\" --component=mail mailto:\%1", NULL);
    g_free (evolutionBinary);
    if ((returnValue = RegSetValueExA (tmp_subkey, NULL, 0, REG_SZ, (const BYTE *)mailtoCommand, strlen(mailtoCommand) + 1))) {
        g_free (mailtoCommand);
        return;
    }
    g_free (mailtoCommand);
    
    RegFlushKey (tmp_subkey);
    
    RegCloseKey (tmp_subkey);
}

static void
_e_win32_register_mailer_impl (WINBOOL system)
{
    LONG returnValue;
    DWORD i, dwDisposition;
    gchar *defaultIcon = NULL;
    gchar *dllPath = NULL;
    gchar *dllShortPath = NULL;
    DWORD dllShortPathLength;
    gchar *evolutionBinary = NULL;
    gchar *openCommand = NULL;
    gchar *setDefaultCommand = NULL;
    gchar *unsetDefaultCommand = NULL;

    static HKEY reg_key = (HKEY) INVALID_HANDLE_VALUE;
    static HKEY reg_subkey = (HKEY) INVALID_HANDLE_VALUE;


    if ((returnValue = RegCreateKeyExA (system ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER,
        "Software\\Clients\\Mail\\" CANONICALNAME, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &reg_key, &dwDisposition)))
        return;
    
    if ((returnValue = RegSetValueExA (reg_key, NULL, 0, REG_SZ, (const BYTE *)CANONICALNAME, strlen(CANONICALNAME) + 1)))
        return;
    
    dllPath = _e_win32_sanitize_path (g_build_path(G_DIR_SEPARATOR_S, _e_get_bindir (), EUTILDLL, NULL));
    if ((returnValue = RegSetValueExA (reg_key, "DLLPath", 0, REG_SZ, (const BYTE *)dllPath, strlen (dllPath) + 1))) {
        g_free (dllPath);
        return;
    }
    g_free(dllPath);

    RegFlushKey (reg_key);
    
    if ((returnValue = RegCreateKeyExA (reg_key, "DefaultIcon", 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &reg_subkey, &dwDisposition)))
        return;
    
    evolutionBinary = _e_win32_sanitize_path (g_build_path (G_DIR_SEPARATOR_S, _e_get_bindir (), EVOBINARY, NULL));
    defaultIcon = g_strconcat(evolutionBinary, ",0", NULL);
    g_free (evolutionBinary);
    if ((returnValue = RegSetValueExA (reg_subkey, NULL, 0, REG_SZ, (const BYTE *)defaultIcon, strlen (defaultIcon) + 1))) {
        g_free (defaultIcon);
        return;
    }
    
    RegFlushKey (reg_subkey);
    
    RegCloseKey (reg_subkey);
    
    if ((returnValue = RegCreateKeyExA (reg_key, "shell\\open\\command", 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &reg_subkey, &dwDisposition)))
        return;

    evolutionBinary = _e_win32_sanitize_path (g_build_path (G_DIR_SEPARATOR_S, _e_get_bindir (), EVOBINARY, NULL));
    openCommand = g_strconcat("\"", evolutionBinary, "\" -component=mail", NULL);
    g_free (evolutionBinary);
    if ((returnValue = RegSetValueExA (reg_subkey, NULL, 0, REG_SZ, (const BYTE *)openCommand, strlen (openCommand) + 1))) {
        g_free (openCommand);
        return;
    }
    g_free (openCommand);
    
    RegFlushKey (reg_subkey);
    
    RegCloseKey (reg_subkey);

    if ((returnValue = RegCreateKeyExA (reg_key, "InstallInfo", 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &reg_subkey, &dwDisposition)))
        return;
    
    dllPath = _e_win32_sanitize_path (g_build_path(G_DIR_SEPARATOR_S, _e_get_bindir (), EUTILDLL, NULL));
    dllShortPathLength = GetShortPathNameA (dllPath, NULL, 0);
    dllShortPath = g_new0 (char, dllShortPathLength);
    GetShortPathNameA (dllPath, dllShortPath, dllShortPathLength);
    g_free (dllPath);
    
    setDefaultCommand = g_strconcat ("rundll32 ", dllShortPath, ",_e_win32_set_default_mailer", NULL);
    unsetDefaultCommand = g_strconcat ("rundll32 ", dllShortPath, ",_e_win32_set_default_mailer", NULL);
    g_free (dllShortPath);

    if ((returnValue = RegSetValueExA (reg_subkey, "ReinstallCommand", 0, REG_SZ, (const BYTE *)setDefaultCommand, strlen (setDefaultCommand) + 1))) {
        g_free (setDefaultCommand);
        g_free (unsetDefaultCommand);
        return;
    }
    
    if ((returnValue = RegSetValueExA (reg_subkey, "ShowIconsCommand", 0, REG_SZ, (const BYTE *)setDefaultCommand, strlen (setDefaultCommand) + 1))) {
        g_free (setDefaultCommand);
        g_free (unsetDefaultCommand);
        return;
    }

    if ((returnValue = RegSetValueExA (reg_subkey, "HideIconsCommand", 0, REG_SZ, (const BYTE *)unsetDefaultCommand, strlen (unsetDefaultCommand) + 1))) {
        g_free (setDefaultCommand);
        g_free (unsetDefaultCommand);
        return;
    }

    g_free (setDefaultCommand);
    g_free (unsetDefaultCommand);
    
    i = 1;  
    if ((returnValue = RegSetValueExA (reg_subkey, "IconsVisible", 0, REG_DWORD, (BYTE*)&i, sizeof (i))))
        return;
    
    RegFlushKey (reg_subkey);
    
    RegCloseKey (reg_subkey);
    
    if ((returnValue = RegCreateKeyExA (reg_key, "Protocols\\mailto", 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &reg_subkey, &dwDisposition)))
        return;

    if ((returnValue = RegSetValueExA (reg_key, NULL, 0, REG_SZ, (const BYTE *)CANONICALNAME, strlen (CANONICALNAME) + 1)))
        return;
    
    _e_register_mailto_structure (reg_subkey);

    RegCloseKey (reg_subkey);
    
    RegCloseKey (reg_key);
}

void
_e_win32_register_mailer (void)
{
    _e_win32_register_mailer_impl (TRUE);
    _e_win32_register_mailer_impl (FALSE);
}

static void
_e_win32_set_default_mailer_impl (WINBOOL system)
{
    LONG returnValue;
    DWORD dwDisposition;
    
    const char client_message[] = "Software\\Clients\\Mail";
    const char mailto_message[] = "Software\\Classes\\mailto";

    static HKEY reg_key = (HKEY) INVALID_HANDLE_VALUE;

    _e_win32_register_mailer_impl (system);
    
    if ((returnValue = RegCreateKeyExA (system ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER,
        client_message, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &reg_key, &dwDisposition)))
        return;
    
    if ((returnValue = RegSetValueExA (reg_key, NULL, 0, REG_SZ, (const BYTE *)CANONICALNAME, strlen (CANONICALNAME) + 1)))
        return;
    
    RegFlushKey (reg_key);

    RegCloseKey (reg_key);
    
    if ((returnValue = RegCreateKeyExA (system ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER,
        mailto_message, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &reg_key, &dwDisposition)))
        return;

    _e_register_mailto_structure (reg_key);
    
    RegFlushKey (reg_key);

    RegCloseKey (reg_key);
    
    SendMessageA (HWND_BROADCAST, WM_SETTINGCHANGE, 0, (LPARAM)&client_message);
    SendMessageA (HWND_BROADCAST, WM_SETTINGCHANGE, 0, (LPARAM)&mailto_message);
}

void
_e_win32_set_default_mailer (void)
{
    _e_win32_set_default_mailer_impl (TRUE);
    _e_win32_set_default_mailer_impl (FALSE);
}

void
_e_win32_unset_default_mailer (void)
{
    /* TODO: implement */
}