void Opeke::saveFileAs ( const QString &outputFileName )
{
KSaveFile *file = new KSaveFile(outputFileName);
file->open();
m_view->saveBricks(file);
file->finalize();
file->close();
fileName = KUrl(outputFileName).fileName();
setCaption(fileName, false);
setWindowModified(false);
}
/* antlarr: KDE 4.0: make the first parameter "const QString &" */
bool KConfigINIBackEnd::writeConfigFile(QString filename, bool bGlobal, bool bMerge)
{
// is the config object read-only?
if(pConfig->isReadOnly())
return true; // pretend we wrote it
KEntryMap aTempMap;
QFile *mergeFile = (bMerge ? new QFile(filename) : 0);
bool bEntriesLeft = getEntryMap(aTempMap, bGlobal, mergeFile);
delete mergeFile;
if(bFileImmutable)
return true; // pretend we wrote it
// OK now the temporary map should be full of ALL entries.
// write it out to disk.
// Check if file exists:
int fileMode = -1;
bool createNew = true;
KDE_struct_stat buf;
if(KDE_stat(QFile::encodeName(filename), &buf) == 0)
{
if(buf.st_uid == getuid())
{
// Preserve file mode if file exists and is owned by user.
fileMode = buf.st_mode & 0777;
}
else
{
// File is not owned by user:
// Don't create new file but write to existing file instead.
createNew = false;
}
}
KSaveFile *pConfigFile = 0;
FILE *pStream = 0;
if(createNew)
{
pConfigFile = new KSaveFile(filename, 0600);
if(pConfigFile->status() != 0)
{
delete pConfigFile;
return bEntriesLeft;
}
if(!bGlobal && (fileMode == -1))
fileMode = mFileMode;
if(fileMode != -1)
{
fchmod(pConfigFile->handle(), fileMode);
}
pStream = pConfigFile->fstream();
}
else
{
// Open existing file.
// We use open() to ensure that we call without O_CREAT.
int fd = KDE_open(QFile::encodeName(filename), O_WRONLY | O_TRUNC);
if(fd < 0)
{
return bEntriesLeft;
}
pStream = KDE_fdopen(fd, "w");
if(!pStream)
{
close(fd);
return bEntriesLeft;
}
}
writeEntries(pStream, aTempMap);
if(pConfigFile)
{
bool bEmptyFile = (ftell(pStream) == 0);
if(bEmptyFile && ((fileMode == -1) || (fileMode == 0600)))
{
// File is empty and doesn't have special permissions: delete it.
::unlink(QFile::encodeName(filename));
pConfigFile->abort();
}
else
{
// Normal case: Close the file
pConfigFile->close();
}
delete pConfigFile;
}
else
{
fclose(pStream);
}
return bEntriesLeft;
}
bool KexiCSVExport::exportData(KexiDB::TableOrQuerySchema& tableOrQuery,
const Options& options, int rowCount, QTextStream *predefinedTextStream)
{
KexiDB::Connection* conn = tableOrQuery.connection();
if (!conn)
return false;
if (rowCount == -1)
rowCount = KexiDB::rowCount(tableOrQuery);
if (rowCount == -1)
return false;
//! @todo move this to non-GUI location so it can be also used via command line
//! @todo add a "finish" page with a progressbar.
//! @todo look at rowCount whether the data is really large;
//! if so: avoid copying to clipboard (or ask user) because of system memory
//! @todo OPTIMIZATION: use fieldsExpanded(true /*UNIQUE*/)
//! @todo OPTIMIZATION? (avoid multiple data retrieving) look for already fetched data within KexiProject..
KexiDB::QuerySchema* query = tableOrQuery.query();
if (!query)
query = tableOrQuery.table()->query();
KexiDB::QueryColumnInfo::Vector fields( query->fieldsExpanded( KexiDB::QuerySchema::WithInternalFields ) );
QString buffer;
KSaveFile *kSaveFile = 0;
QTextStream *stream = 0;
const bool copyToClipboard = options.mode==Clipboard;
if (copyToClipboard) {
//! @todo (during exporting): enlarge bufSize by factor of 2 when it became too small
uint bufSize = QMIN((rowCount<0 ? 10 : rowCount) * fields.count() * 20, 128000);
buffer.reserve( bufSize );
if (buffer.capacity() < bufSize) {
kdWarning() << "KexiCSVExportWizard::exportData() cannot allocate memory for " << bufSize
<< " characters" << endl;
return false;
}
}
else {
if (predefinedTextStream) {
stream = predefinedTextStream;
}
else {
if (options.fileName.isEmpty()) {//sanity
kdWarning() << "KexiCSVExportWizard::exportData(): fname is empty" << endl;
return false;
}
kSaveFile = new KSaveFile(options.fileName);
if (0 == kSaveFile->status())
stream = kSaveFile->textStream();
if (0 != kSaveFile->status() || !stream) {//sanity
kdWarning() << "KexiCSVExportWizard::exportData(): status != 0 or stream == 0" << endl;
delete kSaveFile;
return false;
}
}
}
//! @todo escape strings
#define _ERR \
delete [] isText; \
if (kSaveFile) { kSaveFile->abort(); delete kSaveFile; } \
return false
#define APPEND(what) \
if (copyToClipboard) buffer.append(what); else (*stream) << (what)
// line endings should be as in RFC 4180
#define CSV_EOLN "\r\n"
// 0. Cache information
const uint fieldsCount = query->fieldsExpanded().count(); //real fields count without internals
const QCString delimiter( options.delimiter.left(1).latin1() );
const bool hasTextQuote = !options.textQuote.isEmpty();
const QString textQuote( options.textQuote.left(1) );
const QCString escapedTextQuote( (textQuote + textQuote).latin1() ); //ok?
//cache for faster checks
bool *isText = new bool[fieldsCount];
bool *isDateTime = new bool[fieldsCount];
bool *isTime = new bool[fieldsCount];
bool *isBLOB = new bool[fieldsCount];
uint *visibleFieldIndex = new uint[fieldsCount];
// bool isInteger[fieldsCount]; //cache for faster checks
// bool isFloatingPoint[fieldsCount]; //cache for faster checks
for (uint i=0; i<fieldsCount; i++) {
KexiDB::QueryColumnInfo* ci;
const int indexForVisibleLookupValue = fields[i]->indexForVisibleLookupValue();
if (-1 != indexForVisibleLookupValue) {
ci = query->expandedOrInternalField( indexForVisibleLookupValue );
visibleFieldIndex[i] = indexForVisibleLookupValue;
}
else {
ci = fields[i];
visibleFieldIndex[i] = i;
}
isText[i] = ci->field->isTextType();
isDateTime[i] = ci->field->type()==KexiDB::Field::DateTime;
isTime[i] = ci->field->type()==KexiDB::Field::Time;
isBLOB[i] = ci->field->type()==KexiDB::Field::BLOB;
// isInteger[i] = fields[i]->field->isIntegerType()
// || fields[i]->field->type()==KexiDB::Field::Boolean;
// isFloatingPoint[i] = fields[i]->field->isFPNumericType();
}
// 1. Output column names
if (options.addColumnNames) {
for (uint i=0; i<fieldsCount; i++) {
if (i>0)
APPEND( delimiter );
if (hasTextQuote){
APPEND( textQuote + fields[i]->captionOrAliasOrName().replace(textQuote, escapedTextQuote) + textQuote );
}
else {
APPEND( fields[i]->captionOrAliasOrName() );
}
}
APPEND(CSV_EOLN);
}
KexiGUIMessageHandler handler;
KexiDB::Cursor *cursor = conn->executeQuery(*query);
if (!cursor) {
handler.showErrorMessage(conn);
_ERR;
}
for (cursor->moveFirst(); !cursor->eof() && !cursor->error(); cursor->moveNext()) {
const uint realFieldCount = QMIN(cursor->fieldCount(), fieldsCount);
for (uint i=0; i<realFieldCount; i++) {
const uint real_i = visibleFieldIndex[i];
if (i>0)
APPEND( delimiter );
if (cursor->value(real_i).isNull())
continue;
if (isText[real_i]) {
if (hasTextQuote)
APPEND( textQuote + QString(cursor->value(real_i).toString()).replace(textQuote, escapedTextQuote) + textQuote );
else
APPEND( cursor->value(real_i).toString() );
}
else if (isDateTime[real_i]) { //avoid "T" in ISO DateTime
APPEND( cursor->value(real_i).toDateTime().date().toString(Qt::ISODate)+" "
+ cursor->value(real_i).toDateTime().time().toString(Qt::ISODate) );
}
else if (isTime[real_i]) { //time is temporarily stored as null date + time...
APPEND( cursor->value(real_i).toTime().toString(Qt::ISODate) );
}
else if (isBLOB[real_i]) { //BLOB is escaped in a special way
if (hasTextQuote)
//! @todo add options to suppport other types from KexiDB::BLOBEscapingType enum...
APPEND( textQuote + KexiDB::escapeBLOB(cursor->value(real_i).toByteArray(), KexiDB::BLOBEscapeHex) + textQuote );
else
APPEND( KexiDB::escapeBLOB(cursor->value(real_i).toByteArray(), KexiDB::BLOBEscapeHex) );
}
else {//other types
APPEND( cursor->value(real_i).toString() );
}
}
APPEND(CSV_EOLN);
}
if (copyToClipboard)
buffer.squeeze();
if (!conn->deleteCursor(cursor)) {
handler.showErrorMessage(conn);
_ERR;
}
if (copyToClipboard)
kapp->clipboard()->setText(buffer, QClipboard::Clipboard);
delete [] isText;
delete [] isDateTime;
delete [] isTime;
delete [] isBLOB;
delete [] visibleFieldIndex;
if (kSaveFile) {
if (!kSaveFile->close()) {
kdWarning() << "KexiCSVExportWizard::exportData(): error close(); status == "
<< kSaveFile->status() << endl;
}
delete kSaveFile;
}
return true;
}
void KSaveFileTest::test_ksavefile()
{
QString targetFile;
{
//This will be the file we eventually write to. Yes, I know you
//should never remove the temporaryfile and then expect the filename
//to continue to be unique, but this is a test for crying out loud. :)
KTemporaryFile file;
file.setPrefix("ksavefiletest");
QVERIFY( file.open() );
targetFile = file.fileName();
}
{
//Test basic functionality
KSaveFile saveFile;
saveFile.setFileName(targetFile);
QVERIFY( saveFile.open() );
QVERIFY( !QFile::exists(targetFile) );
QTextStream ts ( &saveFile );
ts << "This is test data one.\n";
ts.flush();
QCOMPARE( saveFile.error(), QFile::NoError );
QVERIFY( !QFile::exists(targetFile) );
QVERIFY( saveFile.finalize() );
QVERIFY( QFile::exists(targetFile) );
QFile::remove(targetFile);
QVERIFY( !QFile::exists(targetFile) );
}
{
//Make sure destructor does what it is supposed to do.
{
KSaveFile saveFile;
saveFile.setFileName(targetFile);
QVERIFY( saveFile.open() );
QVERIFY( !QFile::exists(targetFile) );
}
QVERIFY( QFile::exists(targetFile) );
QFile::remove(targetFile);
QVERIFY( !QFile::exists(targetFile) );
}
{
//Test some error conditions
KSaveFile saveFile;
QVERIFY( !saveFile.open() ); //no filename
saveFile.setFileName(targetFile);
QVERIFY( saveFile.open() );
QVERIFY( !QFile::exists(targetFile) );
QVERIFY( !saveFile.open() ); //already open
QVERIFY( saveFile.finalize() );
QVERIFY( QFile::exists(targetFile) );
QVERIFY( !saveFile.finalize() ); //already finalized
QFile::remove(targetFile);
QVERIFY( !QFile::exists(targetFile) );
}
{
//Do it again, aborting this time
KSaveFile saveFile ( targetFile );
QVERIFY( saveFile.open() );
QVERIFY( !QFile::exists(targetFile) );
QTextStream ts ( &saveFile );
ts << "This is test data two.\n";
ts.flush();
QCOMPARE( saveFile.error(), QFile::NoError );
QVERIFY( !QFile::exists(targetFile) );
saveFile.abort();
QVERIFY( !QFile::exists(targetFile) );
}
QFile file ( targetFile );
QVERIFY( file.open(QIODevice::WriteOnly | QIODevice::Text) );
QVERIFY( file.setPermissions( file.permissions() | QFile::ExeUser ) );
file.close();
{
//Test how it works when the file already exists
//Also check for special permissions
KSaveFile saveFile ( targetFile );
QVERIFY( saveFile.open() );
QVERIFY( QFile::exists(targetFile) );
QFileInfo fi ( targetFile );
#ifndef Q_WS_WIN
// Windows: qt_ntfs_permission_lookup is not set by default in
// qfsfileengine_win.cpp, could change in future Qt versions.
QVERIFY( fi.permission( QFile::ExeUser ) );
#endif
QVERIFY( fi.size() == 0 );
QTextStream ts ( &saveFile );
ts << "This is test data three.\n";
ts.flush();
fi.refresh();
QVERIFY( fi.size() == 0 );
QVERIFY( saveFile.finalize() );
fi.refresh();
QVERIFY( fi.size() != 0 );
#ifndef Q_WS_WIN
QVERIFY( fi.permission( QFile::ExeUser ) );
#endif
QFile::remove(targetFile);
}
{
QFileInfo fi ( targetFile );
targetFile = fi.fileName();
QDir::setCurrent(fi.path());
//one more time, this time with relative filenames
KSaveFile saveFile ( targetFile );
QVERIFY( saveFile.open() );
QVERIFY( !QFile::exists(targetFile) );
QTextStream ts ( &saveFile );
ts << "This is test data four.\n";
ts.flush();
QCOMPARE( saveFile.error(), QFile::NoError );
QVERIFY( !QFile::exists(targetFile) );
QVERIFY( saveFile.finalize() );
QVERIFY( QFile::exists(targetFile) );
QFile::remove(targetFile);
}
}