OdbcGen/GenWindow.cpp at master · lovefc3s/OdbcGen · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
#include "stdafx.h"
#include "GenWindow.h"

using namespace std;

GenWindow::GenWindow() :
	m_l001(" "),
	m_l002(" "),
	m_l003("Driver"),
	m_l004("Server"),
	m_l005("Database"),
	m_l006("User ID"),
	m_l007("PassWord"),
	m_l008("File"),
	m_find("..."),
	m_Ok("OK"),
	m_Close("閉じる")
{
	set_title("ODBC Class Generation");
	set_default_size(400,400);
	m_grid.set_row_spacing(4);
	m_grid.set_column_spacing(4);
	m_grid.attach(m_l001,0,0,1,1);
	m_grid.attach_next_to(m_l002,m_l001, Gtk::POS_RIGHT,1,1);
	m_grid.attach_next_to(m_l003,m_l002, Gtk::POS_BOTTOM,1,1);
	m_grid.attach_next_to(m_driver,m_l003, Gtk::POS_RIGHT,2,1);
	m_grid.attach_next_to(m_l004,m_l003, Gtk::POS_BOTTOM,1,1);
	m_grid.attach_next_to(m_server,m_l004, Gtk::POS_RIGHT,2,1);
	m_grid.attach_next_to(m_l005,m_l004, Gtk::POS_BOTTOM,1,1);
	m_grid.attach_next_to(m_database,m_l005, Gtk::POS_RIGHT,2,1);
	m_grid.attach_next_to(m_l006,m_l005, Gtk::POS_BOTTOM,1,1);
	m_grid.attach_next_to(m_user,m_l006, Gtk::POS_RIGHT,2,1);
	m_grid.attach_next_to(m_l007,m_l006, Gtk::POS_BOTTOM,1,1);
	m_grid.attach_next_to(m_pass,m_l007, Gtk::POS_RIGHT,2,1);
	m_grid.attach_next_to(m_l008,m_l007, Gtk::POS_BOTTOM,1,1);
	m_grid.attach_next_to(m_file,m_l008, Gtk::POS_RIGHT,2,1);
	m_grid.attach_next_to(m_find,m_file, Gtk::POS_RIGHT,1,1);
	m_find.signal_clicked().connect(sigc::mem_fun(this, &GenWindow::on_find_clicked));
	m_hbox.add(m_Ok);
	m_hbox.add(m_Close);
	m_grid.attach_next_to(m_hbox,m_l008, Gtk::POS_BOTTOM,4,1);
	add(m_grid);
	m_Ok.signal_clicked().connect(sigc::mem_fun(this, &GenWindow::on_Ok_clicked));
	m_Close.signal_clicked().connect(sigc::mem_fun(this, &GenWindow::on_Close_clicked));
	show_all_children();
}

GenWindow::~GenWindow() {

}

void GenWindow::on_find_clicked() {
	Gtk::FileChooserDialog dialog(*this,"File Name",Gtk::FILE_CHOOSER_ACTION_SAVE );
    dialog.add_button("_Cancel", Gtk::RESPONSE_CANCEL);
    dialog.add_button("_Save", Gtk::RESPONSE_OK);
	dialog.set_select_multiple(false);
	int ret = dialog.run();
	if(ret == Gtk::RESPONSE_OK){
		m_file.set_text(dialog.get_filename());
	}
	show_all_children();
}

void GenWindow::on_Close_clicked() {
	this->close();
}

void GenWindow::on_Ok_clicked() {
	OdbcConnection *pcon = new OdbcConnection();
	pcon->Set_Driver(m_driver.get_text());
	pcon->Set_Server(m_server.get_text());
	pcon->Set_Database(m_database.get_text());
	pcon->Set_UserID(m_user.get_text());
	pcon->Set_Password(m_pass.get_text());
	SQLRETURN rtn = pcon->DriverConnect();
	std::string con = pcon->Get_ConnectionString();
	COdbcsql *_sql = pcon->m_psql;
	OdbcCommand	*com = new OdbcCommand(pcon);
	com->m_CommandString = "SELECT TABLE_CATALOG, TABLE_SCHEMA, TABLE_NAME, TABLE_TYPE FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_CATALOG = '" +
	pcon->Get_Database() + "';";
	SQLHSTMT  _hstmt = com->Get_StatementHandle();
	SQLRETURN retcode = com->Direct();
	SQLCHAR TABLE_CATALOG[256], TABLE_SCHEMA[256], TABLE_NAME[256], TABLE_TYPE[256];
	retcode = _sql->CSQLBindCol(_hstmt, 1, SQL_C_CHAR, TABLE_CATALOG, 256, 0);
	retcode = _sql->CSQLBindCol(_hstmt, 2, SQL_C_CHAR, TABLE_SCHEMA, 256, 0);
	retcode = _sql->CSQLBindCol(_hstmt, 3, SQL_C_CHAR, TABLE_NAME, 256, 0);
	retcode = _sql->CSQLBindCol(_hstmt, 4, SQL_C_CHAR, TABLE_TYPE, 256, 0);
	vector<std::string> _tbl;
	std::string filename = m_file.get_text();
	if(filename.length() < 1 ){
		filename = pcon->Get_Database() + ".h";
		m_file.set_text( filename);
	}
	std::ofstream outputfile(filename);
	outputfile << "//" << dnew;
	outputfile << "// This File is OdbcGen Generation." << dnew;
	outputfile << "//" << dnew;
	outputfile << "#pragma once" << dnew;
	outputfile << "#include <sql.h>" << dnew;
	outputfile << "#include <sqlext.h>"  << dnew;
	outputfile << "#include \"stdafx.h\"" << dnew;
	outputfile << dnew;
	outputfile.close();
	delete com;
	delete pcon;
}