GRASS Programmer's Manual  6.4.2(2012)
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Pages
form.c
Go to the documentation of this file.
1 #include <stdlib.h>
2 #include <string.h>
3 #include <stdio.h>
4 #include <unistd.h>
5 #include <fcntl.h>
6 #include <sys/time.h>
7 #include <sys/types.h>
8 
9 /* hack for tcl 8.6 */
10 #define USE_INTERP_RESULT
11 #include <tcl.h>
12 #include <tk.h>
13 
14 #include <locale.h>
15 #include <grass/gis.h>
16 #include <grass/dbmi.h>
17 #include <grass/form.h>
18 
19 /* Structure to store column names and values */
20 typedef struct
21 {
22  char *name;
23  int ctype;
24  char *value;
25 } COLUMN;
26 
27 static char *Drvname, *Dbname, *Tblname, *Key;
28 
29 static COLUMN *Columns = NULL;
30 static int allocatedRows = 0; /* allocated space */
31 static int nRows = 0;
32 
33 /* Start new sql update */
34 int reset_values(ClientData cdata, Tcl_Interp * interp, int argc,
35  char *argv[])
36 {
37  nRows = 0;
38  Drvname = NULL;
39  Dbname = NULL;
40  Tblname = NULL;
41  Key = NULL;
42 
43  return TCL_OK;
44 }
45 
46 int set_value(ClientData cdata, Tcl_Interp * interp, int argc, char *argv[])
47 {
48  G_debug(2, "set_value(): %s %s", argv[1], argv[2]);
49 
50  if (strcmp(argv[1], F_DRIVER_FNAME) == 0) {
51  Drvname = G_store(argv[2]);
52  }
53  else if (strcmp(argv[1], F_DATABASE_FNAME) == 0) {
54  Dbname = G_store(argv[2]);
55  }
56  else if (strcmp(argv[1], F_TABLE_FNAME) == 0) {
57  Tblname = G_store(argv[2]);
58  }
59  else if (strcmp(argv[1], F_KEY_FNAME) == 0) {
60  Key = G_store(argv[2]);
61  }
62  else {
63  if (nRows == allocatedRows) {
64  allocatedRows += 100;
65  Columns =
66  (COLUMN *) G_realloc(Columns,
67  (allocatedRows) * sizeof(COLUMN));
68  }
69  Columns[nRows].name = G_store(argv[1]);
70  Columns[nRows].value = G_store(argv[2]);
71  nRows++;
72  }
73 
74  return TCL_OK;
75 }
76 
77 /* Update table, use the data previously stored by set_value() */
78 int submit(ClientData cdata, Tcl_Interp * interp, int argc, char *argv[])
79 {
80  int i, first, ncols, found, col, sqltype, keyval = 0, ret;
81  char buf[2001];
82  dbString sql, table_name, strval;
83  dbDriver *driver;
84  dbHandle handle;
85  dbTable *table;
86  dbColumn *column;
87 
88  G_debug(2, "submit()");
89 
90  db_init_string(&sql);
91  db_init_string(&table_name);
92  db_init_string(&strval);
93 
94  /* Check if all internal values are set */
95  if (Drvname == NULL || Dbname == NULL || Tblname == NULL || Key == NULL) {
96  G_warning("db connection was not set by form");
97  sprintf(buf, "set submit_msg \"db connection was not set by form.\"");
98  Tcl_Eval(interp, buf);
99  Tcl_Eval(interp, "set submit_result 0");
100  return TCL_OK;
101  }
102 
103  /* Get column types */
104  G_debug(2, "Open driver");
105  driver = db_start_driver(Drvname);
106  if (driver == NULL) {
107  G_warning("Cannot open driver");
108  sprintf(buf, "set submit_msg \"Cannot open driver '%s'\"", Drvname);
109  Tcl_Eval(interp, buf);
110  Tcl_Eval(interp, "set submit_result 0");
111  return TCL_OK;
112  }
113  G_debug(2, "Driver opened");
114 
115  db_init_handle(&handle);
116  db_set_handle(&handle, Dbname, NULL);
117  G_debug(2, "Open database");
118  if (db_open_database(driver, &handle) != DB_OK) {
119  G_warning("Cannot open database");
120  db_shutdown_driver(driver);
121  sprintf(buf,
122  "set submit_msg \"Cannot open database '%s' by driver '%s'\"",
123  Dbname, Drvname);
124 
125  Tcl_Eval(interp, buf);
126  Tcl_Eval(interp, "set submit_result 0");
127  return TCL_OK;
128  }
129  G_debug(2, "Database opened");
130 
131  db_set_string(&table_name, Tblname);
132  if (db_describe_table(driver, &table_name, &table) != DB_OK) {
133  G_warning("Cannot describe table");
134  db_shutdown_driver(driver);
135  db_close_database(driver);
136  sprintf(buf, "set submit_msg \"Cannot describe table '%s'\"",
137  Tblname);
138  Tcl_Eval(interp, buf);
139  Tcl_Eval(interp, "set submit_result 0");
140  return TCL_OK;
141  }
142  ncols = db_get_table_number_of_columns(table);
143 
144  /* For each column get ctype */
145  for (i = 0; i < nRows; i++) {
146  found = 0;
147  for (col = 0; col < ncols; col++) {
148  /* get keyval */
149  if (G_strcasecmp(Columns[i].name, Key) == 0) {
150  keyval = atoi(Columns[i].value);
151  }
152  column = db_get_table_column(table, col);
153  if (G_strcasecmp(db_get_column_name(column), Columns[i].name) ==
154  0) {
155  sqltype = db_get_column_sqltype(column);
156  Columns[i].ctype = db_sqltype_to_Ctype(sqltype);
157  found = 1;
158  break;
159  }
160  }
161  if (!found && (G_strcasecmp(Columns[i].name, F_ENCODING) != 0)) {
162  G_warning("Cannot find column type");
163  db_close_database(driver);
164  db_shutdown_driver(driver);
165  sprintf(buf, "set submit_msg \"Cannot find column type\"");
166  Tcl_Eval(interp, buf);
167  Tcl_Eval(interp, "set submit_result 0");
168  return TCL_OK;
169  }
170  }
171 
172  /* Construct update statement */
173  sprintf(buf, "update %s set ", Tblname);
174  db_set_string(&sql, buf);
175 
176  first = 1;
177  for (i = 0; i < nRows; i++) {
178  G_debug(3, "Index = %d of %d Name = %s, Key = %s", i, nRows,
179  Columns[i].name, Key);
180  if (G_strcasecmp(Columns[i].name, Key) == 0)
181  continue;
182 
183  if (G_strcasecmp(Columns[i].name, F_ENCODING) == 0) {
184 
185  G_debug(3, "GRASS_DB_ENCODING env-var is '%s', col val is '%s'",
186  G__getenv("GRASS_DB_ENCODING"), Columns[i].value);
187 
188  if ((strlen(Columns[i].value) == 0) ||
189  G_strcasecmp(Columns[i].value,
190  G__getenv("GRASS_DB_ENCODING")) == 0)
191  continue;
192  else {
193  G_setenv("GRASS_DB_ENCODING", Columns[i].value);
194  G_debug(3, "Set env var GRASS_DB_ENCODING to '%s'",
195  Columns[i].value);
196  if (Tcl_SetSystemEncoding(interp, Columns[i].value) ==
197  TCL_ERROR) {
198  G_warning
199  ("Could not set Tcl system encoding to '%s' (%s)",
200  Columns[i].value, interp->result);
201  }
202  }
203  continue;
204  }
205 
206  if (!first) {
207  db_append_string(&sql, ", ");
208  }
209  if (strlen(Columns[i].value) == 0) {
210  sprintf(buf, "%s = null", Columns[i].name);
211  }
212  else {
213  if (Columns[i].ctype == DB_C_TYPE_INT ||
214  Columns[i].ctype == DB_C_TYPE_DOUBLE) {
215  sprintf(buf, "%s = %s", Columns[i].name, Columns[i].value);
216  }
217  else {
218  memset(buf, '\0', strlen(buf));
219  ret = Tcl_UtfToExternal(interp,
220  Tcl_GetEncoding(interp,
221  G__getenv
222  ("GRASS_DB_ENCODING")),
223  Columns[i].value,
224  strlen(Columns[i].value), 0, NULL,
225  buf, 2000, NULL, NULL, NULL);
226 
227  if (ret != TCL_OK) {
228  G_warning("Could not convert UTF to external.");
229  db_set_string(&strval, Columns[i].value);
230  }
231  else {
232  db_set_string(&strval, buf);
233  }
234 
235  db_double_quote_string(&strval);
236  sprintf(buf, "%s = '%s'", Columns[i].name,
237  db_get_string(&strval));
238  }
239  }
240  db_append_string(&sql, buf);
241  first = 0;
242  }
243 
244  sprintf(buf, " where %s = %d", Key, keyval);
245  db_append_string(&sql, buf);
246 
247  G_debug(2, "SQL: %s", db_get_string(&sql));
248 
249  /* Update table */
250  ret = db_execute_immediate(driver, &sql);
251 
252  db_close_database(driver);
253  db_shutdown_driver(driver);
254 
255  if (ret != DB_OK) {
256  G_warning("Cannot update table");
257  Tcl_VarEval(interp, "set submit_msg \"Cannot update table:\n",
258  db_get_error_msg(), "\"", NULL);
259  Tcl_Eval(interp, "set submit_result 0");
260  }
261  else {
262  Tcl_Eval(interp, "set submit_msg \"Record successfully updated\"");
263  Tcl_Eval(interp, "set submit_result 1");
264  }
265 
266  return TCL_OK;
267 }
268 
269 /*
270  * Form
271  */
272 int Tcl_AppInit(Tcl_Interp * interp)
273 {
274  if (Tcl_Init(interp) == TCL_ERROR)
275  return TCL_ERROR;
276 
277  if (Tk_Init(interp) == TCL_ERROR)
278  return TCL_ERROR;
279 
280  Tcl_StaticPackage(interp, "Tk", Tk_Init, Tk_SafeInit);
281 
282  /*
283  * Call Tcl_CreateCommand for application-specific commands, if
284  * they weren't already created by the init procedures called above.
285  */
286 
287  Tcl_CreateCommand(interp, "submit", (Tcl_CmdProc *) submit,
288  (ClientData) NULL, (Tcl_CmdDeleteProc *) NULL);
289  Tcl_CreateCommand(interp, "set_value",
290  (Tcl_CmdProc *) set_value,
291  (ClientData) NULL, (Tcl_CmdDeleteProc *) NULL);
292  Tcl_CreateCommand(interp, "reset_values",
293  (Tcl_CmdProc *) reset_values,
294  (ClientData) NULL, (Tcl_CmdDeleteProc *) NULL);
295  /*
296  * Specify a user-specific startup file to invoke if the application
297  * is run interactively. Typically the startup file is "~/.apprc"
298  * where "app" is the name of the application. If this line is deleted
299  * then no user-specific startup file will be run under any conditions.
300  */
301 
302  Tcl_SetVar(interp, "tcl_rcFileName", "~/.grassformrc", TCL_GLOBAL_ONLY);
303  return TCL_OK;
304 }
305 
306 int main(int argc, char *argv[])
307 {
308  G_gisinit("form");
309  G_debug(2, "Form: main()");
310 
311  Tk_Main(argc, argv, Tcl_AppInit);
312  return 0;
313 }