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
|
/* $Header$ */
#include "pool.h"
#include "pool_memory.h"
#include "parsenodes.h"
#include "gramparse.h"
#include "parser.h"
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
enum
{
TEST_CONTINUE,
TEST_QUIT
};
extern int GetDatabaseEncoding(void);
int
command(const char *cmd)
{
char name[1024],
value[1024];
if (*cmd == 'q')
return TEST_QUIT;
if (sscanf(cmd, "pset %s %s", name, value) == 2)
{
if (strcmp(name, "standard_conforming_strings") == 0)
{
parser_set_param(name, value);
fprintf(stdout, "pset %d\n", standard_conforming_strings);
}
else if (strcmp(name, "server_version") == 0)
{
parser_set_param(name, value);
fprintf(stdout, "pset %d\n", server_version_num);
}
else if (strcmp(name, "server_encoding") == 0)
{
parser_set_param(name, value);
fprintf(stdout, "pset %d\n", GetDatabaseEncoding());
}
else
fprintf(stdout, "ERROR: pset %s\n", name);
}
return TEST_CONTINUE;
}
int
main(int argc, char **argv)
{
List *tree;
ListCell *l;
int state = TEST_CONTINUE;
int notty = (!isatty(fileno(stdin)) || !isatty(fileno(stdout)));
char line[1024];
while (state != TEST_QUIT)
{
if (!notty)
fprintf(stdout, "> ");
if (!fgets(line, 1024, stdin))
break;
*(strchr(line, (int) '\n')) = '\0';
if (line[0] == '#' || line[0] == '\0')
continue;
if (line[0] == '\\')
{
state = command(line + 1);
continue;
}
tree = raw_parser(line);
if (tree == NULL)
{
printf("syntax error: %s\n", line);
}
else
{
foreach(l, tree)
{
Node *node = (Node *) lfirst(l);
printf("%s\n", nodeToString(node));
}
}
free_parser();
}
return 0;
}
|
