source: trunk/gsinstaller/gsProgman.cpp@ 1537

#include "gsProgman.h"

#include <stdio.h>

// include shell for explorer-based OS'es
#include <shlobj.h>
// include dde for old (win 3.1/winNT 3.5) OS'es
#include <ddeml.h>

#include "FilePath.h"

extern HINSTANCE app_instance;

/**
 * A private function for Dde callbacks; in practice it needs to be here,
 * but does nothing whatsoever
 */
HDDEDATA CALLBACK DdeCallBack(
    UINT uType, // transaction type
    UINT uFmt,  // clipboard data format
    HCONV hconv,    // handle to the conversation
    HSZ hsz1,   // handle to a string
    HSZ hsz2,   // handle to a string
    HDDEDATA hdata, // handle to a global memory object
    DWORD dwData1,  // transaction-specific data
    DWORD dwData2   // transaction-specific data
   )
{   return NULL;
}

 /**
 * A private little function to do DDE activity
 */
bool DDEShellCommand(DWORD instance, char *command)
{   HSZ             serviceName;
    HCONV           conversation;
  int               commandLength;
  HDDEDATA  ddeData;
  DWORD         result;
  bool          reply   = false;

    // create DDE string from service request
    serviceName = DdeCreateStringHandle(instance, "PROGMAN", CP_WINANSI);
  if (serviceName   == NULL)
  { /*char buffer[20];
    sprintf(buffer, "%d %d", DdeGetLastError(instance), instance);
    MessageBox(0, buffer, "GSInstall", MB_OK);*/
    return FALSE;
  }

  // strike up a conversation with the DDE service
  if ((conversation = DdeConnect(instance, serviceName, serviceName, NULL)) != NULL)
  { commandLength   = lstrlen(command);

        // send the message via DDE
//    MessageBox(0, command, "Executing", MB_OK);
    ddeData = DdeClientTransaction((LPBYTE) command, commandLength + 1,
                                                                    conversation, NULL, CF_TEXT, XTYP_EXECUTE,
                                    1000, &result);
        // if our reply isn't NULL, it succeeded
    reply = (ddeData != NULL);
    if (reply == false)
    { /*
        char buffer[20];

        sprintf(buffer, "%d", DdeGetLastError(instance));
      MessageBox(0, buffer, "DDE Error", MB_OK);*/
    }

    // disconnect the conversation
    DdeDisconnect(conversation);
  }
  else
  { MessageBox(0, "DDE Connect failed", "GSInstall", MB_OK);
  }

  // free DDE string
  DdeFreeStringHandle(instance, serviceName);
  return reply;
}

gsProgramManager::gsProgramManager(installManager &manager)
    : installAgent(manager)
{   this->platform  = new gsPlatform();
    this->connected = false;
}

void gsProgramManager::logAction(string actionName, string group)
{ unInstallCommand command(actionName);
    command.addParameter(group);
    manager->storeCommand(command);
}

void gsProgramManager::logAction(string actionName, string group, string item, string parameter)
{ unInstallCommand command(actionName);
  command.addParameter(group);
    command.addParameter(item);
  command.addParameter(parameter);
  manager->storeCommand(command);
}

bool gsProgramManager::undoAction(string actionName, stringArray &params)
{ if (actionName == "ProgManCreateGroup")
    {   this->removeProgramGroup(params[0]);
    return true;
  }
  else if (actionName == "ProgManAddItem")
  { this->removeIcon(params[0], params[1]);
    return true;
  }
  return false;
}

bool gsProgramManager::connect()
{ HRESULT response;

    if (this->platform->isExplorerShell())
  { // Initialise the shell connection
    response = CoInitialize(NULL);
    if (response < 0)
    {   return false;
    }
  }
  else
  { FARPROC procPtr;

    procPtr = MakeProcInstance((FARPROC) DdeCallBack, app_instance);

    instance    = 0L;
    if (DdeInitialize(&instance, (PFNCALLBACK) procPtr, APPCMD_CLIENTONLY, 0L) != DMLERR_NO_ERROR)
    { MessageBox(0, "DDE failed to initialise", "GSInstall", MB_OK);
        return false;
    }
  }
  this->connected   = true;
  return true;
}

bool gsProgramManager::disconnect()
{ if (this->connected)
    {   if (this->platform->isExplorerShell())
        {   CoUninitialize();
      }
    else
      { DdeUninitialize(instance);
    }
  }
  this->connected   = false;
  return true;
}

bool gsProgramManager::addProgramGroup(string groupName)
{   //CreateGroup(groupName);
    char buffer[256];

  if (this->platform->isExplorerShell())
  { // add using shell controls
    LPITEMIDLIST pidl;

    if (SHGetSpecialFolderLocation(0, CSIDL_PROGRAMS, &pidl) == NOERROR)
    { SHGetPathFromIDList(pidl, buffer);

        // Create the folder as required
        FilePath *path  = new FilePath(buffer, groupName);
      if (!CreateDirectory((char *) path->cString(), NULL))
      { // it probably existed already - if so skip it!
        DWORD reason = GetLastError();
        if (reason != ERROR_ALREADY_EXISTS)
        {   return false;
        }
      }
    }
  }
  else
  { // add using program manager DDE
    sprintf(buffer, "[CreateGroup (%s)]", groupName.c_str());
    if (!DDEShellCommand(this->instance, buffer))
    {   // flag error by returning false
      return false;
    }
  }

  this->logAction("ProgManCreateGroup", groupName);
    return true;
}

