mirror of
https://github.com/ultimatepp/ultimatepp.git
synced 2026-05-15 22:02:49 -06:00
392 lines
10 KiB
C++
392 lines
10 KiB
C++
#include "SqlCtrl.h"
|
|
#include "SqlDlg.h"
|
|
|
|
NAMESPACE_UPP
|
|
|
|
class DlgSqlExport : public WithSqlExportLayout<TopWindow>
|
|
{
|
|
public:
|
|
typedef DlgSqlExport CLASSNAME;
|
|
DlgSqlExport();
|
|
|
|
void Run(Sql& cursor, String command, String tablename);
|
|
void Serialize(Stream& stream);
|
|
|
|
private:
|
|
void SyncUI();
|
|
void Toggle();
|
|
|
|
private:
|
|
ArrayOption exp;
|
|
enum {
|
|
FMT_TEXT,
|
|
FMT_SQL,
|
|
};
|
|
enum {
|
|
DELIM_TAB,
|
|
DELIM_SEMICOLON,
|
|
};
|
|
String recent_file;
|
|
};
|
|
|
|
void RunDlgSqlExport(Sql& cursor, String command, String tablename)
|
|
{
|
|
DlgSqlExport().Run(cursor, command, tablename);
|
|
}
|
|
|
|
DlgSqlExport::DlgSqlExport()
|
|
{
|
|
CtrlLayoutOKCancel(*this, "");
|
|
columns.AddColumn(t_("Column name"));
|
|
columns.AddColumn(t_("Type"));
|
|
columns.AddColumn(t_("Width"));
|
|
HeaderCtrl::Column& hc = exp.AddColumn(columns, "").Margin(0).HeaderTab();
|
|
hc.SetMargin(0).Fixed(18).SetAlign(ALIGN_CENTER).SetImage(CtrlImg::smallcheck());
|
|
hc.WhenAction = THISBACK(Toggle);
|
|
format <<= FMT_TEXT;
|
|
delimiters <<= DELIM_SEMICOLON;
|
|
quote <<= true;
|
|
format <<= THISBACK(SyncUI);
|
|
}
|
|
|
|
void DlgSqlExport::SyncUI()
|
|
{
|
|
int f = ~format;
|
|
delimiters.Enable(f == FMT_TEXT);
|
|
quote.Enable(f == FMT_TEXT);
|
|
identity_insert.Enable(f == FMT_SQL);
|
|
object_name.Enable(f == FMT_SQL);
|
|
}
|
|
|
|
void DlgSqlExport::Toggle()
|
|
{
|
|
bool state = true;
|
|
if(columns.IsCursor())
|
|
state = !(bool)columns.Get(3);
|
|
for(int i = 0; i < columns.GetCount(); i++)
|
|
columns.Set(i, 3, state);
|
|
}
|
|
|
|
void DlgSqlExport::Run(Sql& cursor, String command, String tablename)
|
|
{
|
|
Title(Nvl(tablename, t_("SQL query")) + t_(" export"));
|
|
object_name <<= tablename;
|
|
if(!cursor.Execute(command)) {
|
|
Exclamation(NFormat(t_("Error executing [* \1%s\1]: \1%s"), command, cursor.GetLastError()));
|
|
return;
|
|
}
|
|
for(int i = 0; i < cursor.GetColumns(); i++) {
|
|
const SqlColumnInfo& sci = cursor.GetColumnInfo(i);
|
|
String type;
|
|
switch(sci.type) {
|
|
case BOOL_V:
|
|
case INT_V: type = t_("integer"); break;
|
|
case DOUBLE_V: type = t_("real number"); break;
|
|
case STRING_V:
|
|
case WSTRING_V: type = t_("string"); break;
|
|
case DATE_V: type = t_("date"); break;
|
|
case TIME_V: type = t_("date/time"); break;
|
|
case /*ORA_BLOB_V*/-1: type = t_("BLOB"); break;
|
|
case /*ORA_CLOB_V*/-2: type = t_("CLOB"); break;
|
|
default: type = FormatInt(sci.type); break;
|
|
}
|
|
columns.Add(sci.name, sci.type, sci.width, 1);
|
|
}
|
|
static String cfg;
|
|
LoadFromString(*this, cfg);
|
|
SyncUI();
|
|
while(TopWindow::Run() == IDOK)
|
|
try {
|
|
String out_table = ~object_name;
|
|
String delim;
|
|
switch((int)~delimiters) {
|
|
case DELIM_TAB: delim = "\t"; break;
|
|
case DELIM_SEMICOLON: delim = ";"; break;
|
|
}
|
|
Vector<int> out;
|
|
String colstr;
|
|
String title;
|
|
for(int i = 0; i < columns.GetCount(); i++)
|
|
if(columns.Get(i, 3)) {
|
|
out.Add(i);
|
|
String cname = cursor.GetColumnInfo(i).name;
|
|
colstr << (i ? ", " : "") << cname;
|
|
if(i) title << delim;
|
|
title << cname;
|
|
}
|
|
if(out.IsEmpty()) {
|
|
throw Exc(t_("No columns selected!"));
|
|
continue;
|
|
}
|
|
String rowbegin, rowend;
|
|
int fmt = ~format;
|
|
FileSel fsel;
|
|
String ext;
|
|
switch(fmt) {
|
|
case FMT_TEXT: {
|
|
rowend = "";
|
|
ext = ".txt";
|
|
fsel.Type(t_("Text files (*.txt)"), "*.txt");
|
|
break;
|
|
}
|
|
case FMT_SQL: {
|
|
if(identity_insert)
|
|
rowbegin << "set identity_insert " << out_table << " on ";
|
|
rowbegin << "insert into " << out_table << "(" << colstr << ") values (";
|
|
rowend = ");";
|
|
ext = ".sql";
|
|
fsel.Type(t_("SQL scripts (*.sql)"), "*.sql");
|
|
break;
|
|
}
|
|
}
|
|
fsel.AllFilesType().DefaultExt(ext.Mid(1));
|
|
if(!IsNull(recent_file))
|
|
fsel <<= ForceExt(recent_file, ext);
|
|
if(!fsel.ExecuteSaveAs(t_("Save export as")))
|
|
continue;
|
|
recent_file = ~fsel;
|
|
FileOut fo;
|
|
if(!fo.Open(recent_file)) {
|
|
Exclamation(NFormat(t_("Error creating file [* \1%s\1]."), recent_file));
|
|
continue;
|
|
}
|
|
if(fmt == FMT_TEXT)
|
|
fo.PutLine(title);
|
|
Progress progress(t_("Exporting row %d"));
|
|
while(cursor.Fetch()) {
|
|
String script = rowbegin;
|
|
for(int i = 0; i < out.GetCount(); i++) {
|
|
Value v = cursor[out[i]];
|
|
switch(fmt) {
|
|
case FMT_TEXT: {
|
|
if(i)
|
|
script.Cat(delim);
|
|
if(IsString(v) && quote) {
|
|
String s = v;
|
|
script << '\"';
|
|
for(const char *p = s, *e = s.End(); p < e; p++)
|
|
if(*p == '\"')
|
|
script.Cat("\"\"");
|
|
else
|
|
script.Cat(*p);
|
|
script << '\"';
|
|
}
|
|
else
|
|
script << StdFormat(v);
|
|
break;
|
|
}
|
|
case FMT_SQL: {
|
|
if(i) script.Cat(", ");
|
|
// script << SqlCompile(SQLD_ORACLE, SqlFormat(v));
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
script << rowend;
|
|
fo.PutLine(script);
|
|
/*
|
|
if(autocommit && --left <= 0) {
|
|
fo.PutLine("commit;");
|
|
left = autocommit;
|
|
}
|
|
*/
|
|
if(progress.StepCanceled()) {
|
|
Exclamation(t_("Export aborted!"));
|
|
return;
|
|
}
|
|
}
|
|
fo.Close();
|
|
if(fo.IsError())
|
|
throw Exc(NFormat(t_("Error writing file %s."), recent_file));
|
|
break;
|
|
}
|
|
catch(Exc e) {
|
|
ShowExc(e);
|
|
}
|
|
|
|
cfg = StoreAsString(*this);
|
|
}
|
|
|
|
void DlgSqlExport::Serialize(Stream& stream)
|
|
{
|
|
int version = 1;
|
|
stream / version % format % recent_file;
|
|
}
|
|
|
|
class SqlObjectTree : public TopWindow {
|
|
public:
|
|
typedef SqlObjectTree CLASSNAME;
|
|
SqlObjectTree(SqlSession& session);
|
|
|
|
void Run();
|
|
|
|
private:
|
|
enum OBJTYPE {
|
|
OBJ_NULL,
|
|
OBJ_SCHEMA,
|
|
OBJ_SCHEMA_OBJECTS,
|
|
OBJ_TABLES,
|
|
OBJ_VIEWS,
|
|
OBJ_SEQUENCES,
|
|
OBJ_TABLE_COLUMNS,
|
|
OBJ_SEQUENCE,
|
|
OBJ_COLUMN,
|
|
OBJ_PRIMARY_KEY,
|
|
OBJ_ROWID,
|
|
};
|
|
struct Item {
|
|
Item(OBJTYPE type = OBJ_NULL, String schema = Null, String object = Null)
|
|
: type(type), schema(schema), object(object) {}
|
|
Item(const Value& v) { if(IsTypeRaw<Item>(v)) *this = ValueTo<Item>(v); }
|
|
operator Value() const { return RawToValue(*this); }
|
|
|
|
int type;
|
|
String schema;
|
|
String object;
|
|
};
|
|
|
|
void Open(int node);
|
|
void OpenSchema(int node);
|
|
void OpenTables(int node, const Item& item);
|
|
void OpenViews(int node, const Item& item);
|
|
void OpenSequences(int node, const Item& item);
|
|
void OpenTableColumns(int node, const Item& item);
|
|
|
|
void ToolLocal(Bar& bar);
|
|
void TableExport(String table_name);
|
|
|
|
private:
|
|
TreeCtrl schema;
|
|
SqlSession& session;
|
|
};
|
|
|
|
void SQLObjectTree(SqlSession& session) { SqlObjectTree(session).Run(); }
|
|
|
|
SqlObjectTree::SqlObjectTree(SqlSession& sess)
|
|
: session(sess)
|
|
{
|
|
Title(t_("SQL object tree"));
|
|
Sizeable().MaximizeBox();
|
|
Add(schema.SizePos());
|
|
schema.WhenOpen = THISBACK(Open);
|
|
schema.SetRoot(CtrlImg::Computer(), Item(OBJ_SCHEMA), t_("Schemas"));
|
|
schema.WhenBar = THISBACK(ToolLocal);
|
|
}
|
|
|
|
void SqlObjectTree::Run() {
|
|
TopWindow::Run();
|
|
}
|
|
|
|
void SqlObjectTree::Open(int node)
|
|
{
|
|
const Item& item = ValueTo<Item>(schema[node]);
|
|
switch(item.type) {
|
|
case OBJ_SCHEMA: OpenSchema(node); break;
|
|
case OBJ_TABLES: OpenTables(node, item); break;
|
|
case OBJ_VIEWS: OpenViews(node, item); break;
|
|
case OBJ_SEQUENCES: OpenSequences(node, item); break;
|
|
case OBJ_TABLE_COLUMNS: OpenTableColumns(node, item); break;
|
|
}
|
|
}
|
|
|
|
void SqlObjectTree::OpenSchema(int node) {
|
|
try {
|
|
schema.RemoveChildren(node);
|
|
Vector<String> schemas = session.EnumDatabases();
|
|
Sort(schemas);
|
|
for(int i = 0; i < schemas.GetCount(); i++) {
|
|
String sname = schemas[i];
|
|
int snode = schema.Add(node, CtrlImg::Dir(), Item(OBJ_SCHEMA_OBJECTS, sname), sname, true);
|
|
schema.Add(snode, CtrlImg::Dir(), Item(OBJ_TABLES, sname), t_("Tables"), true);
|
|
schema.Add(snode, CtrlImg::Dir(), Item(OBJ_VIEWS, sname), t_("Views"), true);
|
|
schema.Add(snode, CtrlImg::Dir(), Item(OBJ_SEQUENCES, sname), t_("Sequences"), true);
|
|
}
|
|
}
|
|
catch(Exc e) {
|
|
ShowExc(e);
|
|
}
|
|
}
|
|
|
|
void SqlObjectTree::OpenTables(int node, const Item& item)
|
|
{
|
|
try {
|
|
schema.RemoveChildren(node);
|
|
Vector<String> tables = session.EnumTables(item.schema);
|
|
Sort(tables);
|
|
for(int i = 0; i < tables.GetCount(); i++)
|
|
schema.Add(node, CtrlImg::File(),
|
|
Item(OBJ_TABLE_COLUMNS, item.schema, tables[i]),
|
|
tables[i], true);
|
|
} catch(Exc e) { ShowExc(e); }
|
|
}
|
|
|
|
void SqlObjectTree::OpenViews(int node, const Item& item) {
|
|
try {
|
|
schema.RemoveChildren(node);
|
|
Vector<String> views = session.EnumViews(item.schema);
|
|
Sort(views);
|
|
for(int i = 0; i < views.GetCount(); i++)
|
|
schema.Add(node, CtrlImg::File(),
|
|
Item(OBJ_TABLE_COLUMNS, item.schema, views[i]),
|
|
views[i], true);
|
|
} catch(Exc e) { ShowExc(e); }
|
|
}
|
|
|
|
void SqlObjectTree::OpenSequences(int node, const Item& item) {
|
|
try {
|
|
schema.RemoveChildren(node);
|
|
Vector<String> sequences = session.EnumSequences(item.schema);
|
|
Sort(sequences);
|
|
for(int i = 0; i < sequences.GetCount(); i++)
|
|
schema.Add(node, CtrlImg::File(),
|
|
Item(OBJ_SEQUENCE, item.schema, sequences[i]),
|
|
sequences[i], true);
|
|
} catch(Exc e) { ShowExc(e); }
|
|
}
|
|
|
|
void SqlObjectTree::OpenTableColumns(int node, const Item& item)
|
|
{
|
|
try {
|
|
schema.RemoveChildren(node);
|
|
Vector<SqlColumnInfo> columns = session.EnumColumns(item.schema, item.object);
|
|
for(int i = 0; i < columns.GetCount(); i++)
|
|
schema.Add(node, CtrlImg::Hd(), Item(OBJ_COLUMN, item.schema, item.object), (Value)columns[i].name);
|
|
Vector<String> pk = session.EnumPrimaryKey(item.schema, item.object);
|
|
if(!pk.IsEmpty()) {
|
|
String pklist;
|
|
for(int i = 0; i < pk.GetCount(); i++)
|
|
pklist << (i ? "; " : t_("Primary key: ")) << pk[i];
|
|
schema.Add(node, Null, Item(OBJ_PRIMARY_KEY, item.schema, item.object), (Value)pklist);
|
|
}
|
|
String rowid = session.EnumRowID(item.schema, item.object);
|
|
if(!IsNull(rowid))
|
|
schema.Add(node, Null, Item(OBJ_ROWID, item.schema, item.object), Value("RowID: " + rowid));
|
|
} catch(Exc e) { ShowExc(e); }
|
|
}
|
|
|
|
void SqlObjectTree::ToolLocal(Bar& bar)
|
|
{
|
|
if(schema.IsCursor()) {
|
|
const Item& item = ValueTo<Item>(~schema);
|
|
switch(item.type) {
|
|
case OBJ_TABLE_COLUMNS: {
|
|
bar.Add(t_("Export"), THISBACK1(TableExport, item.schema + "." + item.object));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void SqlObjectTree::TableExport(String table_name)
|
|
{
|
|
String schema, table = table_name;
|
|
int f = table.Find('.');
|
|
if(f >= 0) {
|
|
schema = table.Left(f);
|
|
table = table.Mid(f + 1);
|
|
}
|
|
Vector<SqlColumnInfo> info = session.EnumColumns(schema, table);
|
|
Sql cursor(session);
|
|
RunDlgSqlExport(cursor, "select * from " + schema + "." + table, table);
|
|
}
|
|
|
|
END_UPP_NAMESPACE
|