source: trunk/gsinstaller/gsRegistry.cpp@ 1475

#include "gsRegistry.h"

gsRegistry::gsRegistry(installManager &manager, configureFile &configFile)
  : installAgent(manager)
{ unsigned int space;

    // Create string for volume key
    volumeKey = "Software\\" + configFile.getString("CompanyName") + "\\" +
                            configFile.getString("CollectionName") + "\\" +
                            configFile.getString("CollectionVolume") + "\\" +
                            configFile.getString("CollectionVersion");

    // Create string for collection key
    collectKey = "Software\\" + configFile.getString("CompanyName") + "\\" +
                            configFile.getString("CollectionName") + "\\" +
                configFile.getString("CollectionVersion");
}

bool gsRegistry::logAction(string action, HKEY base, string path, string item, string value)
{   manager->writeString(action);
    manager->writeSeparator();
  manager->writeString(path);
    if (item != "" || value != "")
  { manager->writeSeparator();
    if (item == "")
    {   manager->writeString("{Default}");
    }
    else
    {   manager->writeString(item);
    }
        manager->writeSeparator();
    manager->writeString(value);
  }
  manager->writeString("\n");
}

bool gsRegistry::undoAction(string actionName, stringArray &params)
{ if (actionName == "RegistryCreateKey")
    {   this->destroyKey(HKEY_LOCAL_MACHINE, params[0]);
  }
  else if (actionName == "RegistrySetString")
  { this->destroyItem(HKEY_LOCAL_MACHINE, params[0], params[1]);
  }
  else
  { return false;
  }
  return true;
}

bool gsRegistry::ensureKeysExist()
{ if (!this->ensureKeyExists(HKEY_LOCAL_MACHINE, this->collectKey))
    {   return false;
  }
  return this->ensureKeyExists(HKEY_LOCAL_MACHINE, this->volumeKey);
}

bool gsRegistry::ensureKeyExists(HKEY base, string key)
{ if (!registry_keyExists(base, key.c_str()))
    {   if (!registry_ensureKeyExists(base, key.c_str()))
    {   return false;
    }
    this->logAction("RegistryCreateKey", base, key, "", "");
  }
  return true;
}

bool gsRegistry::destroyKey(HKEY base, string key)
{   registry_deletekey(base, key.c_str());
}

bool gsRegistry::storeKeyString(HKEY base, string path, string item, string value)
{   HKEY local;
    bool    reply;

  if (!registry_openkey(base, path.c_str(), &local))
    {   return false;
  }
  reply = registry_setstring(local, item.c_str(), value.c_str());
  if (reply)
  { this->logAction("RegistrySetString", base, path, item, value);
  }
  registry_closekey(local);
  return reply;
}

bool gsRegistry::destroyItem(HKEY base, string path, string item)
{ HKEY local;

    if (!registry_openkey(base, path.c_str(), &local))
  { return false;
  }
  registry_deleteitem(local, item.c_str());
  registry_closekey(local);
}

bool gsRegistry::collectionInstalled()
{   return registry_keyExists(HKEY_LOCAL_MACHINE, this->collectKey.c_str());
}

bool gsRegistry::volumeInstalled()
{   return registry_keyExists(HKEY_LOCAL_MACHINE, this->volumeKey.c_str());
}

string gsRegistry::collectKeyId()
{   return this->collectKey;
}

string gsRegistry::volumeKeyId()
{   return this->volumeKey;
}

string gsRegistry::exeKeyId(string exename)
{   string reply = "Software\\Microsoft\\Windows\\CurrentVersion\\App Paths\\";
    reply += exename;
  return reply;
}

string gsRegistry::uninstallKeyId(string collectName)
{   return "Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\" + collectName;
}

FilePath *gsRegistry::collectionPath()
{ char *path = NULL;
    FilePath *reply = NULL;
    HKEY key;

  if (registry_openkey(HKEY_LOCAL_MACHINE, this->collectKey.c_str(), &key))
  { if (registry_getstring(key, "library", &path))
    { // get leaf (\library.exe) and remove it to give a path
        char *leaf;

        leaf    = strrchr(path, '\\');
            if (leaf != NULL)
      { *leaf = '\0';
      }
      reply = new FilePath(path);
      free(path);
    }
    registry_closekey(key);
  }
  return reply;
}