bool gsProgramManager::addIcon( string groupName,               // group of the icon
                                                                string iconName,                // name of the icon
                                string iconDestination, // destination file of the icon
                                                            string description)         // the textual description
                                                                                // of the icon (not usually displayed)
{   bool                    reply = false;
    char                    buffer[MAX_PATH];

  if (this->platform->isExplorerShell())
  { // create item using the shell
    LPITEMIDLIST    pidl;
    HRESULT hres;
    IShellLink* psl;
    FilePath *  fullPath;

      if (SHGetSpecialFolderLocation(0, CSIDL_PROGRAMS, &pidl) != NOERROR)
    { return false;
      }

        SHGetPathFromIDList(pidl, buffer);
        fullPath    = new FilePath(4, buffer, groupName.c_str(), iconName.c_str(), ".lnk");

    // check if the shortcut already exists then don't create this again
    if (fullPath->exists())
    { delete fullPath;
        return true;
    }

        // Get a pointer to the IShellLink interface.
    hres = CoCreateInstance(CLSID_ShellLink, NULL,
                                                 CLSCTX_INPROC_SERVER, IID_IShellLink, (void **) &psl);
      if (SUCCEEDED(hres))
    { IPersistFile* ppf;

        // Set the path to the shortcut target, and add the
        // description.
      psl->SetPath(iconDestination.c_str());

        psl->SetDescription(description.c_str());

        // Query IShellLink for the IPersistFile interface for saving the
      // shortcut in persistent storage.
        hres = psl->QueryInterface(IID_IPersistFile, (void **) &ppf);

        if (SUCCEEDED(hres))
      { WCHAR wsz[MAX_PATH];

          // Ensure that the string is ANSI.
        MultiByteToWideChar(CP_ACP, 0, fullPath->cString(), -1, wsz, MAX_PATH);

          // store the link in the "persistent" file
        reply = ppf->Save(wsz, TRUE);

          // release the persistent file handle
        ppf->Release();
        }
      psl->Release();
      }
    delete fullPath;
  }
  else
  { // Back to the dark ages with win 3.1!

    // ensure that the requisite group is active; the add icon command will
    // use the active group.
    sprintf(buffer, "[ShowGroup(%s,5)]", groupName.c_str());
    if (!DDEShellCommand(this->instance, buffer))
    { // TODO: articulate/determine possible errors; expand error handling
            return false;
    }

    // and now add the item itself to the active (given) group
    sprintf(buffer, "[AddItem(%s,%s)]", iconDestination.c_str(), iconName.c_str());
    if (!DDEShellCommand(this->instance, buffer))
    {   // TODO: again work out what the problems could be
        return false;
    }
    reply = true;
  }

  if (reply)
  { this->logAction("ProgManAddItem", groupName, iconName, iconDestination);
  }
  return reply;
}

bool gsProgramManager::removeProgramGroup(string groupName)
{   //CreateGroup(groupName);
    char buffer[256];

  if (this->platform->isExplorerShell())
  { // add using shell controls
    LPITEMIDLIST pidl;

    if (SHGetSpecialFolderLocation(0, CSIDL_PROGRAMS, &pidl) == NOERROR)
    { SHGetPathFromIDList(pidl, buffer);

        // Create the folder as required
        FilePath *path  = new FilePath(buffer, groupName);
      if (!RemoveDirectory((char *) path->cString()))
      { // TODO: find cause of failure to remove
        DWORD reason = GetLastError();
        if (reason != 0)
        {   return false;
        }
      }
    }
  }
  else
  { // add using program manager DDE
    sprintf(buffer, "[DeleteGroup (%s)]", groupName.c_str());
    if (!DDEShellCommand(this->instance, buffer))
    {   // flag error by returning false
      return false;
    }
  }
    return true;
}

bool gsProgramManager::removeIcon(string groupName,             // group of the icon
                                                                    string iconName)                    // name of the icon
{   bool                    reply = false;
    char                    buffer[MAX_PATH];

  if (this->platform->isExplorerShell())
  { // create item using the shell
    LPITEMIDLIST    pidl;
    HRESULT hres;
    IShellLink* psl;
    FilePath *  fullPath;

      if (SHGetSpecialFolderLocation(0, CSIDL_PROGRAMS, &pidl) != NOERROR)
    { return false;
      }

    // get path of folder from shell
        SHGetPathFromIDList(pidl, buffer);

    // delete file
    fullPath    = new FilePath(4, buffer, groupName.c_str(), iconName.c_str(), "!.lnk");
        if (!DeleteFile(fullPath->cString()))
    { reply = false;
    }
    delete fullPath;
  }
  else
  { // Back to the dark ages with win 3.1!

    // ensure that the requisite group is active; the add icon command will
    // use the active group.
    sprintf(buffer, "[ShowGroup(%s,5)]", groupName.c_str());
    if (!DDEShellCommand(this->instance, buffer))
    { // TODO: articulate/determine possible errors; expand error handling
            return false;
    }

    // and now add the item itself to the active (given) group
    sprintf(buffer, "[DeleteItem(%s)]", iconName.c_str());
    if (!DDEShellCommand(this->instance, buffer))
    {   // TODO: again work out what the problems could be
        return false;
    }
  }
  return reply;
}