123 lines
3 KiB
C++
123 lines
3 KiB
C++
|
/////////////////////////////////////////////////////////////////////
|
||
|
// DBconvert - A simple command line tool to convert between the two
|
||
|
// prompt database formats.
|
||
|
/////////////////////////////////////////////////////////// SS //////
|
||
|
|
||
|
#include <stdlib.h>
|
||
|
#include <stdio.h>
|
||
|
#include <atlbase.h>
|
||
|
#include <wchar.h>
|
||
|
|
||
|
#include "msprompteng.h"
|
||
|
#include "msprompteng_i.c"
|
||
|
|
||
|
void Error(const WCHAR* msg)
|
||
|
{
|
||
|
wprintf(L"ERROR - %s\n", msg);
|
||
|
}
|
||
|
|
||
|
// TODO: should use multibyte text convert.
|
||
|
// a quick hack to convert commandline arguments to WCHAR's.
|
||
|
WCHAR* PlainToWide(const char* str)
|
||
|
{
|
||
|
WCHAR* tmp;
|
||
|
unsigned int i;
|
||
|
tmp = (WCHAR*) malloc(sizeof(WCHAR) * (strlen(str) + 1));
|
||
|
|
||
|
for (i = 0; i < strlen(str); i++)
|
||
|
tmp[i] = (WCHAR) str[i];
|
||
|
tmp[i] = 0;
|
||
|
|
||
|
return tmp;
|
||
|
}
|
||
|
|
||
|
bool FileExists(WCHAR* filename)
|
||
|
{
|
||
|
FILE* fp;
|
||
|
fp = _wfopen(filename, L"rb");
|
||
|
|
||
|
if (fp) {
|
||
|
fclose(fp);
|
||
|
return true;
|
||
|
}
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
void main(int argc, char** argv)
|
||
|
{
|
||
|
IPromptDb* db;
|
||
|
WCHAR* source;
|
||
|
WCHAR* destination;
|
||
|
HRESULT hr = E_FAIL;
|
||
|
int f_convertToNew = 0;
|
||
|
int f_convertToOld = 0;
|
||
|
|
||
|
if (argc == 4) {
|
||
|
|
||
|
// process command line arguments
|
||
|
f_convertToNew = (0 == strcmp("-new", argv[1]));
|
||
|
f_convertToOld = (0 == strcmp("-old", argv[1]));
|
||
|
|
||
|
source = PlainToWide(argv[2]);
|
||
|
destination = PlainToWide(argv[3]);
|
||
|
|
||
|
// check for file existence
|
||
|
if (! FileExists(source)) {
|
||
|
wprintf(L"ERROR - %s doesn't exist\n", source);
|
||
|
} else if (FileExists(destination)) {
|
||
|
wprintf(L"ERROR - %s exists\n", destination);
|
||
|
} else {
|
||
|
|
||
|
if (f_convertToNew || f_convertToOld) {
|
||
|
wprintf(L"Prompt Engine Database Conversion ");
|
||
|
|
||
|
if (SUCCEEDED(::CoInitialize(NULL))) {
|
||
|
if (SUCCEEDED(::CoCreateInstance(CLSID_PromptDb, NULL, CLSCTX_ALL, IID_IPromptDb, (void**) &db))) {
|
||
|
|
||
|
// TODO: I should use DB_LOAD_WITH_WRITE defined in CPromptDb.
|
||
|
|
||
|
// I use IPromptDb directly instead of MsPromptEng, and so some enums are invisible
|
||
|
// with simply including promptdb.h .
|
||
|
if (SUCCEEDED(db->AddDb(L"DEFAULT", source, 2 /* = DB_LOAD_WITH_WRITE */))) { // read database
|
||
|
db->ActivateDbName(L"DEFAULT");
|
||
|
|
||
|
wprintf(L"%s -> %s ", source, destination);
|
||
|
|
||
|
if (f_convertToNew)
|
||
|
hr = db->UpdateDb(destination); // save to new format
|
||
|
else
|
||
|
hr = db->UpdateDbOldFormat(destination); // save to old format
|
||
|
|
||
|
if (SUCCEEDED(hr))
|
||
|
wprintf(L"success\n");
|
||
|
else
|
||
|
wprintf(L"failure\n");
|
||
|
} else {
|
||
|
Error(L"bad database format");
|
||
|
}
|
||
|
|
||
|
db->Release();
|
||
|
} else {
|
||
|
Error(L"cannot load PromptDb COM object");
|
||
|
}
|
||
|
|
||
|
::CoUninitialize();
|
||
|
} else {
|
||
|
Error(L"cannot initialize COM library");
|
||
|
}
|
||
|
} else {
|
||
|
printf("ERROR - illegal switch \"%s\". use \"-new\" or \"-old\".\n", argv[1]);
|
||
|
}
|
||
|
|
||
|
}
|
||
|
|
||
|
free(source);
|
||
|
free(destination);
|
||
|
} else {
|
||
|
// help message
|
||
|
wprintf(L"DBconvert usage:\n");
|
||
|
wprintf(L"dbconvert -new <source (new/old)> <destination (new)>\n");
|
||
|
wprintf(L"dbconvert -old <source (new/old)> <destination (old)>\n");
|
||
|
}
|
||
|
}
|