// load specified profile.
bool DataHandler::loadProfile(const std::string& pf)
{
if (currentFile != pProfiles)
throw StringException("Cannot load profile, profile list not loaded.");
std::string filename = pf + ".dat";
if (pProfile != NULL)
fclose(pProfile);
Node *n = NULL;
// Find it.
for (unsigned int i=0; i < res->size(); i++)
{
if ( res->getNode(i)->getString() == pf )
n = res->getNode(i);
}
if ( n == NULL )
throw StringException("Couldn't find that profile.");
pProfile = openFile(filename.c_str(), "r+");
if (pProfile == NULL)
{
pProfile = openFile(filename.c_str(), "w+");
}
if (pProfile == NULL)
throw StringException("Couldn't open the file for some reason...");
profile = filename;
readFile(filename, pProfile, PROFILE);
return true;
}
bool DataHandler::readFile(const std::string& filename, FILE *fp, int type)
{
currentFilename = filename;
currentFile = fp;
currentType = type;
if (currentFile == NULL)
throw StringException("Bad file pointer.");
rewind(currentFile);
if ( feof(currentFile) )
throw StringException("End of file.");
res->clear();
char read[256];
char *newline;
// Read the file.
while (fgets(read, sizeof(read), currentFile))
{
newline = strchr(read, '\n');
if (newline != NULL)
*newline = '\0';
res->addNode( read , currentType );
}
return true;
}
bool DataHandler::addProfile(const std::string& profileName)
{
if (currentFile != pProfiles)
throw StringException("Profile isn't loaded.");
for (unsigned int i=0; i < res->size(); i++)
{
if ( res->getNode(i)->getString() == profileName )
throw StringException("That profile already exists.\n");
}
return writeLineToFile(profileName, pProfiles, MAX_LINE_IN_FILE);
}
void Timestamp::update()
{
#if defined(_WIN32)
FILETIME ft;
GetSystemTimeAsFileTime(&ft);
ULARGE_INTEGER epoch; // UNIX epoch (1970-01-01 00:00:00) expressed in Windows NT FILETIME
epoch.LowPart = 0xD53E8000;
epoch.HighPart = 0x019DB1DE;
ULARGE_INTEGER ts;
ts.LowPart = ft.dwLowDateTime;
ts.HighPart = ft.dwHighDateTime;
ts.QuadPart -= epoch.QuadPart;
_ts = ts.QuadPart/10;
#else
struct timeval tv;
if (gettimeofday(&tv, NULL))
throw StringException("cannot get time of day");
_ts = TimeVal(tv.tv_sec)*resolution() + tv.tv_usec;
#endif
}
bool DataHandler::writeLineToFile(const std::string& line, FILE* file, unsigned int maxLine)
{
if (file == NULL)
{
throw StringException("Error: trying to write to NULL file.\n");
return false;
}
else if (line.length() > maxLine)
{
throw StringException("Error: Line too long.\n");
return false;
}
fseek(file, 0, SEEK_END);
printf("Writing <%s> to file...", line.c_str());
fprintf(file, "%s\n", line.c_str()); //puts(line, file);
//fputs("\n", file);
fflush(file);
return true;
}
bool DataHandler::addDatabase(const std::string& databaseString)
{
if (currentFile != pProfile)
throw StringException("Profile isn't loaded.");
ProfileNode pn;
// Utilize ProfileNode's validator
if (!pn.setString(databaseString))
throw StringException("Invalid databaseString.");
for (unsigned int i=0; i < res->size(); i++)
{
// Utilize ProfileNode's parsing of the database name.
if ( ((ProfileNode*)res->getNode(i))->getDatabase() == pn.getDatabase() )
throw StringException("That database already exists.\n");
}
return writeLineToFile(databaseString, pProfile, MAX_LINE_IN_FILE);
}
bool DataHandler::loadDatabase(std::string& db)
{
if (currentFile != pProfile)
throw StringException("Cannot load database, profile not loaded.");
std::string filename = db + ".dat";
if (pDatabase != NULL)
fclose(pDatabase);
Node *pn = NULL;
// Find it.
for (unsigned int i=0; i < res->size(); i++)
{
if ( ((ProfileNode*)res->getNode(i))->getDatabase() == db )
pn = res->getNode(i);
}
if (pn == NULL)
throw StringException("Couldn't find that database.\n");
// Try opening the file.
pDatabase = openFile(filename.c_str(), "r+");
// Try creating the file if needed.
if (pDatabase == NULL)
{
pDatabase = openFile(filename.c_str(), "w+");
}
if (pDatabase == NULL)
throw StringException("Couldn't open the file for some reason...");
database = filename;
readFile(filename, pDatabase, ((ProfileNode*)pn)->getType());
return true;
}
// load all profiles.dat to Results.
bool DataHandler::loadProfiles()
{
// Locate profiles.dat
// look in current directory only.
pProfiles = openFile("profiles.dat", "r+");
if (pProfiles == NULL)
{
pProfiles = openFile("profiles.dat", "w+");
if (pProfiles != NULL)
{
fputs("Default\n", pProfiles);
fputs("Owen\n", pProfiles);
}
}
if (pProfiles == NULL)
{
throw StringException("Problem opening / creating profiles.dat");
}
readFile("profiles.dat", pProfiles, TEXT);
return true;
}
bool DataHandler::writeToDatabase(const std::string& lineInDatabase)
{
if (currentFile != pDatabase)
throw StringException("Database isn't loaded.");
return writeLineToFile(lineInDatabase, pDatabase, MAX_LINE_IN_FILE);
}
