void ResourceNet::init( const KUrl &url, const QString &format )
{
d->mLoadJob = 0;
d->mIsLoading = false;
d->mSaveJob = 0;
d->mIsSaving = false;
mFormatName = format;
FormatFactory *factory = FormatFactory::self();
mFormat = factory->format( mFormatName );
if ( !mFormat ) {
mFormatName = QLatin1String( "vcard" );
mFormat = factory->format( mFormatName );
}
setUrl( url );
}
void registerOutputFormatNull(FormatFactory & factory)
{
factory.registerOutputFormat("Null", [](
WriteBuffer &,
const Block & sample,
const Context &,
const FormatSettings &)
{
return std::make_shared<NullBlockOutputStream>(sample);
});
}
void registerOutputFormatJSONCompact(FormatFactory & factory)
{
factory.registerOutputFormat("JSONCompact", [](
WriteBuffer & buf,
const Block & sample,
const Context &,
const FormatSettings & format_settings)
{
return std::make_shared<BlockOutputStreamFromRowOutputStream>(
std::make_shared<JSONCompactRowOutputStream>(buf, sample, format_settings), sample);
});
}
void registerOutputFormatTSKV(FormatFactory & factory)
{
factory.registerOutputFormat("TSKV", [](
WriteBuffer & buf,
const Block & sample,
const Context &,
const FormatSettings & settings)
{
return std::make_shared<BlockOutputStreamFromRowOutputStream>(
std::make_shared<TSKVRowOutputStream>(buf, sample, settings), sample);
});
}
void registerOutputFormatRowBinary(FormatFactory & factory)
{
factory.registerOutputFormat("RowBinary", [](
WriteBuffer & buf,
const Block & sample,
const Context &,
const FormatSettings &)
{
return std::make_shared<BlockOutputStreamFromRowOutputStream>(
std::make_shared<BinaryRowOutputStream>(buf, sample, false, false), sample);
});
factory.registerOutputFormat("RowBinaryWithNamesAndTypes", [](
WriteBuffer & buf,
const Block & sample,
const Context &,
const FormatSettings &)
{
return std::make_shared<BlockOutputStreamFromRowOutputStream>(
std::make_shared<BinaryRowOutputStream>(buf, sample, true, true), sample);
});
}
void registerInputFormatValues(FormatFactory & factory)
{
factory.registerInputFormat("Values", [](
ReadBuffer & buf,
const Block & sample,
const Context & context,
size_t max_block_size,
const FormatSettings & settings)
{
return std::make_shared<BlockInputStreamFromRowInputStream>(
std::make_shared<ValuesRowInputStream>(buf, sample, context, settings),
sample, max_block_size, settings);
});
}
void registerInputFormatCSV(FormatFactory & factory)
{
for (bool with_names : {false, true})
{
factory.registerInputFormat(with_names ? "CSVWithNames" : "CSV", [=](
ReadBuffer & buf,
const Block & sample,
const Context &,
UInt64 max_block_size,
const FormatSettings & settings)
{
return std::make_shared<BlockInputStreamFromRowInputStream>(
std::make_shared<CSVRowInputStream>(buf, sample, with_names, settings),
sample, max_block_size, settings);
});
}
}
// Main program loop
int Convertor::start()
{
out << tr("Contact convertor by Mikhail Y. Zvyozdochkin\n");
// Parse command line
if (arguments().count()<2) {
printUsage();
return 1;
}
QString inPath, outPath, outFormat;
bool forceOverwrite = false;
bool forceSingleFile = false;
bool forceDirectory = false;
for (int i=1; i<arguments().count(); i++) {
if (arguments()[i]=="-i") {
i++;
if (i==arguments().count()) {
out << tr("linvert error: -i option present, but file path is missing\n");
printUsage();
return 2;
}
inPath = arguments()[i];
continue;
}
else if (arguments()[i]=="-o") {
i++;
if (i==arguments().count()) {
out << tr("linvert error: -o option present, but file path is missing\n");
printUsage();
return 3;
}
outPath = arguments()[i];
continue;
}
else if (arguments()[i]=="-f") {
i++;
if (i==arguments().count()) {
out << tr("linvert error: -f option present, but format name is missing\n");
printUsage();
return 4;
}
outFormat = arguments()[i];
if (outFormat!="vcf21" && outFormat!="vcf30" && outFormat!="vcfauto" && outFormat!="udx" && outFormat!="copy") {
out << tr("Unknown output format: %1\n").arg(outFormat);
printUsage();
return 5;
}
continue;
}
else if (arguments()[i]=="-w")
forceOverwrite = true;
else if (arguments()[i]=="-s")
forceSingleFile = true;
else if (arguments()[i]=="-d")
forceDirectory = true;
else {
out << tr("Unknown option: %1\n").arg(arguments()[i]);
printUsage();
return 6;
}
}
// Check input data completion
if (inPath.isEmpty()) {
out << tr("Input path is missing\n");
printUsage();
return 7;
}
if (outPath.isEmpty()) {
out << tr("Output path is missing\n");
printUsage();
return 8;
}
if (outFormat.isEmpty()) {
out << tr("Output format name is missing\n");
printUsage();
return 9;
}
if (forceSingleFile && forceDirectory) {
out << tr("Options -s and -d are not compatible\n");
printUsage();
return 10;
}
if (forceDirectory && !outFormat.contains("vcf") && outFormat!="copy") {
out << tr("-d option applicable only for vCard format");
return 11;
}
// Check if output file exists
QFile of(outPath);
if (of.exists() && !forceOverwrite && !QFileInfo(outPath).isDir()) {
out << tr("Output file already exists, use -w if necessary\n");
printUsage();
return 12;
}
// Define, create file or directory at output
// (default: as input)
FormatType ift = QFileInfo(inPath).isDir() ? ftDirectory : ftFile;
FormatType oft = ift;
if (forceSingleFile)
oft = ftFile;
else if (forceDirectory)
oft = ftDirectory;
// Read
IFormat* iFormat = 0;
FormatFactory factory;
if (ift==ftFile)
iFormat = factory.createObject(inPath);
else
iFormat = new VCFDirectory();
if (!iFormat) {
out << factory.error << "\n";
return 13;
}
ContactList items;
bool res = iFormat->importRecords(inPath, items, false);
logFormat(iFormat);
delete iFormat;
if (!res)
return 14;
out << tr("%1 records read\n").arg(items.count());
//Define output format
gd.preferredVCFVersion = GlobalConfig::VCF21;
IFormat* oFormat = 0;
if (oft==ftDirectory)
oFormat = new VCFDirectory();
else if (outFormat.contains("vcf")) {
if (outFormat=="vcf30")
gd.preferredVCFVersion = GlobalConfig::VCF30;
gd.useOriginalFileVersion = (outFormat=="vcfauto");
oFormat = new VCFFile();
}
else if (outFormat.contains("udx"))
oFormat = new UDXFile();
else { // copy input format
if (VCFFile::detect(inPath)) {
gd.useOriginalFileVersion = true;
oFormat = new VCFFile();
}
else if (UDXFile::detect(inPath))
oFormat = new UDXFile();
// else if (MPBFile::detect(inPath)) { TODO m.b. implement?
else if (inPath.contains(".mpb", Qt::CaseInsensitive)) { // temp hack
out << "MPB format is read-only and incompatible with 'copy' option. Use VCF or UDX for output file\n";
return 15;
}
else {
out << "Can't autodetect input format\n";
return 16;
}
}
// Write
res = oFormat->exportRecords(outPath, items);
logFormat(oFormat);
delete oFormat;
out << "Output file successfully written\n";
return res ? 0 : 17;
}
